mirror of
https://github.com/weloe/token-go.git
synced 2025-10-06 07:56:58 +08:00
feat: update CreateTempToken,add CreateTempTokenByStyle
This commit is contained in:
@@ -2,16 +2,20 @@ package token_go
|
||||
|
||||
import (
|
||||
"github.com/weloe/token-go/model"
|
||||
"reflect"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestEnforcer_TempToken(t *testing.T) {
|
||||
enforcer, _ := NewTestEnforcer(t)
|
||||
service := "code"
|
||||
tempToken, err := enforcer.CreateTempToken("uuid", service, "1234", -1)
|
||||
tempToken, err := enforcer.CreateTempToken("tempToken", service, "1234", -1)
|
||||
if err != nil {
|
||||
t.Fatalf("CreateTempToken() failed: %v", err)
|
||||
}
|
||||
if !reflect.DeepEqual(tempToken, "tempToken") {
|
||||
t.Fatalf("token error, want is %v", "1234")
|
||||
}
|
||||
timeout := enforcer.GetTempTokenTimeout(service, tempToken)
|
||||
if timeout != -1 {
|
||||
t.Errorf("GetTempTokenTimeout() failed, unexpected timeout: %v", timeout)
|
||||
@@ -33,6 +37,16 @@ func TestEnforcer_TempToken(t *testing.T) {
|
||||
if codeValue != "" {
|
||||
t.Errorf("ParseTempToken() failed, unexpected codeValue: %v", codeValue)
|
||||
}
|
||||
|
||||
// create token
|
||||
tempToken, err = enforcer.CreateTempTokenByStyle("uuid", service, "1234", -1)
|
||||
if err != nil {
|
||||
t.Fatalf("CreateTempTokenByStyle() failed: %v", err)
|
||||
}
|
||||
// delete
|
||||
if enforcer.DeleteTempToken(service, tempToken) != nil {
|
||||
t.Fatalf("DeleteTempToken() failed: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestEnforcer_ConfirmQRCode(t *testing.T) {
|
||||
|
Reference in New Issue
Block a user