Create scanning script to determine header dependencies.

build/scan_sources_test.py

+#!/usr/bin/python
+# Copyright (c) 2011 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+import os
+import scan_sources
+import unittest
+
+
+class MockSimplePath(object):
+  def exists(self, pathname):
+    return True
+
+  def getcwd(self):
+    return '.'
+
+  def isdir(self, pathname):
+    return True
+
+  def realpath(self, pathname):
+    return pathname
+
+
+class MockScanner(object):
+  def __init__(self, filelists):
+    self.filelists = filelists
+
+  def ScanFile(self, filename):
+    if not self.filelists:
+      return []
+    return self.filelists.pop()
+
+
+class MockResolver(object):
+  def FindFile(self, filename):
+    return filename
+
+
+class ScannerUnitTest(unittest.TestCase):
+
+  def testScanData(self):
+    scanner = scan_sources.Scanner()
+    test = """
+#This is not an include
+#include is <bogus>
+#include <x1>
+ #include "x2"
+#include     <x3>
+#include     "x4"
+# include     <x5>
+# include  "x6"
+# include "x7"
+Not
+"""
+    results = scanner.ScanData(test)
+    self.assertEqual(results, ['x1', 'x2', 'x3', 'x4', 'x5', 'x6', 'x7'])
+
+  def testResolverAddDirectories(self):
+    resolver = scan_sources.Resolver(MockSimplePath())
+    resolver.AddDirectories(['Dir1 Dir2', 'Dir3', 'Dir4', 'Dir5'])
+    results = resolver.GetDirectories()
+    self.assertEqual(results, ['Dir1','Dir2','Dir3','Dir4','Dir5'])
+
+  def testResolverRelative(self):
+    resolver = scan_sources.Resolver()
+    tests = [
+        ('/foo/bar','/foo/bar/file1','file1'),
+        ('/foo/bar/extra', '/foo/bar/file2', '../file2'),
+        ('/foo/bar', '/foo/bar/extra/file3', 'extra/file3'),
+    ]
+    for (base, full, rel) in tests:
+      self.assertEqual(rel, resolver.RealToRelative(full, base))
+
+  def testWorkQ(self):
+    filelists = [['file1', 'file4', 'file2'], ['file3'], ['file5', 'file2']]
+    resolver = MockResolver()
+    scanner = MockScanner(filelists)
+    workq = scan_sources.WorkQueue(resolver, scanner)
+    workq.PushIfNew('file3')
+    result = workq.Run()
+
+    flat = set([item for item in [sublist for sublist in filelists]])
+    flat = [item for item in flat]
+
+    flat = sorted(flat)
+    result = sorted(flat)
+    self.assertEqual(flat, result)
+
+if __name__ == '__main__':
+  unittest.main()
+