toki
c46874055a
- Add edge node unit tests and transport package - Add node test client and related artifacts - Update bootstrap, node, and config modules - Add proto generated files - Update Makefile and configuration files
65 lines
1.6 KiB
Go
65 lines
1.6 KiB
Go
package transport
|
|
|
|
import (
|
|
"testing"
|
|
|
|
toki "git.toki-labs.com/toki/common-proto-socket/go"
|
|
"google.golang.org/protobuf/proto"
|
|
|
|
iop "iop/proto/gen/iop"
|
|
)
|
|
|
|
func TestNodeParserMap_RunRequest(t *testing.T) {
|
|
parsers := nodeParserMap()
|
|
original := &iop.RunRequest{RunId: "run-1", Adapter: "mock", Model: "v1"}
|
|
payload, err := proto.Marshal(original)
|
|
if err != nil {
|
|
t.Fatalf("marshal: %v", err)
|
|
}
|
|
|
|
parsed, err := parsers[toki.TypeNameOf(original)](payload)
|
|
if err != nil {
|
|
t.Fatalf("parse: %v", err)
|
|
}
|
|
got := parsed.(*iop.RunRequest)
|
|
if got.GetRunId() != original.GetRunId() ||
|
|
got.GetAdapter() != original.GetAdapter() ||
|
|
got.GetModel() != original.GetModel() {
|
|
t.Fatalf("unexpected run request: %+v", got)
|
|
}
|
|
}
|
|
|
|
func TestNodeParserMap_CancelRequest(t *testing.T) {
|
|
parsers := nodeParserMap()
|
|
original := &iop.CancelRequest{RunId: "run-1"}
|
|
payload, err := proto.Marshal(original)
|
|
if err != nil {
|
|
t.Fatalf("marshal: %v", err)
|
|
}
|
|
|
|
parsed, err := parsers[toki.TypeNameOf(original)](payload)
|
|
if err != nil {
|
|
t.Fatalf("parse: %v", err)
|
|
}
|
|
got := parsed.(*iop.CancelRequest)
|
|
if got.GetRunId() != original.GetRunId() {
|
|
t.Fatalf("unexpected cancel request: %+v", got)
|
|
}
|
|
}
|
|
|
|
func TestNodeParserMap_CapabilityRequest(t *testing.T) {
|
|
parsers := nodeParserMap()
|
|
original := &iop.CapabilityRequest{}
|
|
payload, err := proto.Marshal(original)
|
|
if err != nil {
|
|
t.Fatalf("marshal: %v", err)
|
|
}
|
|
|
|
parsed, err := parsers[toki.TypeNameOf(original)](payload)
|
|
if err != nil {
|
|
t.Fatalf("parse: %v", err)
|
|
}
|
|
if _, ok := parsed.(*iop.CapabilityRequest); !ok {
|
|
t.Fatalf("unexpected capability request type: %T", parsed)
|
|
}
|
|
}
|