Refactor: Consolidate scrobbling logic in play_tracker

tests/mock_mediafile_repo.go

@@ -2,6 +2,7 @@ package tests
 
 import (
 	"errors"
+	"time"
 
 	"github.com/google/uuid"
 
@@ -10,13 +11,13 @@ import (
 
 func CreateMockMediaFileRepo() *MockMediaFileRepo {
 	return &MockMediaFileRepo{
-		data: make(map[string]model.MediaFile),
+		data: make(map[string]*model.MediaFile),
 	}
 }
 
 type MockMediaFileRepo struct {
 	model.MediaFileRepository
-	data map[string]model.MediaFile
+	data map[string]*model.MediaFile
 	err  bool
 }
 
@@ -25,9 +26,9 @@ func (m *MockMediaFileRepo) SetError(err bool) {
 }
 
 func (m *MockMediaFileRepo) SetData(mfs model.MediaFiles) {
-	m.data = make(map[string]model.MediaFile)
-	for _, mf := range mfs {
-		m.data[mf.ID] = mf
+	m.data = make(map[string]*model.MediaFile)
+	for i, mf := range mfs {
+		m.data[mf.ID] = &mfs[i]
 	}
 }
 
@@ -44,22 +45,34 @@ func (m *MockMediaFileRepo) Get(id string) (*model.MediaFile, error) {
 		return nil, errors.New("Error!")
 	}
 	if d, ok := m.data[id]; ok {
-		return &d, nil
+		return d, nil
 	}
 	return nil, model.ErrNotFound
 }
 
 func (m *MockMediaFileRepo) Put(mf *model.MediaFile) error {
 	if m.err {
-		return errors.New("error!")
+		return errors.New("error")
 	}
 	if mf.ID == "" {
 		mf.ID = uuid.NewString()
 	}
-	m.data[mf.ID] = *mf
+	m.data[mf.ID] = mf
 	return nil
 }
 
+func (m *MockMediaFileRepo) IncPlayCount(id string, timestamp time.Time) error {
+	if m.err {
+		return errors.New("error")
+	}
+	if d, ok := m.data[id]; ok {
+		d.PlayCount++
+		d.PlayDate = timestamp
+		return nil
+	}
+	return model.ErrNotFound
+}
+
 func (m *MockMediaFileRepo) FindByAlbum(artistId string) (model.MediaFiles, error) {
 	if m.err {
 		return nil, errors.New("Error!")
@@ -68,7 +81,7 @@ func (m *MockMediaFileRepo) FindByAlbum(artistId string) (model.MediaFiles, erro
 	i := 0
 	for _, a := range m.data {
 		if a.AlbumID == artistId {
-			res[i] = a
+			res[i] = *a
 			i++
 		}
 	}