func testCleanup(t *testing.T, g *graph.Graph, cmds []string, ints []string) {
// cleanup side on the test
testPassed := false
onChange := func(ws *websocket.Conn) {
g.Lock()
defer g.Unlock()
if !testPassed {
clean := true
for _, intf := range ints {
n := g.LookupFirstNode(graph.Metadatas{"Name": intf})
if n != nil {
clean = false
break
}
}
if clean {
testPassed = true
ws.Close()
}
}
}
testTopology(t, g, cmds, onChange)
if !testPassed {
t.Error("test not executed or failed")
}
}
func processGraphMessage(g *graph.Graph, m []byte) error {
g.Lock()
defer g.Unlock()
msg, err := shttp.UnmarshalWSMessage(m)
if err != nil {
return err
}
if msg.Namespace != "Graph" {
return nil
}
msg, err = graph.UnmarshalWSMessage(msg)
if err != nil {
return err
}
switch msg.Type {
case "NodeUpdated":
n := msg.Obj.(*graph.Node)
node := g.GetNode(n.ID)
if node != nil {
g.SetMetadata(node, n.Metadata())
}
case "NodeDeleted":
g.DelNode(msg.Obj.(*graph.Node))
case "NodeAdded":
n := msg.Obj.(*graph.Node)
if g.GetNode(n.ID) == nil {
g.AddNode(n)
}
case "EdgeUpdated":
e := msg.Obj.(*graph.Edge)
edge := g.GetEdge(e.ID)
if edge != nil {
g.SetMetadata(edge, e.Metadata())
}
case "EdgeDeleted":
g.DelEdge(msg.Obj.(*graph.Edge))
case "EdgeAdded":
e := msg.Obj.(*graph.Edge)
if g.GetEdge(e.ID) == nil {
g.AddEdge(e)
}
}
return nil
}
