From 695dcf3c49dba0be867ad4dffc4837edb8805a97 Mon Sep 17 00:00:00 2001
From: lzh <18320341470>
Date: Wed, 18 Jun 2025 16:45:01 +0800
Subject: [PATCH] sms_tool

---
 sms_tool/sms_client_test.go | 25 ++++++++++++++++++++++++-
 1 file changed, 24 insertions(+), 1 deletion(-)

diff --git a/sms_tool/sms_client_test.go b/sms_tool/sms_client_test.go
index a6e9dc1..e3d44b7 100644
--- a/sms_tool/sms_client_test.go
+++ b/sms_tool/sms_client_test.go
@@ -137,7 +137,30 @@ func TestSmsClient_SaveCode(t *testing.T) {
 	goRedis := NewRedisCacheAdapter(addr, redisPassword, db)
 	smsClient.Cache = goRedis
 
-	err = smsClient.SaveCode(context.Background(), codeKey, "123456", time.Minute*5, time.Minute)
+	err = smsClient.SaveCode(context.Background(), codeKey, "123456", time.Minute*5)
+	if err != nil {
+		fmt.Println(err.Error())
+		return
+	}
+	fmt.Println("保存成功")
+}
+
+func TestSmsClient_SaveCodeWithFrequency(t *testing.T) {
+	config := &SmsConfig{
+		AccessKeyId:     accessKeyId,
+		AccessKeySecret: accessKeySecret,
+	}
+	smsClient, err := NewSmsClient(config)
+	if err != nil {
+		fmt.Println(err.Error())
+		return
+	}
+	addr := fmt.Sprintf("%s:%s", redisHost, redisPort)
+	db, _ := strconv.Atoi(redisDb)
+	goRedis := NewRedisCacheAdapter(addr, redisPassword, db)
+	smsClient.Cache = goRedis
+
+	err = smsClient.SaveCodeWithFrequency(context.Background(), codeKey, "123456", time.Minute*5, time.Minute)
 	if err != nil {
 		fmt.Println(err.Error())
 		return