Code generation for extensions api

tools/json_schema_compiler/code_test.py

+# 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.
+
+from code import Code
+import unittest
+
+class CodeTest(unittest.TestCase):
+  def testAppend(self):
+    c = Code()
+    c.append('line')
+    self.assertEquals('line', c.render())
+
+  def testBlock(self):
+    c = Code()
+    (c.append('line')
+      .sblock('sblock')
+        .append('inner')
+        .append('moreinner')
+        .sblock('moresblock')
+          .append('inner')
+        .eblock('out')
+        .append('inner')
+      .eblock('out')
+    )
+    self.assertEquals(
+      'line\n'
+      'sblock\n'
+      '  inner\n'
+      '  moreinner\n'
+      '  moresblock\n'
+      '    inner\n'
+      '  out\n'
+      '  inner\n'
+      'out',
+      c.render())
+
+  def testConcat(self):
+    b = Code()
+    (b.sblock('2')
+        .append('2')
+      .eblock('2')
+    )
+    c = Code()
+    (c.sblock('1')
+        .concat(b)
+        .append('1')
+      .eblock('1')
+    )
+    self.assertEquals(
+      '1\n'
+      '  2\n'
+      '    2\n'
+      '  2\n'
+      '  1\n'
+      '1',
+      c.render())
+    d = Code()
+    a = Code()
+    a.concat(d)
+    self.assertEquals('', a.render())
+    a.concat(c)
+    self.assertEquals(
+      '1\n'
+      '  2\n'
+      '    2\n'
+      '  2\n'
+      '  1\n'
+      '1',
+      a.render())
+
+  def testConcatErrors(self):
+    c = Code()
+    d = Code()
+    d.append('%s')
+    self.assertRaises(TypeError, c.concat, d)
+    d = Code()
+    d.append('%(classname)s')
+    self.assertRaises(TypeError, c.concat, d)
+    d = 'line of code'
+    self.assertRaises(TypeError, c.concat, d)
+
+  def testSubstitute(self):
+    c = Code()
+    c.append('%(var1)s %(var2)s %(var1)s')
+    c.substitute({'var1': 'one', 'var2': 'two'})
+    self.assertEquals('one two one', c.render())
+    c.append('%(var1)s %(var2)s %(var3)s')
+    c.append('%(var2)s %(var1)s %(var3)s')
+    c.substitute({'var1': 'one', 'var2': 'two', 'var3': 'three'})
+    self.assertEquals(
+        'one two one\n'
+        'one two three\n'
+        'two one three',
+        c.render())
+
+  def testSubstituteErrors(self):
+    # No unnamed placeholders allowed when substitute is run
+    c = Code()
+    c.append('%s %s')
+    self.assertRaises(TypeError, c.substitute, ('var1', 'one'))
+    c = Code()
+    c.append('%s %(var1)s')
+    self.assertRaises(TypeError, c.substitute, {'var1': 'one'})
+    c = Code()
+    c.append('%s %(var1)s')
+    self.assertRaises(TypeError, c.substitute, {'var1': 'one'})
+    c = Code()
+    c.append('%(var1)s')
+    self.assertRaises(KeyError, c.substitute, {'clearlynotvar1': 'one'})
+
+  def testComment(self):
+    long_comment = ('This comment is eighty nine characters in longness, '
+        'that is, to use another word, length')
+    c = Code()
+    c.comment(long_comment)
+    self.assertEquals(
+        '// This comment is eighty nine characters '
+        'in longness, that is, to use another\n'
+        '// word, length',
+        c.render())
+    c = Code()
+    c.sblock('sblock')
+    c.comment(long_comment)
+    c.eblock('eblock')
+    c.comment(long_comment)
+    self.assertEquals(
+        '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',
+        c.render())
+    long_word = 'x' * 100
+    c = Code()
+    c.comment(long_word)
+    self.assertEquals(
+        '// ' + 'x' * 77 + '\n'
+        '// ' + 'x' * 23,
+        c.render())
+
+if __name__ == '__main__':
+  unittest.main()