Function TestCompactJob_merge_compact_merge_fail

kv/compact_job_test.go:98–146

(t *testing.T)

}

func TestCompactJob_merge_compact_merge_fail(t *testing.T) {
	ctrl := gomock.NewController(t)
	defer ctrl.Finish()

	snapshot := version.NewMockSnapshot(ctrl)
	reader := table.NewMockReader(ctrl)
	gomock.InOrder(
		reader.EXPECT().Iterator().Return(generateIterator(ctrl, map[uint32][]byte{
			1: []byte("value1"),
		})),
		reader.EXPECT().Iterator().Return(generateIterator(ctrl, map[uint32][]byte{})),
	)
	snapshot.EXPECT().GetReader(gomock.Any()).Return(reader, nil).MaxTimes(2)
	merge := NewMockMerger(ctrl)
	merge.EXPECT().Merge(gomock.Any(), gomock.Any()).Return(fmt.Errorf("err"))
	family := generateMockFamily(ctrl, func(flusher Flusher) (Merger, error) {
		return merge, nil
	})

	family.EXPECT().familyInfo().Return("family").AnyTimes()
	f1 := version.NewFileMeta(1, 1, 10, 100)
	f4 := version.NewFileMeta(4, 30, 100, 100)
	compaction := version.NewCompaction(1, 0, []*version.FileMeta{f1}, []*version.FileMeta{f4})
	state := newCompactionState(1000, snapshot, compaction)
	compactJob1 := newCompactJob(family, state, nil)
	err := compactJob1.Run()
	assert.Error(t, err)

	gomock.InOrder(
		reader.EXPECT().Iterator().Return(generateIterator(ctrl, map[uint32][]byte{
			1: []byte("value1"),
		})),
		reader.EXPECT().Iterator().Return(generateIterator(ctrl, map[uint32][]byte{2: []byte("value2")})),
	)
	snapshot.EXPECT().GetReader(gomock.Any()).Return(reader, nil).MaxTimes(2)
	merge.EXPECT().Merge(uint32(1), gomock.Any()).Return(fmt.Errorf("err"))
	compactJob1 = newCompactJob(family, state, nil)
	err = compactJob1.Run()
	assert.Error(t, err)

	// create merger failure
	compactJob1 = newCompactJob(family, state, nil)
	compactJob2 := compactJob1.(*compactJob)
	compactJob2.newMerger = func(flusher Flusher) (Merger, error) {
		return nil, fmt.Errorf("err")
	}
	err = compactJob2.Run()
	assert.Error(t, err)
}

func TestCompactJob_merge_doMerge_fail(t *testing.T) {
	ctrl := gomock.NewController(t)