lotus/blockstore/timed_test.go

// stm: #unit
package blockstore

import (
	"context"
	"testing"
	"time"

	"github.com/ipfs/go-cid"
	blocks "github.com/ipfs/go-libipfs/blocks"
	"github.com/raulk/clock"
	"github.com/stretchr/testify/require"
)

func TestTimedCacheBlockstoreSimple(t *testing.T) {
	//stm: @SPLITSTORE_TIMED_BLOCKSTORE_START_001
	//stm: @SPLITSTORE_TIMED_BLOCKSTORE_PUT_001, @SPLITSTORE_TIMED_BLOCKSTORE_HAS_001, @SPLITSTORE_TIMED_BLOCKSTORE_GET_001
	//stm: @SPLITSTORE_TIMED_BLOCKSTORE_ALL_KEYS_CHAN_001
	tc := NewTimedCacheBlockstore(10 * time.Millisecond)
	mClock := clock.NewMock()
	mClock.Set(time.Now())
	tc.clock = mClock
	tc.doneRotatingCh = make(chan struct{})

	ctx := context.Background()

	_ = 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(ctx, b1))

	b2 := blocks.NewBlock([]byte("bar"))
	require.NoError(t, tc.Put(ctx, b2))

	b3 := blocks.NewBlock([]byte("baz"))

	b1out, err := tc.Get(ctx, b1.Cid())
	require.NoError(t, err)
	require.Equal(t, b1.RawData(), b1out.RawData())

	has, err := tc.Has(ctx, 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(ctx, b1.Cid())
	require.NoError(t, err)
	require.True(t, has)

	has, err = tc.Has(ctx, b2.Cid())
	require.NoError(t, err)
	require.True(t, has)

	// extend b2, add b3.
	require.NoError(t, tc.Put(ctx, b2))
	require.NoError(t, tc.Put(ctx, 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(ctx, b1.Cid())
	require.NoError(t, err)
	require.False(t, has)

	has, err = tc.Has(ctx, b2.Cid())
	require.NoError(t, err)
	require.True(t, has)

	has, err = tc.Has(ctx, b3.Cid())
	require.NoError(t, err)
	require.True(t, has)
}