24 lines
508 B
Go
24 lines
508 B
Go
package api
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestWebsocket(t *testing.T) {
|
|
t.Run("bad request", func(t *testing.T) {
|
|
app, router, _ := NewApiTest()
|
|
WebSocket(router)
|
|
r := PerformRequest(app, "GET", "/api/v1/ws")
|
|
assert.Equal(t, http.StatusBadRequest, r.Code)
|
|
})
|
|
|
|
t.Run("router nil", func(t *testing.T) {
|
|
app, _, _ := NewApiTest()
|
|
WebSocket(nil)
|
|
r := PerformRequest(app, "GET", "/api/v1/ws")
|
|
assert.Equal(t, http.StatusNotFound, r.Code)
|
|
})
|
|
}
|