func startTopologyClient(t *testing.T, g *graph.Graph, onReady func(*websocket.Conn), onChange func(*websocket.Conn)) error { // ready when got a first ping ws, err := connectToAgent(5, onReady) if err != nil { return err } for { _, m, err := ws.ReadMessage() if err != nil { break } err = processGraphMessage(g, m) if err != nil { return err } logging.GetLogger().Debug("%s", string(m)) logging.GetLogger().Debug("%s", g.String()) onChange(ws) } return nil }