tor

commit 86d3d310f529bc1e13ce1ca5013c674837c9856b
parent 4c095329964c47c7a37c812fdaf58b9238cc9ff5
Author: Nick Mathewson <nickm@torproject.org>
Date:   Wed, 17 Jul 2019 14:02:17 +0200

Practracker: allow tabs in include lines

This isn't actually something that Tor does, but it's cleaner to do
it this way.  Part of 29746.

Diffstat:
M
scripts/maint/practracker/metrics.py
 | 
2
+-
M
scripts/maint/practracker/practracker_tests.py
 | 
10
++++++++++

2 files changed, 11 insertions(+), 1 deletion(-)
diff --git a/scripts/maint/practracker/metrics.py b/scripts/maint/practracker/metrics.py
@@ -16,7 +16,7 @@ def get_include_count(f):
     """Get number of #include statements in the file"""
     include_count = 0
     for line in f:
-        if re.match(r' *# *include', line):
+        if re.match(r'\s*#\s*include', line):
             include_count += 1
     return include_count
 
diff --git a/scripts/maint/practracker/practracker_tests.py b/scripts/maint/practracker/practracker_tests.py
@@ -48,5 +48,15 @@ class TestFunctionLength(unittest.TestCase):
         for name, lines in metrics.get_function_lines(funcs):
             self.assertEqual(lines, 4)
 
+class TestIncludeCount(unittest.TestCase):
+    def test_include_count(self):
+        f = StringIO.StringIO("""
+  #   include <abc.h>
+  #   include "def.h"
+#include "ghi.h"
+\t#\t include "jkl.h"
+""")
+        self.assertEqual(metrics.get_include_count(f),4)
+
 if __name__ == '__main__':
     unittest.main()