From b85a297b4eae1ee730059be277d7aa83658c1fe4 Mon Sep 17 00:00:00 2001
From: weloe <1345895607@qq.com>
Date: Mon, 16 Oct 2023 21:48:26 +0800
Subject: [PATCH] feat: add QRCode state test case

---
 enforcer_manager_api_test.go | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/enforcer_manager_api_test.go b/enforcer_manager_api_test.go
index 4e44c72..2a5c509 100644
--- a/enforcer_manager_api_test.go
+++ b/enforcer_manager_api_test.go
@@ -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)
+	}
+}