| Index: test/message_generator_test.dart
 | 
| diff --git a/test/message_generator_test.dart b/test/message_generator_test.dart
 | 
| index d007e3bbb5ae657efe117fe76dc043e8f0c6da20..78d48c39e1850635af86f5de910ad4bdce28429e 100755
 | 
| --- a/test/message_generator_test.dart
 | 
| +++ b/test/message_generator_test.dart
 | 
| @@ -63,6 +63,7 @@ class PhoneNumber extends GeneratedMessage {
 | 
|  }
 | 
|  
 | 
|  ''';
 | 
| +    FileDescriptorProto fd = new FileDescriptorProto();
 | 
|      EnumDescriptorProto ed = new EnumDescriptorProto()
 | 
|          ..name = 'PhoneType'
 | 
|          ..value.addAll([
 | 
| @@ -93,7 +94,7 @@ class PhoneNumber extends GeneratedMessage {
 | 
|                  ..number = 2
 | 
|                  ..label = FieldDescriptorProto_Label.LABEL_OPTIONAL
 | 
|                  ..type = FieldDescriptorProto_Type.TYPE_ENUM
 | 
| -                ..typeName = 'PhoneNumber.PhoneType',
 | 
| +                ..typeName = '.PhoneNumber.PhoneType',
 | 
|              new FieldDescriptorProto()
 | 
|                  ..name = 'name'
 | 
|                  ..number = 3
 | 
| @@ -107,9 +108,9 @@ class PhoneNumber extends GeneratedMessage {
 | 
|      var options =
 | 
|          new GenerationOptions(
 | 
|              new CodeGeneratorRequest(), new CodeGeneratorResponse());
 | 
| -    MessageGenerator mg =
 | 
| -        new MessageGenerator(
 | 
| -            md, null, new GenerationContext(options));
 | 
| +    var context = new GenerationContext(options);
 | 
| +    FileGenerator fg = new FileGenerator(fd, null, context);
 | 
| +    MessageGenerator mg = new MessageGenerator(md, fg, context);
 | 
|      mg.initializeFields();
 | 
|      mg.generate(writer);
 | 
|      expect(buffer.toString(), expected);
 | 
| 
 |