mirror of
https://github.com/danielmiessler/Fabric.git
synced 2026-02-12 23:15:05 -05:00
ce59999503
CHANGES - Update PrintWithVendor signature to accept default vendor and model - Mark default vendor/model with asterisk in non-shell output - Compare vendor and model case-insensitively when marking - Pass registry defaults to PrintWithVendor from CLI - Add test ensuring default selection appears with asterisk - Keep shell completion output unchanged without default markers
57 lines
1.4 KiB
Go
57 lines
1.4 KiB
Go
package ai
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestNewVendorsModels(t *testing.T) {
|
|
vendors := NewVendorsModels()
|
|
if vendors == nil {
|
|
t.Fatalf("NewVendorsModels() returned nil")
|
|
}
|
|
if len(vendors.GroupsItems) != 0 {
|
|
t.Fatalf("NewVendorsModels() returned non-empty VendorsModels map")
|
|
}
|
|
}
|
|
|
|
func TestFindVendorsByModelFirst(t *testing.T) {
|
|
vendors := NewVendorsModels()
|
|
vendors.AddGroupItems("vendor1", []string{"model1", "model2"}...)
|
|
vendor := vendors.FindGroupsByItemFirst("model1")
|
|
if vendor != "vendor1" {
|
|
t.Fatalf("FindVendorsByModelFirst() = %v, want %v", vendor, "vendor1")
|
|
}
|
|
}
|
|
|
|
func TestFindVendorsByModel(t *testing.T) {
|
|
vendors := NewVendorsModels()
|
|
vendors.AddGroupItems("vendor1", []string{"model1", "model2"}...)
|
|
foundVendors := vendors.FindGroupsByItem("model1")
|
|
if len(foundVendors) != 1 || foundVendors[0] != "vendor1" {
|
|
t.Fatalf("FindVendorsByModel() = %v, want %v", foundVendors, []string{"vendor1"})
|
|
}
|
|
}
|
|
|
|
func TestPrintWithVendorMarksDefault(t *testing.T) {
|
|
vendors := NewVendorsModels()
|
|
vendors.AddGroupItems("vendor1", []string{"model1"}...)
|
|
vendors.AddGroupItems("vendor2", []string{"model2"}...)
|
|
|
|
r, w, _ := os.Pipe()
|
|
oldStdout := os.Stdout
|
|
os.Stdout = w
|
|
|
|
vendors.PrintWithVendor(false, "vendor2", "model2")
|
|
|
|
w.Close()
|
|
os.Stdout = oldStdout
|
|
out, _ := io.ReadAll(r)
|
|
|
|
if !strings.Contains(string(out), " *\t[2]\tvendor2|model2") {
|
|
t.Fatalf("default model not marked: %s", out)
|
|
}
|
|
}
|