test

celestiaorg · Jun 25, 2024 · 9cb6944 · 9cb6944
1 parent bbd0e5d
commit 9cb6944
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/sync/sync_test.go b/sync/sync_test.go
@@ -36,7 +36,7 @@ func TestSyncSimpleRequestingHead(t *testing.T) {
 	require.NoError(t, err)
 	require.Equal(t, exp.Height(), h101.Height())
 
-	localStore := newTestStore(t, ctx, head)
+	localStore := newTestStore(t, ctx, head, store.WithWriteBatchSize(10))
 	syncer, err := NewSyncer(
 		local.NewExchange(remoteStore),
 		localStore,
@@ -403,7 +403,9 @@ func (d *delayedGetter[H]) GetRangeByHeight(ctx context.Context, from H, to uint
 }
 
 // newTestStore creates initialized and started in memory header Store which is useful for testing.
-func newTestStore(tb testing.TB, ctx context.Context, head *headertest.DummyHeader) header.Store[*headertest.DummyHeader] {
+func newTestStore(tb testing.TB, ctx context.Context, head *headertest.DummyHeader,
+	opts ...store.Option,
+) header.Store[*headertest.DummyHeader] {
 	ds := sync.MutexWrap(datastore.NewMapDatastore())
-	return store.NewTestStore(tb, ctx, ds, head)
+	return store.NewTestStore(tb, ctx, ds, head, opts...)
 }