62 lines
1.8 KiB
Go
62 lines
1.8 KiB
Go
// Package testcontract 在回归中校验实际报文与公开 Schema，服务二进制不依赖此包。
|
||
package testcontract
|
||
|
||
import (
|
||
"strings"
|
||
"sync"
|
||
"testing"
|
||
|
||
"git.yuntuo.hk/k2safe/orbit/protocol/workerv1"
|
||
"github.com/santhosh-tekuri/jsonschema/v6"
|
||
)
|
||
|
||
var workerOnce sync.Once
|
||
var workerSchemas map[string]*jsonschema.Schema
|
||
var workerError error
|
||
|
||
// WorkerSchema 编译并缓存两个方向的 Schema；开启 date-time 等格式断言。
|
||
func WorkerSchema(t testing.TB, direction string) *jsonschema.Schema {
|
||
t.Helper()
|
||
workerOnce.Do(func() {
|
||
compiler := jsonschema.NewCompiler()
|
||
compiler.AssertFormat()
|
||
doc, err := jsonschema.UnmarshalJSON(strings.NewReader(workerv1.Schema))
|
||
if err != nil {
|
||
workerError = err
|
||
return
|
||
}
|
||
const resource = "https://git.yuntuo.hk/k2safe/orbit/schema/worker.v1.json"
|
||
if err := compiler.AddResource(resource, doc); err != nil {
|
||
workerError = err
|
||
return
|
||
}
|
||
workerSchemas = make(map[string]*jsonschema.Schema)
|
||
for _, side := range []string{"client", "server"} {
|
||
workerSchemas[side], workerError = compiler.Compile(resource + "#/$defs/" + side + "Frame")
|
||
if workerError != nil {
|
||
return
|
||
}
|
||
}
|
||
})
|
||
if workerError != nil {
|
||
t.Fatal(workerError)
|
||
}
|
||
schema := workerSchemas[direction]
|
||
if schema == nil {
|
||
t.Fatal("unknown Worker protocol direction")
|
||
}
|
||
return schema
|
||
}
|
||
|
||
// WorkerFrame 验证真实传输的 JSON，不在错误中输出可能包含秘密的报文正文。
|
||
func WorkerFrame(t testing.TB, direction string, raw []byte) {
|
||
t.Helper()
|
||
value, err := jsonschema.UnmarshalJSON(strings.NewReader(string(raw)))
|
||
if err != nil {
|
||
t.Fatal("Worker frame is not valid JSON")
|
||
}
|
||
if WorkerSchema(t, direction).Validate(value) != nil {
|
||
t.Fatalf("%s Worker frame violates published Schema", direction)
|
||
}
|
||
}
|