From 777e36bcf311dc911c761fdf3ac51f95278f7777 Mon Sep 17 00:00:00 2001 From: chrislusf Date: Wed, 10 Dec 2025 21:33:11 -0800 Subject: [PATCH] test: add flag validation to ec_balance_targets_correct_disk_type Add assertNoFlagError calls after ec.balance commands to ensure -diskType flag is properly recognized for both SSD and HDD disk types. --- test/erasure_coding/ec_integration_test.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/test/erasure_coding/ec_integration_test.go b/test/erasure_coding/ec_integration_test.go index 6ba54b5f4..c5fd97153 100644 --- a/test/erasure_coding/ec_integration_test.go +++ b/test/erasure_coding/ec_integration_test.go @@ -1329,7 +1329,7 @@ func TestECDiskTypeSupport(t *testing.T) { args := []string{ "-collection", "ssd_test", "-sourceDiskType", "ssd", // Filter source volumes by SSD - "-diskType", "ssd", // Place EC shards on SSD + "-diskType", "ssd", // Place EC shards on SSD "-force", } @@ -1612,6 +1612,7 @@ func TestECDiskTypeMixedCluster(t *testing.T) { ssdErr := ecBalanceCmd.Do(ssdArgs, commandEnv, &ssdOutput) t.Logf("EC balance for SSD: %v, output: %s", ssdErr, ssdOutput.String()) + assertNoFlagError(t, ssdErr, ssdOutput.String(), "ec.balance -diskType=ssd") // Run ec.balance for HDD collection with -diskType=hdd var hddOutput bytes.Buffer @@ -1622,6 +1623,7 @@ func TestECDiskTypeMixedCluster(t *testing.T) { hddErr := ecBalanceCmd.Do(hddArgs, commandEnv, &hddOutput) t.Logf("EC balance for HDD: %v, output: %s", hddErr, hddOutput.String()) + assertNoFlagError(t, hddErr, hddOutput.String(), "ec.balance -diskType=hdd") }) }