diff --git a/pkg/search/mock_client.go b/pkg/search/mock_client.go
index 228c22dc..4a36fec5 100644
--- a/pkg/search/mock_client.go
+++ b/pkg/search/mock_client.go
@@ -5,7 +5,10 @@ import (
 )
 
 type MockClient struct {
-	Err            error
+	Err error
+
+	Searches []Entity[Value]
+
 	SearchResponse SearchResponse
 }
 
@@ -19,5 +22,8 @@ func (c *MockClient) SearchByEntity(ctx context.Context, entity Entity[Value], o
 	if c.Err != nil {
 		return SearchResponse{}, c.Err
 	}
+
+	c.Searches = append(c.Searches, entity)
+
 	return c.SearchResponse, c.Err
 }