mirror of
https://github.com/stashapp/stash.git
synced 2025-12-18 04:44:37 +03:00
Restructure data layer (#2532)
* Add new txn manager interface * Add txn management to sqlite * Rename get to getByID * Add contexts to repository methods * Update query builders * Add context to reader writer interfaces * Use repository in resolver * Tighten interfaces * Tighten interfaces in dlna * Tighten interfaces in match package * Tighten interfaces in scraper package * Tighten interfaces in scan code * Tighten interfaces on autotag package * Remove ReaderWriter usage * Merge database package into sqlite
This commit is contained in:
@@ -3,6 +3,8 @@
|
||||
package mocks
|
||||
|
||||
import (
|
||||
context "context"
|
||||
|
||||
models "github.com/stashapp/stash/pkg/models"
|
||||
mock "github.com/stretchr/testify/mock"
|
||||
)
|
||||
@@ -12,13 +14,13 @@ type ScrapedItemReaderWriter struct {
|
||||
mock.Mock
|
||||
}
|
||||
|
||||
// All provides a mock function with given fields:
|
||||
func (_m *ScrapedItemReaderWriter) All() ([]*models.ScrapedItem, error) {
|
||||
ret := _m.Called()
|
||||
// All provides a mock function with given fields: ctx
|
||||
func (_m *ScrapedItemReaderWriter) All(ctx context.Context) ([]*models.ScrapedItem, error) {
|
||||
ret := _m.Called(ctx)
|
||||
|
||||
var r0 []*models.ScrapedItem
|
||||
if rf, ok := ret.Get(0).(func() []*models.ScrapedItem); ok {
|
||||
r0 = rf()
|
||||
if rf, ok := ret.Get(0).(func(context.Context) []*models.ScrapedItem); ok {
|
||||
r0 = rf(ctx)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]*models.ScrapedItem)
|
||||
@@ -26,8 +28,8 @@ func (_m *ScrapedItemReaderWriter) All() ([]*models.ScrapedItem, error) {
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func() error); ok {
|
||||
r1 = rf()
|
||||
if rf, ok := ret.Get(1).(func(context.Context) error); ok {
|
||||
r1 = rf(ctx)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
@@ -35,13 +37,13 @@ func (_m *ScrapedItemReaderWriter) All() ([]*models.ScrapedItem, error) {
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// Create provides a mock function with given fields: newObject
|
||||
func (_m *ScrapedItemReaderWriter) Create(newObject models.ScrapedItem) (*models.ScrapedItem, error) {
|
||||
ret := _m.Called(newObject)
|
||||
// Create provides a mock function with given fields: ctx, newObject
|
||||
func (_m *ScrapedItemReaderWriter) Create(ctx context.Context, newObject models.ScrapedItem) (*models.ScrapedItem, error) {
|
||||
ret := _m.Called(ctx, newObject)
|
||||
|
||||
var r0 *models.ScrapedItem
|
||||
if rf, ok := ret.Get(0).(func(models.ScrapedItem) *models.ScrapedItem); ok {
|
||||
r0 = rf(newObject)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, models.ScrapedItem) *models.ScrapedItem); ok {
|
||||
r0 = rf(ctx, newObject)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).(*models.ScrapedItem)
|
||||
@@ -49,8 +51,8 @@ func (_m *ScrapedItemReaderWriter) Create(newObject models.ScrapedItem) (*models
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(models.ScrapedItem) error); ok {
|
||||
r1 = rf(newObject)
|
||||
if rf, ok := ret.Get(1).(func(context.Context, models.ScrapedItem) error); ok {
|
||||
r1 = rf(ctx, newObject)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user