Function TestIteratorSingleDone

iterator_test.go:487–584

(t *testing.T)

}

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

	store, m, err := OpenStoreCollection(tmpDir,
		StoreOptions{}, StorePersistOptions{CompactionConcern: CompactionForce})
	if err != nil || m == nil || store == nil {
		t.Errorf("expected open empty store collection to work")
	}
	defer m.Close()
	defer store.Close()

	mc := m.(*collection)

	setVal := func(s string) {
		b, err2 := m.NewBatch(0, 0)
		if err2 != nil || b == nil {
			t.Errorf("expected b ok")
		}
		b.Set([]byte(s), []byte(s))
		err2 = m.ExecuteBatch(b, WriteOptions{})
		if err2 != nil {
			t.Errorf("expected execute batch ok")
		}
	}

	for _, mv := range []string{"a", "b"} {
		setVal(mv)
	}

	mc.NotifyMerger("mergeAll", true)

	waitUntilClean := func() error {
		for {
			stats, err2 := m.Stats()
			if err2 != nil {
				return err2
			}

			if stats.CurDirtyOps <= 0 &&
				stats.CurDirtyBytes <= 0 &&
				stats.CurDirtySegments <= 0 {
				break
			}

			time.Sleep(200 * time.Millisecond)
		}

		return nil
	}

	waitUntilClean()

	ss, err := m.Snapshot()
	if err != nil {
		t.Errorf("expected ss ok")
	}
	iter, err := ss.StartIterator([]byte("b"), nil, IteratorOptions{})