Preparing for new scanner

scanner_legacy/importer_test.go

@@ -0,0 +1,74 @@
+package scanner_legacy
+
+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)
+		}
+	})
+}