diff options
Diffstat (limited to 'internal/service/mock.go')
| -rw-r--r-- | internal/service/mock.go | 23 |
1 files changed, 20 insertions, 3 deletions
diff --git a/internal/service/mock.go b/internal/service/mock.go index 82a25de..50e2e53 100644 --- a/internal/service/mock.go +++ b/internal/service/mock.go @@ -30,7 +30,9 @@ type MockMediaService struct { DownloadMediaFunc func(ctx context.Context, mediaID, userID int64) (*FileResult, error) GetThumbnailFunc func(ctx context.Context, mediaID, userID int64) (*FileResult, error) RegenerateThumbnailFunc func(ctx context.Context, mediaID, userID int64) error - RegenerateSetCoverFunc func(ctx context.Context, setID, userID int64) error + RegenerateSetCoverFunc func(ctx context.Context, setID int64, folder string, userID int64) error + BrowseSetFunc func(ctx context.Context, setID, userID int64, parent string) (*BrowseResult, error) + GetSetCoverFunc func(ctx context.Context, setID int64, folder string, userID int64) (*FileResult, error) ToggleFavoriteFunc func(ctx context.Context, userID, mediaID int64) (bool, error) AssignTagFunc func(ctx context.Context, mediaID, userID int64, tagName string) error RemoveTagFunc func(ctx context.Context, mediaID, userID int64, tagName string) error @@ -90,12 +92,24 @@ func (m *MockMediaService) RegenerateThumbnail(ctx context.Context, mediaID, use } return nil } -func (m *MockMediaService) RegenerateSetCover(ctx context.Context, setID, userID int64) error { +func (m *MockMediaService) RegenerateSetCover(ctx context.Context, setID int64, folder string, userID int64) error { if m.RegenerateSetCoverFunc != nil { - return m.RegenerateSetCoverFunc(ctx, setID, userID) + return m.RegenerateSetCoverFunc(ctx, setID, folder, userID) } return nil } +func (m *MockMediaService) BrowseSet(ctx context.Context, setID, userID int64, parent string) (*BrowseResult, error) { + if m.BrowseSetFunc != nil { + return m.BrowseSetFunc(ctx, setID, userID, parent) + } + return nil, nil +} +func (m *MockMediaService) GetSetCover(ctx context.Context, setID int64, folder string, userID int64) (*FileResult, error) { + if m.GetSetCoverFunc != nil { + return m.GetSetCoverFunc(ctx, setID, folder, userID) + } + return nil, errors.New("not implemented") +} func (m *MockMediaService) ToggleFavorite(ctx context.Context, userID, mediaID int64) (bool, error) { if m.ToggleFavoriteFunc != nil { return m.ToggleFavoriteFunc(ctx, userID, mediaID) @@ -168,6 +182,9 @@ func (m *MockMediaService) GetSharedMedia(ctx context.Context, token string) (*G } return nil, errors.New("not implemented") } +func (m *MockMediaService) GetSharedThumbnail(ctx context.Context, token string) (*FileResult, error) { + return nil, errors.New("not implemented") +} func (m *MockMediaService) GetNote(ctx context.Context, mediaID, userID int64) (*model.Note, error) { if m.GetNoteFunc != nil { return m.GetNoteFunc(ctx, mediaID, userID) |