c7606f3d5f
After we introduced the linter we haven't really touched the old code. This commit now fixes all linting issue that exist right now.
92 lines
3.0 KiB
Go
92 lines
3.0 KiB
Go
package auth
|
|
|
|
import (
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/sirupsen/logrus/hooks/test"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/suite"
|
|
"gitlab.hpi.de/codeocean/codemoon/poseidon/config"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
)
|
|
|
|
const testToken = "C0rr3ctT0k3n"
|
|
|
|
type AuthenticationMiddlewareTestSuite struct {
|
|
suite.Suite
|
|
request *http.Request
|
|
recorder *httptest.ResponseRecorder
|
|
httpAuthenticationMiddleware http.Handler
|
|
}
|
|
|
|
func (s *AuthenticationMiddlewareTestSuite) SetupTest() {
|
|
correctAuthenticationToken = []byte(testToken)
|
|
s.recorder = httptest.NewRecorder()
|
|
request, err := http.NewRequest(http.MethodGet, "/api/v1/test", nil)
|
|
if err != nil {
|
|
s.T().Fatal(err)
|
|
}
|
|
s.request = request
|
|
s.httpAuthenticationMiddleware = HTTPAuthenticationMiddleware(
|
|
http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusOK)
|
|
}))
|
|
}
|
|
|
|
func (s *AuthenticationMiddlewareTestSuite) TearDownTest() {
|
|
correctAuthenticationToken = []byte(nil)
|
|
}
|
|
|
|
func (s *AuthenticationMiddlewareTestSuite) TestReturns401WhenHeaderUnset() {
|
|
s.httpAuthenticationMiddleware.ServeHTTP(s.recorder, s.request)
|
|
assert.Equal(s.T(), http.StatusUnauthorized, s.recorder.Code)
|
|
}
|
|
|
|
func (s *AuthenticationMiddlewareTestSuite) TestReturns401WhenTokenWrong() {
|
|
s.request.Header.Set(TokenHeader, "Wr0ngT0k3n")
|
|
s.httpAuthenticationMiddleware.ServeHTTP(s.recorder, s.request)
|
|
assert.Equal(s.T(), http.StatusUnauthorized, s.recorder.Code)
|
|
}
|
|
|
|
func (s *AuthenticationMiddlewareTestSuite) TestWarnsWhenUnauthorized() {
|
|
var hook *test.Hook
|
|
logger, hook := test.NewNullLogger()
|
|
log = logger.WithField("pkg", "api/auth")
|
|
|
|
s.request.Header.Set(TokenHeader, "Wr0ngT0k3n")
|
|
s.httpAuthenticationMiddleware.ServeHTTP(s.recorder, s.request)
|
|
|
|
assert.Equal(s.T(), http.StatusUnauthorized, s.recorder.Code)
|
|
assert.Equal(s.T(), logrus.WarnLevel, hook.LastEntry().Level)
|
|
assert.Equal(s.T(), hook.LastEntry().Data["token"], "Wr0ngT0k3n")
|
|
}
|
|
|
|
func (s *AuthenticationMiddlewareTestSuite) TestPassesWhenTokenCorrect() {
|
|
s.request.Header.Set(TokenHeader, testToken)
|
|
s.httpAuthenticationMiddleware.ServeHTTP(s.recorder, s.request)
|
|
|
|
assert.Equal(s.T(), http.StatusOK, s.recorder.Code)
|
|
}
|
|
|
|
func TestHTTPAuthenticationMiddleware(t *testing.T) {
|
|
suite.Run(t, new(AuthenticationMiddlewareTestSuite))
|
|
}
|
|
|
|
func TestInitializeAuthentication(t *testing.T) {
|
|
t.Run("if token unset", func(t *testing.T) {
|
|
config.Config.Server.Token = ""
|
|
initialized := InitializeAuthentication()
|
|
assert.Equal(t, false, initialized)
|
|
assert.Equal(t, []byte(nil), correctAuthenticationToken, "it should not set correctAuthenticationToken")
|
|
})
|
|
t.Run("if token set", func(t *testing.T) {
|
|
config.Config.Server.Token = testToken
|
|
initialized := InitializeAuthentication()
|
|
assert.Equal(t, true, initialized)
|
|
assert.Equal(t, []byte(testToken), correctAuthenticationToken, "it should set correctAuthenticationToken")
|
|
config.Config.Server.Token = ""
|
|
correctAuthenticationToken = []byte(nil)
|
|
})
|
|
}
|