Add codegen support in smoke

pkg/smoke/test/codegen/generator_test.dart

Index: pkg/smoke/test/codegen/generator_test.dart
diff --git a/pkg/smoke/test/codegen/generator_test.dart b/pkg/smoke/test/codegen/generator_test.dart
new file mode 100644
index 0000000000000000000000000000000000000000..0e2cae5bb0a4d047fb590621f3fb71b2dcdc9c62
--- /dev/null
+++ b/pkg/smoke/test/codegen/generator_test.dart
@@ -0,0 +1,135 @@
+// Copyright (c) 2014, the Dart project authors.  Please see the AUTHORS file
+// for details. All rights reserved. Use of this source code is governed by a
+// BSD-style license that can be found in the LICENSE file.
+
+library smoke.test.codegen.generator_test;
+
+import 'package:smoke/codegen/generator.dart';
+import 'package:unittest/unittest.dart';
+
+main() {
+  test('getters', () {
+    var generator = new SmokeCodeGenerator();
+    generator.addGetter('i');
+    _checkResults(generator, initCall:
+        'useGeneratedCode(new StaticConfiguration(\n'
+        '    checkedMode: false,\n'
+        '    getters: {\n'
+        '      #i: (o) => o.i,\n'
+        '    }));\n');
+
+    generator.addGetter('foo');
+    _checkResults(generator, initCall:
+        'useGeneratedCode(new StaticConfiguration(\n'
+        '    checkedMode: false,\n'
+        '    getters: {\n'
+        '      #foo: (o) => o.foo,\n'
+        '      #i: (o) => o.i,\n'
+        '    }));\n');
+  });
+
+  test('setters', () {
+    var generator = new SmokeCodeGenerator();
+    generator.addSetter('i');
+    _checkResults(generator, initCall:
+        'useGeneratedCode(new StaticConfiguration(\n'
+        '    checkedMode: false,\n'
+        '    setters: {\n'
+        '      #i: (o, v) { o.i = v; },\n'
+        '    }));\n');
+
+    generator.addSetter('foo');
+    _checkResults(generator, initCall:
+        'useGeneratedCode(new StaticConfiguration(\n'
+        '    checkedMode: false,\n'
+        '    setters: {\n'
+        '      #foo: (o, v) { o.foo = v; },\n'
+        '      #i: (o, v) { o.i = v; },\n'
+        '    }));\n');
+  });
+
+  test('names/symbols', () {
+    var generator = new SmokeCodeGenerator();
+    generator.addSymbol('i');
+    generator.addSymbol('foo');
+    _checkResults(generator, initCall:
+        'useGeneratedCode(new StaticConfiguration(\n'
+        '    checkedMode: false,\n'
+        '    names: {\n'
+        '      #foo: \'foo\',\n'
+        '      #i: \'i\',\n'
+        '    }));\n');
+  });
+
+  test('getters, setters, and names', () {
+    var generator = new SmokeCodeGenerator();
+    generator.addGetter('i');
+    generator.addSetter('i');
+    generator.addSetter('foo');
+    generator.addSymbol('foo');
+    _checkResults(generator, initCall:
+        'useGeneratedCode(new StaticConfiguration(\n'
+        '    checkedMode: false,\n'
+        '    getters: {\n'
+        '      #i: (o) => o.i,\n'
+        '    },\n'
+        '    setters: {\n'
+        '      #foo: (o, v) { o.foo = v; },\n'
+        '      #i: (o, v) { o.i = v; },\n'
+        '    },\n'
+        '    names: {\n'
+        '      #foo: \'foo\',\n'
+        '    }));\n');
+  });
+
+  test('parents', () {
+    var generator = new SmokeCodeGenerator();
+    generator.addParent(new TypeIdentifier('a.dart', 'A'),
+        new TypeIdentifier('b.dart', 'B'));
+    generator.addParent(new TypeIdentifier('a.dart', 'C'),
+        new TypeIdentifier('a.dart', 'A'));
+    _checkResults(generator,
+        imports: [
+          "import 'a.dart' as smoke_0;",
+          "import 'b.dart' as smoke_1;"
+        ],
+        initCall:
+          'useGeneratedCode(new StaticConfiguration(\n'
+          '    checkedMode: false,\n'
+          '    parents: {\n'
+          '      smoke_0.A: smoke_1.B,\n'
+          '      smoke_0.C: smoke_0.A,\n'
+          '    }));\n');
+  });
+
+  test('declarations', () {
+    var generator = new SmokeCodeGenerator();
+    generator.addDeclaration(new TypeIdentifier('a.dart', 'A'), 'foo',
+        new TypeIdentifier('dart:core', 'int'), isField: true, isFinal: true);
+    generator.addDeclaration(new TypeIdentifier('a.dart', 'A'), 'bar',
+        new TypeIdentifier('dart:core', 'Function'), isMethod: true,
+        annotations: [new ConstExpression.constructor(null, 'Annotation',
+          [new ConstExpression.string("hi")], const {})]);
+    _checkResults(generator,
+        imports: ["import 'a.dart' as smoke_0;"],
+        initCall:
+            'useGeneratedCode(new StaticConfiguration(\n'
+            '    checkedMode: false,\n'
+            '    declarations: {\n'
+            '      smoke_0.A: {\n'
+            '        #bar: const Declaration(#bar, Function, kind: METHOD, '
+                           'annotations: const [const Annotation(\'hi\')]),\n'
+            '        #foo: const Declaration(#foo, int, isFinal: true),\n'
+            '      },\n'
+            '    }));\n');
+  });
+}
+
+_checkResults(SmokeCodeGenerator generator, {List<String> imports: const [],
+    String topLevel: '', String initCall}) {
+  var allImports = []..addAll(DEFAULT_IMPORTS)..addAll(imports);
+  expect(generator.generateImports(), allImports);
+  expect(generator.generateTopLevelDeclarations(), topLevel);
+  var indentedCode = initCall.replaceAll("\n", "\n  ").trim();
+  expect(generator.generateInitCall(), '  $indentedCode\n');
+}