package timedbs import ( "context" "testing" "time" "github.com/raulk/clock" "github.com/stretchr/testify/require" blocks "github.com/ipfs/go-block-format" "github.com/ipfs/go-cid" ) func TestTimedBSSimple(t *testing.T) { tc := NewTimedCacheBS(10 * time.Millisecond) mClock := clock.NewMock() mClock.Set(time.Now()) tc.clock = mClock tc.doneRotatingCh = make(chan struct{}) _ = tc.Start(context.Background()) mClock.Add(1) // IDK why it is needed but it makes it work defer func() { _ = tc.Stop(context.Background()) }() b1 := blocks.NewBlock([]byte("foo")) require.NoError(t, tc.Put(b1)) b2 := blocks.NewBlock([]byte("bar")) require.NoError(t, tc.Put(b2)) b3 := blocks.NewBlock([]byte("baz")) b1out, err := tc.Get(b1.Cid()) require.NoError(t, err) require.Equal(t, b1.RawData(), b1out.RawData()) has, err := tc.Has(b1.Cid()) require.NoError(t, err) require.True(t, has) mClock.Add(10 * time.Millisecond) <-tc.doneRotatingCh // We should still have everything. has, err = tc.Has(b1.Cid()) require.NoError(t, err) require.True(t, has) has, err = tc.Has(b2.Cid()) require.NoError(t, err) require.True(t, has) // extend b2, add b3. require.NoError(t, tc.Put(b2)) require.NoError(t, tc.Put(b3)) // all keys once. allKeys, err := tc.AllKeysChan(context.Background()) var ks []cid.Cid for k := range allKeys { ks = append(ks, k) } require.NoError(t, err) require.ElementsMatch(t, ks, []cid.Cid{b1.Cid(), b2.Cid(), b3.Cid()}) mClock.Add(10 * time.Millisecond) <-tc.doneRotatingCh // should still have b2, and b3, but not b1 has, err = tc.Has(b1.Cid()) require.NoError(t, err) require.False(t, has) has, err = tc.Has(b2.Cid()) require.NoError(t, err) require.True(t, has) has, err = tc.Has(b3.Cid()) require.NoError(t, err) require.True(t, has) }