1818import static junit .framework .Assert .assertEquals ;
1919
2020import com .google .api .generator .engine .writer .JavaWriterVisitor ;
21+ import com .google .api .generator .gapic .composer .common .TestProtoLoader ;
2122import com .google .api .generator .gapic .model .GapicClass ;
23+ import com .google .api .generator .gapic .model .GapicContext ;
2224import com .google .api .generator .gapic .model .Message ;
2325import com .google .api .generator .gapic .model .ResourceName ;
2426import com .google .api .generator .gapic .model .Service ;
@@ -101,7 +103,9 @@ public void generateResourceNameClass_echoFoobarMultiplePatterns() {
101103 ResourceName foobarResname = resourceNames .get ("showcase.googleapis.com/Foobar" );
102104 assertThat (outputResourceNames ).contains (foobarResname );
103105
104- GapicClass clazz = ResourceNameHelperClassComposer .instance ().generate (foobarResname );
106+ GapicClass clazz =
107+ ResourceNameHelperClassComposer .instance ()
108+ .generate (foobarResname , TestProtoLoader .instance ().parseShowcaseEcho ());
105109
106110 JavaWriterVisitor visitor = new JavaWriterVisitor ();
107111 clazz .classDefinition ().accept (visitor );
@@ -149,7 +153,8 @@ public void generateResourceNameClass_loggingOnePatternMultipleVariables() {
149153 assertThat (outputResourceNames ).contains (billingAccountLocationResname );
150154
151155 GapicClass clazz =
152- ResourceNameHelperClassComposer .instance ().generate (billingAccountLocationResname );
156+ ResourceNameHelperClassComposer .instance ()
157+ .generate (billingAccountLocationResname , TestProtoLoader .instance ().parseLogging ());
153158
154159 JavaWriterVisitor visitor = new JavaWriterVisitor ();
155160 clazz .classDefinition ().accept (visitor );
@@ -179,7 +184,9 @@ public void generateResourceNameClass_testingSessionOnePattern() {
179184 ResourceName sessionResname = resourceNames .get ("showcase.googleapis.com/Session" );
180185 assertThat (outputResourceNames ).contains (sessionResname );
181186
182- GapicClass clazz = ResourceNameHelperClassComposer .instance ().generate (sessionResname );
187+ GapicClass clazz =
188+ ResourceNameHelperClassComposer .instance ()
189+ .generate (sessionResname , TestProtoLoader .instance ().parseShowcaseTesting ());
183190
184191 JavaWriterVisitor visitor = new JavaWriterVisitor ();
185192 clazz .classDefinition ().accept (visitor );
@@ -208,7 +215,9 @@ public void generateResourceNameClass_testingBlueprintPatternWithNonSlashSeparat
208215 ResourceName testResname = resourceNames .get ("showcase.googleapis.com/Test" );
209216 assertThat (outputResourceNames ).contains (testResname );
210217
211- GapicClass clazz = ResourceNameHelperClassComposer .instance ().generate (testResname );
218+ GapicClass clazz =
219+ ResourceNameHelperClassComposer .instance ()
220+ .generate (testResname , TestProtoLoader .instance ().parseShowcaseTesting ());
212221
213222 JavaWriterVisitor visitor = new JavaWriterVisitor ();
214223 clazz .classDefinition ().accept (visitor );
@@ -231,7 +240,9 @@ public void generateResourceNameClass_childSingleton() {
231240 .setDescription ("This is a description" )
232241 .build ();
233242
234- GapicClass clazz = ResourceNameHelperClassComposer .instance ().generate (agentResname );
243+ GapicContext irrelevantContext = TestProtoLoader .instance ().parseShowcaseEcho ();
244+ GapicClass clazz =
245+ ResourceNameHelperClassComposer .instance ().generate (agentResname , irrelevantContext );
235246 JavaWriterVisitor visitor = new JavaWriterVisitor ();
236247 clazz .classDefinition ().accept (visitor );
237248 Utils .saveCodegenToFile (this .getClass (), "AgentName.golden" , visitor .write ());
0 commit comments