tests to assert panics and getStoreType

sei-protocol · Oct 4, 2023 · 3199d07 · 3199d07
1 parent 478d177
commit 3199d07
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/store/multiversion/mvkv.go b/store/multiversion/mvkv.go
@@ -168,7 +168,7 @@ func (v *VersionIndexedStore) ReverseIterator(start []byte, end []byte) dbm.Iter
 
 // GetStoreType implements types.KVStore.
 func (v *VersionIndexedStore) GetStoreType() types.StoreType {
-	panic("unimplemented")
+	return v.parent.GetStoreType()
 }
 
 // CacheWrap implements types.KVStore.

diff --git a/store/multiversion/mvkv_test.go b/store/multiversion/mvkv_test.go
@@ -108,3 +108,20 @@ func TestVersionIndexedStoreSetters(t *testing.T) {
 	require.Equal(t, []byte("value3"), vis.Get([]byte("key2")))
 	require.Zero(t, len(vis.GetReadset()))
 }
+
+func TestVersionIndexedStoreBoilerplateFunctions(t *testing.T) {
+	mem := dbadapter.Store{DB: dbm.NewMemDB()}
+	parentKVStore := cachekv.NewStore(mem, types.NewKVStoreKey("mock"), 1000)
+	mvs := multiversion.NewMultiVersionStore()
+	// initialize a new VersionIndexedStore
+	vis := multiversion.NewVersionIndexedStore(parentKVStore, mvs, 1, 2, make(chan scheduler.Abort))
+
+	// asserts panics where appropriate
+	require.Panics(t, func() { vis.CacheWrap(types.NewKVStoreKey("mock")) })
+	require.Panics(t, func() { vis.CacheWrapWithListeners(types.NewKVStoreKey("mock"), nil) })
+	require.Panics(t, func() { vis.CacheWrapWithTrace(types.NewKVStoreKey("mock"), nil, nil) })
+	require.Panics(t, func() { vis.GetWorkingHash() })
+
+	// assert properly returns store type
+	require.Equal(t, types.StoreTypeDB, vis.GetStoreType())
+}