navidrome/engine/mock_nowplaying_repo.go

package engine

import (
	"errors"
	"time"
)

func CreateMockNowPlayingRepo() *MockNowPlaying {
	return &MockNowPlaying{}
}

type MockNowPlaying struct {
	NowPlayingRepository
	data []NowPlayingInfo
	t    time.Time
	err  bool
}

func (m *MockNowPlaying) SetError(err bool) {
	m.err = err
}

func (m *MockNowPlaying) Enqueue(info *NowPlayingInfo) error {
	if m.err {
		return errors.New("Error!")
	}

	m.data = append(m.data, NowPlayingInfo{})
	copy(m.data[1:], m.data[0:])
	m.data[0] = *info

	if !m.t.IsZero() {
		m.data[0].Start = m.t
		m.t = time.Time{}
	}

	return nil
}

func (m *MockNowPlaying) Dequeue(playerId int) (*NowPlayingInfo, error) {
	if len(m.data) == 0 {
		return nil, nil
	}
	l := len(m.data)
	info := m.data[l-1]
	m.data = m.data[:l-1]

	return &info, nil
}

func (m *MockNowPlaying) Count(playerId int) (int64, error) {
	return int64(len(m.data)), nil
}

func (m *MockNowPlaying) GetAll() ([]*NowPlayingInfo, error) {
	np, err := m.Head(1)
	if np == nil || err != nil {
		return nil, err
	}
	return []*NowPlayingInfo{np}, err
}

func (m *MockNowPlaying) Head(playerId int) (*NowPlayingInfo, error) {
	if len(m.data) == 0 {
		return nil, nil
	}
	info := m.data[0]
	return &info, nil
}

func (m *MockNowPlaying) Tail(playerId int) (*NowPlayingInfo, error) {
	if len(m.data) == 0 {
		return nil, nil
	}
	info := m.data[len(m.data)-1]
	return &info, nil
}

func (m *MockNowPlaying) ClearAll() {
	m.data = make([]NowPlayingInfo, 0)
	m.err = false
}

func (m *MockNowPlaying) OverrideNow(t time.Time) {
	m.t = t
}