Test quota project

.gitignore

@@ -25,7 +25,10 @@ tmp
 *.iml
 *.ipr
 *.iws
+.ijwb
 
 # Showcase src directories
 showcase-protos
 showcase-protos-patched
+google
+googleapis

showcase/java/build.gradle

@@ -41,7 +41,9 @@ repositories {
 dependencies {
     implementation 'com.google.cloud:gapic-google-cloud-showcase-v1beta1:0.0.0-SNAPSHOT'
     implementation "io.grpc:grpc-netty-shaded:1.13.1"
+    compile 'com.google.api:gax:1.57.0'
 
+    testCompile 'com.google.api:gax:1.57.0'
     testImplementation 'junit:junit:4.13'
     testImplementation "com.google.truth:truth:1.0"
 }

showcase/java/src/test/java/com/google/api/showcase/ShowcaseTest.java

@@ -19,12 +19,17 @@
 import static com.google.common.truth.Truth.assertThat;
 import static org.junit.Assert.fail;
 
+import com.google.api.gax.core.CredentialsProvider;
+import com.google.api.gax.core.NoCredentialsProvider;
 import com.google.api.gax.rpc.AbortedException;
 import com.google.api.gax.rpc.ApiStreamObserver;
 import com.google.api.gax.rpc.BidiStreamObserver;
 import com.google.api.gax.rpc.ClientStream;
+import com.google.api.gax.rpc.HeaderProvider;
 import com.google.api.gax.rpc.ServerStream;
 import com.google.api.gax.rpc.StreamController;
+import com.google.auth.Credentials;
+import com.google.auth.oauth2.GoogleCredentials;
 import com.google.protobuf.Duration;
 import com.google.rpc.Code;
 import com.google.rpc.Status;
@@ -36,14 +41,20 @@
 import com.google.showcase.v1beta1.EchoSettings;
 import com.google.showcase.v1beta1.ExpandRequest;
 import io.grpc.StatusRuntimeException;
+import java.io.ByteArrayInputStream;
+import java.io.IOException;
+import java.io.InputStream;
 import java.util.ArrayList;
+import java.util.Collections;
 import java.util.List;
+import java.util.Map;
 import java.util.Random;
 import java.util.concurrent.CountDownLatch;
 import java.util.concurrent.TimeUnit;
 import java.util.stream.Collectors;
 import java.util.stream.IntStream;
 import org.junit.After;
+import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
@@ -54,13 +65,17 @@
 public class ShowcaseTest {
 
   private EchoClient client;
+  private ShowcaseTransportChannelProvider channelProvider;
 
   @Before
   public void setup() throws Exception {
     String host = System.getenv("HOST");
     if (host == null) host = "localhost";
     String port = System.getenv("PORT");
     if (port == null) port = "7469";
+    channelProvider =
+        new ShowcaseTransportChannelProvider(
+            host, Integer.parseInt(port), new ShowcaseHeaderProvider());
 
     // init client for all tests
     client =
@@ -258,4 +273,121 @@ public void block() {
                 .build());
     assertThat(result.getContent()).isEqualTo("Hello, World!");
   }
+
+  static GoogleCredentials loadCredentials(String credentialFile) {
+    try {
+      InputStream keyStream = new ByteArrayInputStream(credentialFile.getBytes());
+      return GoogleCredentials.fromStream(keyStream);
+    } catch (IOException e) {
+      fail("Couldn't create fake JSON credentials.");
+    }
+    return null;
+  }
+
+  @Test
+  public void quotaProjectIdTest() throws Exception {
+    final String QUOTA_PROJECT_ID_KEY = "x-google-user-project";
+    final String QUOTA_PROJECT_ID = "quota_project_id";
+    final String QUOTA_PROJECT_ID_FROM_HEADER_VALUE = "quota_project_id_from_headers";
+    final String QUOTA_PROJECT_ID_FROM_CREDENTIALS_VALUE = "quota_project_id_from_credentials";
+    final String JSON_KEY_QUOTA_PROJECT_ID =
+        "{\n"
+            + "  \"private_key_id\": \"somekeyid\",\n"
+            + "  \"private_key\": \"-----BEGIN PRIVATE KEY-----\\nMIIEvgIBADANBgkqhkiG9w0BAQEFAASCBKgwggS"
+            + "kAgEAAoIBAQC+K2hSuFpAdrJI\\nnCgcDz2M7t7bjdlsadsasad+fvRSW6TjNQZ3p5LLQY1kSZRqBqylRkzteMOyHg"
+            + "aR\\n0Pmxh3ILCND5men43j3h4eDbrhQBuxfEMalkG92sL+PNQSETY2tnvXryOvmBRwa/\\nQP/9dJfIkIDJ9Fw9N4"
+            + "Bhhhp6mCcRpdQjV38H7JsyJ7lih/oNjECgYAt\\nknddadwkwewcVxHFhcZJO+XWf6ofLUXpRwiTZakGMn8EE1uVa2"
+            + "LgczOjwWHGi99MFjxSer5m9\\n1tCa3/KEGKiS/YL71JvjwX3mb+cewlkcmweBKZHM2JPTk0ZednFSpVZMtycjkbLa"
+            + "\\ndYOS8V85AgMBewECggEBAKksaldajfDZDV6nGqbFjMiizAKJolr/M3OQw16K6o3/\\n0S31xIe3sSlgW0+UbYlF"
+            + "4U8KifhManD1apVSC3csafaspP4RZUHFhtBywLO9pR5c\\nr6S5aLp+gPWFyIp1pfXbWGvc5VY/v9x7ya1VEa6rXvL"
+            + "sKupSeWAW4tMj3eo/64ge\\nsdaceaLYw52KeBYiT6+vpsnYrEkAHO1fF/LavbLLOFJmFTMxmsNaG0tuiJHgjshB\\"
+            + "n82DpMCbXG9YcCgI/DbzuIjsdj2JC1cascSP//3PmefWysucBQe7Jryb6NQtASmnv\\nCdDw/0jmZTEjpe4S1lxfHp"
+            + "lAhHFtdgYTvyYtaLZiVVkCgYEA8eVpof2rceecw/I6\\n5ng1q3Hl2usdWV/4mZMvR0fOemacLLfocX6IYxT1zA1FF"
+            + "JlbXSRsJMf/Qq39mOR2\\nSpW+hr4jCoHeRVYLgsbggtrevGmILAlNoqCMpGZ6vDmJpq6ECV9olliDvpPgWOP+\\nm"
+            + "YPDreFBGxWvQrADNbRt2dmGsrsCgYEAyUHqB2wvJHFqdmeBsaacewzV8x9WgmeX\\ngUIi9REwXlGDW0Mz50dxpxcK"
+            + "CAYn65+7TCnY5O/jmL0VRxU1J2mSWyWTo1C+17L0\\n3fUqjxL1pkefwecxwecvC+gFFYdJ4CQ/MHHXU81Lwl1iWdF"
+            + "Cd2UoGddYaOF+KNeM\\nHC7cmqra+JsCgYEAlUNywzq8nUg7282E+uICfCB0LfwejuymR93CtsFgb7cRd6ak\\nECR"
+            + "8FGfCpH8ruWJINllbQfcHVCX47ndLZwqv3oVFKh6pAS/vVI4dpOepP8++7y1u\\ncoOvtreXCX6XqfrWDtKIvv0vjl"
+            + "HBhhhp6mCcRpdQjV38H7JsyJ7lih/oNjECgYAt\\nkndj5uNl5SiuVxHFhcZJO+XWf6ofLUregtevZakGMn8EE1uVa"
+            + "2AY7eafmoU/nZPT\\n00YB0TBATdCbn/nBSuKDESkhSg9s2GEKQZG5hBmL5uCMfo09z3SfxZIhJdlerreP\\nJ7gSi"
+            + "dI12N+EZxYd4xIJh/HFDgp7RRO87f+WJkofMQKBgGTnClK1VMaCRbJZPriw\\nEfeFCoOX75MxKwXs6xgrw4W//AYG"
+            + "GUjDt83lD6AZP6tws7gJ2IwY/qP7+lyhjEqN\\nHtfPZRGFkGZsdaksdlaksd323423d+15/UvrlRSFPNj1tWQmNKk"
+            + "XyRDW4IG1Oa2p\\nrALStNBx5Y9t0/LQnFI4w3aG\\n-----END PRIVATE KEY-----\\n\",\n"
+            + "  \"project_id\": \"someprojectid\",\n"
+            + "  \"client_email\": \"someclientid@developer.gserviceaccount.com\",\n"
+            + "  \"client_id\": \"someclientid.apps.googleusercontent.com\",\n"
+            + "  \"type\": \"service_account\",\n"
+            + "  \"quota_project_id\": \""
+            + QUOTA_PROJECT_ID_FROM_CREDENTIALS_VALUE
+            + "\"\n"
+            + "}";
+    final GoogleCredentials credentialsWithQuotaProject =
+        loadCredentials(JSON_KEY_QUOTA_PROJECT_ID);
+    final CredentialsProvider credentialsProviderWithQuota =
+        new CredentialsProvider() {
+          @Override
+          public Credentials getCredentials() throws IOException {
+            return credentialsWithQuotaProject;
+          }
+        };
+    final HeaderProvider headerProviderWithQuota =
+        new HeaderProvider() {
+          @Override
+          public Map<String, String> getHeaders() {
+            return Collections.singletonMap(
+                QUOTA_PROJECT_ID_KEY, QUOTA_PROJECT_ID_FROM_HEADER_VALUE);
+          }
+        };
+
+    EchoSettings settingsSetQuota =
+        EchoSettings.newBuilder()
+            .setTransportChannelProvider(channelProvider)
+            .setCredentialsProvider(NoCredentialsProvider.create())
+            .setQuotaProjectId(QUOTA_PROJECT_ID)
+            .build();
+    EchoSettings settingsHeader =
+        EchoSettings.newBuilder()
+            .setTransportChannelProvider(channelProvider)
+            .setCredentialsProvider(NoCredentialsProvider.create())
+            .setHeaderProvider(headerProviderWithQuota)
+            .build();
+    EchoSettings settingsHeaderAndQuota =
+        EchoSettings.newBuilder()
+            .setTransportChannelProvider(channelProvider)
+            .setCredentialsProvider(NoCredentialsProvider.create())
+            .setHeaderProvider(headerProviderWithQuota)
+            .setQuotaProjectId(QUOTA_PROJECT_ID)
+            .build();
+    EchoSettings settingsCredentials =
+        EchoSettings.newBuilder()
+            .setTransportChannelProvider(channelProvider)
+            .setCredentialsProvider(credentialsProviderWithQuota)
+            .build();
+    EchoSettings settingsCredentialsAndQuota =
+        EchoSettings.newBuilder()
+            .setTransportChannelProvider(channelProvider)
+            .setCredentialsProvider(credentialsProviderWithQuota)
+            .setQuotaProjectId(QUOTA_PROJECT_ID)
+            .build();
+    EchoSettings settingsQuotaAll =
+        EchoSettings.newBuilder()
+            .setTransportChannelProvider(channelProvider)
+            .setCredentialsProvider(credentialsProviderWithQuota)
+            .setQuotaProjectId(QUOTA_PROJECT_ID)
+            .setHeaderProvider(headerProviderWithQuota)
+            .build();
+
+    validQuotaProjectId(settingsSetQuota, QUOTA_PROJECT_ID);
+    validQuotaProjectId(settingsHeader, QUOTA_PROJECT_ID_FROM_HEADER_VALUE);
+    validQuotaProjectId(settingsHeaderAndQuota, QUOTA_PROJECT_ID);
+    validQuotaProjectId(settingsCredentials, QUOTA_PROJECT_ID_FROM_CREDENTIALS_VALUE);
+    validQuotaProjectId(settingsCredentialsAndQuota, QUOTA_PROJECT_ID);
+    validQuotaProjectId(client.getSettings(), null);
+    validQuotaProjectId(settingsQuotaAll, QUOTA_PROJECT_ID);
+  }
+
+  private void validQuotaProjectId(EchoSettings settings, String expectId) throws Exception {
+    EchoClient clientSetQuota = EchoClient.create(settings);
+    Assert.assertEquals(clientSetQuota.getSettings().getQuotaProjectId(), expectId);
+  }
 }