Integrate ArtistIndex logic into Artist

persistence/artist_repository_test.go

@@ -13,18 +13,47 @@ var _ = Describe("ArtistRepository", func() {
 		repo = NewArtistRepository()
 	})
 
-	It("saves and retrieves data", func() {
-		Expect(repo.Get("1")).To(Equal(&model.Artist{ID: "1", Name: "Saara Saara", AlbumCount: 2}))
+	Describe("Put/Get", func() {
+		It("saves and retrieves data", func() {
+			Expect(repo.Get("1")).To(Equal(&artistSaaraSaara))
+		})
+
+		It("overrides data if ID already exists", func() {
+			Expect(repo.Put(&model.Artist{ID: "1", Name: "Saara Saara is The Best!", AlbumCount: 3})).To(BeNil())
+			Expect(repo.Get("1")).To(Equal(&model.Artist{ID: "1", Name: "Saara Saara is The Best!", AlbumCount: 3}))
+		})
+
+		It("returns ErrNotFound when the ID does not exist", func() {
+			_, err := repo.Get("999")
+			Expect(err).To(MatchError(model.ErrNotFound))
+		})
 	})
 
-	It("overrides data if ID already exists", func() {
-		Expect(repo.Put(&model.Artist{ID: "1", Name: "Saara Saara is The Best!", AlbumCount: 3})).To(BeNil())
-		Expect(repo.Get("1")).To(Equal(&model.Artist{ID: "1", Name: "Saara Saara is The Best!", AlbumCount: 3}))
-	})
-
-	It("returns ErrNotFound when the ID does not exist", func() {
-		_, err := repo.Get("999")
-		Expect(err).To(MatchError(model.ErrNotFound))
+	Describe("GetIndex", func() {
+		It("returns the index", func() {
+			idx, err := repo.GetIndex()
+			Expect(err).To(BeNil())
+			Expect(idx).To(Equal(model.ArtistIndexes{
+				{
+					ID: "B",
+					Artists: model.Artists{
+						artistBeatles,
+					},
+				},
+				{
+					ID: "K",
+					Artists: model.Artists{
+						artistKraftwerk,
+					},
+				},
+				{
+					ID: "S",
+					Artists: model.Artists{
+						{ID: "1", Name: "Saara Saara is The Best!", AlbumCount: 3},
+					},
+				},
+			}))
+		})
 	})
 
 	Describe("PurgeInactive", func() {