search: record Searches in MockClient

moov-io · Jan 17, 2025 · e82d0b5 · e82d0b5
1 parent 0e34ede
commit e82d0b5
Showing 1 changed file with 7 additions and 1 deletion.
diff --git 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
 }