package captcha import ( "bytes" "golang.org/x/image/font/gofont/goregular" "image/color" "math/rand" "testing" ) func TestNewCaptcha(t *testing.T) { data, err := New(150, 50) if err != nil { t.Fatal(err) } buf := new(bytes.Buffer) data.WriteTo(buf) } func TestSmallCaptcha(t *testing.T) { _, err := New(36, 12) if err != nil { t.Fatal(err) } } func TestNewCaptchaOptions(t *testing.T) { New(100, 34, func(options *Options) { options.BackgroundColor = color.Opaque options.CharPreset = "1234567890" options.CurveNumber = 0 options.TextLength = 6 }) NewMathExpr(100, 34, func(options *Options) { options.BackgroundColor = color.Black }) } func TestNewMathExpr(t *testing.T) { _, err := NewMathExpr(150, 50) if err != nil { t.Fatal(err) } } func TestCovNilFontError(t *testing.T) { temp := ttfFont ttfFont = nil _, err := New(150, 50) if err == nil { t.Fatal("Expect to get nil font error") } _, err = NewMathExpr(150, 50) if err == nil { t.Fatal("Expect to get nil font error") } ttfFont = temp } func TestLoadFont(t *testing.T) { err := LoadFont(goregular.TTF) if err != nil { t.Fatal("Fail to load go font") } err = LoadFont([]byte("invalid")) if err == nil { t.Fatal("LoadFont incorrectly parse an invalid font") } } func TestMaxColor(t *testing.T) { var result uint32 result = maxColor() if result != 0 { t.Fatalf("Expect max color to be 0, got %v", result) } result = maxColor(1) if result != 1 { t.Fatalf("Expect max color to be 1, got %v", result) } result = maxColor(52428, 65535) if result != 255 { t.Fatalf("Expect max color to be 255, got %v", result) } var rng = rand.New(rand.NewSource(0)) for i := 0; i < 10; i++ { result = maxColor(rng.Uint32(), rng.Uint32(), rng.Uint32()) if result > 255 { t.Fatalf("Number out of range: %v", result) } } } func TestMinColor(t *testing.T) { var result uint32 result = minColor() if result != 255 { t.Fatalf("Expect min color to be 255, got %v", result) } result = minColor(1) if result != 1 { t.Fatalf("Expect min color to be 1, got %v", result) } result = minColor(52428, 65535) if result != 204 { t.Fatalf("Expect min color to be 1, got %v", result) } var rng = rand.New(rand.NewSource(0)) for i := 0; i < 10; i++ { result = minColor(rng.Uint32(), rng.Uint32(), rng.Uint32()) if result > 255 { t.Fatalf("Number out of range: %v", result) } } }