79453663b3
* Annotate api,proxy_util,blockstore_badger, policy tests * Annotate splitstore: bsbadger / markset * Annotate splitstore feature * Annotate union/timed blockstore tests * Annotate openrpc, diff_adt tests * Annotate error,drand,events tests * Annotate predicates_test * Fix annotations * Annotate tscache, gen tests * Annotate fundmanager test * Annotate repub and selection tests * Annotate statetree_test * Annotate forks_test * Annotate searchwait_test.go * Fix duplicated @@ symbols * Annotate chain stmgr/store tests * Annotate more (types) tests * More tests annotated * Annotate conformance chaos actor tests * Annotate more integration tests * Annotate journal system tests * Annotate more tests. * Annotate gas,head buffer behaviors * Fix markset annotations * doc: test annotations for the markets dagstore wrapper * Annotate miner_api test in dagstore * Annotate more test files * Remove bad annotations from fsrepo * Annotate wdpost system * Remove bad annotations * Renamce "conformance" to "chaos_actor" tests * doc: stm annotations for blockheader & election proof tests * Annotate remaining "A" tests * annotate: stm for error_test * memrepo_test.go * Annotate "b" file tests * message_test.go * doc: stm annotate for fsrepo_test * Annotate "c" file tests * Annotate "D" test files * message_test.go * doc: stm annotate for chain, node/config & client * docs: stm annotate node_test * Annotate u,v,wl tests * doc: stm annotations for various test files * Annotate "T" test files * doc: stm annotate for proxy_util_test & policy_test * doc: stm annotate for various tests * doc: final few stm annotations * Add mempool unit tests * Add two more memPool Add tests * Update submodules * Add check function tests * Add stm annotations, refactor test helper * Annotate api,proxy_util,blockstore_badger, policy tests * Annotate splitstore: bsbadger / markset solving merge conflicts * Annotate splitstore feature * Annotate union/timed blockstore tests * Annotate openrpc, diff_adt tests * Annotate error,drand,events tests * Annotate predicates_test * Fix annotations * Annotate tscache, gen tests * Annotate fundmanager test * Annotate statetree_test * Annotate forks_test * Annotate searchwait_test.go * Fix duplicated @@ symbols * Annotate chain stmgr/store tests * Annotate more (types) tests * More tests annotated * Annotate conformance chaos actor tests * Annotate more integration tests * Annotate journal system tests * Annotate more tests. * Annotate gas,head buffer behaviors solve merge conflict * Fix markset annotations * Annotate miner_api test in dagstore * Annotate more test files * doc: test annotations for the markets dagstore wrapper * Annotate wdpost system * Renamce "conformance" to "chaos_actor" tests * Annotate remaining "A" tests * doc: stm annotations for blockheader & election proof tests * annotate: stm for error_test * Annotate "b" file tests * memrepo_test.go * Annotate "c" file tests * message_test.go * Annotate "D" test files * doc: stm annotate for fsrepo_test * Annotate u,v,wl tests * message_test.go * doc: stm annotate for chain, node/config & client * docs: stm annotate node_test * Annotate "T" test files * doc: stm annotations for various test files * Add mempool unit tests solve merge conflict * doc: stm annotate for proxy_util_test & policy_test * doc: stm annotate for various tests * doc: final few stm annotations * Add two more memPool Add tests * Update submodules * Add check function tests solve conflict * Add stm annotations, refactor test helper solve merge conflict * Change CLI test kinds to "unit" * Fix double merged test * Fix ccupgrade_test merge * Fix lint issues * Add stm annotation to types_Test * Test vectors submodule * Add file annotation to burn_test Co-authored-by: Nikola Divic <divicnikola@gmail.com> Co-authored-by: TheMenko <themenkoprojects@gmail.com>
191 lines
5.1 KiB
Go
191 lines
5.1 KiB
Go
//stm: #unit
|
|
package cli
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/filecoin-project/go-state-types/abi"
|
|
"github.com/filecoin-project/lotus/api"
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
"github.com/filecoin-project/lotus/chain/types/mock"
|
|
"github.com/golang/mock/gomock"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestSyncStatus(t *testing.T) {
|
|
app, mockApi, buf, done := NewMockAppWithFullAPI(t, WithCategory("sync", SyncStatusCmd))
|
|
defer done()
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
ts1 := mock.TipSet(mock.MkBlock(nil, 0, 0))
|
|
ts2 := mock.TipSet(mock.MkBlock(ts1, 0, 0))
|
|
|
|
start := time.Now()
|
|
end := start.Add(time.Minute)
|
|
|
|
state := &api.SyncState{
|
|
ActiveSyncs: []api.ActiveSync{{
|
|
WorkerID: 1,
|
|
Base: ts1,
|
|
Target: ts2,
|
|
Stage: api.StageMessages,
|
|
Height: abi.ChainEpoch(0),
|
|
Start: start,
|
|
End: end,
|
|
Message: "whatever",
|
|
}},
|
|
VMApplied: 0,
|
|
}
|
|
|
|
mockApi.EXPECT().SyncState(ctx).Return(state, nil)
|
|
|
|
//stm: @CLI_SYNC_STATUS_001
|
|
err := app.Run([]string{"sync", "status"})
|
|
assert.NoError(t, err)
|
|
|
|
out := buf.String()
|
|
|
|
// output is plaintext, had to do string matching
|
|
assert.Contains(t, out, fmt.Sprintf("Base:\t[%s]", ts1.Blocks()[0].Cid().String()))
|
|
assert.Contains(t, out, fmt.Sprintf("Target:\t[%s]", ts2.Blocks()[0].Cid().String()))
|
|
assert.Contains(t, out, "Height diff:\t1")
|
|
assert.Contains(t, out, "Stage: message sync")
|
|
assert.Contains(t, out, "Height: 0")
|
|
assert.Contains(t, out, "Elapsed: 1m0s")
|
|
}
|
|
|
|
func TestSyncMarkBad(t *testing.T) {
|
|
app, mockApi, _, done := NewMockAppWithFullAPI(t, WithCategory("sync", SyncMarkBadCmd))
|
|
defer done()
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
blk := mock.MkBlock(nil, 0, 0)
|
|
|
|
mockApi.EXPECT().SyncMarkBad(ctx, blk.Cid()).Return(nil)
|
|
|
|
//stm: @CLI_SYNC_MARK_BAD_001
|
|
err := app.Run([]string{"sync", "mark-bad", blk.Cid().String()})
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestSyncUnmarkBad(t *testing.T) {
|
|
t.Run("one-block", func(t *testing.T) {
|
|
app, mockApi, _, done := NewMockAppWithFullAPI(t, WithCategory("sync", SyncUnmarkBadCmd))
|
|
defer done()
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
blk := mock.MkBlock(nil, 0, 0)
|
|
|
|
mockApi.EXPECT().SyncUnmarkBad(ctx, blk.Cid()).Return(nil)
|
|
|
|
//stm: @CLI_SYNC_UNMARK_BAD_001
|
|
err := app.Run([]string{"sync", "unmark-bad", blk.Cid().String()})
|
|
assert.NoError(t, err)
|
|
})
|
|
|
|
t.Run("all", func(t *testing.T) {
|
|
app, mockApi, _, done := NewMockAppWithFullAPI(t, WithCategory("sync", SyncUnmarkBadCmd))
|
|
defer done()
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
mockApi.EXPECT().SyncUnmarkAllBad(ctx).Return(nil)
|
|
|
|
//stm: @CLI_SYNC_UNMARK_BAD_002
|
|
err := app.Run([]string{"sync", "unmark-bad", "-all"})
|
|
assert.NoError(t, err)
|
|
})
|
|
}
|
|
|
|
func TestSyncCheckBad(t *testing.T) {
|
|
t.Run("not-bad", func(t *testing.T) {
|
|
app, mockApi, buf, done := NewMockAppWithFullAPI(t, WithCategory("sync", SyncCheckBadCmd))
|
|
defer done()
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
blk := mock.MkBlock(nil, 0, 0)
|
|
|
|
mockApi.EXPECT().SyncCheckBad(ctx, blk.Cid()).Return("", nil)
|
|
|
|
//stm: @CLI_SYNC_CHECK_BAD_002
|
|
err := app.Run([]string{"sync", "check-bad", blk.Cid().String()})
|
|
assert.NoError(t, err)
|
|
|
|
assert.Contains(t, buf.String(), "block was not marked as bad")
|
|
})
|
|
|
|
t.Run("bad", func(t *testing.T) {
|
|
app, mockApi, buf, done := NewMockAppWithFullAPI(t, WithCategory("sync", SyncCheckBadCmd))
|
|
defer done()
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
blk := mock.MkBlock(nil, 0, 0)
|
|
reason := "whatever"
|
|
|
|
mockApi.EXPECT().SyncCheckBad(ctx, blk.Cid()).Return(reason, nil)
|
|
|
|
//stm: @CLI_SYNC_CHECK_BAD_001
|
|
err := app.Run([]string{"sync", "check-bad", blk.Cid().String()})
|
|
assert.NoError(t, err)
|
|
|
|
assert.Contains(t, buf.String(), reason)
|
|
})
|
|
}
|
|
|
|
func TestSyncCheckpoint(t *testing.T) {
|
|
t.Run("tipset", func(t *testing.T) {
|
|
app, mockApi, _, done := NewMockAppWithFullAPI(t, WithCategory("sync", SyncCheckpointCmd))
|
|
defer done()
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
blk := mock.MkBlock(nil, 0, 0)
|
|
ts := mock.TipSet(blk)
|
|
|
|
gomock.InOrder(
|
|
mockApi.EXPECT().ChainGetBlock(ctx, blk.Cid()).Return(blk, nil),
|
|
mockApi.EXPECT().SyncCheckpoint(ctx, ts.Key()).Return(nil),
|
|
)
|
|
|
|
//stm: @CLI_SYNC_CHECKPOINT_001
|
|
err := app.Run([]string{"sync", "checkpoint", blk.Cid().String()})
|
|
assert.NoError(t, err)
|
|
})
|
|
|
|
t.Run("epoch", func(t *testing.T) {
|
|
app, mockApi, _, done := NewMockAppWithFullAPI(t, WithCategory("sync", SyncCheckpointCmd))
|
|
defer done()
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
epoch := abi.ChainEpoch(0)
|
|
blk := mock.MkBlock(nil, 0, 0)
|
|
ts := mock.TipSet(blk)
|
|
|
|
gomock.InOrder(
|
|
mockApi.EXPECT().ChainGetTipSetByHeight(ctx, epoch, types.EmptyTSK).Return(ts, nil),
|
|
mockApi.EXPECT().SyncCheckpoint(ctx, ts.Key()).Return(nil),
|
|
)
|
|
|
|
//stm: @CLI_SYNC_CHECKPOINT_002
|
|
err := app.Run([]string{"sync", "checkpoint", fmt.Sprintf("-epoch=%d", epoch)})
|
|
assert.NoError(t, err)
|
|
})
|
|
}
|