ADD: Testcase that demonstrates a caching issue with prepared statements

Author: rPraml

pom.xml

@@ -44,6 +44,14 @@
       <scope>test</scope>
     </dependency>
 
+    <!-- Override com.h2database version to get schema support -->    
+    <dependency>
+      <groupId>com.h2database</groupId>
+      <artifactId>h2</artifactId>
+      <version>1.4.193</version>
+      <scope>test</scope>
+    </dependency>
+
   </dependencies>
 
   <build>

src/test/java/org/avaje/datasource/pool/ConnectionPoolTest.java

@@ -8,6 +8,9 @@
 import java.sql.Connection;
 import java.sql.PreparedStatement;
 import java.sql.SQLException;
+import java.sql.Statement;
+
+import javax.sql.DataSource;
 
 import static org.assertj.core.api.AssertionsForClassTypes.assertThat;
 
@@ -116,4 +119,29 @@ public void getDelegate() throws SQLException {
     assertThat(underlying).isInstanceOf(org.h2.jdbc.JdbcConnection.class);
     connection.close();
   }
+  
+  @Test
+  public void testSchemaSwitch() throws SQLException {
+    Connection conn = pool.getConnection();
+    Statement stmt = conn.createStatement();
+    stmt.executeUpdate("CREATE SCHEMA TENANT_1");
+    stmt.executeUpdate("CREATE SCHEMA TENANT_2");
+    stmt.executeUpdate("CREATE TABLE TENANT_1.LOCAL_MODEL (id integer)");
+    stmt.executeUpdate("CREATE TABLE TENANT_2.LOCAL_MODEL (id integer)");
+    stmt.close();
+    
+    conn.setSchema("TENANT_1");
+    PreparedStatement ps1 = conn.prepareStatement("SELECT * from local_model");
+    ps1.close();
+    PreparedStatement ps2 = conn.prepareStatement("SELECT * from local_model");
+    ps2.close();
+
+    conn.setSchema("TENANT_2");
+    PreparedStatement ps3 = conn.prepareStatement("SELECT * from local_model");
+    ps3.close();
+
+
+    assertThat(ps1).isSameAs(ps2);  // test if pstmtCache is working
+    assertThat(ps1).isNotSameAs(ps3); // test if datasource recognize schema change
+  }
 }