add tests

firebolt-db · Oct 18, 2024 · a9e785a · a9e785a
1 parent fb94f40
commit a9e785a
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/test/java/com/firebolt/jdbc/type/JavaTypeToFireboltSQLStringTest.java b/src/test/java/com/firebolt/jdbc/type/JavaTypeToFireboltSQLStringTest.java
@@ -80,6 +80,21 @@ void shouldEscapeCharactersWhenTransformingFromString() throws SQLException {
 		assertEquals("'105\\'", JavaTypeToFireboltSQLString.transformAny("105\\"));
 	}
 
+	@Test
+	void shouldEscapeCharactersWhenTransformingFromStringLegacy() throws SQLException {
+		// quotes are escaped
+		assertEquals("'105'' OR 1=1--'' '",
+				JavaTypeToFireboltSQLString.STRING.transform("105' OR 1=1--' ", FireboltVersion.LEGACY));
+		assertEquals("'105'' OR 1=1--'' '",
+				JavaTypeToFireboltSQLString.transformAny("105' OR 1=1--' ", FireboltVersion.LEGACY));
+		// \0 is escaped
+		assertEquals("'105\\\\0'", JavaTypeToFireboltSQLString.STRING.transform("105\0", FireboltVersion.LEGACY));
+		assertEquals("'105\\\\0'", JavaTypeToFireboltSQLString.transformAny("105\0", FireboltVersion.LEGACY));
+		// backslashes are escaped
+		assertEquals("'105\\\\'", JavaTypeToFireboltSQLString.STRING.transform("105\\", FireboltVersion.LEGACY));
+		assertEquals("'105\\\\'", JavaTypeToFireboltSQLString.transformAny("105\\", FireboltVersion.LEGACY));
+	}
+
 	@Test
 	void shouldTransformLongToString() throws SQLException {
 		assertEquals("105", JavaTypeToFireboltSQLString.LONG.transform(105L));