Add methods to Playlist model

Also, don't load genres for Playlists tracks (not necessary for now)
2021-10-16 22:44:44 -04:00
parent d200933b68
commit c72add516a
7 changed files with 102 additions and 53 deletions
@@ -70,16 +70,9 @@ var (
 )

 var (
-	plsBest = model.Playlist{
-		Name:      "Best",
-		Comment:   "No Comments",
-		Owner:     "userid",
-		Public:    true,
-		SongCount: 2,
-		Tracks:    model.MediaFiles{{ID: "1001"}, {ID: "1003"}},
-	}
-	plsCool       = model.Playlist{Name: "Cool", Owner: "userid", Tracks: model.MediaFiles{{ID: "1004"}}}
-	testPlaylists = []*model.Playlist{&plsBest, &plsCool}
+	plsBest       model.Playlist
+	plsCool       model.Playlist
+	testPlaylists []*model.Playlist
 )

 func P(path string) string {
@@ -130,6 +123,18 @@ var _ = Describe("Initialize test DB", func() {
 			}
 		}

+		plsBest = model.Playlist{
+			Name:      "Best",
+			Comment:   "No Comments",
+			Owner:     "userid",
+			Public:    true,
+			SongCount: 2,
+		}
+		plsBest.AddTracks([]string{"1001", "1003"})
+		plsCool = model.Playlist{Name: "Cool", Owner: "userid"}
+		plsCool.AddTracks([]string{"1004"})
+		testPlaylists = []*model.Playlist{&plsBest, &plsCool}
+
 		pr := NewPlaylistRepository(ctx, o)
 		for i := range testPlaylists {
 			err := pr.Put(testPlaylists[i])
@@ -162,6 +167,5 @@ var _ = Describe("Initialize test DB", func() {
 		songComeTogether.Starred = true
 		songComeTogether.StarredAt = mf.StarredAt
 		testSongs[1] = songComeTogether
-
 	})
 })