34 lines
564 B
Go
34 lines
564 B
Go
package tests
|
|
|
|
import (
|
|
"github.com/deluan/rest"
|
|
"github.com/navidrome/navidrome/model"
|
|
)
|
|
|
|
type MockPlaylistRepo struct {
|
|
model.PlaylistRepository
|
|
|
|
Entity *model.Playlist
|
|
Error error
|
|
}
|
|
|
|
func (m *MockPlaylistRepo) Get(_ string) (*model.Playlist, error) {
|
|
if m.Error != nil {
|
|
return nil, m.Error
|
|
}
|
|
if m.Entity == nil {
|
|
return nil, model.ErrNotFound
|
|
}
|
|
return m.Entity, nil
|
|
}
|
|
|
|
func (m *MockPlaylistRepo) Count(_ ...rest.QueryOptions) (int64, error) {
|
|
if m.Error != nil {
|
|
return 0, m.Error
|
|
}
|
|
if m.Entity == nil {
|
|
return 0, nil
|
|
}
|
|
return 1, nil
|
|
}
|