test: adds test for default user agent

thiagotnunes · thiagotnunes · commit 6ff8f445e6c9 · 2021-02-17T13:51:29.000+11:00
Tests if the default user agent is present in the user-agent header set
in the GapicSpannerRpc class.
diff --git a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/spi/v1/GapicSpannerRpcTest.java b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/spi/v1/GapicSpannerRpcTest.java
@@ -24,6 +24,7 @@
 import static org.junit.Assume.assumeTrue;
 
 import com.google.api.core.ApiFunction;
+import com.google.api.gax.core.GaxProperties;
 import com.google.api.gax.rpc.ApiCallContext;
 import com.google.api.gax.rpc.HeaderProvider;
 import com.google.auth.oauth2.AccessToken;
@@ -153,6 +154,7 @@ public class GapicSpannerRpcTest {
   private InetSocketAddress address;
   private final Map<SpannerRpc.Option, Object> optionsMap = new HashMap<>();
   private Metadata seenHeaders;
+  private String defaultUserAgent;
 
   @BeforeClass
   public static void checkNotEmulator() {
@@ -163,6 +165,7 @@ public static void checkNotEmulator() {
 
   @Before
   public void startServer() throws IOException {
+    defaultUserAgent = "spanner-java/" + GaxProperties.getLibraryVersion(GapicSpannerRpc.class);
     mockSpanner = new MockSpannerServiceImpl();
     mockSpanner.setAbortProbability(0.0D); // We don't want any unpredictable aborted transactions.
     mockSpanner.putStatementResult(StatementResult.query(SELECT1AND2, SELECT1_RESULTSET));
@@ -505,6 +508,21 @@ public void testAdminRequestsLimitExceededRetryAlgorithm() {
     assertThat(alg.shouldRetry(new Exception("random exception"), null)).isFalse();
   }
 
+  @Test
+  public void testDefaultUserAgent() {
+    final SpannerOptions options = createSpannerOptions();
+    final Spanner spanner = options.getService();
+    final DatabaseClient databaseClient =
+        spanner.getDatabaseClient(DatabaseId.of("[PROJECT]", "[INSTANCE]", "[DATABASE]"));
+
+    try (final ResultSet rs = databaseClient.singleUse().executeQuery(SELECT1AND2)) {
+      rs.next();
+    }
+
+    assertThat(seenHeaders.get(Key.of("user-agent", Metadata.ASCII_STRING_MARSHALLER)))
+        .contains(defaultUserAgent);
+  }
+
   @Test
   public void testCustomUserAgent() {
     final HeaderProvider userAgentHeaderProvider =
@@ -521,12 +539,13 @@ public Map<String, String> getHeaders() {
     final Spanner spanner = options.getService();
     final DatabaseClient databaseClient =
         spanner.getDatabaseClient(DatabaseId.of("[PROJECT]", "[INSTANCE]", "[DATABASE]"));
+
     try (final ResultSet rs = databaseClient.singleUse().executeQuery(SELECT1AND2)) {
       rs.next();
     }
 
     assertThat(seenHeaders.get(Key.of("user-agent", Metadata.ASCII_STRING_MARSHALLER)))
-        .contains("test-agent");
+        .contains("test-agent " + defaultUserAgent);
   }
 
   @SuppressWarnings("rawtypes")
