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