| Index: third_party/protobuf/src/google/protobuf/compiler/python/python_plugin_unittest.cc
|
| diff --git a/third_party/protobuf/src/google/protobuf/compiler/python/python_plugin_unittest.cc b/third_party/protobuf/src/google/protobuf/compiler/python/python_plugin_unittest.cc
|
| index fde88761c8ebb566b6b7d6ece21cc93f49a15623..da619ad3e91fe7a408c391044708b60ab6ee2df1 100644
|
| --- a/third_party/protobuf/src/google/protobuf/compiler/python/python_plugin_unittest.cc
|
| +++ b/third_party/protobuf/src/google/protobuf/compiler/python/python_plugin_unittest.cc
|
| @@ -56,19 +56,19 @@ class TestGenerator : public CodeGenerator {
|
|
|
| virtual bool Generate(const FileDescriptor* file,
|
| const string& parameter,
|
| - OutputDirectory* output_directory,
|
| + GeneratorContext* context,
|
| string* error) const {
|
| - TryInsert("test_pb2.py", "imports", output_directory);
|
| - TryInsert("test_pb2.py", "module_scope", output_directory);
|
| - TryInsert("test_pb2.py", "class_scope:foo.Bar", output_directory);
|
| - TryInsert("test_pb2.py", "class_scope:foo.Bar.Baz", output_directory);
|
| + TryInsert("test_pb2.py", "imports", context);
|
| + TryInsert("test_pb2.py", "module_scope", context);
|
| + TryInsert("test_pb2.py", "class_scope:foo.Bar", context);
|
| + TryInsert("test_pb2.py", "class_scope:foo.Bar.Baz", 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);
|
| }
|
|
|