2020-10-27 16:01:40 +01:00
|
|
|
package tests
|
2020-01-19 21:37:41 +01:00
|
|
|
|
2020-01-27 15:41:33 +01:00
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
|
|
|
|
"github.com/navidrome/navidrome/model"
|
|
|
|
)
|
2020-01-19 21:37:41 +01:00
|
|
|
|
|
|
|
type MockDataStore struct {
|
2021-06-24 06:01:05 +02:00
|
|
|
MockedGenre model.GenreRepository
|
|
|
|
MockedAlbum model.AlbumRepository
|
|
|
|
MockedArtist model.ArtistRepository
|
|
|
|
MockedMediaFile model.MediaFileRepository
|
|
|
|
MockedUser model.UserRepository
|
|
|
|
MockedProperty model.PropertyRepository
|
|
|
|
MockedPlayer model.PlayerRepository
|
|
|
|
MockedShare model.ShareRepository
|
|
|
|
MockedTranscoding model.TranscodingRepository
|
|
|
|
MockedUserProps model.UserPropsRepository
|
|
|
|
MockedScrobbleBuffer model.ScrobbleBufferRepository
|
2020-01-19 21:37:41 +01:00
|
|
|
}
|
|
|
|
|
2020-01-27 15:41:33 +01:00
|
|
|
func (db *MockDataStore) Album(context.Context) model.AlbumRepository {
|
2020-01-19 21:37:41 +01:00
|
|
|
if db.MockedAlbum == nil {
|
|
|
|
db.MockedAlbum = CreateMockAlbumRepo()
|
|
|
|
}
|
|
|
|
return db.MockedAlbum
|
|
|
|
}
|
|
|
|
|
2020-01-27 15:41:33 +01:00
|
|
|
func (db *MockDataStore) Artist(context.Context) model.ArtistRepository {
|
2020-01-19 21:37:41 +01:00
|
|
|
if db.MockedArtist == nil {
|
|
|
|
db.MockedArtist = CreateMockArtistRepo()
|
|
|
|
}
|
|
|
|
return db.MockedArtist
|
|
|
|
}
|
|
|
|
|
2020-01-27 15:41:33 +01:00
|
|
|
func (db *MockDataStore) MediaFile(context.Context) model.MediaFileRepository {
|
2020-01-19 21:37:41 +01:00
|
|
|
if db.MockedMediaFile == nil {
|
|
|
|
db.MockedMediaFile = CreateMockMediaFileRepo()
|
|
|
|
}
|
|
|
|
return db.MockedMediaFile
|
|
|
|
}
|
|
|
|
|
2020-01-27 15:41:33 +01:00
|
|
|
func (db *MockDataStore) MediaFolder(context.Context) model.MediaFolderRepository {
|
2020-01-19 21:37:41 +01:00
|
|
|
return struct{ model.MediaFolderRepository }{}
|
|
|
|
}
|
|
|
|
|
2020-01-27 15:41:33 +01:00
|
|
|
func (db *MockDataStore) Genre(context.Context) model.GenreRepository {
|
2020-01-19 21:37:41 +01:00
|
|
|
if db.MockedGenre != nil {
|
|
|
|
return db.MockedGenre
|
|
|
|
}
|
|
|
|
return struct{ model.GenreRepository }{}
|
|
|
|
}
|
|
|
|
|
2020-01-27 15:41:33 +01:00
|
|
|
func (db *MockDataStore) Playlist(context.Context) model.PlaylistRepository {
|
2020-01-19 21:37:41 +01:00
|
|
|
return struct{ model.PlaylistRepository }{}
|
|
|
|
}
|
|
|
|
|
2020-07-31 19:07:39 +02:00
|
|
|
func (db *MockDataStore) PlayQueue(context.Context) model.PlayQueueRepository {
|
|
|
|
return struct{ model.PlayQueueRepository }{}
|
|
|
|
}
|
|
|
|
|
2021-06-23 22:47:32 +02:00
|
|
|
func (db *MockDataStore) UserProps(context.Context) model.UserPropsRepository {
|
|
|
|
if db.MockedUserProps == nil {
|
|
|
|
db.MockedUserProps = &MockedUserPropsRepo{}
|
|
|
|
}
|
|
|
|
return db.MockedUserProps
|
|
|
|
}
|
|
|
|
|
2020-01-27 15:41:33 +01:00
|
|
|
func (db *MockDataStore) Property(context.Context) model.PropertyRepository {
|
2021-05-02 20:13:17 +02:00
|
|
|
if db.MockedProperty == nil {
|
2021-06-08 22:30:19 +02:00
|
|
|
db.MockedProperty = &MockedPropertyRepo{}
|
2021-05-02 20:13:17 +02:00
|
|
|
}
|
|
|
|
return db.MockedProperty
|
2020-01-19 21:37:41 +01:00
|
|
|
}
|
|
|
|
|
2021-05-30 17:50:35 +02:00
|
|
|
func (db *MockDataStore) Share(context.Context) model.ShareRepository {
|
2021-06-08 22:30:19 +02:00
|
|
|
if db.MockedShare == nil {
|
|
|
|
db.MockedShare = &MockShareRepo{}
|
|
|
|
}
|
|
|
|
return db.MockedShare
|
2021-05-30 17:50:35 +02:00
|
|
|
}
|
|
|
|
|
2020-01-27 15:41:33 +01:00
|
|
|
func (db *MockDataStore) User(context.Context) model.UserRepository {
|
2020-01-20 19:35:59 +01:00
|
|
|
if db.MockedUser == nil {
|
2021-05-16 19:25:38 +02:00
|
|
|
db.MockedUser = CreateMockUserRepo()
|
2020-01-20 19:35:59 +01:00
|
|
|
}
|
|
|
|
return db.MockedUser
|
2020-01-20 04:36:15 +01:00
|
|
|
}
|
|
|
|
|
2020-03-01 02:01:09 +01:00
|
|
|
func (db *MockDataStore) Transcoding(context.Context) model.TranscodingRepository {
|
2020-03-16 17:56:15 +01:00
|
|
|
if db.MockedTranscoding != nil {
|
|
|
|
return db.MockedTranscoding
|
|
|
|
}
|
2020-03-01 02:01:09 +01:00
|
|
|
return struct{ model.TranscodingRepository }{}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (db *MockDataStore) Player(context.Context) model.PlayerRepository {
|
|
|
|
if db.MockedPlayer != nil {
|
|
|
|
return db.MockedPlayer
|
|
|
|
}
|
|
|
|
return struct{ model.PlayerRepository }{}
|
|
|
|
}
|
|
|
|
|
2021-06-24 06:01:05 +02:00
|
|
|
func (db *MockDataStore) ScrobbleBuffer(ctx context.Context) model.ScrobbleBufferRepository {
|
|
|
|
if db.MockedScrobbleBuffer == nil {
|
|
|
|
db.MockedScrobbleBuffer = CreateMockedScrobbleBufferRepo()
|
|
|
|
}
|
|
|
|
return db.MockedScrobbleBuffer
|
|
|
|
}
|
|
|
|
|
2020-01-19 21:37:41 +01:00
|
|
|
func (db *MockDataStore) WithTx(block func(db model.DataStore) error) error {
|
|
|
|
return block(db)
|
|
|
|
}
|
2020-01-20 02:40:18 +01:00
|
|
|
|
2020-01-27 15:41:33 +01:00
|
|
|
func (db *MockDataStore) Resource(ctx context.Context, m interface{}) model.ResourceRepository {
|
2020-01-20 02:40:18 +01:00
|
|
|
return struct{ model.ResourceRepository }{}
|
|
|
|
}
|
2020-01-20 19:35:59 +01:00
|
|
|
|
2020-10-02 22:18:45 +02:00
|
|
|
func (db *MockDataStore) GC(ctx context.Context, rootFolder string) error {
|
2020-02-01 02:17:36 +01:00
|
|
|
return nil
|
|
|
|
}
|