1 files changed, 71 insertions, 0 deletions
diff --git a/site_scons/site_tools/mongo_unittest.py b/site_scons/site_tools/mongo_unittest.py
new file mode 100644
index 00000000000..70d66a02833
--- /dev/null
+++ b/site_scons/site_tools/mongo_unittest.py
@@ -0,0 +1,71 @@
+"""Pseudo-builders for building and registering unit tests.
+"""
+
+import os
+import json
+
+from buildscripts import smoke
+
+def exists(env):
+    return True
+
+def register_unit_test(env, test):
+    env['UNITTEST_LIST_ENV']._UnitTestList('$UNITTEST_LIST', test)
+    env.Alias('$UNITTEST_ALIAS', test)
+
+def unit_test_list_builder_action(env, target, source):
+    print "Generating " + str(target[0])
+    ofile = open(str(target[0]), 'wb')
+    try:
+        for s in source:
+            print '\t' + str(s)
+            ofile.write('%s\n' % s)
+    finally:
+        ofile.close()
+        
+    dir, filename = os.path.split(str(target[0]))
+    filebase, ext = os.path.splitext(filename)
+    
+    # Generate metadata file for unit tests
+    metadata_filename = os.path.join(dir, ".".join([filebase, "json"]))        
+    print "Generating metadata file %s" % metadata_filename
+    
+    tests = []
+    for s in source:
+        tests.append(smoke.tests.Test(filename=str(s), test_type="exe_test"))
+        print '\t' + str(s)
+    
+    # For now, write JSON to avoid YAML parsing dependency
+    smoke.tests.write_metadata(tests, metadata_filename, json_only=True)
+
+def build_cpp_unit_test(env, target, source, **kwargs):
+
+    libdeps = kwargs.get('LIBDEPS', [])
+    libdeps.append( '$BUILD_DIR/mongo/unittest/unittest_main' )
+
+    includeCrutch = True
+    if "NO_CRUTCH" in kwargs:
+        includeCrutch = not kwargs["NO_CRUTCH"]
+
+    if includeCrutch:
+        libdeps.append( '$BUILD_DIR/mongo/unittest/unittest_crutch' )
+
+    kwargs['LIBDEPS'] = libdeps
+
+    result = env.Program(target, source, **kwargs)
+    env.RegisterUnitTest(result[0])
+    env.Install("#/build/unittests/", target)
+    return result
+
+def generate(env):
+    # Capture the top level env so we can use it to generate the unit test list file
+    # indepenently of which environment CppUnitTest was called in. Otherwise we will get "Two
+    # different env" warnings for the unit_test_list_builder_action.
+    env['UNITTEST_LIST_ENV'] = env;
+    unit_test_list_builder = env.Builder(
+        action=env.Action(unit_test_list_builder_action, "Generating $TARGET"),
+        multi=True)
+    env.Append(BUILDERS=dict(_UnitTestList=unit_test_list_builder))
+    env.AddMethod(register_unit_test, 'RegisterUnitTest')
+    env.AddMethod(build_cpp_unit_test, 'CppUnitTest')
+    env.Alias('$UNITTEST_ALIAS', '$UNITTEST_LIST')