From 1a4308134aa19198d711bfff66cdf93e7f3a9bcf Mon Sep 17 00:00:00 2001
From: Petro Tiurin <93913847+ptiurin@users.noreply.github.com>
Date: Mon, 23 Dec 2024 11:48:52 +0000
Subject: [PATCH] test: relax version checks (#489)

---
 src/test/java/com/firebolt/FireboltDriverTest.java          | 5 ++++-
 .../jdbc/metadata/FireboltDatabaseMetadataTest.java         | 5 +++--
 src/test/java/com/firebolt/jdbc/util/VersionUtilTest.java   | 6 ++++--
 3 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/src/test/java/com/firebolt/FireboltDriverTest.java b/src/test/java/com/firebolt/FireboltDriverTest.java
index 608cd9b1..8f80ffff 100644
--- a/src/test/java/com/firebolt/FireboltDriverTest.java
+++ b/src/test/java/com/firebolt/FireboltDriverTest.java
@@ -21,6 +21,7 @@
 import java.util.stream.Collector;
 import java.util.stream.Collectors;
 
+import static org.junit.Assert.assertTrue;
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertFalse;
 import static org.junit.jupiter.api.Assertions.assertNotNull;
@@ -93,7 +94,9 @@ void jdbcCompliant() {
 	void version() {
 		FireboltDriver fireboltDriver = new FireboltDriver();
 		assertEquals(3, fireboltDriver.getMajorVersion());
-		assertEquals(5, fireboltDriver.getMinorVersion());
+		int minorVersion = fireboltDriver.getMinorVersion();
+		// Sanity check for minor version
+		assertTrue(minorVersion >= 0 && minorVersion < 100);
 	}
 
 	@ParameterizedTest
diff --git a/src/test/java/com/firebolt/jdbc/metadata/FireboltDatabaseMetadataTest.java b/src/test/java/com/firebolt/jdbc/metadata/FireboltDatabaseMetadataTest.java
index 2a245535..1dfe82b1 100644
--- a/src/test/java/com/firebolt/jdbc/metadata/FireboltDatabaseMetadataTest.java
+++ b/src/test/java/com/firebolt/jdbc/metadata/FireboltDatabaseMetadataTest.java
@@ -356,12 +356,13 @@ void shouldGetDriverMajorVersion() {
 
 	@Test
 	void shouldGetDriverMinorVersion() {
-		assertEquals(5, fireboltDatabaseMetadata.getDriverMinorVersion());
+		int minorVersion = fireboltDatabaseMetadata.getDriverMinorVersion();
+		assertTrue(minorVersion >= 0 && minorVersion < 100);
 	}
 
 	@Test
 	void shouldGetDriverVersion() throws SQLException {
-		assertEquals("3.5.0", fireboltDatabaseMetadata.getDriverVersion());
+		assertTrue(fireboltDatabaseMetadata.getDriverVersion().matches("3\\.\\d+\\.\\d+"));
 	}
 
 	@Test
diff --git a/src/test/java/com/firebolt/jdbc/util/VersionUtilTest.java b/src/test/java/com/firebolt/jdbc/util/VersionUtilTest.java
index 2b243b8c..c6c0601e 100644
--- a/src/test/java/com/firebolt/jdbc/util/VersionUtilTest.java
+++ b/src/test/java/com/firebolt/jdbc/util/VersionUtilTest.java
@@ -1,6 +1,7 @@
 package com.firebolt.jdbc.util;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertTrue;
 
 import org.junit.jupiter.api.Test;
 
@@ -13,12 +14,13 @@ void shouldGetDriverMajorVersion() {
 
 	@Test
 	void shouldGetDriverMinorVersion() {
-		assertEquals(5, VersionUtil.getDriverMinorVersion());
+		int minorVersion = VersionUtil.getDriverMinorVersion();
+		assertTrue(minorVersion >= 0 && minorVersion < 100);
 	}
 
 	@Test
 	void shouldGetProjectVersion() {
-		assertEquals("3.5.0", VersionUtil.getDriverVersion());
+		assertTrue(VersionUtil.getDriverVersion().matches("3\\.\\d+\\.\\d+"));
 	}
 
 	@Test