Function testTaskInspectMgrDo

blobstore/blobnode/task_inspect_mgr_test.go:41–117

(t *testing.T, mode codemode.CodeMode)

}

func testTaskInspectMgrDo(t *testing.T, mode codemode.CodeMode) {
	replicas := genMockVol(1, mode)
	bids := []proto.BlobID{1, 2, 3, 4, 5, 6, 7}
	sizes := []int64{10, 1024, 1024, 1024, 1024, 1024, 1024}
	getter := NewMockGetterWithBids(replicas, mode, bids, sizes)
	mgr := NewInspectTaskMgr(1, getter, newMockReporter(t))
	task := proto.VolumeInspectTask{
		TaskID:   "InspectTask_XXX",
		Mode:     mode,
		Replicas: replicas,
	}
	ret := mgr.doInspect(context.Background(), &task)
	require.NoError(t, ret.Err())
	require.Equal(t, 0, len(ret.MissedShards))

	getter.MarkDelete(context.Background(), replicas[0].Vuid, 1)
	ret = mgr.doInspect(context.Background(), &task)
	require.NoError(t, ret.Err())
	require.Equal(t, 0, len(ret.MissedShards))

	// normal
	var expectMissedShard []*proto.MissedShard
	ret = mgr.doInspect(context.Background(), &task)
	require.NoError(t, ret.Err())
	verifyInspectResult(t, expectMissedShard, ret.MissedShards)

	// delete one bid
	getter.Delete(context.Background(), replicas[0].Vuid, 1)
	expectMissedShard = append(expectMissedShard, &proto.MissedShard{Vuid: replicas[0].Vuid, Bid: 1})
	ret = mgr.doInspect(context.Background(), &task)
	require.NoError(t, ret.Err())
	verifyInspectResult(t, expectMissedShard, ret.MissedShards)

	// delete  other bid 2
	getter.Delete(context.Background(), replicas[1].Vuid, 2)
	expectMissedShard = append(expectMissedShard, &proto.MissedShard{Vuid: replicas[1].Vuid, Bid: 2})
	ret = mgr.doInspect(context.Background(), &task)
	require.NoError(t, ret.Err())
	verifyInspectResult(t, expectMissedShard, ret.MissedShards)

	// delete m
	delStartIdx := 1
	delEndIdx := mode.Tactic().M
	for _, replica := range replicas[delStartIdx:delEndIdx] {
		getter.Delete(context.Background(), replica.Vuid, 1)
		expectMissedShard = append(expectMissedShard, &proto.MissedShard{Vuid: replica.Vuid, Bid: 1})
	}

	ret = mgr.doInspect(context.Background(), &task)
	require.NoError(t, ret.Err())
	verifyInspectResult(t, expectMissedShard, ret.MissedShards)

	delStartIdx = delEndIdx
	delEndIdx = delEndIdx + 1
	for _, replica := range replicas[delStartIdx:delEndIdx] {
		getter.Delete(context.Background(), replica.Vuid, 1)
		expectMissedShard = append(expectMissedShard, &proto.MissedShard{Vuid: replica.Vuid, Bid: 1})
	}