fix mocks

internal/pool/hooks_test.go

@@ -10,12 +10,13 @@ import (
 
 // TestHook for testing hook functionality
 type TestHook struct {
-	OnGetCalled  int
+	OnGetCalled    int
-	OnPutCalled  int
+	OnPutCalled    int
-	GetError     error
+	OnRemoveCalled int
-	PutError     error
+	GetError       error
-	ShouldPool   bool
+	PutError       error
-	ShouldRemove bool
+	ShouldPool     bool
+	ShouldRemove   bool
 }
 
 func (th *TestHook) OnGet(ctx context.Context, conn *Conn, isNewConn bool) error {
@@ -28,6 +29,10 @@ func (th *TestHook) OnPut(ctx context.Context, conn *Conn) (shouldPool bool, sho
 	return th.ShouldPool, th.ShouldRemove, th.PutError
 }
 
+func (th *TestHook) OnRemove(ctx context.Context, conn *Conn, reason error) {
+	th.OnRemoveCalled++
+}
+
 func TestPoolHookManager(t *testing.T) {
 	manager := NewPoolHookManager()
 

maintnotifications/pool_hook_test.go

@@ -92,6 +92,10 @@ func (mp *mockPool) Stats() *pool.Stats {
 	return &pool.Stats{}
 }
 
+func (mp *mockPool) Size() int {
+	return 0
+}
+
 func (mp *mockPool) AddPoolHook(hook pool.PoolHook) {
 	// Mock implementation - do nothing
 }