[geode] branch feature/GEODE-6225 updated: added test for multiple primary key table

[ds...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

dschneider pushed a commit to branch feature/GEODE-6225
in repository https://gitbox.apache.org/repos/asf/geode.git


The following commit(s) were added to refs/heads/feature/GEODE-6225 by this push:
     new 8f039cb  added test for multiple primary key table
8f039cb is described below

commit 8f039cb8c366273fd1a1ab9111edc5a3b71782e6
Author: Darrel Schneider <ds...@pivotal.io>
AuthorDate: Wed Jan 2 16:26:18 2019 -0800

    added test for multiple primary key table
---
 .../internal/TableMetaDataManagerIntegrationTest.java  | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/geode-connectors/src/acceptanceTest/java/org/apache/geode/connectors/jdbc/internal/TableMetaDataManagerIntegrationTest.java b/geode-connectors/src/acceptanceTest/java/org/apache/geode/connectors/jdbc/internal/TableMetaDataManagerIntegrationTest.java
index 225f0e5..620a041 100644
--- a/geode-connectors/src/acceptanceTest/java/org/apache/geode/connectors/jdbc/internal/TableMetaDataManagerIntegrationTest.java
+++ b/geode-connectors/src/acceptanceTest/java/org/apache/geode/connectors/jdbc/internal/TableMetaDataManagerIntegrationTest.java
@@ -87,6 +87,14 @@ public abstract class TableMetaDataManagerIntegrationTest {
         + "age" + quote + " int)");
   }
 
+  protected void createTableWithMultiplePrimaryKeys() throws SQLException {
+    DatabaseMetaData metaData = connection.getMetaData();
+    String quote = metaData.getIdentifierQuoteString();
+    statement.execute("CREATE TABLE " + REGION_TABLE_NAME + " (" + quote + "id" + quote
+        + " VARCHAR(10)," + quote + "name" + quote + " VARCHAR(10)," + quote
+        + "age" + quote + " int, PRIMARY KEY (id, name))");
+  }
+
   @Test
   public void validateKeyColumnName() throws SQLException {
     createTable();
@@ -98,6 +106,16 @@ public abstract class TableMetaDataManagerIntegrationTest {
   }
 
   @Test
+  public void validateMultipleKeyColumnNames() throws SQLException {
+    createTableWithMultiplePrimaryKeys();
+    TableMetaDataView metaData = manager.getTableMetaDataView(connection, regionMapping);
+
+    List<String> keyColumnNames = metaData.getKeyColumnNames();
+
+    assertThat(keyColumnNames).isEqualTo(Arrays.asList("id", "name"));
+  }
+
+  @Test
   public void validateKeyColumnNameOnNonPrimaryKey() throws SQLException {
     createTableWithNoPrimaryKey();
     regionMapping.setIds("nonprimaryid");