138 lines
3.7 KiB
Go
138 lines
3.7 KiB
Go
package engine_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/deluan/gosonic/engine"
|
|
"github.com/deluan/gosonic/itunesbridge"
|
|
"github.com/deluan/gosonic/persistence"
|
|
. "github.com/deluan/gosonic/tests"
|
|
. "github.com/smartystreets/goconvey/convey"
|
|
"github.com/syndtr/goleveldb/leveldb/errors"
|
|
)
|
|
|
|
func TestScrobbler(t *testing.T) {
|
|
|
|
Init(t, false)
|
|
|
|
mfRepo := persistence.CreateMockMediaFileRepo()
|
|
npRepo := engine.CreateMockNowPlayingRepo()
|
|
itCtrl := &mockItunesControl{}
|
|
|
|
scrobbler := engine.NewScrobbler(itCtrl, mfRepo, npRepo)
|
|
|
|
Convey("Given a DB with one song", t, func() {
|
|
mfRepo.SetData(`[{"Id":"2","Title":"Hands Of Time"}]`, 1)
|
|
|
|
Convey("When I scrobble an existing song", func() {
|
|
now := time.Now()
|
|
mf, err := scrobbler.Register(1, "2", now)
|
|
|
|
Convey("Then I get the scrobbled song back", func() {
|
|
So(err, ShouldBeNil)
|
|
So(mf.Title, ShouldEqual, "Hands Of Time")
|
|
})
|
|
|
|
Convey("And iTunes is notified", func() {
|
|
So(itCtrl.played, ShouldContainKey, "2")
|
|
So(itCtrl.played["2"].Equal(now), ShouldBeTrue)
|
|
})
|
|
|
|
})
|
|
|
|
Convey("When the ID is not in the DB", func() {
|
|
_, err := scrobbler.Register(1, "3", time.Now())
|
|
|
|
Convey("Then I receive an error", func() {
|
|
So(err, ShouldNotBeNil)
|
|
})
|
|
|
|
Convey("And iTunes is not notified", func() {
|
|
So(itCtrl.played, ShouldNotContainKey, "3")
|
|
})
|
|
})
|
|
|
|
Convey("When I inform the song that is now playing", func() {
|
|
mf, err := scrobbler.NowPlaying(1, "DSub", "2", "deluan")
|
|
|
|
Convey("Then I get the song for that id back", func() {
|
|
So(err, ShouldBeNil)
|
|
So(mf.Title, ShouldEqual, "Hands Of Time")
|
|
})
|
|
|
|
Convey("And it saves the song as the one current playing", func() {
|
|
info, _ := npRepo.Head(1)
|
|
So(info.TrackId, ShouldEqual, "2")
|
|
So(info.Start, ShouldHappenBefore, time.Now())
|
|
So(info.Username, ShouldEqual, "deluan")
|
|
So(info.PlayerName, ShouldEqual, "DSub")
|
|
})
|
|
|
|
Convey("And iTunes is not notified", func() {
|
|
So(itCtrl.played, ShouldNotContainKey, "2")
|
|
})
|
|
})
|
|
|
|
Reset(func() {
|
|
itCtrl.played = make(map[string]time.Time)
|
|
itCtrl.skipped = make(map[string]time.Time)
|
|
})
|
|
|
|
})
|
|
Convey("Given a DB with two songs", t, func() {
|
|
mfRepo.SetData(`[{"Id":"1","Title":"Femme Fatale"},{"Id":"2","Title":"Here She Comes Now"},{"Id":"3","Title":"Lady Godiva's Operation"}]`, 3)
|
|
itCtrl.skipped = make(map[string]time.Time)
|
|
npRepo.ClearAll()
|
|
Convey("When I play one song", func() {
|
|
scrobbler.NowPlaying(1, "DSub", "1", "deluan")
|
|
Convey("And I play the other song without scrobbling the first one", func() {
|
|
scrobbler.NowPlaying(1, "DSub", "2", "deluan")
|
|
mf, err := scrobbler.Register(1, "2", time.Now())
|
|
Convey("Then the first song should be marked as skipped", func() {
|
|
So(mf.Id, ShouldEqual, "2")
|
|
So(itCtrl.skipped, ShouldContainKey, "1")
|
|
So(err, ShouldBeNil)
|
|
})
|
|
})
|
|
Convey("And I scrobble it before starting to play the other song", func() {
|
|
mf, err := scrobbler.Register(1, "1", time.Now())
|
|
Convey("Then the first song should NOT marked as skipped", func() {
|
|
So(mf.Id, ShouldEqual, "1")
|
|
So(itCtrl.skipped, ShouldBeEmpty)
|
|
So(err, ShouldBeNil)
|
|
})
|
|
})
|
|
})
|
|
})
|
|
}
|
|
|
|
type mockItunesControl struct {
|
|
itunesbridge.ItunesControl
|
|
played map[string]time.Time
|
|
skipped map[string]time.Time
|
|
error bool
|
|
}
|
|
|
|
func (m *mockItunesControl) MarkAsPlayed(id string, playDate time.Time) error {
|
|
if m.error {
|
|
return errors.New("ID not found")
|
|
}
|
|
if m.played == nil {
|
|
m.played = make(map[string]time.Time)
|
|
}
|
|
m.played[id] = playDate
|
|
return nil
|
|
}
|
|
|
|
func (m *mockItunesControl) MarkAsSkipped(id string, skipDate time.Time) error {
|
|
if m.error {
|
|
return errors.New("ID not found")
|
|
}
|
|
if m.skipped == nil {
|
|
m.skipped = make(map[string]time.Time)
|
|
}
|
|
m.skipped[id] = skipDate
|
|
return nil
|
|
}
|