mirror of
https://github.com/danielmiessler/Fabric.git
synced 2026-02-13 15:34:59 -05:00
34 lines
937 B
Go
34 lines
937 B
Go
package ai
|
|
|
|
import (
|
|
"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"})
|
|
}
|
|
}
|