refactor: new persistence, more SQL, less ORM

persistence/mediafile_repository_test.go

@@ -1,29 +1,86 @@
 package persistence
 
 import (
-	"os"
-	"path/filepath"
+	"context"
 
 	"github.com/astaxie/beego/orm"
 	"github.com/deluan/navidrome/model"
+	"github.com/google/uuid"
 	. "github.com/onsi/ginkgo"
 	. "github.com/onsi/gomega"
 )
 
-var _ = Describe("MediaFileRepository", func() {
-	var repo model.MediaFileRepository
+var _ = Describe("MediaRepository", func() {
+	var mr model.MediaFileRepository
 
 	BeforeEach(func() {
-		repo = NewMediaFileRepository(orm.NewOrm())
+		ctx := context.WithValue(context.Background(), "user", &model.User{ID: "userid"})
+		mr = NewMediaFileRepository(ctx, orm.NewOrm())
 	})
 
-	Describe("FindByPath", func() {
-		It("returns all records from a given ArtistID", func() {
-			path := string(os.PathSeparator) + filepath.Join("beatles", "1")
-			Expect(repo.FindByPath(path)).To(Equal(model.MediaFiles{
-				songComeTogether,
-			}))
-		})
+	It("gets mediafile from the DB", func() {
+		Expect(mr.Get("4")).To(Equal(&songAntenna))
 	})
 
+	It("returns ErrNotFound", func() {
+		_, err := mr.Get("56")
+		Expect(err).To(MatchError(model.ErrNotFound))
+	})
+
+	It("counts the number of mediafiles in the DB", func() {
+		Expect(mr.CountAll()).To(Equal(int64(4)))
+	})
+
+	It("checks existence of mediafiles in the DB", func() {
+		Expect(mr.Exists(songAntenna.ID)).To(BeTrue())
+		Expect(mr.Exists("666")).To(BeFalse())
+	})
+
+	It("find mediafiles by album", func() {
+		Expect(mr.FindByAlbum("3")).To(Equal(model.MediaFiles{
+			songRadioactivity,
+			songAntenna,
+		}))
+	})
+
+	It("returns empty array when no tracks are found", func() {
+		Expect(mr.FindByAlbum("67")).To(Equal(model.MediaFiles{}))
+	})
+
+	It("finds tracks by path", func() {
+		Expect(mr.FindByPath(P("/beatles/1/sgt"))).To(Equal(model.MediaFiles{
+			songDayInALife,
+		}))
+	})
+
+	It("returns starred tracks", func() {
+		Expect(mr.GetStarred("userid")).To(Equal(model.MediaFiles{
+			songComeTogether,
+		}))
+	})
+
+	It("delete tracks by id", func() {
+		random, _ := uuid.NewRandom()
+		id := random.String()
+		Expect(mr.Put(&model.MediaFile{ID: id})).To(BeNil())
+
+		Expect(mr.Delete(id)).To(BeNil())
+
+		_, err := mr.Get(id)
+		Expect(err).To(MatchError(model.ErrNotFound))
+	})
+
+	It("delete tracks by path", func() {
+		id1 := "1111"
+		Expect(mr.Put(&model.MediaFile{ID: id1, Path: P("/abc/123/" + id1 + ".mp3")})).To(BeNil())
+		id2 := "2222"
+		Expect(mr.Put(&model.MediaFile{ID: id2, Path: P("/abc/123/" + id2 + ".mp3")})).To(BeNil())
+
+		Expect(mr.DeleteByPath(P("/abc"))).To(BeNil())
+
+		_, err := mr.Get(id1)
+		Expect(err).To(MatchError(model.ErrNotFound))
+		_, err = mr.Get(id2)
+		Expect(err).To(MatchError(model.ErrNotFound))
+	})
 })