Merge "RIC:1060: Change in PTL"

[ric-plt/sdlgo.git] / internal / mocks / db_mocks_private_testing.go

diff --git a/internal/mocks/db_mocks_private_testing.go b/internal/mocks/db_mocks_private_testing.go
index 45fbb78..1a4e29b 100644 (file)
--- 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
 }
@@ -54,3 +64,13 @@ 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)
+}