Dependancy check for .proto files is added.

checkdeps/checkdeps_test.py

 #!/usr/bin/env python
 # Copyright (c) 2012 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.
 
 """Tests for checkdeps.
 """
 
 import os
 import unittest
@@ skipping 180 matching lines @@
     dirs_traversed = []
     for rules, filenames in self.deps_checker.GetAllRulesAndFiles():
       self.failUnlessEqual(type(filenames), list)
       self.failUnlessEqual(filenames, sorted(filenames))
       if filenames:
         dir_names = set(os.path.dirname(file) for file in filenames)
         self.failUnlessEqual(1, len(dir_names))
         dirs_traversed.append(dir_names.pop())
     self.failUnlessEqual(dirs_traversed, sorted(dirs_traversed))
 
+  def testCheckAddedImportsAllowed(self):
+    problems = self.deps_checker.CheckAddedProtoImports(
+      [['checkdeps/testdata/disallowed/test.proto',
+        ['import "checkdeps/testdata/allowed/good.proto"']
+      ]])
+    self.failIf(problems)
+
+  def testCheckAddedImportsDisallowed(self):
+    problems = self.deps_checker.CheckAddedProtoImports(
+      [['checkdeps/testdata/allowed/test.proto',
+        ['import "checkdeps/testdata/disallowed/bad.proto"']
+      ]])
+    self.failUnless(problems)
+
+  def testCheckAddedImportsManyGarbageLines(self):
+    garbage_lines = ["My name is Sam%d\n" % num for num in range(50)]
+    problems = self.deps_checker.CheckAddedProtoImports(
+      [['checkdeps/testdata/allowed/test.proto', garbage_lines]])
+    self.failIf(problems)
+
+  def testCheckAddedIncludesNoRule(self):
+    problems = self.deps_checker.CheckAddedProtoImports(
+      [['checkdeps/testdata/allowed/test.proto',
+        ['import "no_rule_for_this/nogood.proto"']
+      ]])
+    self.failUnless(problems)
 
 if __name__ == '__main__':
   unittest.main()