[DAT-18785] Fix test setup, handle case when connection isn't establi…

…shed.
liquibase · Oct 4, 2024 · f81d6a7 · f81d6a7
1 parent 1881378
commit f81d6a7
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/src/main/java/liquibase/ext/mongodb/database/MongoLiquibaseDatabase.java b/src/main/java/liquibase/ext/mongodb/database/MongoLiquibaseDatabase.java
@@ -77,8 +77,15 @@ public String getDatabaseProductName() {
 
     @Override
     public String getDatabaseProductVersion() {
-        Document document = getMongoDatabase().runCommand(new Document("buildInfo",1));
-        return document.getString("version") != null ? document.getString("version") : "Unknown";
+        String unknownVersion = "Unknown";
+        try {
+            Document document = getMongoDatabase().runCommand(new Document("buildInfo", 1));
+            String version = document.getString("version");
+            return version != null ? version : unknownVersion;
+        } catch (Exception unableToDetermineVersion) {
+            Scope.getCurrentScope().getLog(getClass()).warning("Unable to determine mongo database version!", unableToDetermineVersion);
+            return unknownVersion;
+        }
     }
 
     /**

diff --git a/src/test/java/liquibase/ext/mongodb/change/AbstractMongoChangeTest.java b/src/test/java/liquibase/ext/mongodb/change/AbstractMongoChangeTest.java
@@ -30,6 +30,11 @@ public abstract class AbstractMongoChangeTest {
 
     @BeforeEach
     void setUp() {
-        database = new MongoLiquibaseDatabase();
+        database = new MongoLiquibaseDatabase() {
+            @Override
+            public String getDatabaseProductVersion() {
+                return "test";
+            }
+        };
     }
 }