Add a test case to TestInvalidCommands

canonical · Oct 18, 2023 · 9815ea4 · 9815ea4
1 parent 30926cc
commit 9815ea4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cmd/ubuntu-image/main_test.go b/cmd/ubuntu-image/main_test.go
@@ -123,7 +123,8 @@ func TestInvalidCommands(t *testing.T) {
 		{"invalid_flag", []string{"classic"}, []string{"--nonexistent"}, "unknown flag `nonexistent'"},
 		{"invalid_validation", []string{"snap"}, []string{"--validation=test"}, "unknown flag `validation'"},
 		{"invalid_sector_size", []string{"snap"}, []string{"--sector_size=123"}, "unknown flag `sector_size'"},
-		{"missing_flag", []string{"pack"}, []string{"--artifact-type=raw"}, "the required flags `--gadget-dir' and `--rootfs-dir' were not specified"},
+		{"missing_one_flag", []string{"pack"}, []string{"--artifact-type=raw"}, "the required flags `--gadget-dir' and `--rootfs-dir' were not specified"},
+		{"missing_flags", []string{"pack"}, []string{"--artifact-type=raw", "--gadget-dir=./test"}, "the required flag `--rootfs-dir' was not specified"},
 	}
 	for _, tc := range testCases {
 		tc := tc // capture range variable for parallel execution