Function TestImageGenerationCompatibilityWarning

internal/cli/chat_test.go:169–269

(t *testing.T)

}

func TestImageGenerationCompatibilityWarning(t *testing.T) {
	// Save original stderr to restore later
	originalStderr := os.Stderr
	defer func() {
		os.Stderr = originalStderr
	}()

	tests := []struct {
		name          string
		model         string
		imageFile     string
		expectWarning bool
		warningSubstr string
		description   string
	}{
		{
			name:          "Compatible model with image",
			model:         "gpt-4o",
			imageFile:     "test.png",
			expectWarning: false,
			description:   "Should not warn for compatible model",
		},
		{
			name:          "Incompatible model with image",
			model:         "o1-mini",
			imageFile:     "test.png",
			expectWarning: true,
			warningSubstr: "Warning: Model 'o1-mini' does not support image generation",
			description:   "Should warn for incompatible model",
		},
		{
			name:          "Incompatible model without image",
			model:         "o1-mini",
			imageFile:     "",
			expectWarning: false,
			description:   "Should not warn when no image file specified",
		},
		{
			name:          "Compatible model without image",
			model:         "gpt-4o-mini",
			imageFile:     "",
			expectWarning: false,
			description:   "Should not warn when no image file specified even for compatible model",
		},
		{
			name:          "Another incompatible model with image",
			model:         "gpt-3.5-turbo",
			imageFile:     "output.jpg",
			expectWarning: true,
			warningSubstr: "Warning: Model 'gpt-3.5-turbo' does not support image generation",
			description:   "Should warn for different incompatible model",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			// Note: In a real integration test, we would capture stderr like this:
			// stderrCapture := &bytes.Buffer{}