wat

site_scons/site_tools/gtest/__init__.py

@@ -0,0 +1,28 @@
+# gtestprogram.py
+# Eryn Wells <eryn@erynwells.me>
+
+import SCons.Util
+import dirs
+
+
+def build_gtest_program(env, target, source=None, *args, **kwargs):
+    if not SCons.Util.is_List(source):
+        source = [source]
+    source.insert(0, dirs.LIB_DIR.Dir('gtest').File('gtest_main.cc'))
+    source.append(env['LIBS']['gtest']['static'])
+    return env.Program(target, source, *args, **kwargs)
+
+
+def generate(env):
+    print 'gtestprogram generate()'
+    try:
+        env.AddMethod(build_gtest_program, 'GTestProgram')
+    except AttributeError:
+        # Old version of SCons
+        from SCons.Script.SConscript import SConsEnvironment
+        SConsEnvironment.GTestProgram = build_gtest_program
+
+
+def exists(env):
+    print 'gtestprogram exists()'
+    return 'gtest' in env['LIBS']