websocket

type Connection struct {
	// contains filtered or unexported fields
}

type ConnectionInterface interface {
	WriteText(body []byte) error
	WriteBinary(body []byte) error
	WriteJSON(body interface{}) error
	Close(status websocket.StatusCode, reason string)
}

type Message struct {
	Type websocket.MessageType `json:"type"`
	Body string                `json:"body"`
}

type MockConnection struct {
	WriteJSONCalled bool
	ReceivedMessage interface{}
	// contains filtered or unexported fields
}

type MockPool struct {
	RegisterCalled      bool
	UnregisterCalled    bool
	BroadcastJSONCalled bool
	BroadcastMsg        interface{}
}

type MockWebsocketConn struct {
	WriteJSONCalled bool
	WriteJSONErr    error
	CloseCalled     bool
}

type Pool struct {
	Register      chan ConnectionInterface
	Unregister    chan ConnectionInterface
	Connections   map[ConnectionInterface]bool
	BroadcastJSON chan interface{}
}

type PoolInterface interface {
	Start()
	RegisterConnection(ConnectionInterface)
	UnregisterConnection(ConnectionInterface)
	BroadcastJSONMessage(interface{})
}

type WebsocketConnWrapperInterface interface {
	Subprotocol() string
	Ping(ctx context.Context) error
	Read(ctx context.Context) (websocket.MessageType, []byte, error)
	Write(ctx context.Context, messageType websocket.MessageType, payload []byte) error
	Close(status websocket.StatusCode, reason string) error
	WriteJSON(ctx context.Context, body interface{}) error
}