1 files changed, 89 insertions, 0 deletions
diff --git a/native_client_sdk/src/tools/tests/fix_deps_test.py b/native_client_sdk/src/tools/tests/fix_deps_test.py
new file mode 100755
index 0000000..d02f8ac
--- /dev/null
+++ b/native_client_sdk/src/tools/tests/fix_deps_test.py
@@ -0,0 +1,89 @@
+#!/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.
+import os
+import sys
+import tempfile
+import unittest
+
+SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
+PARENT_DIR = os.path.dirname(SCRIPT_DIR)
+DATA_DIR = os.path.join(SCRIPT_DIR, 'data')
+CHROME_SRC = os.path.dirname(os.path.dirname(os.path.dirname(PARENT_DIR)))
+MOCK_DIR = os.path.join(CHROME_SRC, "third_party", "pymock")
+
+# For the mock library
+sys.path.append(MOCK_DIR)
+sys.path.append(PARENT_DIR)
+
+import fix_deps
+import mock
+
+
+class TestFixDeps(unittest.TestCase):
+  def setUp(self):
+    self.tempfile = None
+
+  def tearDown(self):
+    if self.tempfile:
+      os.remove(self.tempfile)
+
+  def testRequiresFile(self):
+    with mock.patch('sys.stderr'):
+      self.assertRaises(SystemExit, fix_deps.main, [])
+
+  def testInvalidOption(self):
+    with mock.patch('sys.stderr'):
+      self.assertRaises(SystemExit, fix_deps.main, ['--foo', 'bar'])
+
+  def testMissingFile(self):
+    with mock.patch('sys.stderr'):
+      self.assertRaises(fix_deps.Error, fix_deps.main, ['nonexistent.file'])
+
+  def testAddsDeps(self):
+    self.tempfile = tempfile.mktemp("_sdktest")
+    with open(self.tempfile, 'w') as out:
+      out.write('foo.o: foo.c foo.h bar.h\n')
+    fix_deps.FixupDepFile(self.tempfile)
+    with open(self.tempfile) as infile:
+      contents = infile.read()
+    lines = contents.splitlines()
+    self.assertEqual(len(lines), 5)
+    self.assertTrue('foo.c:' in lines)
+    self.assertTrue('foo.h:' in lines)
+    self.assertTrue('bar.h:' in lines)
+
+  def testSpacesInFilenames(self):
+    self.tempfile = tempfile.mktemp("_sdktest")
+    with open(self.tempfile, 'w') as out:
+      out.write('foo.o: foo\\ bar.h\n')
+    fix_deps.FixupDepFile(self.tempfile)
+    with open(self.tempfile) as infile:
+      contents = infile.read()
+    lines = contents.splitlines()
+    self.assertEqual(len(lines), 3)
+    self.assertEqual(lines[2], 'foo\\ bar.h:')
+
+  def testColonInFilename(self):
+    self.tempfile = tempfile.mktemp("_sdktest")
+    with open(self.tempfile, 'w') as out:
+      out.write('foo.o: c:foo.c\\\n c:bar.h\n')
+    fix_deps.FixupDepFile(self.tempfile)
+    with open(self.tempfile) as infile:
+      contents = infile.read()
+    lines = contents.splitlines()
+    self.assertEqual(len(lines), 5)
+    self.assertEqual(lines[3], 'c:foo.c:')
+    self.assertEqual(lines[4], 'c:bar.h:')
+
+  def testDoubleInvoke(self):
+    self.tempfile = tempfile.mktemp("_sdktest")
+    with open(self.tempfile, 'w') as out:
+      out.write('foo.o: foo\\ bar.h\n')
+    fix_deps.FixupDepFile(self.tempfile)
+    self.assertRaises(fix_deps.Error, fix_deps.FixupDepFile, self.tempfile)
+
+
+if __name__ == '__main__':
+  unittest.main()