Add unit tests for stub, stub settings,service settings

Author: cindy-peng

gapic-generator-java/src/test/java/com/google/api/generator/gapic/composer/grpc/ServiceSettingsClassComposerTest.java

@@ -32,12 +32,20 @@ static Stream<Arguments> data() {
             "EchoSettings",
             TestProtoLoader.instance().parseShowcaseEcho(),
             "localhost:7469",
-            "v1beta1"),
+            "v1beta1",
+            0),
         Arguments.of(
             "DeprecatedServiceSettings",
             TestProtoLoader.instance().parseDeprecatedService(),
             "localhost:7469",
-            "v1"));
+            "v1",
+            0),
+        Arguments.of(
+            "EchoServiceSelectiveGapicServiceSettings",
+            TestProtoLoader.instance().parseSelectiveGenerationTesting(),
+            "localhost:7469",
+            "v1beta1",
+            1));
   }
 
   @ParameterizedTest
@@ -46,8 +54,9 @@ void generateServiceSettingsClasses(
       String name,
       GapicContext context,
       String apiShortNameExpected,
-      String packageVersionExpected) {
-    Service service = context.services().get(0);
+      String packageVersionExpected,
+      int serviceIndex) {
+    Service service = context.services().get(serviceIndex);
     GapicClass clazz = ServiceSettingsClassComposer.instance().generate(context, service);
 
     Assert.assertGoldenClass(this.getClass(), clazz, name + ".golden");

gapic-generator-java/src/test/java/com/google/api/generator/gapic/composer/grpc/ServiceStubClassComposerTest.java

@@ -27,9 +27,19 @@
 class ServiceStubClassComposerTest {
   static Stream<Arguments> data() {
     return Stream.of(
-        Arguments.of("EchoStub", TestProtoLoader.instance().parseShowcaseEcho(), "", ""),
+        Arguments.of("EchoStub", TestProtoLoader.instance().parseShowcaseEcho(), "", "", 0),
         Arguments.of(
-            "DeprecatedServiceStub", TestProtoLoader.instance().parseDeprecatedService(), "", ""));
+            "DeprecatedServiceStub",
+            TestProtoLoader.instance().parseDeprecatedService(),
+            "",
+            "",
+            0),
+        Arguments.of(
+            "EchoServiceSelectiveGapicClientStub",
+            TestProtoLoader.instance().parseSelectiveGenerationTesting(),
+            "",
+            "",
+            1));
   }
 
   @ParameterizedTest
@@ -38,8 +48,9 @@ void generateServiceStubClasses(
       String name,
       GapicContext context,
       String apiShortNameExpected,
-      String packageVersionExpected) {
-    Service service = context.services().get(0);
+      String packageVersionExpected,
+      int serviceIndex) {
+    Service service = context.services().get(serviceIndex);
     GapicClass clazz = ServiceStubClassComposer.instance().generate(context, service);
 
     Assert.assertGoldenClass(this.getClass(), clazz, name + ".golden");

gapic-generator-java/src/test/java/com/google/api/generator/gapic/composer/grpc/ServiceStubSettingsClassComposerTest.java

@@ -31,27 +31,38 @@ static Stream<Arguments> data() {
             "LoggingServiceV2StubSettings",
             GrpcTestProtoLoader.instance().parseLogging(),
             "logging",
-            "v2"),
+            "v2",
+            0),
         Arguments.of(
             "PublisherStubSettings",
             GrpcTestProtoLoader.instance().parsePubSubPublisher(),
             "pubsub",
-            "v1"),
+            "v1",
+            0),
         Arguments.of(
             "EchoStubSettings",
             GrpcTestProtoLoader.instance().parseShowcaseEcho(),
             "localhost:7469",
-            "v1beta1"),
+            "v1beta1",
+            0),
         Arguments.of(
             "DeprecatedServiceStubSettings",
             GrpcTestProtoLoader.instance().parseDeprecatedService(),
             "localhost:7469",
-            "v1"),
+            "v1",
+            0),
         Arguments.of(
             "ApiVersionTestingStubSettings",
             GrpcTestProtoLoader.instance().parseApiVersionTesting(),
             "localhost:7469",
-            "v1"));
+            "v1",
+            0),
+        Arguments.of(
+            "EchoServiceSelectiveGapicStubSettings",
+            GrpcTestProtoLoader.instance().parseSelectiveGenerationTesting(),
+            "localhost:7469",
+            "v1beta1",
+            1));
   }
 
   @ParameterizedTest
@@ -60,8 +71,9 @@ void generateServiceStubSettingsClasses(
       String name,
       GapicContext context,
       String apiShortNameExpected,
-      String packageVersionExpected) {
-    Service service = context.services().get(0);
+      String packageVersionExpected,
+      int serviceIndex) {
+    Service service = context.services().get(serviceIndex);
     GapicClass clazz = ServiceStubSettingsClassComposer.instance().generate(context, service);
 
     Assert.assertGoldenClass(this.getClass(), clazz, name + ".golden");

gapic-generator-java/src/test/java/com/google/api/generator/gapic/composer/grpc/goldens/EchoServiceSelectiveGapicClientStub.golden

@@ -0,0 +1,49 @@
+package com.google.selective.generate.v1beta1.stub;
+
+import com.google.api.core.BetaApi;
+import com.google.api.core.InternalApi;
+import com.google.api.gax.core.BackgroundResource;
+import com.google.api.gax.rpc.BidiStreamingCallable;
+import com.google.api.gax.rpc.UnaryCallable;
+import com.google.selective.generate.v1beta1.EchoRequest;
+import com.google.selective.generate.v1beta1.EchoResponse;
+import javax.annotation.Generated;
+
+// AUTO-GENERATED DOCUMENTATION AND CLASS.
+/**
+ * Base stub class for the EchoServiceShouldGeneratePartialUsual service API.
+ *
+ * <p>This class is for advanced usage and reflects the underlying API directly.
+ */
+@BetaApi
+@Generated("by gapic-generator-java")
+public abstract class EchoServiceShouldGeneratePartialUsualStub implements BackgroundResource {
+
+  public UnaryCallable<EchoRequest, EchoResponse> echoShouldGenerateAsUsualCallable() {
+    throw new UnsupportedOperationException("Not implemented: echoShouldGenerateAsUsualCallable()");
+  }
+
+  public BidiStreamingCallable<EchoRequest, EchoResponse> chatShouldGenerateAsUsualCallable() {
+    throw new UnsupportedOperationException("Not implemented: chatShouldGenerateAsUsualCallable()");
+  }
+
+  public BidiStreamingCallable<EchoRequest, EchoResponse> chatAgainShouldGenerateAsUsualCallable() {
+    throw new UnsupportedOperationException(
+        "Not implemented: chatAgainShouldGenerateAsUsualCallable()");
+  }
+
+  @InternalApi("Internal API. This API is not intended for public consumption.")
+  public UnaryCallable<EchoRequest, EchoResponse> chatShouldGenerateAsInternalCallable() {
+    throw new UnsupportedOperationException(
+        "Not implemented: chatShouldGenerateAsInternalCallable()");
+  }
+
+  @InternalApi("Internal API. This API is not intended for public consumption.")
+  public BidiStreamingCallable<EchoRequest, EchoResponse> echoShouldGenerateAsInternalCallable() {
+    throw new UnsupportedOperationException(
+        "Not implemented: echoShouldGenerateAsInternalCallable()");
+  }
+
+  @Override
+  public abstract void close();
+}