Code generation for extensions api

tools/json_schema_compiler/code_test.py

+# 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 unittest
+import model
+import code
+import json
+
+class TestCppName(unittest.TestCase):
+  def test_cpp_name(self):
+    self.assertEquals(code.cpp_name('permissions'), 'Permissions')
+    self.assertEquals(code.cpp_name('updateAllTheThings'), 'UpdateAllTheThings')
+    self.assertEquals(code.cpp_name('aa.bb.cc'), 'Aa_Bb_Cc')
+
+class TestCode(unittest.TestCase):
+  def test_append(self):
+    c = code.Code()
+    c.append('line')
+    self.assertEquals(c.render(), 'line')
+
+  def test_block(self):
+    c = code.Code()
+    c.append('line')
+    c.sblock('sblock')
+    c.append('inner')
+    c.append('moreinner')
+    c.sblock('moresblock')
+    c.append('inner')
+    c.eblock('out')
+    c.append('inner')
+    c.eblock('out')
+    self.assertEquals(c.render(),
+      'line\n'
+      'sblock\n'
+      '  inner\n'
+      '  moreinner\n'
+      '  moresblock\n'
+      '    inner\n'
+      '  out\n'
+      '  inner\n'
+      'out')
+
+  def test_add(self):
+    b = code.Code()
+    b.sblock('2')
+    b.append('2')
+    b.eblock('2')
+    c = code.Code()
+    c.sblock('1')
+    c.add(b)
+    c.append('1')
+    c.eblock('1')
+    self.assertEquals(c.render(),
+      '1\n'
+      '  2\n'
+      '    2\n'
+      '  2\n'
+      '  1\n'
+      '1')
+    d = code.Code()
+    a = code.Code()
+    a.add(d)
+    self.assertEquals(a.render(), '')
+    a.add(c)
+    self.assertEquals(a.render(),
+      '1\n'
+      '  2\n'
+      '    2\n'
+      '  2\n'
+      '  1\n'
+      '1')
+
+  def test_add_errors(self):
+    c = code.Code()
+    d = code.Code()
+    d.append('%s')
+    self.assertRaises(TypeError, c.add, d)
+    d = code.Code()
+    d.append('%(classname)s')
+    self.assertRaises(TypeError, c.add, d)
+    d = 'line of code'
+    self.assertRaises(TypeError, c.add, d)
+
+  def test_substitute(self):
+    c = code.Code()
+    c.append('%(var1)s %(var2)s %(var1)s')
+    c.substitute({'var1': 'one', 'var2': 'two'})
+    self.assertEquals(c.render(), 'one two one')
+    c.append('%(var1)s %(var2)s %(var3)s')
+    c.append('%(var1)s %(var2)s %(var3)s')
+    c.substitute({'var1': 'one', 'var2': 'two', 'var3': 'three'})
+    self.assertEquals(c.render(),
+        'one two one\n'
+        'one two three\n'
+        'one two three')
+
+  def test_substitute_errors(self):
+    # No unnamed placeholders allowed when substitute is run
+    c = code.Code()
+    c.append('%s %s')
+    self.assertRaises(TypeError, c.substitute, ('var1', 'one'))
+    c = code.Code()
+    c.append('%s %(var1)s')
+    self.assertRaises(TypeError, c.substitute, {'var1': 'one'})
+    c = code.Code()
+    c.append('%s %(var1)s')
+    self.assertRaises(TypeError, c.substitute, {'var1': 'one'})
+    c = code.Code()
+    c.append('%(var1)s')
+    self.assertRaises(KeyError, c.substitute, {'clearlynotvar1': 'one'})
+
+  def test_comment(self):
+    long_comment = ('This comment is eighty nine characters in longness, '
+        'that is, to use another word, length')
+    c = code.Code()
+    c.comment(long_comment)
+    self.assertEquals(c.render(),
+        '// This comment is eighty nine characters in longness, that is, to use another\n'
+        '// word, length')
+    c = code.Code()
+    c.sblock('sblock')
+    c.comment(long_comment)
+    c.eblock('eblock')
+    c.comment(long_comment)
+    self.assertEquals(c.render(),
+        'sblock\n'
+        '  // This comment is eighty nine characters in longness, that is, to use\n'
+        '  // another word, length\n'
+        'eblock\n'
+        '// This comment is eighty nine characters in longness, that is, to use another\n'
+        '// word, length')
+    long_word = 'x' * 100
+    c = code.Code()
+    c.comment(long_word)
+    self.assertEquals(c.render(),
+        '// ' + 'x' * 77 + '\n'
+        '// ' + 'x' * 23)
+
+if __name__ == '__main__':
+  unittest.main()