services/syncbase/store/test/stream.go

// Copyright 2015 The Vanadium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package test

import (
	"bytes"
	"testing"

	"v.io/v23/verror"
	"v.io/x/ref/services/syncbase/store"
)

// RunStreamTest verifies store.Stream operations.
func RunStreamTest(t *testing.T, st store.Store) {
	// Test that advancing or canceling a stream that has reached its end
	// doesn't cause a panic.
	s := st.Scan([]byte("a"), []byte("z"))
	verifyAdvance(t, s, nil, nil)
	verifyAdvance(t, s, nil, nil)
	if s.Err() != nil {
		t.Fatalf("unexpected error: %v", s.Err())
	}
	s.Cancel()
	if s.Err() != nil {
		t.Fatalf("unexpected error: %v", s.Err())
	}

	key1, value1 := []byte("key1"), []byte("value1")
	st.Put(key1, value1)
	key2, value2 := []byte("key2"), []byte("value2")
	st.Put(key2, value2)
	key3, value3 := []byte("key3"), []byte("value3")
	st.Put(key3, value3)
	s = st.Scan([]byte("a"), []byte("z"))
	verifyAdvance(t, s, key1, value1)
	if !s.Advance() {
		t.Fatalf("can't advance the stream")
	}
	s.Cancel()
	for i := 0; i < 2; i++ {
		var key, value []byte
		if key = s.Key(key); !bytes.Equal(key, key2) {
			t.Fatalf("unexpected key: got %q, want %q", key, key2)
		}
		if value = s.Value(value); !bytes.Equal(value, value2) {
			t.Fatalf("unexpected value: got %q, want %q", value, value2)
		}
	}
	verifyAdvance(t, s, nil, nil)
	verifyError(t, s.Err(), verror.ErrCanceled.ID, store.ErrMsgCanceledStream)
}