From edcf800dd62a49d6f2c8dd6a11fa4ab6ebb9efb0 Mon Sep 17 00:00:00 2001 From: Samuel Laferriere Date: Fri, 6 Sep 2024 11:37:29 -0700 Subject: [PATCH] fix: memstore tests --- store/memory_test.go | 21 +++++++++------------ 1 file changed, 9 insertions(+), 12 deletions(-) diff --git a/store/memory_test.go b/store/memory_test.go index 0724a11..d524460 100644 --- a/store/memory_test.go +++ b/store/memory_test.go @@ -19,7 +19,7 @@ const ( func getDefaultTestConfig() MemStoreConfig { return MemStoreConfig{ MaxBlobSizeBytes: 1024 * 1024, - BlobExpiration: time.Hour * 1000, + BlobExpiration: 0, PutLatency: 0, GetLatency: 0, } @@ -87,11 +87,13 @@ func TestExpiration(t *testing.T) { verifier, err := verify.NewVerifier(cfg, nil) require.NoError(t, err) + memstoreConfig := getDefaultTestConfig() + memstoreConfig.BlobExpiration = 10 * time.Millisecond ms, err := NewMemStore( ctx, verifier, log.New(), - getDefaultTestConfig(), + memstoreConfig, ) require.NoError(t, err) @@ -134,12 +136,10 @@ func TestLatency(t *testing.T) { verifier, err := verify.NewVerifier(cfg, nil) require.NoError(t, err) - ms, err := NewMemStore( - ctx, - verifier, - log.New(), - getDefaultTestConfig(), - ) + config := getDefaultTestConfig() + config.PutLatency = putLatency + config.GetLatency = getLatency + ms, err := NewMemStore(ctx, verifier, log.New(), config) require.NoError(t, err) @@ -149,12 +149,9 @@ func TestLatency(t *testing.T) { require.NoError(t, err) require.GreaterOrEqual(t, time.Since(timeBeforePut), putLatency) - // sleep 1 second and verify that older blob entries are removed - time.Sleep(time.Second * 1) - timeBeforeGet := time.Now() _, err = ms.Get(ctx, key) - require.Error(t, err) + require.NoError(t, err) require.GreaterOrEqual(t, time.Since(timeBeforeGet), getLatency) }