@@ -261,11 +261,11 @@ func (p *store[K, V]) Cleanup(ctx context.Context, maxAge time.Duration) (int, e
261261// Flush removes all entries from Datastore.
262262// Returns the number of entries removed and any error.
263263func (p * store [K , V ]) Flush (ctx context.Context ) (int , error ) {
264- // Query for all keys
264+ // Query for all keys (use empty slice for mock compatibility)
265265 query := ds .NewQuery (p .kind ).KeysOnly ()
266266
267- var entries []entry
268- keys , err := p .client .GetAll (ctx , query , & entries )
267+ var dst []entry
268+ keys , err := p .client .GetAll (ctx , query , & dst )
269269 if err != nil {
270270 return 0 , fmt .Errorf ("query all entries: %w" , err )
271271 }
@@ -285,7 +285,8 @@ func (p *store[K, V]) Flush(ctx context.Context) (int, error) {
285285// Len returns the number of entries in Datastore.
286286func (p * store [K , V ]) Len (ctx context.Context ) (int , error ) {
287287 query := ds .NewQuery (p .kind ).KeysOnly ()
288- keys , err := p .client .GetAll (ctx , query , nil )
288+ var dst []entry
289+ keys , err := p .client .GetAll (ctx , query , & dst )
289290 if err != nil {
290291 return 0 , fmt .Errorf ("query all entries: %w" , err )
291292 }
0 commit comments