unit test for line-wrapping in ninja.py
authorEvan Martin <martine@danga.com>
Wed, 27 Jul 2011 16:35:50 +0000 (09:35 -0700)
committerEvan Martin <martine@danga.com>
Wed, 27 Jul 2011 17:03:13 +0000 (10:03 -0700)
From a patch from Elazar Leibovich <elazarl@gmail.com>.

misc/ninja_test.py [new file with mode: 0755]

diff --git a/misc/ninja_test.py b/misc/ninja_test.py
new file mode 100755 (executable)
index 0000000..555f189
--- /dev/null
@@ -0,0 +1,45 @@
+#!/usr/bin/env python
+
+# Copyright 2011 Google Inc. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import unittest
+from StringIO import StringIO
+
+import ninja
+
+LONGWORD = 'a' * 10
+
+class TestLineWordWrap(unittest.TestCase):
+    def setUp(self):
+        self.out = StringIO()
+        self.n = ninja.Writer(self.out, width=5)
+
+    def test_single_long_word(self):
+        # We shouldn't wrap a single long word.
+        self.n._line(LONGWORD)
+        self.assertEqual(LONGWORD, self.out.getvalue())
+
+    def test_few_long_words(self):
+        # We should wrap a line where the second word is overlong.
+        self.n._line(' '.join(['x', LONGWORD, 'y']))
+        self.assertEqual('x \n' + LONGWORD + '\ny', self.out.getvalue())
+
+    def test_few_long_words_space(self):
+        # We should also wrap indented lines.
+        self.n._line(' '.join(['x', LONGWORD, 'y']), indent=2)
+        self.assertEqual('x \n  ' + LONGWORD + '\n  y', self.out.getvalue())
+
+if __name__ == '__main__':
+    unittest.main()