diff --git a/synth.py b/synth.py index 96aa7f6f92..be86b40f42 100644 --- a/synth.py +++ b/synth.py @@ -17,6 +17,7 @@ import synthtool as s import synthtool.gcp as gcp import synthtool.languages.java as java +import os AUTOSYNTH_MULTIPLE_COMMITS = True @@ -35,3 +36,26 @@ ) java.common_templates() + +# TODO: Remove all below s.replace() logic when upstream correction is made in gapic https://github.com/googleapis/gapic-generator/issues/3181 +# Remove line added by gapic generator +s.replace("google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1alpha2/MockBigQueryWriteImpl.java", + "final Object response = responses.remove();", + "") + +# Add back lines removed by gapic generator +s.replace("google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1alpha2/MockBigQueryWriteImpl.java", + """ + public void onNext(AppendRowsRequest value) { + if (response instanceof AppendRowsResponse) { + """, + """ + public void onNext(AppendRowsRequest value) { + requests.add(value); + final Object response = responses.remove(); + if (response instanceof AppendRowsResponse) { + """ + ) + +# Re-run java code formatter after making hacky code change +os.system("mvn com.coveo:fmt-maven-plugin:format") \ No newline at end of file