github.com/kubeshop/testkube@v1.17.23/pkg/dbmigrator/mock_database.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: github.com/kubeshop/testkube/pkg/dbmigrator (interfaces: Database)
     3  
     4  // Package dbmigrator is a generated GoMock package.
     5  package dbmigrator
     6  
     7  import (
     8  	context "context"
     9  	reflect "reflect"
    10  
    11  	gomock "github.com/golang/mock/gomock"
    12  	primitive "go.mongodb.org/mongo-driver/bson/primitive"
    13  )
    14  
    15  // MockDatabase is a mock of Database interface.
    16  type MockDatabase struct {
    17  	ctrl     *gomock.Controller
    18  	recorder *MockDatabaseMockRecorder
    19  }
    20  
    21  // MockDatabaseMockRecorder is the mock recorder for MockDatabase.
    22  type MockDatabaseMockRecorder struct {
    23  	mock *MockDatabase
    24  }
    25  
    26  // NewMockDatabase creates a new mock instance.
    27  func NewMockDatabase(ctrl *gomock.Controller) *MockDatabase {
    28  	mock := &MockDatabase{ctrl: ctrl}
    29  	mock.recorder = &MockDatabaseMockRecorder{mock}
    30  	return mock
    31  }
    32  
    33  // EXPECT returns an object that allows the caller to indicate expected use.
    34  func (m *MockDatabase) EXPECT() *MockDatabaseMockRecorder {
    35  	return m.recorder
    36  }
    37  
    38  // DeleteMigrationState mocks base method.
    39  func (m *MockDatabase) DeleteMigrationState(arg0 context.Context, arg1 *DbMigration) error {
    40  	m.ctrl.T.Helper()
    41  	ret := m.ctrl.Call(m, "DeleteMigrationState", arg0, arg1)
    42  	ret0, _ := ret[0].(error)
    43  	return ret0
    44  }
    45  
    46  // DeleteMigrationState indicates an expected call of DeleteMigrationState.
    47  func (mr *MockDatabaseMockRecorder) DeleteMigrationState(arg0, arg1 interface{}) *gomock.Call {
    48  	mr.mock.ctrl.T.Helper()
    49  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteMigrationState", reflect.TypeOf((*MockDatabase)(nil).DeleteMigrationState), arg0, arg1)
    50  }
    51  
    52  // GetAppliedMigrations mocks base method.
    53  func (m *MockDatabase) GetAppliedMigrations(arg0 context.Context) ([]DbMigration, error) {
    54  	m.ctrl.T.Helper()
    55  	ret := m.ctrl.Call(m, "GetAppliedMigrations", arg0)
    56  	ret0, _ := ret[0].([]DbMigration)
    57  	ret1, _ := ret[1].(error)
    58  	return ret0, ret1
    59  }
    60  
    61  // GetAppliedMigrations indicates an expected call of GetAppliedMigrations.
    62  func (mr *MockDatabaseMockRecorder) GetAppliedMigrations(arg0 interface{}) *gomock.Call {
    63  	mr.mock.ctrl.T.Helper()
    64  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAppliedMigrations", reflect.TypeOf((*MockDatabase)(nil).GetAppliedMigrations), arg0)
    65  }
    66  
    67  // InsertMigrationState mocks base method.
    68  func (m *MockDatabase) InsertMigrationState(arg0 context.Context, arg1 *DbMigration) error {
    69  	m.ctrl.T.Helper()
    70  	ret := m.ctrl.Call(m, "InsertMigrationState", arg0, arg1)
    71  	ret0, _ := ret[0].(error)
    72  	return ret0
    73  }
    74  
    75  // InsertMigrationState indicates an expected call of InsertMigrationState.
    76  func (mr *MockDatabaseMockRecorder) InsertMigrationState(arg0, arg1 interface{}) *gomock.Call {
    77  	mr.mock.ctrl.T.Helper()
    78  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertMigrationState", reflect.TypeOf((*MockDatabase)(nil).InsertMigrationState), arg0, arg1)
    79  }
    80  
    81  // RunCommands mocks base method.
    82  func (m *MockDatabase) RunCommands(arg0 context.Context, arg1 []primitive.D) error {
    83  	m.ctrl.T.Helper()
    84  	ret := m.ctrl.Call(m, "RunCommands", arg0, arg1)
    85  	ret0, _ := ret[0].(error)
    86  	return ret0
    87  }
    88  
    89  // RunCommands indicates an expected call of RunCommands.
    90  func (mr *MockDatabaseMockRecorder) RunCommands(arg0, arg1 interface{}) *gomock.Call {
    91  	mr.mock.ctrl.T.Helper()
    92  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RunCommands", reflect.TypeOf((*MockDatabase)(nil).RunCommands), arg0, arg1)
    93  }