Dependancy check for .proto files is added.

checkdeps/checkdeps_test.py

Index: checkdeps/checkdeps_test.py
diff --git a/checkdeps/checkdeps_test.py b/checkdeps/checkdeps_test.py
index 760668ce800e8210e26f5faa379712d14d589572..f295e0c5064638b785c6c6b6bb0f744d2c0f7bc6 100755
--- a/checkdeps/checkdeps_test.py
+++ b/checkdeps/checkdeps_test.py
@@ -198,6 +198,32 @@ class CheckDepsTest(unittest.TestCase):
         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()