15 lines
385 B
Go
15 lines
385 B
Go
package ws
|
|
|
|
import (
|
|
"io"
|
|
)
|
|
|
|
// Connection is an internal interface for websocket.Conn in order to mock it for unit tests.
|
|
type Connection interface {
|
|
WriteMessage(messageType int, data []byte) error
|
|
Close() error
|
|
NextReader() (messageType int, r io.Reader, err error)
|
|
CloseHandler() func(code int, text string) error
|
|
SetCloseHandler(handler func(code int, text string) error)
|
|
}
|