FIR-31966 implemented nativeSQL()

firebolt-db · Apr 15, 2024 · 4dde5b8 · 4dde5b8
1 parent 72fc714
commit 4dde5b8
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/src/main/java/com/firebolt/jdbc/connection/FireboltConnection.java b/src/main/java/com/firebolt/jdbc/connection/FireboltConnection.java
@@ -507,9 +507,8 @@ public int getConnectionTimeout() {
 	}
 
 	@Override
-	@NotImplemented
-	public String nativeSQL(String sql) throws SQLException {
-		throw new FireboltUnsupportedOperationException();
+	public String nativeSQL(String sql) {
+		return sql;
 	}
 
 	@Override

diff --git a/src/test/java/com/firebolt/jdbc/connection/FireboltConnectionTest.java b/src/test/java/com/firebolt/jdbc/connection/FireboltConnectionTest.java
@@ -133,8 +133,7 @@ private static Stream<Arguments> unsupported() {
 
 				Arguments.of("setTypeMap", (Executable) () -> connection.setTypeMap(Map.of())),
 
-				Arguments.of("rollback(savepoint)", (Executable) () -> connection.rollback(mock(Savepoint.class))),
-				Arguments.of("nativeSQL", (Executable) () -> connection.nativeSQL("select 1"))
+				Arguments.of("rollback(savepoint)", (Executable) () -> connection.rollback(mock(Savepoint.class)))
 		);
 	}
 
@@ -641,5 +640,12 @@ void shouldGetEngineUrlWhenEngineIsProvided() throws SQLException {
 		}
 	}
 
+	@Test
+	void nativeSql() throws SQLException {
+		try (FireboltConnection connection = createConnection(URL, connectionProperties)) {
+			assertEquals("SELECT 1", connection.nativeSQL("SELECT 1"));
+		}
+	}
+
 	protected abstract FireboltConnection createConnection(String url, Properties props) throws SQLException;
 }