diff --git a/systemtest/tests/auth.go b/systemtest/tests/auth.go index 25039dd..7d9bf24 100644 --- a/systemtest/tests/auth.go +++ b/systemtest/tests/auth.go @@ -9,6 +9,7 @@ import ( "github.com/EternisAI/silo-proxy/internal/api/http/dto" "github.com/EternisAI/silo-proxy/internal/auth" + "github.com/EternisAI/silo-proxy/internal/db/sqlc" "github.com/gin-gonic/gin" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" @@ -24,7 +25,7 @@ func TestRegister(t *testing.T, router *gin.Engine, jwtSecret string) { var resp dto.RegisterResponse require.NoError(t, json.Unmarshal(rr.Body.Bytes(), &resp)) assert.Equal(t, "testuser", resp.Username) - assert.Equal(t, "Student", resp.Role) + assert.Equal(t, sqlc.UserRoleUser, resp.Role) assert.NotEmpty(t, resp.ID) }) @@ -69,7 +70,7 @@ func TestLogin(t *testing.T, router *gin.Engine, jwtSecret string) { claims, err := auth.ValidateToken(jwtSecret, resp.Token) require.NoError(t, err) assert.Equal(t, "loginuser", claims.Username) - assert.Equal(t, "Student", claims.Role) + assert.Equal(t, sqlc.UserRoleUser, claims.Role) }) t.Run("wrong password", func(t *testing.T) {