Index: third_party/protobuf/src/google/protobuf/compiler/java/java_plugin_unittest.cc |
diff --git a/third_party/protobuf/src/google/protobuf/compiler/java/java_plugin_unittest.cc b/third_party/protobuf/src/google/protobuf/compiler/java/java_plugin_unittest.cc |
index cfe01885ffaa4583bd29a089a477bcd957936999..ccc94c9dfe3edc33bd1791110b31e58cb66d27b3 100644 |
--- a/third_party/protobuf/src/google/protobuf/compiler/java/java_plugin_unittest.cc |
+++ b/third_party/protobuf/src/google/protobuf/compiler/java/java_plugin_unittest.cc |
@@ -56,21 +56,22 @@ class TestGenerator : public CodeGenerator { |
virtual bool Generate(const FileDescriptor* file, |
const string& parameter, |
- OutputDirectory* output_directory, |
+ GeneratorContext* context, |
string* error) const { |
- TryInsert("Test.java", "outer_class_scope", output_directory); |
- TryInsert("Test.java", "class_scope:foo.Bar", output_directory); |
- TryInsert("Test.java", "class_scope:foo.Bar.Baz", output_directory); |
- TryInsert("Test.java", "builder_scope:foo.Bar", output_directory); |
- TryInsert("Test.java", "builder_scope:foo.Bar.Baz", output_directory); |
- TryInsert("Test.java", "enum_scope:foo.Qux", output_directory); |
+ string filename = "Test.java"; |
+ TryInsert(filename, "outer_class_scope", context); |
+ TryInsert(filename, "class_scope:foo.Bar", context); |
+ TryInsert(filename, "class_scope:foo.Bar.Baz", context); |
+ TryInsert(filename, "builder_scope:foo.Bar", context); |
+ TryInsert(filename, "builder_scope:foo.Bar.Baz", context); |
+ TryInsert(filename, "enum_scope:foo.Qux", context); |
return true; |
} |
void TryInsert(const string& filename, const string& insertion_point, |
- OutputDirectory* output_directory) const { |
+ GeneratorContext* context) const { |
scoped_ptr<io::ZeroCopyOutputStream> output( |
- output_directory->OpenForInsert(filename, insertion_point)); |
+ context->OpenForInsert(filename, insertion_point)); |
io::Printer printer(output.get(), '$'); |
printer.Print("// inserted $name$\n", "name", insertion_point); |
} |