mirror of
https://github.com/stashapp/stash.git
synced 2025-12-18 04:44:37 +03:00
Model refactor, part 3 (#4152)
* Remove manager.Repository * Refactor other repositories * Fix tests and add database mock * Add AssertExpectations method * Refactor routes * Move default movie image to internal/static and add convenience methods * Refactor default performer image boxes
This commit is contained in:
107
pkg/models/mocks/database.go
Normal file
107
pkg/models/mocks/database.go
Normal file
@@ -0,0 +1,107 @@
|
||||
package mocks
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/stashapp/stash/pkg/models"
|
||||
"github.com/stashapp/stash/pkg/txn"
|
||||
"github.com/stretchr/testify/mock"
|
||||
)
|
||||
|
||||
type Database struct {
|
||||
File *FileReaderWriter
|
||||
Folder *FolderReaderWriter
|
||||
Gallery *GalleryReaderWriter
|
||||
GalleryChapter *GalleryChapterReaderWriter
|
||||
Image *ImageReaderWriter
|
||||
Movie *MovieReaderWriter
|
||||
Performer *PerformerReaderWriter
|
||||
Scene *SceneReaderWriter
|
||||
SceneMarker *SceneMarkerReaderWriter
|
||||
Studio *StudioReaderWriter
|
||||
Tag *TagReaderWriter
|
||||
SavedFilter *SavedFilterReaderWriter
|
||||
}
|
||||
|
||||
func (*Database) Begin(ctx context.Context, exclusive bool) (context.Context, error) {
|
||||
return ctx, nil
|
||||
}
|
||||
|
||||
func (*Database) WithDatabase(ctx context.Context) (context.Context, error) {
|
||||
return ctx, nil
|
||||
}
|
||||
|
||||
func (*Database) Commit(ctx context.Context) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (*Database) Rollback(ctx context.Context) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (*Database) Complete(ctx context.Context) {
|
||||
}
|
||||
|
||||
func (*Database) AddPostCommitHook(ctx context.Context, hook txn.TxnFunc) {
|
||||
}
|
||||
|
||||
func (*Database) AddPostRollbackHook(ctx context.Context, hook txn.TxnFunc) {
|
||||
}
|
||||
|
||||
func (*Database) IsLocked(err error) bool {
|
||||
return false
|
||||
}
|
||||
|
||||
func (*Database) Reset() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func NewDatabase() *Database {
|
||||
return &Database{
|
||||
File: &FileReaderWriter{},
|
||||
Folder: &FolderReaderWriter{},
|
||||
Gallery: &GalleryReaderWriter{},
|
||||
GalleryChapter: &GalleryChapterReaderWriter{},
|
||||
Image: &ImageReaderWriter{},
|
||||
Movie: &MovieReaderWriter{},
|
||||
Performer: &PerformerReaderWriter{},
|
||||
Scene: &SceneReaderWriter{},
|
||||
SceneMarker: &SceneMarkerReaderWriter{},
|
||||
Studio: &StudioReaderWriter{},
|
||||
Tag: &TagReaderWriter{},
|
||||
SavedFilter: &SavedFilterReaderWriter{},
|
||||
}
|
||||
}
|
||||
|
||||
func (db *Database) AssertExpectations(t mock.TestingT) {
|
||||
db.File.AssertExpectations(t)
|
||||
db.Folder.AssertExpectations(t)
|
||||
db.Gallery.AssertExpectations(t)
|
||||
db.GalleryChapter.AssertExpectations(t)
|
||||
db.Image.AssertExpectations(t)
|
||||
db.Movie.AssertExpectations(t)
|
||||
db.Performer.AssertExpectations(t)
|
||||
db.Scene.AssertExpectations(t)
|
||||
db.SceneMarker.AssertExpectations(t)
|
||||
db.Studio.AssertExpectations(t)
|
||||
db.Tag.AssertExpectations(t)
|
||||
db.SavedFilter.AssertExpectations(t)
|
||||
}
|
||||
|
||||
func (db *Database) Repository() models.Repository {
|
||||
return models.Repository{
|
||||
TxnManager: db,
|
||||
File: db.File,
|
||||
Folder: db.Folder,
|
||||
Gallery: db.Gallery,
|
||||
GalleryChapter: db.GalleryChapter,
|
||||
Image: db.Image,
|
||||
Movie: db.Movie,
|
||||
Performer: db.Performer,
|
||||
Scene: db.Scene,
|
||||
SceneMarker: db.SceneMarker,
|
||||
Studio: db.Studio,
|
||||
Tag: db.Tag,
|
||||
SavedFilter: db.SavedFilter,
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user