mlogclub
119 lines
4.2 KiB
Go
119 lines
4.2 KiB
Go
package registry
|
|
|
|
import "testing"
|
|
|
|
func TestDefaultRegistryVariablesHaveBusinessLabels(t *testing.T) {
|
|
for _, spec := range DefaultRegistry().List() {
|
|
for _, variable := range append(spec.InputSchema, spec.OutputSchema...) {
|
|
if variable.Label == "" {
|
|
t.Fatalf("%s.%s is missing business label", spec.Type, variable.Name)
|
|
}
|
|
if variable.Description == "" || variable.Description == variable.Name {
|
|
t.Fatalf("%s.%s is missing readable description", spec.Type, variable.Name)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDefaultRegistryExposesStartOutputs(t *testing.T) {
|
|
spec, ok := DefaultRegistry().Get(NodeTypeStart)
|
|
if !ok {
|
|
t.Fatalf("start node spec not found")
|
|
}
|
|
if !hasVariable(spec.OutputSchema, "userMessage", VariableTypeString) {
|
|
t.Fatalf("expected start output userMessage:string, got %#v", spec.OutputSchema)
|
|
}
|
|
if !hasVariable(spec.OutputSchema, "knowledgeBaseIds", VariableTypeIntegerArray) {
|
|
t.Fatalf("expected start output knowledgeBaseIds:array<int>, got %#v", spec.OutputSchema)
|
|
}
|
|
}
|
|
|
|
func TestDefaultRegistryExposesKnowledgeRetrieveInputsAndOutputs(t *testing.T) {
|
|
spec, ok := DefaultRegistry().Get(NodeTypeKnowledgeRetrieve)
|
|
if !ok {
|
|
t.Fatalf("knowledge_retrieve node spec not found")
|
|
}
|
|
if !hasRequiredVariable(spec.InputSchema, "query", VariableTypeString) {
|
|
t.Fatalf("expected knowledge_retrieve required input query:string, got %#v", spec.InputSchema)
|
|
}
|
|
if !hasVariable(spec.OutputSchema, "items", VariableTypeObjectArray) {
|
|
t.Fatalf("expected knowledge_retrieve output items:array<object>, got %#v", spec.OutputSchema)
|
|
}
|
|
}
|
|
|
|
func TestDefaultRegistryExposesSendReplyRequiredInput(t *testing.T) {
|
|
spec, ok := DefaultRegistry().Get(NodeTypeSendReply)
|
|
if !ok {
|
|
t.Fatalf("send_reply node spec not found")
|
|
}
|
|
if !hasRequiredVariable(spec.InputSchema, "replyText", VariableTypeString) {
|
|
t.Fatalf("expected send_reply required input replyText:string, got %#v", spec.InputSchema)
|
|
}
|
|
if !hasVariable(spec.OutputSchema, "sent", VariableTypeBoolean) {
|
|
t.Fatalf("expected send_reply output sent:boolean, got %#v", spec.OutputSchema)
|
|
}
|
|
}
|
|
|
|
func TestDefaultRegistryExposesConversationUnderstandingOutputs(t *testing.T) {
|
|
spec, ok := DefaultRegistry().Get(NodeTypeConversationUnderstanding)
|
|
if !ok {
|
|
t.Fatalf("conversation_understanding node spec not found")
|
|
}
|
|
if !hasRequiredVariable(spec.InputSchema, "userMessage", VariableTypeString) {
|
|
t.Fatalf("expected conversation_understanding required input userMessage:string, got %#v", spec.InputSchema)
|
|
}
|
|
for _, want := range []string{"messageIntent", "answerScope", "riskSignals", "reason"} {
|
|
if !hasVariableName(spec.OutputSchema, want) {
|
|
t.Fatalf("expected conversation_understanding output %s, got %#v", want, spec.OutputSchema)
|
|
}
|
|
}
|
|
if !hasVariable(spec.OutputSchema, "confidence", VariableTypeNumber) {
|
|
t.Fatalf("expected conversation_understanding output confidence:number, got %#v", spec.OutputSchema)
|
|
}
|
|
}
|
|
|
|
func TestDefaultRegistryExposesReplyPolicyOutputs(t *testing.T) {
|
|
spec, ok := DefaultRegistry().Get(NodeTypeReplyPolicy)
|
|
if !ok {
|
|
t.Fatalf("reply_policy node spec not found")
|
|
}
|
|
if !hasRequiredVariable(spec.InputSchema, "messageIntent", VariableTypeString) {
|
|
t.Fatalf("expected reply_policy required input messageIntent:string, got %#v", spec.InputSchema)
|
|
}
|
|
if !hasRequiredVariable(spec.InputSchema, "answerScope", VariableTypeString) {
|
|
t.Fatalf("expected reply_policy required input answerScope:string, got %#v", spec.InputSchema)
|
|
}
|
|
for _, want := range []string{"action", "replyText", "reason", "finalReplySource"} {
|
|
if !hasVariableName(spec.OutputSchema, want) {
|
|
t.Fatalf("expected reply_policy output %s, got %#v", want, spec.OutputSchema)
|
|
}
|
|
}
|
|
}
|
|
|
|
func hasRequiredVariable(items []VariableSpec, name string, variableType VariableType) bool {
|
|
for _, item := range items {
|
|
if item.Name == name && item.Type == variableType && item.Required {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func hasVariableName(items []VariableSpec, name string) bool {
|
|
for _, item := range items {
|
|
if item.Name == name {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func hasVariable(items []VariableSpec, name string, variableType VariableType) bool {
|
|
for _, item := range items {
|
|
if item.Name == name && item.Type == variableType {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|