hook up our own IsTestFilename for cpplint

1 files changed, 9 insertions, 1 deletions

diff --git a/SConstruct b/SConstruct
index 927c3ba2189..a89196dfdf4 100644
--- a/SConstruct
+++ b/SConstruct
@@ -964,10 +964,18 @@ def doLint( env , target , source ):
     filters.append( '-whitespace/tab' ) # errors found: 233
 
     sourceFiles = utils.getAllSourceFiles( prefix="src/mongo/" )
-    sourceFiles.
     args = [ "--filter=" + ",".join( filters ) , "--counting=detailed" ] + sourceFiles
     filenames = buildscripts.cpplint.ParseArguments( args  )
+
+    def _ourIsTestFilename(fn):
+        if fn.find( "dbtests" ) >= 0:
+            return True
+        if fn.endswith( "_test.cpp" ):
+            return True
+        return False
     
+    buildscripts.cpplint._IsTestFilename = _ourIsTestFilename
+
     # Change stderr to write with replacement characters so we don't die
     # if we try to print something containing non-ASCII characters.
     sys.stderr = codecs.StreamReaderWriter(sys.stderr,