98 lines
2.9 KiB
Go
98 lines
2.9 KiB
Go
package persistence
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/astaxie/beego/orm"
|
|
"github.com/deluan/navidrome/conf"
|
|
"github.com/deluan/navidrome/log"
|
|
"github.com/deluan/navidrome/model"
|
|
"github.com/deluan/navidrome/tests"
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
func TestPersistence(t *testing.T) {
|
|
tests.Init(t, true)
|
|
log.SetLevel(log.LevelCritical)
|
|
RegisterFailHandler(Fail)
|
|
RunSpecs(t, "Persistence Suite")
|
|
}
|
|
|
|
var artistSaaraSaara = model.Artist{ID: "1", Name: "Saara Saara", AlbumCount: 2}
|
|
var artistKraftwerk = model.Artist{ID: "2", Name: "Kraftwerk"}
|
|
var artistBeatles = model.Artist{ID: "3", Name: "The Beatles"}
|
|
var testArtists = model.Artists{
|
|
artistSaaraSaara,
|
|
artistKraftwerk,
|
|
artistBeatles,
|
|
}
|
|
|
|
var albumSgtPeppers = model.Album{ID: "1", Name: "Sgt Peppers", Artist: "The Beatles", ArtistID: "1", Genre: "Rock"}
|
|
var albumAbbeyRoad = model.Album{ID: "2", Name: "Abbey Road", Artist: "The Beatles", ArtistID: "1", Genre: "Rock"}
|
|
var albumRadioactivity = model.Album{ID: "3", Name: "Radioactivity", Artist: "Kraftwerk", ArtistID: "2", Genre: "Electronic"}
|
|
var testAlbums = model.Albums{
|
|
albumSgtPeppers,
|
|
albumAbbeyRoad,
|
|
albumRadioactivity,
|
|
}
|
|
|
|
var annRadioactivity = model.Annotation{AnnotationID: "1", UserID: "userid", ItemType: model.AlbumItemType, ItemID: "3", Starred: true}
|
|
var testAnnotations = []model.Annotation{
|
|
annRadioactivity,
|
|
}
|
|
|
|
var songDayInALife = model.MediaFile{ID: "1", Title: "A Day In A Life", ArtistID: "3", AlbumID: "1", Genre: "Rock", Path: P("/beatles/1/sgt/a day.mp3")}
|
|
var songComeTogether = model.MediaFile{ID: "2", Title: "Come Together", ArtistID: "3", AlbumID: "2", Genre: "Rock", Path: P("/beatles/1/come together.mp3")}
|
|
var songRadioactivity = model.MediaFile{ID: "3", Title: "Radioactivity", ArtistID: "2", AlbumID: "3", Genre: "Electronic", Path: P("/kraft/radio/radio.mp3")}
|
|
var songAntenna = model.MediaFile{ID: "4", Title: "Antenna", ArtistID: "2", AlbumID: "3", Genre: "Electronic", Path: P("/kraft/radio/antenna.mp3")}
|
|
var testSongs = model.MediaFiles{
|
|
songDayInALife,
|
|
songComeTogether,
|
|
songRadioactivity,
|
|
songAntenna,
|
|
}
|
|
|
|
func P(path string) string {
|
|
return strings.ReplaceAll(path, "/", string(os.PathSeparator))
|
|
}
|
|
|
|
var _ = Describe("Initialize test DB", func() {
|
|
BeforeSuite(func() {
|
|
conf.Server.DbPath = ":memory:"
|
|
ds := New()
|
|
artistRepo := ds.Artist()
|
|
for _, a := range testArtists {
|
|
err := artistRepo.Put(&a)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
albumRepository := ds.Album()
|
|
for _, a := range testAlbums {
|
|
err := albumRepository.Put(&a)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
mediaFileRepository := ds.MediaFile()
|
|
for _, s := range testSongs {
|
|
err := mediaFileRepository.Put(&s)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
o := orm.NewOrm()
|
|
for _, a := range testAnnotations {
|
|
ann := annotation(a)
|
|
_, err := o.Insert(&ann)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
})
|
|
})
|