mirror of
https://github.com/danielmiessler/Fabric.git
synced 2026-02-12 23:15:05 -05:00
ebc59ee82a
## CHANGES - Move domain types from common to domain package - Move utility functions from common to util package - Update all import statements across codebase - Reorganize OAuth storage functionality into util package - Move file management functions to domain package - Update test files to use new package structure - Maintain backward compatibility for existing functionality
436 lines
12 KiB
Go
436 lines
12 KiB
Go
package cli
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/danielmiessler/fabric/internal/domain"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestInit(t *testing.T) {
|
|
args := []string{"--copy"}
|
|
expectedFlags := &Flags{Copy: true}
|
|
oldArgs := os.Args
|
|
defer func() { os.Args = oldArgs }()
|
|
os.Args = append([]string{"cmd"}, args...)
|
|
|
|
flags, err := Init()
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, expectedFlags.Copy, flags.Copy)
|
|
}
|
|
|
|
func TestReadStdin(t *testing.T) {
|
|
input := "test input"
|
|
stdin := io.NopCloser(strings.NewReader(input))
|
|
// No need to cast stdin to *os.File, pass it as io.ReadCloser directly
|
|
content, err := ReadStdin(stdin)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if content != input {
|
|
t.Fatalf("expected %q, got %q", input, content)
|
|
}
|
|
}
|
|
|
|
// ReadStdin function assuming it's part of `cli` package
|
|
func ReadStdin(reader io.ReadCloser) (string, error) {
|
|
defer reader.Close()
|
|
buf := new(bytes.Buffer)
|
|
_, err := buf.ReadFrom(reader)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return buf.String(), nil
|
|
}
|
|
|
|
func TestBuildChatOptions(t *testing.T) {
|
|
flags := &Flags{
|
|
Temperature: 0.8,
|
|
TopP: 0.9,
|
|
PresencePenalty: 0.1,
|
|
FrequencyPenalty: 0.2,
|
|
Seed: 1,
|
|
}
|
|
|
|
expectedOptions := &domain.ChatOptions{
|
|
Temperature: 0.8,
|
|
TopP: 0.9,
|
|
PresencePenalty: 0.1,
|
|
FrequencyPenalty: 0.2,
|
|
Raw: false,
|
|
Seed: 1,
|
|
}
|
|
options, err := flags.BuildChatOptions()
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, expectedOptions, options)
|
|
}
|
|
|
|
func TestBuildChatOptionsDefaultSeed(t *testing.T) {
|
|
flags := &Flags{
|
|
Temperature: 0.8,
|
|
TopP: 0.9,
|
|
PresencePenalty: 0.1,
|
|
FrequencyPenalty: 0.2,
|
|
}
|
|
|
|
expectedOptions := &domain.ChatOptions{
|
|
Temperature: 0.8,
|
|
TopP: 0.9,
|
|
PresencePenalty: 0.1,
|
|
FrequencyPenalty: 0.2,
|
|
Raw: false,
|
|
Seed: 0,
|
|
}
|
|
options, err := flags.BuildChatOptions()
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, expectedOptions, options)
|
|
}
|
|
|
|
func TestInitWithYAMLConfig(t *testing.T) {
|
|
// Create a temporary YAML config file
|
|
configContent := `
|
|
temperature: 0.9
|
|
model: gpt-4
|
|
pattern: analyze
|
|
stream: true
|
|
`
|
|
tmpfile, err := os.CreateTemp("", "config.*.yaml")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.Remove(tmpfile.Name())
|
|
|
|
if _, err := tmpfile.Write([]byte(configContent)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := tmpfile.Close(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Test 1: Basic YAML loading
|
|
t.Run("Load YAML config", func(t *testing.T) {
|
|
oldArgs := os.Args
|
|
defer func() { os.Args = oldArgs }()
|
|
os.Args = []string{"cmd", "--config", tmpfile.Name()}
|
|
|
|
flags, err := Init()
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 0.9, flags.Temperature)
|
|
assert.Equal(t, "gpt-4", flags.Model)
|
|
assert.Equal(t, "analyze", flags.Pattern)
|
|
assert.True(t, flags.Stream)
|
|
})
|
|
|
|
// Test 2: CLI overrides YAML
|
|
t.Run("CLI overrides YAML", func(t *testing.T) {
|
|
oldArgs := os.Args
|
|
defer func() { os.Args = oldArgs }()
|
|
os.Args = []string{"cmd", "--config", tmpfile.Name(), "--temperature", "0.7", "--model", "gpt-3.5-turbo"}
|
|
|
|
flags, err := Init()
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 0.7, flags.Temperature)
|
|
assert.Equal(t, "gpt-3.5-turbo", flags.Model)
|
|
assert.Equal(t, "analyze", flags.Pattern) // unchanged from YAML
|
|
assert.True(t, flags.Stream) // unchanged from YAML
|
|
})
|
|
|
|
// Test 3: Invalid YAML config
|
|
t.Run("Invalid YAML config", func(t *testing.T) {
|
|
badConfig := `
|
|
temperature: "not a float"
|
|
model: 123 # should be string
|
|
`
|
|
badfile, err := os.CreateTemp("", "bad-config.*.yaml")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.Remove(badfile.Name())
|
|
|
|
if _, err := badfile.Write([]byte(badConfig)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := badfile.Close(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
oldArgs := os.Args
|
|
defer func() { os.Args = oldArgs }()
|
|
os.Args = []string{"cmd", "--config", badfile.Name()}
|
|
|
|
_, err = Init()
|
|
assert.Error(t, err)
|
|
})
|
|
}
|
|
|
|
func TestValidateImageFile(t *testing.T) {
|
|
t.Run("Empty path should be valid", func(t *testing.T) {
|
|
err := validateImageFile("")
|
|
assert.NoError(t, err)
|
|
})
|
|
|
|
t.Run("Valid extensions should pass", func(t *testing.T) {
|
|
validExtensions := []string{".png", ".jpeg", ".jpg", ".webp"}
|
|
for _, ext := range validExtensions {
|
|
filename := "/tmp/test" + ext
|
|
err := validateImageFile(filename)
|
|
assert.NoError(t, err, "Extension %s should be valid", ext)
|
|
}
|
|
})
|
|
|
|
t.Run("Invalid extensions should fail", func(t *testing.T) {
|
|
invalidExtensions := []string{".gif", ".bmp", ".tiff", ".svg", ".txt", ""}
|
|
for _, ext := range invalidExtensions {
|
|
filename := "/tmp/test" + ext
|
|
err := validateImageFile(filename)
|
|
assert.Error(t, err, "Extension %s should be invalid", ext)
|
|
assert.Contains(t, err.Error(), "invalid image file extension")
|
|
}
|
|
})
|
|
|
|
t.Run("Existing file should fail", func(t *testing.T) {
|
|
// Create a temporary file
|
|
tempFile, err := os.CreateTemp("", "test*.png")
|
|
assert.NoError(t, err)
|
|
defer os.Remove(tempFile.Name())
|
|
tempFile.Close()
|
|
|
|
// Validation should fail because file exists
|
|
err = validateImageFile(tempFile.Name())
|
|
assert.Error(t, err)
|
|
assert.Contains(t, err.Error(), "image file already exists")
|
|
})
|
|
|
|
t.Run("Non-existing file with valid extension should pass", func(t *testing.T) {
|
|
nonExistentFile := filepath.Join(os.TempDir(), "non_existent_file.png")
|
|
// Make sure the file doesn't exist
|
|
os.Remove(nonExistentFile)
|
|
|
|
err := validateImageFile(nonExistentFile)
|
|
assert.NoError(t, err)
|
|
})
|
|
}
|
|
|
|
func TestBuildChatOptionsWithImageFileValidation(t *testing.T) {
|
|
t.Run("Valid image file should pass", func(t *testing.T) {
|
|
flags := &Flags{
|
|
ImageFile: "/tmp/output.png",
|
|
}
|
|
|
|
options, err := flags.BuildChatOptions()
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "/tmp/output.png", options.ImageFile)
|
|
})
|
|
|
|
t.Run("Invalid extension should fail", func(t *testing.T) {
|
|
flags := &Flags{
|
|
ImageFile: "/tmp/output.gif",
|
|
}
|
|
|
|
options, err := flags.BuildChatOptions()
|
|
assert.Error(t, err)
|
|
assert.Nil(t, options)
|
|
assert.Contains(t, err.Error(), "invalid image file extension")
|
|
})
|
|
|
|
t.Run("Existing file should fail", func(t *testing.T) {
|
|
// Create a temporary file
|
|
tempFile, err := os.CreateTemp("", "existing*.png")
|
|
assert.NoError(t, err)
|
|
defer os.Remove(tempFile.Name())
|
|
tempFile.Close()
|
|
|
|
flags := &Flags{
|
|
ImageFile: tempFile.Name(),
|
|
}
|
|
|
|
options, err := flags.BuildChatOptions()
|
|
assert.Error(t, err)
|
|
assert.Nil(t, options)
|
|
assert.Contains(t, err.Error(), "image file already exists")
|
|
})
|
|
}
|
|
|
|
func TestValidateImageParameters(t *testing.T) {
|
|
t.Run("No image file and no parameters should pass", func(t *testing.T) {
|
|
err := validateImageParameters("", "", "", "", 0)
|
|
assert.NoError(t, err)
|
|
})
|
|
|
|
t.Run("Image parameters without image file should fail", func(t *testing.T) {
|
|
// Test each parameter individually
|
|
err := validateImageParameters("", "1024x1024", "", "", 0)
|
|
assert.Error(t, err)
|
|
assert.Contains(t, err.Error(), "image parameters")
|
|
assert.Contains(t, err.Error(), "can only be used with --image-file")
|
|
|
|
err = validateImageParameters("", "", "high", "", 0)
|
|
assert.Error(t, err)
|
|
assert.Contains(t, err.Error(), "image parameters")
|
|
|
|
err = validateImageParameters("", "", "", "transparent", 0)
|
|
assert.Error(t, err)
|
|
assert.Contains(t, err.Error(), "image parameters")
|
|
|
|
err = validateImageParameters("", "", "", "", 50)
|
|
assert.Error(t, err)
|
|
assert.Contains(t, err.Error(), "image parameters")
|
|
|
|
// Test multiple parameters
|
|
err = validateImageParameters("", "1024x1024", "high", "transparent", 50)
|
|
assert.Error(t, err)
|
|
assert.Contains(t, err.Error(), "image parameters")
|
|
})
|
|
|
|
t.Run("Valid size values should pass", func(t *testing.T) {
|
|
validSizes := []string{"1024x1024", "1536x1024", "1024x1536", "auto"}
|
|
for _, size := range validSizes {
|
|
err := validateImageParameters("/tmp/test.png", size, "", "", 0)
|
|
assert.NoError(t, err, "Size %s should be valid", size)
|
|
}
|
|
})
|
|
|
|
t.Run("Invalid size should fail", func(t *testing.T) {
|
|
err := validateImageParameters("/tmp/test.png", "invalid", "", "", 0)
|
|
assert.Error(t, err)
|
|
assert.Contains(t, err.Error(), "invalid image size")
|
|
})
|
|
|
|
t.Run("Valid quality values should pass", func(t *testing.T) {
|
|
validQualities := []string{"low", "medium", "high", "auto"}
|
|
for _, quality := range validQualities {
|
|
err := validateImageParameters("/tmp/test.png", "", quality, "", 0)
|
|
assert.NoError(t, err, "Quality %s should be valid", quality)
|
|
}
|
|
})
|
|
|
|
t.Run("Invalid quality should fail", func(t *testing.T) {
|
|
err := validateImageParameters("/tmp/test.png", "", "invalid", "", 0)
|
|
assert.Error(t, err)
|
|
assert.Contains(t, err.Error(), "invalid image quality")
|
|
})
|
|
|
|
t.Run("Valid background values should pass", func(t *testing.T) {
|
|
validBackgrounds := []string{"opaque", "transparent"}
|
|
for _, background := range validBackgrounds {
|
|
err := validateImageParameters("/tmp/test.png", "", "", background, 0)
|
|
assert.NoError(t, err, "Background %s should be valid", background)
|
|
}
|
|
})
|
|
|
|
t.Run("Invalid background should fail", func(t *testing.T) {
|
|
err := validateImageParameters("/tmp/test.png", "", "", "invalid", 0)
|
|
assert.Error(t, err)
|
|
assert.Contains(t, err.Error(), "invalid image background")
|
|
})
|
|
|
|
t.Run("Compression for JPEG should pass", func(t *testing.T) {
|
|
err := validateImageParameters("/tmp/test.jpg", "", "", "", 75)
|
|
assert.NoError(t, err)
|
|
})
|
|
|
|
t.Run("Compression for WebP should pass", func(t *testing.T) {
|
|
err := validateImageParameters("/tmp/test.webp", "", "", "", 50)
|
|
assert.NoError(t, err)
|
|
})
|
|
|
|
t.Run("Compression for PNG should fail", func(t *testing.T) {
|
|
err := validateImageParameters("/tmp/test.png", "", "", "", 75)
|
|
assert.Error(t, err)
|
|
assert.Contains(t, err.Error(), "image compression can only be used with JPEG and WebP formats")
|
|
})
|
|
|
|
t.Run("Invalid compression range should fail", func(t *testing.T) {
|
|
err := validateImageParameters("/tmp/test.jpg", "", "", "", 150)
|
|
assert.Error(t, err)
|
|
assert.Contains(t, err.Error(), "image compression must be between 0 and 100")
|
|
|
|
err = validateImageParameters("/tmp/test.jpg", "", "", "", -10)
|
|
assert.Error(t, err)
|
|
assert.Contains(t, err.Error(), "image compression must be between 0 and 100")
|
|
})
|
|
|
|
t.Run("Transparent background for PNG should pass", func(t *testing.T) {
|
|
err := validateImageParameters("/tmp/test.png", "", "", "transparent", 0)
|
|
assert.NoError(t, err)
|
|
})
|
|
|
|
t.Run("Transparent background for WebP should pass", func(t *testing.T) {
|
|
err := validateImageParameters("/tmp/test.webp", "", "", "transparent", 0)
|
|
assert.NoError(t, err)
|
|
})
|
|
|
|
t.Run("Transparent background for JPEG should fail", func(t *testing.T) {
|
|
err := validateImageParameters("/tmp/test.jpg", "", "", "transparent", 0)
|
|
assert.Error(t, err)
|
|
assert.Contains(t, err.Error(), "transparent background can only be used with PNG and WebP formats")
|
|
})
|
|
}
|
|
|
|
func TestBuildChatOptionsWithImageParameters(t *testing.T) {
|
|
t.Run("Valid image parameters should pass", func(t *testing.T) {
|
|
flags := &Flags{
|
|
ImageFile: "/tmp/test.png",
|
|
ImageSize: "1024x1024",
|
|
ImageQuality: "high",
|
|
ImageBackground: "transparent",
|
|
ImageCompression: 0, // Not set for PNG
|
|
}
|
|
|
|
options, err := flags.BuildChatOptions()
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, options)
|
|
assert.Equal(t, "/tmp/test.png", options.ImageFile)
|
|
assert.Equal(t, "1024x1024", options.ImageSize)
|
|
assert.Equal(t, "high", options.ImageQuality)
|
|
assert.Equal(t, "transparent", options.ImageBackground)
|
|
assert.Equal(t, 0, options.ImageCompression)
|
|
})
|
|
|
|
t.Run("Invalid image parameters should fail", func(t *testing.T) {
|
|
flags := &Flags{
|
|
ImageFile: "/tmp/test.png",
|
|
ImageSize: "invalid",
|
|
ImageQuality: "high",
|
|
ImageBackground: "transparent",
|
|
}
|
|
|
|
options, err := flags.BuildChatOptions()
|
|
assert.Error(t, err)
|
|
assert.Nil(t, options)
|
|
assert.Contains(t, err.Error(), "invalid image size")
|
|
})
|
|
|
|
t.Run("JPEG with compression should pass", func(t *testing.T) {
|
|
flags := &Flags{
|
|
ImageFile: "/tmp/test.jpg",
|
|
ImageSize: "1536x1024",
|
|
ImageQuality: "medium",
|
|
ImageBackground: "opaque",
|
|
ImageCompression: 80,
|
|
}
|
|
|
|
options, err := flags.BuildChatOptions()
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, options)
|
|
assert.Equal(t, 80, options.ImageCompression)
|
|
})
|
|
|
|
t.Run("Image parameters without image file should fail in BuildChatOptions", func(t *testing.T) {
|
|
flags := &Flags{
|
|
ImageSize: "1024x1024", // Image parameter without ImageFile
|
|
}
|
|
|
|
options, err := flags.BuildChatOptions()
|
|
assert.Error(t, err)
|
|
assert.Nil(t, options)
|
|
assert.Contains(t, err.Error(), "image parameters")
|
|
assert.Contains(t, err.Error(), "can only be used with --image-file")
|
|
})
|
|
}
|