Test classes

DataDog · Jul 13, 2024 · 59fca78 · 59fca78
1 parent 16c3197
commit 59fca78
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/dd-java-agent/instrumentation/jdbc/src/test/groovy/test/TestConnection.groovy b/dd-java-agent/instrumentation/jdbc/src/test/groovy/test/TestConnection.groovy
@@ -28,6 +28,7 @@ class TestConnection implements Connection {
     }
   }
 
+  private DatabaseMetaData metadata
 
   @Override
   Statement createStatement() throws SQLException {
@@ -77,7 +78,14 @@ class TestConnection implements Connection {
 
   @Override
   DatabaseMetaData getMetaData() throws SQLException {
-    return new TestDatabaseMetaData()
+    if (metadata == null) {
+      return new TestDatabaseMetaData()
+    }
+    return metadata
+  }
+
+  void setMetaData(DatabaseMetaData metadata) {
+    this.metadata = metadata
   }
 
   @Override

diff --git a/dd-java-agent/instrumentation/jdbc/src/test/groovy/test/TestDatabaseMetaData.groovy b/dd-java-agent/instrumentation/jdbc/src/test/groovy/test/TestDatabaseMetaData.groovy
@@ -12,14 +12,23 @@ class TestDatabaseMetaData implements DatabaseMetaData {
     return false
   }
 
+  private String url
+
   @Override
   boolean allTablesAreSelectable() throws SQLException {
     return false
   }
 
   @Override
   String getURL() throws SQLException {
-    return "jdbc:testdb://localhost"
+    if (url == null) {
+      return "jdbc:testdb://localhost"
+    }
+    return url
+  }
+
+  void setURL(String url) {
+    this.url = url
   }
 
   @Override