fix: reject --image when multiple score files are provided

Signed-off-by: Ben Meier <[email protected]>

main_generate.go

@@ -85,8 +85,8 @@ manifests. All resources and links between Workloads will be resolved and provis
 		}
 		state := &sd.State
 
-		if len(args) != 1 && (cmd.Flags().Lookup(generateCmdOverridesFileFlag).Changed || cmd.Flags().Lookup(generateCmdOverridePropertyFlag).Changed) {
-			return errors.Errorf("cannot use --%s or --%s when 0 or more than 1 score files are provided", generateCmdOverridePropertyFlag, generateCmdOverridesFileFlag)
+		if len(args) != 1 && (cmd.Flags().Lookup(generateCmdOverridesFileFlag).Changed || cmd.Flags().Lookup(generateCmdOverridePropertyFlag).Changed || cmd.Flags().Lookup(generateCmdImageFlag).Changed) {
+			return errors.Errorf("cannot use --%s, --%s, or --%s when 0 or more than 1 score files are provided", generateCmdOverridePropertyFlag, generateCmdOverridesFileFlag, generateCmdImageFlag)
 		}
 
 		slices.Sort(args)

main_generate_test.go

@@ -208,3 +208,31 @@ resources:
 
 	assert.Contains(t, string(raw), string(raw2))
 }
+
+func TestGenerateMultipleSpecsWithImage(t *testing.T) {
+	td := changeToTempDir(t)
+	stdout, _, err := executeAndResetCommand(context.Background(), rootCmd, []string{"init"})
+	assert.NoError(t, err)
+	assert.Equal(t, "", stdout)
+	assert.NoError(t, os.WriteFile(filepath.Join(td, "scoreA.yaml"), []byte(`
+apiVersion: score.dev/v1b1
+metadata:
+  name: example-a
+containers:
+  hello:
+    image: foo
+`), 0644))
+	assert.NoError(t, os.WriteFile(filepath.Join(td, "scoreB.yaml"), []byte(`
+apiVersion: score.dev/v1b1
+metadata:
+  name: example-b
+containers:
+  hello:
+    image: foo
+`), 0644))
+	stdout, _, err = executeAndResetCommand(context.Background(), rootCmd, []string{
+		"generate", "--image", "nginx:latest", "scoreA.yaml", "scoreB.yaml",
+	})
+	assert.EqualError(t, err, "cannot use --override-property, --overrides-file, or --image when 0 or more than 1 score files are provided")
+	assert.Equal(t, "", stdout)
+}