// Code generated by mockery v2.10.0. DO NOT EDIT. package mocks import ( context "context" models "github.com/stashapp/stash/pkg/models" mock "github.com/stretchr/testify/mock" ) // StudioReaderWriter is an autogenerated mock type for the StudioReaderWriter type type StudioReaderWriter struct { mock.Mock } // All provides a mock function with given fields: ctx func (_m *StudioReaderWriter) All(ctx context.Context) ([]*models.Studio, error) { ret := _m.Called(ctx) var r0 []*models.Studio if rf, ok := ret.Get(0).(func(context.Context) []*models.Studio); ok { r0 = rf(ctx) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]*models.Studio) } } var r1 error if rf, ok := ret.Get(1).(func(context.Context) error); ok { r1 = rf(ctx) } else { r1 = ret.Error(1) } return r0, r1 } // Count provides a mock function with given fields: ctx func (_m *StudioReaderWriter) Count(ctx context.Context) (int, error) { ret := _m.Called(ctx) var r0 int if rf, ok := ret.Get(0).(func(context.Context) int); ok { r0 = rf(ctx) } else { r0 = ret.Get(0).(int) } var r1 error if rf, ok := ret.Get(1).(func(context.Context) error); ok { r1 = rf(ctx) } else { r1 = ret.Error(1) } return r0, r1 } // Create provides a mock function with given fields: ctx, newStudio func (_m *StudioReaderWriter) Create(ctx context.Context, newStudio models.Studio) (*models.Studio, error) { ret := _m.Called(ctx, newStudio) var r0 *models.Studio if rf, ok := ret.Get(0).(func(context.Context, models.Studio) *models.Studio); ok { r0 = rf(ctx, newStudio) } else { if ret.Get(0) != nil { r0 = ret.Get(0).(*models.Studio) } } var r1 error if rf, ok := ret.Get(1).(func(context.Context, models.Studio) error); ok { r1 = rf(ctx, newStudio) } else { r1 = ret.Error(1) } return r0, r1 } // Destroy provides a mock function with given fields: ctx, id func (_m *StudioReaderWriter) Destroy(ctx context.Context, id int) error { ret := _m.Called(ctx, id) var r0 error if rf, ok := ret.Get(0).(func(context.Context, int) error); ok { r0 = rf(ctx, id) } else { r0 = ret.Error(0) } return r0 } // DestroyImage provides a mock function with given fields: ctx, studioID func (_m *StudioReaderWriter) DestroyImage(ctx context.Context, studioID int) error { ret := _m.Called(ctx, studioID) var r0 error if rf, ok := ret.Get(0).(func(context.Context, int) error); ok { r0 = rf(ctx, studioID) } else { r0 = ret.Error(0) } return r0 } // Find provides a mock function with given fields: ctx, id func (_m *StudioReaderWriter) Find(ctx context.Context, id int) (*models.Studio, error) { ret := _m.Called(ctx, id) var r0 *models.Studio if rf, ok := ret.Get(0).(func(context.Context, int) *models.Studio); ok { r0 = rf(ctx, id) } else { if ret.Get(0) != nil { r0 = ret.Get(0).(*models.Studio) } } var r1 error if rf, ok := ret.Get(1).(func(context.Context, int) error); ok { r1 = rf(ctx, id) } else { r1 = ret.Error(1) } return r0, r1 } // FindByName provides a mock function with given fields: ctx, name, nocase func (_m *StudioReaderWriter) FindByName(ctx context.Context, name string, nocase bool) (*models.Studio, error) { ret := _m.Called(ctx, name, nocase) var r0 *models.Studio if rf, ok := ret.Get(0).(func(context.Context, string, bool) *models.Studio); ok { r0 = rf(ctx, name, nocase) } else { if ret.Get(0) != nil { r0 = ret.Get(0).(*models.Studio) } } var r1 error if rf, ok := ret.Get(1).(func(context.Context, string, bool) error); ok { r1 = rf(ctx, name, nocase) } else { r1 = ret.Error(1) } return r0, r1 } // FindByStashID provides a mock function with given fields: ctx, stashID func (_m *StudioReaderWriter) FindByStashID(ctx context.Context, stashID models.StashID) ([]*models.Studio, error) { ret := _m.Called(ctx, stashID) var r0 []*models.Studio if rf, ok := ret.Get(0).(func(context.Context, models.StashID) []*models.Studio); ok { r0 = rf(ctx, stashID) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]*models.Studio) } } var r1 error if rf, ok := ret.Get(1).(func(context.Context, models.StashID) error); ok { r1 = rf(ctx, stashID) } else { r1 = ret.Error(1) } return r0, r1 } // FindChildren provides a mock function with given fields: ctx, id func (_m *StudioReaderWriter) FindChildren(ctx context.Context, id int) ([]*models.Studio, error) { ret := _m.Called(ctx, id) var r0 []*models.Studio if rf, ok := ret.Get(0).(func(context.Context, int) []*models.Studio); ok { r0 = rf(ctx, id) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]*models.Studio) } } var r1 error if rf, ok := ret.Get(1).(func(context.Context, int) error); ok { r1 = rf(ctx, id) } else { r1 = ret.Error(1) } return r0, r1 } // FindMany provides a mock function with given fields: ctx, ids func (_m *StudioReaderWriter) FindMany(ctx context.Context, ids []int) ([]*models.Studio, error) { ret := _m.Called(ctx, ids) var r0 []*models.Studio if rf, ok := ret.Get(0).(func(context.Context, []int) []*models.Studio); ok { r0 = rf(ctx, ids) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]*models.Studio) } } var r1 error if rf, ok := ret.Get(1).(func(context.Context, []int) error); ok { r1 = rf(ctx, ids) } else { r1 = ret.Error(1) } return r0, r1 } // GetAliases provides a mock function with given fields: ctx, studioID func (_m *StudioReaderWriter) GetAliases(ctx context.Context, studioID int) ([]string, error) { ret := _m.Called(ctx, studioID) var r0 []string if rf, ok := ret.Get(0).(func(context.Context, int) []string); ok { r0 = rf(ctx, studioID) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]string) } } var r1 error if rf, ok := ret.Get(1).(func(context.Context, int) error); ok { r1 = rf(ctx, studioID) } else { r1 = ret.Error(1) } return r0, r1 } // GetImage provides a mock function with given fields: ctx, studioID func (_m *StudioReaderWriter) GetImage(ctx context.Context, studioID int) ([]byte, error) { ret := _m.Called(ctx, studioID) var r0 []byte if rf, ok := ret.Get(0).(func(context.Context, int) []byte); ok { r0 = rf(ctx, studioID) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]byte) } } var r1 error if rf, ok := ret.Get(1).(func(context.Context, int) error); ok { r1 = rf(ctx, studioID) } else { r1 = ret.Error(1) } return r0, r1 } // GetStashIDs provides a mock function with given fields: ctx, relatedID func (_m *StudioReaderWriter) GetStashIDs(ctx context.Context, relatedID int) ([]models.StashID, error) { ret := _m.Called(ctx, relatedID) var r0 []models.StashID if rf, ok := ret.Get(0).(func(context.Context, int) []models.StashID); ok { r0 = rf(ctx, relatedID) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]models.StashID) } } var r1 error if rf, ok := ret.Get(1).(func(context.Context, int) error); ok { r1 = rf(ctx, relatedID) } else { r1 = ret.Error(1) } return r0, r1 } // HasImage provides a mock function with given fields: ctx, studioID func (_m *StudioReaderWriter) HasImage(ctx context.Context, studioID int) (bool, error) { ret := _m.Called(ctx, studioID) var r0 bool if rf, ok := ret.Get(0).(func(context.Context, int) bool); ok { r0 = rf(ctx, studioID) } else { r0 = ret.Get(0).(bool) } var r1 error if rf, ok := ret.Get(1).(func(context.Context, int) error); ok { r1 = rf(ctx, studioID) } else { r1 = ret.Error(1) } return r0, r1 } // Query provides a mock function with given fields: ctx, studioFilter, findFilter func (_m *StudioReaderWriter) Query(ctx context.Context, studioFilter *models.StudioFilterType, findFilter *models.FindFilterType) ([]*models.Studio, int, error) { ret := _m.Called(ctx, studioFilter, findFilter) var r0 []*models.Studio if rf, ok := ret.Get(0).(func(context.Context, *models.StudioFilterType, *models.FindFilterType) []*models.Studio); ok { r0 = rf(ctx, studioFilter, findFilter) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]*models.Studio) } } var r1 int if rf, ok := ret.Get(1).(func(context.Context, *models.StudioFilterType, *models.FindFilterType) int); ok { r1 = rf(ctx, studioFilter, findFilter) } else { r1 = ret.Get(1).(int) } var r2 error if rf, ok := ret.Get(2).(func(context.Context, *models.StudioFilterType, *models.FindFilterType) error); ok { r2 = rf(ctx, studioFilter, findFilter) } else { r2 = ret.Error(2) } return r0, r1, r2 } // QueryForAutoTag provides a mock function with given fields: ctx, words func (_m *StudioReaderWriter) QueryForAutoTag(ctx context.Context, words []string) ([]*models.Studio, error) { ret := _m.Called(ctx, words) var r0 []*models.Studio if rf, ok := ret.Get(0).(func(context.Context, []string) []*models.Studio); ok { r0 = rf(ctx, words) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]*models.Studio) } } var r1 error if rf, ok := ret.Get(1).(func(context.Context, []string) error); ok { r1 = rf(ctx, words) } else { r1 = ret.Error(1) } return r0, r1 } // Update provides a mock function with given fields: ctx, updatedStudio func (_m *StudioReaderWriter) Update(ctx context.Context, updatedStudio models.StudioPartial) (*models.Studio, error) { ret := _m.Called(ctx, updatedStudio) var r0 *models.Studio if rf, ok := ret.Get(0).(func(context.Context, models.StudioPartial) *models.Studio); ok { r0 = rf(ctx, updatedStudio) } else { if ret.Get(0) != nil { r0 = ret.Get(0).(*models.Studio) } } var r1 error if rf, ok := ret.Get(1).(func(context.Context, models.StudioPartial) error); ok { r1 = rf(ctx, updatedStudio) } else { r1 = ret.Error(1) } return r0, r1 } // UpdateAliases provides a mock function with given fields: ctx, studioID, aliases func (_m *StudioReaderWriter) UpdateAliases(ctx context.Context, studioID int, aliases []string) error { ret := _m.Called(ctx, studioID, aliases) var r0 error if rf, ok := ret.Get(0).(func(context.Context, int, []string) error); ok { r0 = rf(ctx, studioID, aliases) } else { r0 = ret.Error(0) } return r0 } // UpdateFull provides a mock function with given fields: ctx, updatedStudio func (_m *StudioReaderWriter) UpdateFull(ctx context.Context, updatedStudio models.Studio) (*models.Studio, error) { ret := _m.Called(ctx, updatedStudio) var r0 *models.Studio if rf, ok := ret.Get(0).(func(context.Context, models.Studio) *models.Studio); ok { r0 = rf(ctx, updatedStudio) } else { if ret.Get(0) != nil { r0 = ret.Get(0).(*models.Studio) } } var r1 error if rf, ok := ret.Get(1).(func(context.Context, models.Studio) error); ok { r1 = rf(ctx, updatedStudio) } else { r1 = ret.Error(1) } return r0, r1 } // UpdateImage provides a mock function with given fields: ctx, studioID, image func (_m *StudioReaderWriter) UpdateImage(ctx context.Context, studioID int, image []byte) error { ret := _m.Called(ctx, studioID, image) var r0 error if rf, ok := ret.Get(0).(func(context.Context, int, []byte) error); ok { r0 = rf(ctx, studioID, image) } else { r0 = ret.Error(0) } return r0 } // UpdateStashIDs provides a mock function with given fields: ctx, studioID, stashIDs func (_m *StudioReaderWriter) UpdateStashIDs(ctx context.Context, studioID int, stashIDs []models.StashID) error { ret := _m.Called(ctx, studioID, stashIDs) var r0 error if rf, ok := ret.Get(0).(func(context.Context, int, []models.StashID) error); ok { r0 = rf(ctx, studioID, stashIDs) } else { r0 = ret.Error(0) } return r0 }