old auth is not working

firebolt-db · Nov 14, 2023 · 2cd29b4 · 2cd29b4
1 parent 8d3f1ff
commit 2cd29b4
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 12 deletions.
diff --git a/src/integrationTest/java/integration/tests/StatementTest.java b/src/integrationTest/java/integration/tests/StatementTest.java
@@ -31,15 +31,15 @@
 @CustomLog
 class StatementTest extends IntegrationTest {
 
-//	@BeforeEach
-//	void beforeEach() {
-//		executeStatementFromFile("/statements/statement/ddl.sql");
-//	}
-//
-//	@AfterEach
-//	void afterEach() {
-//		executeStatementFromFile("/statements/statement/cleanup.sql");
-//	}
+	@BeforeEach
+	void beforeEach() {
+		executeStatementFromFile("/statements/statement/ddl.sql");
+	}
+
+	@AfterEach
+	void afterEach() {
+		executeStatementFromFile("/statements/statement/cleanup.sql");
+	}
 
 	@Test
 	void shouldSelect1() throws SQLException {

diff --git a/src/integrationTest/java/integration/tests/SystemEngineTest.java b/src/integrationTest/java/integration/tests/SystemEngineTest.java
@@ -33,7 +33,7 @@ public class SystemEngineTest extends IntegrationTest {
 	@BeforeAll
 	void beforeAll() {
 		try {
-			executeStatementFromFile("/statements/system/ddl.sql", null);
+			executeStatementFromFile("/statements/system/ddl.sql", getSystemEngineName());
 		} catch (Exception e) {
 			log.warn("Could not execute statement", e);
 		}
@@ -42,7 +42,7 @@ void beforeAll() {
 	@AfterAll
 	void afterAll() {
 		try {
-			executeStatementFromFile("/statements/system/cleanup.sql", null);
+			executeStatementFromFile("/statements/system/cleanup.sql", getSystemEngineName());
 		} catch (Exception e) {
 			log.warn("Could not execute statement", e);
 		}

diff --git a/src/main/java/com/firebolt/jdbc/connection/FireboltConnectionUserPasswordAuthentication.java b/src/main/java/com/firebolt/jdbc/connection/FireboltConnectionUserPasswordAuthentication.java
@@ -57,7 +57,8 @@ protected void authenticate() throws SQLException {
     protected FireboltProperties extractFireboltProperties(String jdbcUri, Properties connectionProperties) {
         FireboltProperties properties = super.extractFireboltProperties(jdbcUri, connectionProperties);
         boolean systemEngine = "system".equals(properties.getEngine());
-        return properties.toBuilder().systemEngine(systemEngine).build();
+        boolean compressed = !systemEngine && properties.isCompress();
+        return properties.toBuilder().systemEngine(systemEngine).compress(compressed).build();
     }
 
     @Override