61 lines
2.0 KiB
Go
61 lines
2.0 KiB
Go
|
package registry
|
||
|
|
|
||
|
|
import "testing"
|
||
|
|
|
||
|
|
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 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 hasVariable(items []VariableSpec, name string, variableType VariableType) bool {
|
||
|
|
for _, item := range items {
|
||
|
|
if item.Name == name && item.Type == variableType {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return false
|
||
|
|
}
|