X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=internal%2Fmocks%2Fdb_mocks_private_testing.go;h=1a4e29b0d971ffa82c3eba1e66fa911685ecae2c;hb=refs%2Fchanges%2F50%2F9150%2F1;hp=7832ee399b93b6a3cc4742efe43c4d386b78afca;hpb=977a55ca96d5dba1c7f9273671747eaf9cd6f894;p=ric-plt%2Fsdlgo.git

diff --git a/internal/mocks/db_mocks_private_testing.go b/internal/mocks/db_mocks_private_testing.go
index 7832ee3..1a4e29b 100644
--- a/internal/mocks/db_mocks_private_testing.go
+++ b/internal/mocks/db_mocks_private_testing.go
@@ -40,3 +40,37 @@ func (m *MockDB) State() (*sdlgoredis.DbState, error) {
 	a := m.Called()
 	return a.Get(0).(*sdlgoredis.DbState), a.Error(1)
 }
+
+func (m *MockDB) Keys(pattern string) ([]string, error) {
+	a := m.Called(pattern)
+	return a.Get(0).([]string), a.Error(1)
+}
+
+func (m *MockDB) Statistics() (*sdlgoredis.DbStatistics, error) {
+	a := m.Called()
+	return a.Get(0).(*sdlgoredis.DbStatistics), a.Error(1)
+}
+
+type MockSdlApi struct {
+	mock.Mock
+}
+
+func (m *MockSdlApi) ListKeys(ns string, pattern string) ([]string, error) {
+	a := m.Called(ns, pattern)
+	return a.Get(0).([]string), a.Error(1)
+}
+
+func (m *MockSdlApi) Set(ns string, pairs ...interface{}) error {
+	a := m.Called(ns, pairs)
+	return a.Error(0)
+}
+
+func (m *MockSdlApi) Remove(ns string, keys []string) error {
+	a := m.Called(ns, keys)
+	return a.Error(0)
+}
+
+func (m *MockSdlApi) Get(ns string, keys []string) (map[string]interface{}, error) {
+	a := m.Called(ns, keys)
+	return a.Get(0).(map[string]interface{}), a.Error(1)
+}