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

correct handling OnPut

Browse Source
This commit is contained in:
Nedyalko Dyakov
2025-10-24 14:13:13 +03:00
parent 5721512a79
commit 663a60e47f
3 changed files with 25 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
maintnotifications/pool_hook_test.go
View File

@@ -39,7 +39,9 @@ func (m *mockAddr) String() string { return m.addr }
func createMockPoolConnection() *pool.Conn {
mockNetConn := &mockNetConn{addr: "test:6379"}
conn := pool.NewConn(mockNetConn)
conn.SetUsable(true) // Make connection usable for testing
conn.SetUsable(true) // Make connection usable for testing (transitions to IDLE)
// Simulate real flow: connection is acquired (IDLE → IN_USE) before OnPut is called
conn.SetUsed(true) // Transition to IN_USE state
return conn
}
@@ -686,7 +688,7 @@ func TestConnectionHook(t *testing.T) {
t.Errorf("OnPut should succeed: %v", err)
}
if !shouldPool || shouldRemove {
t.Error("Connection should be pooled after handoff")
t.Errorf("Connection should be pooled after handoff (shouldPool=%v, shouldRemove=%v)", shouldPool, shouldRemove)
}
// Wait for handoff to complete