Function TestStorePartialCompactionWithMergeOperator

store_test.go:2385–2512

(t *testing.T)

}

func TestStorePartialCompactionWithMergeOperator(t *testing.T) {
	tmpDir, _ := ioutil.TempDir("", "mossStore")
	defer os.RemoveAll(tmpDir)

	var err error
	var store *Store
	var coll Collection

	spo := StorePersistOptions{CompactionConcern: CompactionDisable}

	store, coll, err = OpenStoreCollection(tmpDir,
		StoreOptions{
			CollectionOptions: CollectionOptions{
				MergeOperator: &MergeOperatorStringAppend{Sep: ":"},
			},
		},
		spo)
	if err != nil || store == nil {
		t.Errorf("OpenStoreCollection failed, err: %v", err)
	}
	defer coll.Close()
	defer store.Close()

	var b Batch

	// Execute first batch, so "k0" ==> "a".

	b, err = coll.NewBatch(0, 0)
	if err != nil {
		t.Errorf("expected NewBatch() to succeed")
	}
	b.Set([]byte("k0"), []byte("a"))

	err = coll.ExecuteBatch(b, WriteOptions{})
	if err != nil {
		t.Errorf("expected ExecuteBatch() to work")
	}
	waitForPersistence(coll)

	store.m.Lock()
	store.footer.m.Lock()
	if len(store.footer.ss.a) != 1 {
		t.Errorf("expected ss height of 1")
	}
	store.footer.m.Unlock()
	store.m.Unlock()

	// Execute second batch, so "k0" ==> "a:b".

	b, err = coll.NewBatch(0, 0)
	if err != nil {
		t.Errorf("expected NewBatch() to succeed")
	}
	b.Merge([]byte("k0"), []byte("b"))

	err = coll.ExecuteBatch(b, WriteOptions{})
	if err != nil {
		t.Errorf("expected ExecuteBatch() to work")