fix: add missing tests

Author: bercianor

cli/flamingock-cli/src/test/java/io/flamingock/cli/SimpleCLITest.java

@@ -39,7 +39,7 @@ void shouldCreateCommandLine() {
     void shouldHaveAuditSubcommands() {
         // When
         CommandLine cmd = new CommandLine(new FlamingockCli());
-        
+
         // Then
         assertThat(cmd.getSubcommands()).containsKey("audit");
         CommandLine auditCmd = cmd.getSubcommands().get("audit");
@@ -52,20 +52,25 @@ void shouldDetectDatabaseTypes() {
         // Given
         FlamingockConfig mongoConfig = TestUtils.createMongoConfig();
         FlamingockConfig dynamoConfig = TestUtils.createDynamoConfig();
+        FlamingockConfig couchbaseConfig = TestUtils.createCouchbaseConfig();
 
         // When/Then
         assertThat(ConfigLoader.detectDatabaseType(mongoConfig))
                 .isEqualTo(ConfigLoader.DatabaseType.MONGODB);
-        
+
         assertThat(ConfigLoader.detectDatabaseType(dynamoConfig))
                 .isEqualTo(ConfigLoader.DatabaseType.DYNAMODB);
+
+        assertThat(ConfigLoader.detectDatabaseType(couchbaseConfig))
+                .isEqualTo(ConfigLoader.DatabaseType.COUCHBASE);
     }
 
     @Test
     void shouldCreateTestConfigs() {
         // When
         FlamingockConfig mongoConfig = TestUtils.createMongoConfig();
         FlamingockConfig dynamoConfig = TestUtils.createDynamoConfig();
+        FlamingockConfig couchbaseoConfig = TestUtils.createCouchbaseConfig();
 
         // Then
         assertThat(mongoConfig).isNotNull();
@@ -75,5 +80,9 @@ void shouldCreateTestConfigs() {
         assertThat(dynamoConfig).isNotNull();
         assertThat(dynamoConfig.getAudit().getDynamodb()).isNotNull();
         assertThat(dynamoConfig.getAudit().getDynamodb().getRegion()).isEqualTo("us-east-1");
+
+        assertThat(couchbaseoConfig).isNotNull();
+        assertThat(couchbaseoConfig.getAudit().getCouchbase()).isNotNull();
+        assertThat(couchbaseoConfig.getAudit().getCouchbase().getBucketName()).isEqualTo("test-bucket");
     }
-}
+}

cli/flamingock-cli/src/test/java/io/flamingock/cli/config/SimpleConfigLoaderTest.java

@@ -45,7 +45,6 @@ void shouldLoadValidMongoConfiguration() throws IOException {
         assertThat(config.getAudit().getMongodb()).isNotNull();
         assertThat(config.getAudit().getMongodb().getConnectionString()).isEqualTo("mongodb://localhost:27017");
         assertThat(config.getAudit().getMongodb().getDatabase()).isEqualTo("test");
-        assertThat(config.getAudit().getDynamodb()).isNull();
     }
 
     @Test
@@ -63,7 +62,25 @@ void shouldLoadValidDynamoConfiguration() throws IOException {
         assertThat(config.getAudit().getDynamodb()).isNotNull();
         assertThat(config.getAudit().getDynamodb().getRegion()).isEqualTo("us-east-1");
         assertThat(config.getAudit().getDynamodb().getEndpoint()).isEqualTo("http://localhost:8000");
-        assertThat(config.getAudit().getMongodb()).isNull();
+    }
+
+    @Test
+    void shouldLoadValidCouchbaseConfiguration() throws IOException {
+        // Given
+        Path configFile = tempDir.resolve("couchbase-config.yml");
+        Files.write(configFile, TestUtils.getValidCouchbaseYaml().getBytes());
+
+        // When
+        FlamingockConfig config = ConfigLoader.loadConfig(configFile.toString());
+
+        // Then
+        assertThat(config.getServiceIdentifier()).isEqualTo("test-cli");
+        assertThat(config.getAudit()).isNotNull();
+        assertThat(config.getAudit().getCouchbase()).isNotNull();
+        assertThat(config.getAudit().getCouchbase().getEndpoint()).isEqualTo("couchbase://localhost:12110");
+        assertThat(config.getAudit().getCouchbase().getUsername()).isEqualTo("test-user");
+        assertThat(config.getAudit().getCouchbase().getPassword()).isEqualTo("test-password");
+        assertThat(config.getAudit().getCouchbase().getBucketName()).isEqualTo("test-bucket");
     }
 
     @Test
@@ -90,6 +107,18 @@ void shouldDetectDynamoDBType() throws IOException {
         assertThat(type).isEqualTo(ConfigLoader.DatabaseType.DYNAMODB);
     }
 
+    @Test
+    void shouldDetectCouchbaseType() throws IOException {
+        // Given
+        FlamingockConfig config = TestUtils.createCouchbaseConfig();
+
+        // When
+        ConfigLoader.DatabaseType type = ConfigLoader.detectDatabaseType(config);
+
+        // Then
+        assertThat(type).isEqualTo(ConfigLoader.DatabaseType.COUCHBASE);
+    }
+
     @Test
     void shouldThrowExceptionForMissingFile() {
         // When/Then

cli/flamingock-cli/src/test/java/io/flamingock/cli/test/TestUtils.java

@@ -28,37 +28,53 @@ public class TestUtils {
     public static FlamingockConfig createMongoConfig() {
         FlamingockConfig config = new FlamingockConfig();
         config.setServiceIdentifier("test-service");
-        
+
         DatabaseConfig databaseConfig = new DatabaseConfig();
         DatabaseConfig.MongoDBConfig mongoConfig = new DatabaseConfig.MongoDBConfig();
         mongoConfig.setConnectionString("mongodb://localhost:27017");
         mongoConfig.setDatabase("test-db");
         databaseConfig.setMongodb(mongoConfig);
-        
+
         config.setAudit(databaseConfig);
         return config;
     }
 
     public static FlamingockConfig createDynamoConfig() {
         FlamingockConfig config = new FlamingockConfig();
         config.setServiceIdentifier("test-service");
-        
+
         DatabaseConfig databaseConfig = new DatabaseConfig();
         DatabaseConfig.DynamoDBConfig dynamoConfig = new DatabaseConfig.DynamoDBConfig();
         dynamoConfig.setRegion("us-east-1");
         dynamoConfig.setEndpoint("http://localhost:8000");
         dynamoConfig.setAccessKey("testKey");
         dynamoConfig.setSecretKey("testSecret");
         databaseConfig.setDynamodb(dynamoConfig);
-        
+
+        config.setAudit(databaseConfig);
+        return config;
+    }
+
+    public static FlamingockConfig createCouchbaseConfig() {
+        FlamingockConfig config = new FlamingockConfig();
+        config.setServiceIdentifier("test-service");
+
+        DatabaseConfig databaseConfig = new DatabaseConfig();
+        DatabaseConfig.CouchbaseConfig couchbaseConfig = new DatabaseConfig.CouchbaseConfig();
+        couchbaseConfig.setEndpoint("couchbase://localhost:12110");
+        couchbaseConfig.setUsername("test-user");
+        couchbaseConfig.setPassword("test-password");
+        couchbaseConfig.setBucketName("test-bucket");
+        databaseConfig.setCouchbase(couchbaseConfig);
+
         config.setAudit(databaseConfig);
         return config;
     }
 
     public static List<AuditEntry> createSampleAuditEntries() {
         AuditEntry entry1 = new AuditEntry(
             "exec-001",
-            "stage-001", 
+            "stage-001",
             "change-001",
             "developer1",
             LocalDateTime.now().minusHours(1),
@@ -76,15 +92,15 @@ public static List<AuditEntry> createSampleAuditEntries() {
         AuditEntry entry2 = new AuditEntry(
             "exec-002",
             "stage-002",
-            "change-002", 
+            "change-002",
             "developer2",
             LocalDateTime.now().minusMinutes(30),
             AuditEntry.Status.FAILED,
             AuditEntry.ExecutionType.EXECUTION,
             "com.example.Change002",
             "rollback",
             800,
-            "server2.example.com", 
+            "server2.example.com",
             null,
             false,
             "Connection timeout"
@@ -122,4 +138,15 @@ public static String getValidDynamoYaml() {
                "      access-key: \"testKey\"\n" +
                "      secret-key: \"testSecret\"\n";
     }
-}
+
+    public static String getValidCouchbaseYaml() {
+        return "flamingock:\n" +
+            "  service-identifier: \"test-cli\"\n" +
+            "  audit:\n" +
+            "    couchbase:\n" +
+            "      endpoint: \"couchbase://localhost:12110\"\n" +
+            "      username: \"test-user\"\n" +
+            "      password: \"test-password\"\n" +
+            "      bucket-name: \"test-bucket\"\n";
+    }
+}