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)
		}
	})
}