322 lines
10 KiB
Go
322 lines
10 KiB
Go
package learsteam_quiz_test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
config "learnsteam/cslms-api/configs"
|
|
"learnsteam/cslms-api/internal/controllers"
|
|
"learnsteam/cslms-api/internal/repositories"
|
|
"learnsteam/cslms-api/internal/services"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/golang-jwt/jwt/v5"
|
|
"github.com/google/go-cmp/cmp"
|
|
"github.com/stretchr/testify/suite"
|
|
"github.com/tj/assert"
|
|
"gorm.io/datatypes"
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/schema"
|
|
)
|
|
|
|
type TokenTestSuite struct {
|
|
suite.Suite
|
|
db *gorm.DB
|
|
repository repositories.TokenRepository
|
|
service services.TokenService
|
|
controller controllers.TokenController
|
|
}
|
|
|
|
func (suite *TokenTestSuite) SetupSuite() {
|
|
// err := os.Remove("test.db")
|
|
// if err != nil {
|
|
// suite.Fail("Failed to remove the test database file")
|
|
// }
|
|
|
|
gorm_config := gorm.Config{NamingStrategy: schema.NamingStrategy{SingularTable: true}}
|
|
db, _ := gorm.Open(sqlite.Open("test.db"), &gorm_config)
|
|
repository := repositories.NewTokenRepository(db)
|
|
service := services.NewTokenService(repository)
|
|
controller := controllers.NewTokenController(service)
|
|
|
|
suite.db = db
|
|
suite.service = service
|
|
suite.repository = repository
|
|
suite.controller = controller
|
|
|
|
// suite.CreateSampleData()
|
|
}
|
|
|
|
// func (suite *TokenTestSuite) CreateSampleData() {
|
|
// suite.db.AutoMigrate(&models.Token{})
|
|
// var tokens []models.Token
|
|
// for i := 1; i < 100; i++ {
|
|
// expire_at := time.Now().Add(time.Hour * 24 * time.Duration(i+45))
|
|
// tokenString, _ := suite.service.Generate("superrichquiz_octet", expire_at.Unix())
|
|
// token := models.Token{
|
|
// ID: uuid.NewString(),
|
|
// Token: tokenString,
|
|
// Status: "valid",
|
|
// ExpireAt: expire_at,
|
|
// UpdatedAt: time.Now(),
|
|
// CreatedAt: time.Now(),
|
|
// }
|
|
// tokens = append(tokens, token)
|
|
// }
|
|
|
|
// for _, token := range tokens {
|
|
// suite.db.Create(&token)
|
|
// }
|
|
// }
|
|
|
|
func (suite *TokenTestSuite) TearDownSuite() {
|
|
// suite.db.Migrator().DropTable(&models.Token{})
|
|
|
|
// err := os.Remove("test.db")
|
|
// if err != nil {
|
|
// suite.Fail("Failed to remove the test database file")
|
|
// }
|
|
}
|
|
|
|
func (suite *TokenTestSuite) SetupTest() {
|
|
// suite.userRepository.DeleteByName("testUserName0001")
|
|
}
|
|
|
|
func (suite *TokenTestSuite) TearDownTest() {
|
|
|
|
}
|
|
|
|
func TestTokenSuite(t *testing.T) {
|
|
suite.Run(t, new(TokenTestSuite))
|
|
}
|
|
|
|
// 토큰 생성 테스트
|
|
func (suite *TokenTestSuite) TestGenerateTokenSuccess() {
|
|
user_id := int64(100000001)
|
|
role := "admin"
|
|
expire_at := time.Now().Add(time.Hour * 24 * 365).Unix()
|
|
token, err := suite.service.Generate(user_id, expire_at, role)
|
|
assert.NoError(suite.T(), err)
|
|
assert.NotNil(suite.T(), token)
|
|
}
|
|
|
|
// // 토큰 생성 테스트
|
|
// func (suite *TokenTestSuite) TestGenerateTokenString() {
|
|
// user_id := "superrichquiz_octet"
|
|
// expire_at := time.Now().Add(time.Hour * 24 * 365 * 100).Unix()
|
|
// tokenString, err := suite.service.Generate(user_id, expire_at)
|
|
// assert.NoError(suite.T(), err)
|
|
|
|
// jwtToken, err := suite.service.VerifyTokenString(tokenString)
|
|
// assert.NoError(suite.T(), err)
|
|
// assert.NotNil(suite.T(), jwtToken)
|
|
// }
|
|
|
|
// // 토큰 검증 테스트
|
|
// func (suite *TokenTestSuite) TestVerifyTokenString() {
|
|
// user_id := "testuser"
|
|
// expire_at := time.Now().Add(time.Hour * 24 * 365).Unix()
|
|
// tokenString, err := suite.service.Generate(user_id, expire_at)
|
|
// assert.NoError(suite.T(), err)
|
|
|
|
// jwtToken, err := suite.service.VerifyTokenString(tokenString)
|
|
// assert.NoError(suite.T(), err)
|
|
// assert.NotNil(suite.T(), jwtToken)
|
|
// assert.Equal(suite.T(), tokenString, jwtToken.Raw)
|
|
// }
|
|
|
|
// // 토큰 유효성 테스트
|
|
// func (suite *TokenTestSuite) TestValidToken() {
|
|
// user_id := "testuser"
|
|
// expire_at := time.Now().Add(time.Hour * 24 * 365).Unix()
|
|
// tokenString, err := suite.service.Generate(user_id, expire_at)
|
|
// assert.NoError(suite.T(), err)
|
|
|
|
// jwtToken, err := suite.service.VerifyTokenString(tokenString)
|
|
// assert.NoError(suite.T(), err)
|
|
// assert.NotNil(suite.T(), jwtToken)
|
|
|
|
// valid, err := suite.service.ValidToken(jwtToken)
|
|
// assert.NoError(suite.T(), err)
|
|
// assert.True(suite.T(), valid)
|
|
// }
|
|
|
|
// // 토큰 유효성 테스트 : Expire
|
|
// func (suite *TokenTestSuite) TestExpiredToken() {
|
|
// user_id := "testuser"
|
|
// expire_at := time.Now().Add(time.Hour * 1).Unix()
|
|
// tokenString, err := suite.service.Generate(user_id, expire_at)
|
|
// assert.NoError(suite.T(), err)
|
|
|
|
// jwtToken, err := suite.service.GetJwtToken(tokenString)
|
|
// assert.NoError(suite.T(), err)
|
|
// // assert.Nil(suite.T(), jwtToken)
|
|
|
|
// expirationTime, err := jwtToken.Claims.GetExpirationTime()
|
|
// assert.NoError(suite.T(), err)
|
|
// assert.Greater(suite.T(), expirationTime.Time, time.Now())
|
|
|
|
// expire_at = time.Now().Add(-time.Hour * 1).Unix()
|
|
// tokenString, err = suite.service.Generate(user_id, expire_at)
|
|
// assert.NoError(suite.T(), err)
|
|
|
|
// jwtToken, err = suite.service.GetJwtToken(tokenString)
|
|
// assert.Error(suite.T(), err)
|
|
// // assert.Nil(suite.T(), jwtToken)
|
|
|
|
// expirationTime, err = jwtToken.Claims.GetExpirationTime()
|
|
// assert.NoError(suite.T(), err)
|
|
// assert.Greater(suite.T(), time.Now(), expirationTime.Time)
|
|
// }
|
|
|
|
// func (suite *TokenTestSuite) TestCreateToken() {
|
|
// dateString := "2023-08-15T01:50:08.706Z"
|
|
// layout := "2006-01-02T15:04:05.000Z"
|
|
// tokenString := "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ0b2tlbklkeCI6MTc2NTQyLCJ0b2tlblR5cGUiOiJXQUxMRVQiLCJ0b2tlbkV4cGlyZWREYXRlIjoiMjAyMy0wOC0xNVQwMTo1MDowOC43MDZaIn0.skgP6ysLNx6KRDBYZy3miZW1Q95iV_Cw0ZVWXj1tJyw"
|
|
// expire_at, _ := time.Parse(layout, dateString)
|
|
// token := models.Token{
|
|
// ID: uuid.NewString(),
|
|
// Token: tokenString,
|
|
// Status: "valid",
|
|
// ExpireAt: expire_at,
|
|
// }
|
|
// result, err := suite.repository.Create(&token)
|
|
|
|
// assert.NoError(suite.T(), err)
|
|
// assert.Equal(suite.T(), tokenString, result.Token)
|
|
// }
|
|
|
|
// // 리프레시할 토큰 목록
|
|
// func (suite *TokenTestSuite) TestListOfTokensToRefresh() {
|
|
// tokens, err := suite.repository.ListRefresh()
|
|
// assert.NoError(suite.T(), err)
|
|
// assert.NotNil(suite.T(), tokens)
|
|
// assert.Greater(suite.T(), len(*tokens), 0)
|
|
// }
|
|
|
|
// // 토큰 리프레시
|
|
// func (suite *TokenTestSuite) TestOldTokenRefresh() {
|
|
// tokens, err := suite.repository.ListRefresh()
|
|
// assert.NoError(suite.T(), err)
|
|
// assert.NotNil(suite.T(), tokens)
|
|
|
|
// for _, token := range *tokens {
|
|
// token.Status = "expired"
|
|
// result, err := suite.repository.Update(&token)
|
|
// assert.NoError(suite.T(), err)
|
|
// assert.Equal(suite.T(), "expired", result.Status)
|
|
|
|
// user_id := "superrichquiz_octet"
|
|
// expire_at := result.ExpireAt.Add(time.Hour * 24 * 90)
|
|
// tokenString, err := suite.service.Generate(user_id, expire_at.Unix())
|
|
// assert.NoError(suite.T(), err)
|
|
|
|
// refreshedToken := models.Token{
|
|
// ID: uuid.NewString(),
|
|
// Token: tokenString,
|
|
// Status: "valid",
|
|
// ExpireAt: expire_at,
|
|
// }
|
|
|
|
// newToken, err := suite.repository.Create(&refreshedToken)
|
|
// assert.NoError(suite.T(), err)
|
|
// assert.Equal(suite.T(), tokenString, newToken.Token)
|
|
// }
|
|
// }
|
|
|
|
// func (suite *TokenTestSuite) TestResponseToken() {
|
|
// respBody := []byte(`{
|
|
// "token": "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ0b2tlbklkeCI6MTc1MzU4LCJ0b2tlblR5cGUiOiJXQUxMRVQiLCJ0b2tlbkV4cGlyZWREYXRlIjoiMjAyMy0wOC0xNFQwNToxMTo1Ni41MzhaIn0.e1i4_y8ItC8Vje13Ew8NHwZTElOMBObIZjpGLgRCdyE",
|
|
// "tokenBody": {
|
|
// "tokenExpiredDate": "2023-08-13T06:06:12.065Z",
|
|
// "tokenIdx": 174025,
|
|
// "tokenType": "WALLET"
|
|
// }
|
|
// }`)
|
|
|
|
// var response map[string]interface{}
|
|
// err := json.Unmarshal(respBody, &response)
|
|
// assert.NoError(suite.T(), err)
|
|
// assert.NotNil(suite.T(), response)
|
|
// assert.Equal(suite.T(), response["token"], "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ0b2tlbklkeCI6MTc1MzU4LCJ0b2tlblR5cGUiOiJXQUxMRVQiLCJ0b2tlbkV4cGlyZWREYXRlIjoiMjAyMy0wOC0xNFQwNToxMTo1Ni41MzhaIn0.e1i4_y8ItC8Vje13Ew8NHwZTElOMBObIZjpGLgRCdyE")
|
|
|
|
// tokenBody := response["tokenBody"].(map[string]interface{})
|
|
// tokenExpiredDate := tokenBody["tokenExpiredDate"].(string)
|
|
// assert.Equal(suite.T(), "2023-08-13T06:06:12.065Z", tokenExpiredDate)
|
|
// }
|
|
|
|
// func (suite *TokenTestSuite) TestResponseError() {
|
|
// respBody := []byte(`{
|
|
// "errorCode": "ERR_0105001",
|
|
// "message": "Invalid token"
|
|
// }`)
|
|
|
|
// var response map[string]interface{}
|
|
// err := json.Unmarshal(respBody, &response)
|
|
// assert.NoError(suite.T(), err)
|
|
// assert.NotNil(suite.T(), response)
|
|
|
|
// assert.NotNil(suite.T(), response["errorCode"])
|
|
// assert.Equal(suite.T(), "ERR_0105001", response["errorCode"].(string))
|
|
|
|
// if response["errorCode"] != nil && response["errorCode"].(string) == "ERR_0105001" {
|
|
// assert.True(suite.T(), true)
|
|
// }
|
|
// }
|
|
|
|
func (suite *TokenTestSuite) TestVerifyValidToken() {
|
|
expire_at := time.Now().Add(time.Hour * 24 * 365 * 100).Unix()
|
|
validToken, err := suite.service.Generate(10001, expire_at, "admin")
|
|
assert.NoError(suite.T(), err)
|
|
|
|
jwtToken, err := jwt.Parse(validToken, func(jwtToken *jwt.Token) (interface{}, error) {
|
|
if _, ok := jwtToken.Method.(*jwt.SigningMethodHMAC); !ok {
|
|
return nil, fmt.Errorf("unexpected signing method: %v", jwtToken.Header["alg"])
|
|
}
|
|
return []byte(config.SECRET_KEY), nil
|
|
})
|
|
|
|
assert.NoError(suite.T(), err)
|
|
assert.True(suite.T(), jwtToken.Valid)
|
|
}
|
|
|
|
func (suite *TokenTestSuite) TestVerifyExpiredToken() {
|
|
expire_at := time.Now().Add(-time.Hour * 24 * 365 * 100).Unix()
|
|
expiredToken, err := suite.service.Generate(10001, expire_at, "admin")
|
|
assert.NoError(suite.T(), err)
|
|
|
|
assert.Equal(suite.T(), "1", expiredToken)
|
|
|
|
jwtToken, err := jwt.Parse(expiredToken, func(jwtToken *jwt.Token) (interface{}, error) {
|
|
if _, ok := jwtToken.Method.(*jwt.SigningMethodHMAC); !ok {
|
|
return nil, fmt.Errorf("unexpected signing method: %v", jwtToken.Header["alg"])
|
|
}
|
|
return []byte(config.SECRET_KEY), nil
|
|
})
|
|
|
|
assert.Error(suite.T(), err)
|
|
assert.False(suite.T(), jwtToken.Valid)
|
|
}
|
|
|
|
func (suite *TokenTestSuite) TestCompareJSON() {
|
|
// a := `["a","b","c"]`
|
|
// b := `["a","b","c"]`
|
|
a := `{"x":1,"y":2}`
|
|
b := `{"y":2,"x":1}`
|
|
|
|
var json1 datatypes.JSON
|
|
json1, err := json.Marshal(a)
|
|
assert.NoError(suite.T(), err)
|
|
|
|
var json2 datatypes.JSON
|
|
json2, err = json.Marshal(b)
|
|
assert.NoError(suite.T(), err)
|
|
|
|
assert.Equal(suite.T(), json2.String(), "")
|
|
|
|
compare := cmp.Equal(json1.String(), json2.String())
|
|
|
|
assert.Equal(suite.T(), true, compare)
|
|
}
|