Improve test coverage

exasol · Feb 22, 2024 · 2e86c0b · 2e86c0b
1 parent 3c3e00c
commit 2e86c0b
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/...t/java/com/exasol/adapter/dialects/exasol/ExasolFromExaWithDataTypeQueryRewriterTest.java b/...t/java/com/exasol/adapter/dialects/exasol/ExasolFromExaWithDataTypeQueryRewriterTest.java
@@ -77,6 +77,18 @@ void rewritePushdownQueryEscapesSingleQuotes() throws AdapterException, SQLExcep
                         + " STATEMENT 'string '' with '''' quotes \"...'"));
     }
 
+    @Test
+    void generateImportStatement() throws SQLException {
+        final Connection connectionMock = mockConnection();
+        when(connectionFactoryMock.getConnection()).thenReturn(connectionMock);
+        final AdapterProperties properties = createAdapterProperties();
+        final SqlDialect dialect = new ExasolSqlDialect(connectionFactoryMock, properties);
+        final ExasolFromExaWithDataTypeQueryRewriter queryRewriter = new ExasolFromExaWithDataTypeQueryRewriter(dialect,
+                new ExasolMetadataReader(connectionMock, properties), connectionFactoryMock);
+        assertThat(queryRewriter.generateImportStatement("connection", "pushdownQuery"),
+                equalTo("IMPORT INTO (c1 DECIMAL(18, 0)) FROM EXA connection STATEMENT 'pushdownQuery'"));
+    }
+
     protected Connection mockConnection() throws SQLException {
         final ResultSetMetaData metadataMock = mock(ResultSetMetaData.class);
         when(metadataMock.getColumnCount()).thenReturn(1);