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=74b5e054dc40ed47cbbbb28f1fc7c80b50efcae6;hp=f6d079188f8308d7774037823b92abdc7ccd3b21;hpb=273d032db02901dba8deeb692f92ed2366dc6ee5;p=ric-plt%2Fsdlgo.git

diff --git a/internal/mocks/db_mocks_private_testing.go b/internal/mocks/db_mocks_private_testing.go
index f6d0791..1a4e29b 100644
--- a/internal/mocks/db_mocks_private_testing.go
+++ b/internal/mocks/db_mocks_private_testing.go
@@ -41,6 +41,16 @@ func (m *MockDB) State() (*sdlgoredis.DbState, error) {
 	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
 }
@@ -49,3 +59,18 @@ 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)
+}