mirror of https://github.com/authelia/authelia.git
97 lines
2.4 KiB
Go
97 lines
2.4 KiB
Go
package handlers
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/stretchr/testify/suite"
|
|
"github.com/valyala/fasthttp"
|
|
|
|
"github.com/authelia/authelia/v4/internal/authentication"
|
|
"github.com/authelia/authelia/v4/internal/mocks"
|
|
)
|
|
|
|
type StateGetSuite struct {
|
|
suite.Suite
|
|
|
|
mock *mocks.MockAutheliaCtx
|
|
}
|
|
|
|
func (s *StateGetSuite) SetupTest() {
|
|
s.mock = mocks.NewMockAutheliaCtx(s.T())
|
|
}
|
|
|
|
func (s *StateGetSuite) TearDownTest() {
|
|
s.mock.Close()
|
|
}
|
|
|
|
func (s *StateGetSuite) TestShouldReturnUsernameFromSession() {
|
|
userSession, err := s.mock.Ctx.GetSession()
|
|
s.Assert().NoError(err)
|
|
|
|
userSession.Username = "username"
|
|
s.Assert().NoError(s.mock.Ctx.SaveSession(userSession))
|
|
|
|
StateGET(s.mock.Ctx)
|
|
|
|
type Response struct {
|
|
Status string
|
|
Data StateResponse
|
|
}
|
|
|
|
expectedBody := Response{
|
|
Status: "OK",
|
|
Data: StateResponse{
|
|
Username: "username",
|
|
DefaultRedirectionURL: "https://www.example.com",
|
|
AuthenticationLevel: authentication.NotAuthenticated,
|
|
},
|
|
}
|
|
actualBody := Response{}
|
|
|
|
err = json.Unmarshal(s.mock.Ctx.Response.Body(), &actualBody)
|
|
require.NoError(s.T(), err)
|
|
assert.Equal(s.T(), fasthttp.StatusOK, s.mock.Ctx.Response.StatusCode())
|
|
assert.Equal(s.T(), []byte("application/json; charset=utf-8"), s.mock.Ctx.Response.Header.ContentType())
|
|
assert.Equal(s.T(), expectedBody, actualBody)
|
|
}
|
|
|
|
func (s *StateGetSuite) TestShouldReturnAuthenticationLevelFromSession() {
|
|
userSession, err := s.mock.Ctx.GetSession()
|
|
s.Assert().NoError(err)
|
|
|
|
userSession.AuthenticationLevel = authentication.OneFactor
|
|
s.Assert().NoError(s.mock.Ctx.SaveSession(userSession))
|
|
require.NoError(s.T(), err)
|
|
|
|
StateGET(s.mock.Ctx)
|
|
|
|
type Response struct {
|
|
Status string
|
|
Data StateResponse
|
|
}
|
|
|
|
expectedBody := Response{
|
|
Status: "OK",
|
|
Data: StateResponse{
|
|
Username: "",
|
|
DefaultRedirectionURL: "https://www.example.com",
|
|
AuthenticationLevel: authentication.OneFactor,
|
|
},
|
|
}
|
|
actualBody := Response{}
|
|
|
|
err = json.Unmarshal(s.mock.Ctx.Response.Body(), &actualBody)
|
|
require.NoError(s.T(), err)
|
|
assert.Equal(s.T(), fasthttp.StatusOK, s.mock.Ctx.Response.StatusCode())
|
|
assert.Equal(s.T(), []byte("application/json; charset=utf-8"), s.mock.Ctx.Response.Header.ContentType())
|
|
assert.Equal(s.T(), expectedBody, actualBody)
|
|
}
|
|
|
|
func TestRunStateGetSuite(t *testing.T) {
|
|
s := new(StateGetSuite)
|
|
suite.Run(t, s)
|
|
}
|