func TestFlowTable_jsonFlowConversationEthernetPath(t *testing.T) {
ft := flow.NewTestFlowTableComplex(t, nil, nil)
fa := &FlowApi{
FlowTable: ft,
}
statStr := fa.jsonFlowConversationEthernetPath(flow.FlowProtocol_ETHERNET)
if statStr == `{"nodes":[],"links":[]}` {
t.Error("stat should not be empty")
}
var decoded interface{}
if err := json.Unmarshal([]byte(statStr), &decoded); err != nil {
t.Error("JSON parsing failed:", err)
}
schema := v.Object(
v.ObjKV("nodes", v.Array(v.ArrEach(v.Object(
v.ObjKV("name", v.String(v.StrMin(1))),
v.ObjKV("group", v.Number(v.NumMin(0), v.NumMax(20))),
)))),
v.ObjKV("links", v.Array(v.ArrEach(v.Object(
v.ObjKV("source", v.Number(v.NumMin(0), v.NumMax(20))),
v.ObjKV("target", v.Number(v.NumMin(0), v.NumMax(20))),
v.ObjKV("value", v.Number(v.NumMin(0), v.NumMax(9999))),
)))),
)
if path, err := schema.Validate(decoded); err != nil {
t.Errorf("Failed (%s). Path: %s", err, path)
}
}
func TestNumber(t *testing.T) {
var np *int
var nnp int = 3
test(t, "basic number", true, v.Number(), 3)
test(t, "non-number", false, v.Number(), "")
test(t, "nil", false, v.Number(), nil)
test(t, "nil int pointer", false, v.Number(), np)
test(t, "int pointer", true, v.Number(), &nnp)
test(t, "numis", true, v.Number(v.NumIs(3)), 3)
test(t, "!numis", false, v.Number(v.NumIs(3)), 2)
test(t, "minlen1", false, v.Number(v.NumMin(3)), 2)
test(t, "minlen2", true, v.Number(v.NumMin(3)), 3)
test(t, "minlen2", true, v.Number(v.NumMin(3)), 4)
test(t, "maxlen1", true, v.Number(v.NumMax(3)), 2)
test(t, "maxlen2", true, v.Number(v.NumMax(3)), 3)
test(t, "maxlen3", false, v.Number(v.NumMax(3)), 4)
test(t, "combination1", false, v.Number(v.NumMin(3), v.NumMax(3)), 2)
test(t, "combination2", true, v.Number(v.NumMin(3), v.NumMax(3)), 3)
test(t, "combination3", false, v.Number(v.NumMin(3), v.NumMax(3)), 4)
}
func test_jsonFlowDiscovery(t *testing.T, DiscoType discoType) {
ft := flow.NewTestFlowTableComplex(t, nil, nil)
fa := &FlowApi{
FlowTable: ft,
}
disco := fa.jsonFlowDiscovery(DiscoType)
if disco == `{"name":"root","children":[]}` {
t.Error("disco should not be empty")
}
var decoded interface{}
if err := json.Unmarshal([]byte(disco), &decoded); err != nil {
t.Error("JSON parsing failed:", err)
}
schema := v.Object(
v.ObjKV("name", v.String(v.StrMin(1))), // root
v.ObjKV("children", v.Array(v.ArrEach(v.Object(
v.ObjKV("name", v.String(v.StrMin(1))), // Ethernet
v.ObjKV("children", v.Array(v.ArrEach(v.Object(
v.ObjKV("name", v.String(v.StrMin(1))), // IPv4
v.ObjKV("children", v.Array(v.ArrEach(v.Object(
v.ObjKV("name", v.String(v.StrMin(1))), // TCP or UDP
v.ObjKV("children", v.Array(v.ArrEach(v.Object(
v.ObjKV("name", v.String(v.StrMin(1))), // Payload
v.ObjKV("size", v.Number(v.NumMin(0), v.NumMax(9999))),
)))),
)))),
)))),
)))),
)
if path, err := schema.Validate(decoded); err != nil {
t.Errorf("Failed (%s). Path: %s", err, path)
}
}
func init() {
// set up raw json data
rawJson := []byte(`
{
"status": true,
"data": {
"token": "CAFED00D",
"debug": 69306,
"items": [
{ "url": "https://news.ycombinator.com/", "comment": "clickbaits" },
{ "url": "http://golang.org/", "comment": "some darn gophers" },
{ "url": "http://www.kickstarter.com/", "comment": "opensource projects. yeah.." }
],
"ghost2": null,
"meta": {
"g": 1,
"xyzzy": 0.25,
"blöö": 0.5
}
}
}`)
// decode json
if err := json.Unmarshal(rawJson, &decoded); err != nil {
log.Panic("JSON parsing failed. Err =", err)
}
// set up a custom validator function
myValidatorFunc := func(data interface{}) (path string, err error) {
path = "myValidatorFunc"
validate, ok := data.(string)
if !ok {
return path, fmt.Errorf("expected string, got %v", reflect.TypeOf(data))
}
if validate != "CAFED00D" {
return path, fmt.Errorf("expected CAFED00D, got %s", validate)
}
return "", nil
}
// construct the schema which is used to validate data
schema = v.Object(
v.ObjKV("status", v.Boolean()),
v.ObjKV("data", v.Object(
v.ObjKV("token", v.Function(myValidatorFunc)),
v.ObjKV("debug", v.Number(v.NumMin(1), v.NumMax(99999))),
v.ObjKV("items", v.Array(v.ArrEach(v.Object(
v.ObjKV("url", v.String(v.StrMin(1))),
v.ObjKV("comment", v.Optional(v.String())),
)))),
v.ObjKV("ghost", v.Optional(v.String())),
v.ObjKV("ghost2", v.Optional(v.String())),
v.ObjKV("meta", v.Object(
v.ObjKeys(v.String()),
v.ObjValues(v.Or(v.Number(v.NumMin(.01), v.NumMax(1.1)), v.String())),
)),
)),
)
}