-
Notifications
You must be signed in to change notification settings - Fork 10
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add test case for local in-memory fallback
- Loading branch information
1 parent
1ab0ec2
commit ca0c260
Showing
3 changed files
with
79 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,70 @@ | ||
| package httprateredis_test | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "math/rand" | ||
| "strconv" | ||
| "testing" | ||
| "time" | ||
|
|
||
| "github.com/alicebob/miniredis/v2" | ||
| httprateredis "github.com/go-chi/httprate-redis" | ||
| ) | ||
|
|
||
| // Test local in-memory counter fallback, which gets activated in case Redis is not available. | ||
| func TestLocalFallback(t *testing.T) { | ||
| redis, err := miniredis.Run() | ||
| redisPort, _ := strconv.Atoi(redis.Port()) | ||
|
|
||
| limitCounter, err := httprateredis.NewRedisLimitCounter(&httprateredis.Config{ | ||
| Host: redis.Host(), | ||
| Port: uint16(redisPort), | ||
| MaxIdle: 0, | ||
| MaxActive: 1, | ||
| ClientName: "httprateredis_test", | ||
| PrefixKey: fmt.Sprintf("httprate:test:%v", rand.Int31n(100000)), // Unique Redis key for each test | ||
| FallbackTimeout: 200 * time.Millisecond, | ||
| }) | ||
| if err != nil { | ||
| t.Fatalf("redis not available: %v", err) | ||
| } | ||
|
|
||
| limitCounter.Config(1000, time.Minute) | ||
|
|
||
| currentWindow := time.Now().UTC().Truncate(time.Minute) | ||
| previousWindow := currentWindow.Add(-time.Minute) | ||
|
|
||
| if limitCounter.IsFallbackActivated() { | ||
| t.Error("fallback should not be activated at the beginning") | ||
| } | ||
|
|
||
| err = limitCounter.IncrementBy("key:fallback", currentWindow, 1) | ||
| if err != nil { | ||
| t.Error(err) | ||
| } | ||
|
|
||
| _, _, err = limitCounter.Get("key:fallback", currentWindow, previousWindow) | ||
| if err != nil { | ||
| t.Error(err) | ||
| } | ||
|
|
||
| if limitCounter.IsFallbackActivated() { | ||
| t.Error("fallback should not be activated before we simulate redis failure") | ||
| } | ||
|
|
||
| redis.Close() | ||
|
|
||
| err = limitCounter.IncrementBy("key:fallback", currentWindow, 1) | ||
| if err != nil { | ||
| t.Error(err) | ||
| } | ||
|
|
||
| _, _, err = limitCounter.Get("key:fallback", currentWindow, previousWindow) | ||
| if err != nil { | ||
| t.Error(err) | ||
| } | ||
|
|
||
| if !limitCounter.IsFallbackActivated() { | ||
| t.Error("fallback should be activated after we simulate redis failure") | ||
| } | ||
| } |