func (m *mockServer) readMessage(conn *websocket.Conn) *messages.Client { _, data, err := conn.ReadMessage() if err != nil { return nil } var msg messages.Client err = msg.Unmarshal(data) Expect(err).ToNot(HaveOccurred()) return &msg }