test: sync unmark-bad cli command

This commit is contained in:
Nikola Divic 2022-02-11 18:00:01 +01:00
parent 7fc2f9dc29
commit 0f79a0024c

View File

@ -69,3 +69,33 @@ func TestSyncMarkBad(t *testing.T) {
err := app.Run([]string{"sync", "mark-bad", blk.Cid().String()}) err := app.Run([]string{"sync", "mark-bad", blk.Cid().String()})
assert.NoError(t, err) 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)
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)
err := app.Run([]string{"sync", "unmark-bad", "-all"})
assert.NoError(t, err)
})
}