redis/go-redis
1
0
Fork 0
mirror of https://github.com/redis/go-redis.git synced 2025-12-02 06:22:31 +03:00
Code Activity

improve reauth state management. fix tests

Browse Source
This commit is contained in:
Nedyalko Dyakov
2025-10-24 15:05:54 +03:00
parent 92433e6f2a
commit 21bd243bf5
3 changed files with 24 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
maintnotifications/pool_hook_test.go
View File

@@ -391,8 +391,8 @@ func TestConnectionHook(t *testing.T) {
ctx := context.Background()
acceptCon, err := processor.OnGet(ctx, conn, false)
if err != ErrConnectionMarkedForHandoff {
t.Errorf("Expected ErrConnectionMarkedForHandoff, got %v", err)
if err != ErrConnectionMarkedForHandoffWithState {
t.Errorf("Expected ErrConnectionMarkedForHandoffWithState, got %v", err)
}
if acceptCon {
t.Error("Connection should not be accepted when marked for handoff")
@@ -425,8 +425,8 @@ func TestConnectionHook(t *testing.T) {
// Test OnGet with pending handoff
ctx := context.Background()
acceptCon, err := processor.OnGet(ctx, conn, false)
if err != ErrConnectionMarkedForHandoff {
t.Error("Should return ErrConnectionMarkedForHandoff for pending connection")
if err != ErrConnectionMarkedForHandoffWithState {
t.Errorf("Should return ErrConnectionMarkedForHandoffWithState for pending connection, got %v", err)
}
if acceptCon {
t.Error("Should not accept connection with pending handoff")
@@ -678,8 +678,8 @@ func TestConnectionHook(t *testing.T) {
if err == nil {
t.Error("OnGet should fail for connection marked for handoff")
}
if err != ErrConnectionMarkedForHandoff {
t.Errorf("Expected ErrConnectionMarkedForHandoff, got %v", err)
if err != ErrConnectionMarkedForHandoffWithState {
t.Errorf("Expected ErrConnectionMarkedForHandoffWithState, got %v", err)
}
if acceptConn {
t.Error("Connection should not be accepted when marked for handoff")