75 lines
1.8 KiB
Go
75 lines
1.8 KiB
Go
package scanner
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/cloudsonic/sonic-server/domain"
|
|
"github.com/cloudsonic/sonic-server/tests"
|
|
"github.com/cloudsonic/sonic-server/utils"
|
|
. "github.com/smartystreets/goconvey/convey"
|
|
)
|
|
|
|
func TestCollectIndex(t *testing.T) {
|
|
tests.Init(t, false)
|
|
|
|
ig := utils.IndexGroups{"A": "A", "B": "B", "Tom": "Tom", "X": "X-Z"}
|
|
|
|
importer := &Importer{}
|
|
|
|
Convey("Simple Name", t, func() {
|
|
a := &domain.Artist{Name: "Björk"}
|
|
artistIndex := make(map[string]tempIndex)
|
|
|
|
importer.collectIndex(ig, a, artistIndex)
|
|
|
|
So(artistIndex, ShouldContainKey, "B")
|
|
So(artistIndex["B"], ShouldContainKey, "björk")
|
|
|
|
for _, k := range []string{"A", "Tom", "X-Z", "#"} {
|
|
So(artistIndex, ShouldNotContainKey, k)
|
|
}
|
|
})
|
|
|
|
Convey("Name not in the index", t, func() {
|
|
a := &domain.Artist{Name: "Kraftwerk"}
|
|
artistIndex := make(map[string]tempIndex)
|
|
|
|
importer.collectIndex(ig, a, artistIndex)
|
|
|
|
So(artistIndex, ShouldContainKey, "#")
|
|
So(artistIndex["#"], ShouldContainKey, "kraftwerk")
|
|
|
|
for _, k := range []string{"A", "B", "Tom", "X-Z"} {
|
|
So(artistIndex, ShouldNotContainKey, k)
|
|
}
|
|
})
|
|
|
|
Convey("Name starts with an article", t, func() {
|
|
a := &domain.Artist{Name: "The The"}
|
|
artistIndex := make(map[string]tempIndex)
|
|
|
|
importer.collectIndex(ig, a, artistIndex)
|
|
|
|
So(artistIndex, ShouldContainKey, "#")
|
|
So(artistIndex["#"], ShouldContainKey, "the")
|
|
|
|
for _, k := range []string{"A", "B", "Tom", "X-Z"} {
|
|
So(artistIndex, ShouldNotContainKey, k)
|
|
}
|
|
})
|
|
|
|
Convey("Name match a multichar entry", t, func() {
|
|
a := &domain.Artist{Name: "Tom Waits"}
|
|
artistIndex := make(map[string]tempIndex)
|
|
|
|
importer.collectIndex(ig, a, artistIndex)
|
|
|
|
So(artistIndex, ShouldContainKey, "Tom")
|
|
So(artistIndex["Tom"], ShouldContainKey, "tom waits")
|
|
|
|
for _, k := range []string{"A", "B", "X-Z", "#"} {
|
|
So(artistIndex, ShouldNotContainKey, k)
|
|
}
|
|
})
|
|
}
|