added Integration tests

Author: gauravsnj

google-cloud-spanner/src/test/java/com/google/cloud/spanner/connection/ITAbstractSpannerTest.java

@@ -18,6 +18,7 @@
 
 import com.google.cloud.NoCredentials;
 import com.google.cloud.spanner.Database;
+import com.google.cloud.spanner.Dialect;
 import com.google.cloud.spanner.ErrorCode;
 import com.google.cloud.spanner.GceTestEnvConfig;
 import com.google.cloud.spanner.IntegrationTestEnv;
@@ -182,7 +183,7 @@ private static StatusRuntimeException createAbortedExceptionWithMinimalRetry() {
 
   @ClassRule public static IntegrationTestEnv env = new IntegrationTestEnv();
   private static final String DEFAULT_KEY_FILE = null;
-  private static Database database;
+  public static Database database;
 
   protected static String getKeyFile() {
     return System.getProperty(GceTestEnvConfig.GCE_CREDENTIALS_FILE, DEFAULT_KEY_FILE);
@@ -322,9 +323,7 @@ protected boolean tableExists(Connection connection, String table) {
     try (ResultSet rs =
         connection.executeQuery(
             Statement.newBuilder(
-                    "SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE UPPER(TABLE_NAME)=@table_name")
-                .bind("table_name")
-                .to(table.toUpperCase())
+                    String.format("SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE UPPER(TABLE_NAME)=UPPER(\'%s\')", table))
                 .build())) {
       while (rs.next()) {
         return true;

google-cloud-spanner/src/test/java/com/google/cloud/spanner/connection/it/ITExplainTest.java

@@ -16,13 +16,20 @@
 
 package com.google.cloud.spanner.connection.it;
 
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
 
+import com.google.cloud.spanner.Dialect;
 import com.google.cloud.spanner.Mutation;
 import com.google.cloud.spanner.ParallelIntegrationTest;
 import com.google.cloud.spanner.ResultSet;
 import com.google.cloud.spanner.Statement;
+import com.google.cloud.spanner.connection.Connection;
 import com.google.cloud.spanner.connection.ITAbstractSpannerTest;
 import java.util.Arrays;
+import java.util.Collections;
+import org.junit.Before;
+import org.junit.BeforeClass;
 import org.junit.Test;
 import org.junit.experimental.categories.Category;
 import org.junit.runner.RunWith;
@@ -36,9 +43,27 @@ public void appendConnectionUri(StringBuilder uri) {
     uri.append(";autocommit=false");
   }
 
-  @Override
-  public boolean doCreateDefaultTestTable() {
-    return true;
+  @BeforeClass
+  public static void setupPostgreSQL(){
+    database = env.getTestHelper().createTestDatabase(Dialect.POSTGRESQL, Collections.emptyList());;
+  }
+
+
+  @Before
+  public void createTestTable() {
+
+    try (Connection connection = createConnection()) {
+      connection.setAutocommit(true);
+      if (!tableExists(connection, "TEST")) {
+        connection.setAutocommit(false);
+        connection.startBatchDdl();
+        connection.execute(
+            Statement.of(
+                "CREATE TABLE TEST (ID INT NOT NULL PRIMARY KEY, NAME VARCHAR(100) NOT NULL)"));
+        connection.runBatch();
+      }
+    }
+
   }
 
   @Test
@@ -52,6 +77,30 @@ public void testExplainStatement() {
 
       ResultSet resultSet =
           connection.execute(Statement.of("EXPLAIN SELECT * from TEST")).getResultSet();
+      while(resultSet.next()){
+        assertNotNull(resultSet.getString("QUERY PLAN"));
+      }
+      assertEquals(1, resultSet.getColumnCount());
+    }
+  }
+
+  @Test
+  public void testExplainAnalyzeStatement() {
+    try (ITConnection connection = createConnection()) {
+      connection.bufferedWrite(
+          Arrays.asList(
+              Mutation.newInsertBuilder("TEST").set("ID").to(1L).set("NAME").to("TEST-1").build(),
+              Mutation.newInsertBuilder("TEST").set("ID").to(2L).set("NAME").to("TEST-2").build()));
+      connection.commit();
+
+      ResultSet resultSet =
+          connection.execute(Statement.of("EXPLAIN ANALYZE SELECT * from TEST")).getResultSet();
+      while(resultSet.next()){
+        assertNotNull(resultSet.getString("QUERY PLAN"));
+        assertNotNull(resultSet.getString("EXECUTION STATS"));
+      }
+      assertEquals(2, resultSet.getColumnCount());
     }
   }
+
 }