import (
"bytes"
"reflect"
+ "regexp"
"strings"
"testing"
)
func TestRedirect(t *testing.T) {
const (
start = "http://google.com/"
- end = "http://www.google.com/"
+ endRe = "^http://www\\.google\\.[a-z.]+/$"
)
+ var end = regexp.MustCompile(endRe)
r, url, err := Get(start)
if err != nil {
t.Fatal(err)
}
r.Body.Close()
- if r.StatusCode != 200 || url != end {
- t.Fatalf("Get(%s) got status %d at %s, want 200 at %s", start, r.StatusCode, url, end)
+ if r.StatusCode != 200 || !end.MatchString(url) {
+ t.Fatalf("Get(%s) got status %d at %q, want 200 matching %q", start, r.StatusCode, url, endRe)
}
}