| #!/usr/bin/env dart |
| // Copyright (c) 2013, 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 enum_generator_test; |
| |
| import 'package:protoc_plugin/indenting_writer.dart'; |
| import 'package:protoc_plugin/protoc.dart'; |
| import 'package:protoc_plugin/src/descriptor.pb.dart'; |
| import 'package:test/test.dart'; |
| |
| import 'golden_file.dart'; |
| |
| void main() { |
| test('testEnumGenerator', () { |
| EnumDescriptorProto ed = new EnumDescriptorProto() |
| ..name = 'PhoneType' |
| ..value.addAll([ |
| new EnumValueDescriptorProto() |
| ..name = 'MOBILE' |
| ..number = 0, |
| new EnumValueDescriptorProto() |
| ..name = 'HOME' |
| ..number = 1, |
| new EnumValueDescriptorProto() |
| ..name = 'WORK' |
| ..number = 2, |
| new EnumValueDescriptorProto() |
| ..name = 'BUSINESS' |
| ..number = 2 |
| ]); |
| IndentingWriter writer = new IndentingWriter(); |
| FileGenerator fg = |
| new FileGenerator(new FileDescriptorProto(), new GenerationOptions()); |
| EnumGenerator eg = new EnumGenerator(ed, fg); |
| eg.generate(writer); |
| expectMatchesGoldenFile(writer.toString(), 'test/goldens/enum'); |
| }); |
| } |