80 lines
2.3 KiB
Go
80 lines
2.3 KiB
Go
package normalize
|
|
|
|
import "testing"
|
|
|
|
func TestNormalizeString_Basic(t *testing.T) {
|
|
tests := []struct {
|
|
input string
|
|
expected string
|
|
}{
|
|
// Lowercase conversion
|
|
{"DARK SIDE OF THE MOON", "dark side of the moon"},
|
|
// Special character removal
|
|
{"Dark Side of the Moon!", "dark side of the moon"},
|
|
{"Dark-Side-of-the-Moon", "dark side of the moon"},
|
|
{"Dark_Side_of_the_Moon", "dark side of the moon"},
|
|
// Bracket removal
|
|
{"Dark Side of the Moon [Deluxe Edition]", "dark side of the moon"},
|
|
{"Dark Side of the Moon [Remastered 2020]", "dark side of the moon"},
|
|
{"Album [2023 Remix]", "album"},
|
|
// Parenthesis removal
|
|
{"Dark Side of the Moon (Deluxe)", "dark side of the moon"},
|
|
{"Album (Remastered)", "album"},
|
|
// Year removal
|
|
{"Dark Side of the Moon 1973", "dark side of the moon"},
|
|
{"Album 2020 Remastered", "album remastered"},
|
|
// Space collapsing
|
|
{"Dark Side of the Moon", "dark side of the moon"},
|
|
// Trim
|
|
{" Dark Side of the Moon ", "dark side of the moon"},
|
|
// Combined
|
|
{"The Dark Side of the Moon [2011 Remaster] (Deluxe Edition)", "the dark side of the moon"},
|
|
// Empty
|
|
{"", ""},
|
|
// Only special chars
|
|
{"!@#$%^&*()", ""},
|
|
// Digits that are not years should stay
|
|
{"30 Seconds to Mars", "30 seconds to mars"},
|
|
{"1941 - The Greatest Hits", "the greatest hits"},
|
|
// Year-only title is preserved (not collapsed to empty) so it can still match
|
|
{"1989", "1989"},
|
|
{"2112", "2112"},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.input, func(t *testing.T) {
|
|
got := NormalizeString(tt.input)
|
|
if got != tt.expected {
|
|
t.Errorf("NormalizeString(%q) = %q, want %q", tt.input, got, tt.expected)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestNormalizeArtistName(t *testing.T) {
|
|
tests := []struct {
|
|
input string
|
|
expected string
|
|
}{
|
|
{"Pink Floyd", "pink floyd"},
|
|
{"The Beatles", "beatles"},
|
|
{"A Perfect Circle", "perfect circle"},
|
|
{"An Orchestra", "orchestra"},
|
|
{" The Who ", "who"},
|
|
{"THE WHO", "who"},
|
|
// No stripping needed
|
|
{"Radiohead", "radiohead"},
|
|
// Already stripped
|
|
{"Beatles", "beatles"},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.input, func(t *testing.T) {
|
|
got := NormalizeArtistName(tt.input)
|
|
if got != tt.expected {
|
|
t.Errorf("NormalizeArtistName(%q) = %q, want %q", tt.input, got, tt.expected)
|
|
}
|
|
})
|
|
}
|
|
}
|