Added the subPrefixDb mock

This commit is contained in:
Alfred Kamanda 2024-10-28 16:23:44 +03:00
parent 5d294b663c
commit 520f5abdcd
Signed by: Alfred-mk
GPG Key ID: 7EA3D01708908703

View File

@ -0,0 +1,21 @@
package mocks
import (
"context"
"github.com/stretchr/testify/mock"
)
type MockSubPrefixDb struct {
mock.Mock
}
func (m *MockSubPrefixDb) Get(ctx context.Context, key []byte) ([]byte, error) {
args := m.Called(ctx, key)
return args.Get(0).([]byte), args.Error(1)
}
func (m *MockSubPrefixDb) Put(ctx context.Context, key, val []byte) error {
args := m.Called(ctx, key, val)
return args.Error(0)
}