You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
1.4 KiB
Go
66 lines
1.4 KiB
Go
package api
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
const (
|
|
token = "123123123"
|
|
)
|
|
|
|
func TestAPI(t *testing.T) {
|
|
RegisterFailHandler(Fail)
|
|
RunSpecs(t, "API Suite")
|
|
}
|
|
|
|
var _ = Describe("API", func() {
|
|
api := New(token)
|
|
|
|
Describe("RequireToken middleware", func() {
|
|
It("should return 401 Unauthorized when token is not provided", func() {
|
|
handlerFunc := api.RequireToken(testHandler)
|
|
|
|
rec := httptest.NewRecorder()
|
|
req := httptest.NewRequest("GET", "/hello", nil)
|
|
|
|
handlerFunc(rec, req)
|
|
|
|
Expect(rec.Code).To(Equal(http.StatusUnauthorized))
|
|
})
|
|
|
|
It("should return 401 Unauthorized when token is invalid", func() {
|
|
handlerFunc := api.RequireToken(testHandler)
|
|
|
|
rec := httptest.NewRecorder()
|
|
req := httptest.NewRequest("GET", "/hello", nil)
|
|
req.Header.Set("Authorization", "Bearer 123")
|
|
|
|
handlerFunc(rec, req)
|
|
|
|
Expect(rec.Code).To(Equal(http.StatusUnauthorized))
|
|
})
|
|
|
|
It("should return 200 OK when token is valid", func() {
|
|
handlerFunc := api.RequireToken(testHandler)
|
|
|
|
rec := httptest.NewRecorder()
|
|
req := httptest.NewRequest("GET", "/hello", nil)
|
|
req.Header.Set("Authorization", "Bearer " + token)
|
|
|
|
handlerFunc(rec, req)
|
|
|
|
Expect(rec.Code).To(Equal(http.StatusOK))
|
|
})
|
|
})
|
|
})
|
|
|
|
func testHandler(w http.ResponseWriter, req *http.Request) {
|
|
_, _ = io.WriteString(w, "Hello!")
|
|
}
|