2626import com .google .api .generator .gapic .composer .grpc .ServiceClientTestClassComposer ;
2727import com .google .api .generator .gapic .composer .grpc .ServiceSettingsClassComposer ;
2828import com .google .api .generator .gapic .composer .grpc .ServiceStubSettingsClassComposer ;
29- import com .google .api .generator .gapic .composer .httpjson .HttpJsonServiceCallableFactoryClassComposer ;
30- import com .google .api .generator .gapic .composer .httpjson .HttpJsonServiceStubClassComposer ;
3129import com .google .api .generator .gapic .composer .resourcename .ResourceNameHelperClassComposer ;
3230import com .google .api .generator .gapic .model .GapicClass ;
3331import com .google .api .generator .gapic .model .GapicClass .Kind ;
3432import com .google .api .generator .gapic .model .GapicContext ;
3533import com .google .api .generator .gapic .model .GapicPackageInfo ;
3634import com .google .api .generator .gapic .model .ResourceName ;
3735import com .google .api .generator .gapic .model .Service ;
38- import com .google .api .generator .gapic .model .Transport ;
3936import com .google .common .annotations .VisibleForTesting ;
4037import java .util .ArrayList ;
4138import java .util .List ;
@@ -83,20 +80,9 @@ public static List<GapicClass> generateStubClasses(GapicContext context) {
8380 .forEach (
8481 s -> {
8582 clazzes .add (ServiceStubClassComposer .instance ().generate (context , s ));
86- if (context .transport () == Transport .REST ) {
87- clazzes .add (
88- com .google .api .generator .gapic .composer .httpjson
89- .ServiceStubSettingsClassComposer .instance ()
90- .generate (context , s ));
91- clazzes .add (
92- HttpJsonServiceCallableFactoryClassComposer .instance ().generate (context , s ));
93- clazzes .add (HttpJsonServiceStubClassComposer .instance ().generate (context , s ));
94- } else {
95- clazzes .add (ServiceStubSettingsClassComposer .instance ().generate (context , s ));
96- clazzes .add (
97- GrpcServiceCallableFactoryClassComposer .instance ().generate (context , s ));
98- clazzes .add (GrpcServiceStubClassComposer .instance ().generate (context , s ));
99- }
83+ clazzes .add (ServiceStubSettingsClassComposer .instance ().generate (context , s ));
84+ clazzes .add (GrpcServiceCallableFactoryClassComposer .instance ().generate (context , s ));
85+ clazzes .add (GrpcServiceStubClassComposer .instance ().generate (context , s ));
10086 });
10187 return clazzes ;
10288 }
@@ -108,14 +94,7 @@ public static List<GapicClass> generateClientSettingsClasses(GapicContext contex
10894 .forEach (
10995 s -> {
11096 clazzes .add (ServiceClientClassComposer .instance ().generate (context , s ));
111- if (context .transport () == Transport .REST ) {
112- clazzes .add (
113- com .google .api .generator .gapic .composer .httpjson .ServiceSettingsClassComposer
114- .instance ()
115- .generate (context , s ));
116- } else {
117- clazzes .add (ServiceSettingsClassComposer .instance ().generate (context , s ));
118- }
97+ clazzes .add (ServiceSettingsClassComposer .instance ().generate (context , s ));
11998 });
12099 return clazzes ;
121100 }
@@ -124,28 +103,15 @@ public static List<GapicClass> generateMockClasses(GapicContext context, List<Se
124103 List <GapicClass > clazzes = new ArrayList <>();
125104 services .forEach (
126105 s -> {
127- if (context .transport () == Transport .REST ) {
128- // REST transport tests donot not use mock services.
129- } else {
130- clazzes .add (MockServiceClassComposer .instance ().generate (context , s ));
131- clazzes .add (MockServiceImplClassComposer .instance ().generate (context , s ));
132- }
106+ clazzes .add (MockServiceClassComposer .instance ().generate (context , s ));
107+ clazzes .add (MockServiceImplClassComposer .instance ().generate (context , s ));
133108 });
134109 return clazzes ;
135110 }
136111
137112 public static List <GapicClass > generateTestClasses (GapicContext context ) {
138113 return context .services ().stream ()
139- .map (
140- s -> {
141- if (context .transport () == Transport .REST ) {
142- return com .google .api .generator .gapic .composer .httpjson
143- .ServiceClientTestClassComposer .instance ()
144- .generate (context , s );
145- } else {
146- return ServiceClientTestClassComposer .instance ().generate (context , s );
147- }
148- })
114+ .map (s -> ServiceClientTestClassComposer .instance ().generate (context , s ))
149115 .collect (Collectors .toList ());
150116 }
151117
0 commit comments