agent/wsclient/mock/client.go
func (m *MockClientServer) WriteMessage(arg0 []byte) error {
	ret := m.ctrl.Call(m, "WriteMessage", arg0)
	ret0, _ := ret[0].(error)
	return ret0
}
func (m *MockClientServer) WriteMessage(arg0 []byte) error {
	ret := m.ctrl.Call(m, "WriteMessage", arg0)
	ret0, _ := ret[0].(error)
	return ret0
}
// WriteMessage mocks base method