feat: Add test case for userAgentHeader

Author: zongsizhang

lib/build.gradle

@@ -30,6 +30,9 @@ dependencies {
     implementation 'org.apache.arrow:arrow-memory-netty:16.1.0'
     implementation 'io.github.resilience4j:resilience4j-retry:2.2.0'
     implementation 'org.slf4j:slf4j-simple:2.0.13'
+    // Test dependencies
+    testImplementation 'org.junit.jupiter:junit-jupiter-api:5.8.1'
+    testRuntimeOnly 'org.junit.jupiter:junit-jupiter-engine:5.8.1'
 }
 
 // Apply a specific Java toolchain to ease working on different environments.
@@ -51,3 +54,7 @@ tasks.named('jar') {
 tasks.withType(Jar) {
     archiveBaseName.set('wherobots-jdbc')
 }
+
+test {
+    useJUnitPlatform()
+}

lib/src/test/java/com/wherobots/db/jdbc/WherobotsJdbcDriverTest.java

@@ -0,0 +1,20 @@
+package com.wherobots.db.jdbc;
+
+
+import org.junit.jupiter.api.Test;
+
+import java.util.Map;
+
+class WherobotsJdbcDriverTest {
+
+    @Test
+    void getUserAgentHeader() {
+        System.setProperty("java.version", "1");
+        System.setProperty("os.name", "os1");
+        WherobotsJdbcDriver driver = new WherobotsJdbcDriver();
+        Map<String, String> header = driver.getUserAgentHeader();
+        assert header.containsKey("User-Agent");
+        String user_agent = header.get("User-Agent");
+        assert user_agent.equals("wherobots-jdbc-driver/unknown os/os1 java/1");
+    }
+}