update

conv/country_test.go

@@ -0,0 +1,29 @@
+package conv
+
+import "testing"
+
+func TestCountryFromHost(t *testing.T) {
+	tests := []struct {
+		host string
+		want string
+	}{
+		{"shop-no.example.com", "no"},
+		{"SHOP-NO.EXAMPLE.COM", "no"},
+		{"shop-se.example.com", "se"},
+		{"shop-se", "se"},
+		{"shop-no", "no"},
+		{"example.com", "se"},
+		{"", "se"},
+		{"localhost", "se"},
+		{"https://shop-no.example.com/page", "no"},
+		{"https://shop-se.example.com", "se"},
+		{"-no-", "no"},
+		{"nose.example.com", "se"}, // no "-no" substring — "nose" has no hyphen
+	}
+	for _, tt := range tests {
+		got := CountryFromHost(tt.host)
+		if got != tt.want {
+			t.Errorf("CountryFromHost(%q) = %q, want %q", tt.host, got, tt.want)
+		}
+	}
+}