19 lines
543 B
Go
19 lines
543 B
Go
package main
|
|
|
|
import "testing"
|
|
|
|
func TestNormalizeDomain(t *testing.T) {
|
|
cases := map[string]string{"example.com": "example.com", "https://WWW.Example.com/a": "www.example.com", "example.com/": "example.com"}
|
|
for in, want := range cases {
|
|
got, err := normalizeDomain(in)
|
|
if err != nil || got != want {
|
|
t.Fatalf("%q: got %q, %v", in, got, err)
|
|
}
|
|
}
|
|
for _, in := range []string{"localhost", "127.0.0.1", "bad_domain.com", "-bad.com"} {
|
|
if _, err := normalizeDomain(in); err == nil {
|
|
t.Fatalf("expected %q invalid", in)
|
|
}
|
|
}
|
|
}
|