navidrome/core/share_test.go

package core

import (
	"context"

	"github.com/deluan/rest"
	"github.com/navidrome/navidrome/model"
	"github.com/navidrome/navidrome/model/request"
	"github.com/navidrome/navidrome/tests"
	. "github.com/onsi/ginkgo/v2"
	. "github.com/onsi/gomega"
)

var _ = Describe("Share", func() {
	var ds model.DataStore
	var share Share
	var mockedRepo rest.Persistable
	ctx := context.Background()

	BeforeEach(func() {
		ds = &tests.MockDataStore{}
		mockedRepo = ds.Share(ctx).(rest.Persistable)
		share = NewShare(ds)
	})

	Describe("NewRepository", func() {
		var repo rest.Persistable

		BeforeEach(func() {
			repo = share.NewRepository(ctx).(rest.Persistable)
			_ = ds.Album(ctx).Put(&model.Album{ID: "123", Name: "Album"})
		})

		Describe("Save", func() {
			It("it sets a random ID", func() {
				entity := &model.Share{Description: "test", ResourceIDs: "123"}
				id, err := repo.Save(entity)
				Expect(err).ToNot(HaveOccurred())
				Expect(id).ToNot(BeEmpty())
				Expect(entity.ID).To(Equal(id))
			})

			It("assigns the logged-in user as owner, ignoring a client-supplied UserID", func() {
				loggedInCtx := request.WithUser(context.Background(), model.User{ID: "logged-in-user"})
				repo := share.NewRepository(loggedInCtx).(rest.Persistable)
				entity := &model.Share{Description: "test", ResourceIDs: "123", UserID: "victim-user"}
				_, err := repo.Save(entity)
				Expect(err).ToNot(HaveOccurred())
				Expect(entity.UserID).To(Equal("logged-in-user"))
			})

			It("does not truncate ASCII labels shorter than 30 characters", func() {
				_ = ds.MediaFile(ctx).Put(&model.MediaFile{ID: "456", Title: "Example Media File"})
				entity := &model.Share{Description: "test", ResourceIDs: "456"}
				_, err := repo.Save(entity)
				Expect(err).ToNot(HaveOccurred())
				Expect(entity.Contents).To(Equal("Example Media File"))
			})

			It("truncates ASCII labels longer than 30 characters", func() {
				_ = ds.MediaFile(ctx).Put(&model.MediaFile{ID: "789", Title: "Example Media File But The Title Is Really Long For Testing Purposes"})
				entity := &model.Share{Description: "test", ResourceIDs: "789"}
				_, err := repo.Save(entity)
				Expect(err).ToNot(HaveOccurred())
				Expect(entity.Contents).To(Equal("Example Media File But The ..."))
			})

			It("does not truncate CJK labels shorter than 30 runes", func() {
				_ = ds.MediaFile(ctx).Put(&model.MediaFile{ID: "456", Title: "青春コンプレックス"})
				entity := &model.Share{Description: "test", ResourceIDs: "456"}
				_, err := repo.Save(entity)
				Expect(err).ToNot(HaveOccurred())
				Expect(entity.Contents).To(Equal("青春コンプレックス"))
			})

			It("truncates CJK labels longer than 30 runes", func() {
				_ = ds.MediaFile(ctx).Put(&model.MediaFile{ID: "789", Title: "私の中の幻想的世界観及びその顕現を想起させたある現実での出来事に関する一考察"})
				entity := &model.Share{Description: "test", ResourceIDs: "789"}
				_, err := repo.Save(entity)
				Expect(err).ToNot(HaveOccurred())
				Expect(entity.Contents).To(Equal("私の中の幻想的世界観及びその顕現を想起させたある現実で..."))
			})

			It("fails when any of the resource IDs does not exist", func() {
				entity := &model.Share{Description: "test", ResourceIDs: "123,missing"}
				_, err := repo.Save(entity)
				Expect(err).To(MatchError(model.ErrNotFound))
			})

			It("fails when the resource IDs are of mixed types", func() {
				_ = ds.MediaFile(ctx).Put(&model.MediaFile{ID: "456", Title: "Example Media File"})
				entity := &model.Share{Description: "test", ResourceIDs: "123,456"}
				_, err := repo.Save(entity)
				Expect(err).To(HaveOccurred())
			})
		})

		Describe("Update", func() {
			It("filters out read-only fields", func() {
				entity := &model.Share{}
				err := repo.Update("id", entity)
				Expect(err).ToNot(HaveOccurred())
				Expect(mockedRepo.(*tests.MockShareRepo).Cols).To(ConsistOf("description", "downloadable"))
			})
		})
	})
})