feat: add QRCode state test case

This commit is contained in:
weloe
2023-10-16 21:48:26 +08:00
parent 5266eaffdf
commit b85a297b4e

View File

@@ -122,3 +122,24 @@ func TestEnforcer_CancelAuthQRCode(t *testing.T) {
t.Logf(" id: [%v] QRCode login is cancelled.", loginId)
}
}
func TestEnforcer_QRCodeState(t *testing.T) {
enforcer, _ := NewTestEnforcer(t)
QRCodeId := "1"
err := enforcer.CreateQRCodeState(QRCodeId, -1)
if err != nil {
t.Fatalf("CreateQRCodeState() failed: %v", err)
}
if timeout := enforcer.getQRCodeTimeout(QRCodeId); timeout != -1 {
t.Fatalf("timeout error, want is -1")
}
err = enforcer.DeleteQRCode(QRCodeId)
if err != nil {
t.Fatalf("DeleteQRCode() failed: %v", err)
}
state := enforcer.GetQRCodeState(QRCodeId)
if state != model.Expired {
t.Fatalf("QR code state error, want is %v", model.Expired)
}
}