c4ecda54e9
- Updated captcha test cases to utilize the new CaptchaGet function for retrieving captcha answers, improving test reliability. - Refactored CaptchaMake and CaptchaValidate functions to leverage a new utils package for better encapsulation and maintainability. - Enhanced error handling and logging in captcha-related processes, contributing to a more robust user experience during authentication. - Streamlined the captcha generation process by integrating with the utils.captcha package, ensuring consistency across the codebase.
66 lines
1.5 KiB
Go
66 lines
1.5 KiB
Go
package helper
|
|
|
|
import (
|
|
"net/url"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/yaoapp/gou/process"
|
|
"github.com/yaoapp/kun/maps"
|
|
)
|
|
|
|
func TestCaptcha(t *testing.T) {
|
|
id, content := CaptchaMake(CaptchaOption{
|
|
Type: "audio",
|
|
Width: 240,
|
|
Height: 80,
|
|
Length: 4,
|
|
Lang: "zh",
|
|
})
|
|
assert.IsType(t, "string", id)
|
|
assert.IsType(t, "string", content)
|
|
assert.True(t, CaptchaValidate(id, CaptchaGet(id)))
|
|
|
|
id, content = CaptchaMake(CaptchaOption{
|
|
Type: "math",
|
|
Width: 240,
|
|
Height: 80,
|
|
Length: 4,
|
|
Lang: "zh",
|
|
})
|
|
assert.IsType(t, "string", id)
|
|
assert.IsType(t, "string", content)
|
|
assert.True(t, CaptchaValidate(id, CaptchaGet(id)))
|
|
|
|
id, content = CaptchaMake(CaptchaOption{
|
|
Type: "digit",
|
|
Width: 240,
|
|
Height: 80,
|
|
Length: 4,
|
|
Lang: "zh",
|
|
})
|
|
assert.IsType(t, "string", id)
|
|
assert.IsType(t, "string", content)
|
|
assert.True(t, CaptchaValidate(id, CaptchaGet(id)))
|
|
}
|
|
|
|
func TestProcessCaptcha(t *testing.T) {
|
|
args := url.Values{}
|
|
args.Add("type", "math")
|
|
args.Add("lang", "zh")
|
|
p := process.New("xiang.helper.Captcha", args)
|
|
res := p.Run().(maps.Map)
|
|
assert.IsType(t, "string", res.Get("id"))
|
|
assert.IsType(t, "string", res.Get("content"))
|
|
|
|
value := CaptchaGet(res.Get("id").(string))
|
|
p = process.New("xiang.helper.CaptchaValidate", res.Get("id"), value)
|
|
assert.True(t, p.Run().(bool))
|
|
assert.Panics(t, func() {
|
|
process.New("xiang.helper.CaptchaValidate", res.Get("id"), "xxx").Run()
|
|
})
|
|
|
|
assert.Panics(t, func() {
|
|
process.New("xiang.helper.CaptchaValidate", res.Get("id"), "").Run()
|
|
})
|
|
}
|