109 lines
2.2 KiB
Go
109 lines
2.2 KiB
Go
|
package validator
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"testing"
|
|||
|
|
|
|||
|
|
"github.com/go-playground/validator/v10"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func TestValidateAuthorizationURL(t *testing.T) {
|
|||
|
|
validate := validator.New()
|
|||
|
|
RegisterCustomValidators(validate)
|
|||
|
|
|
|||
|
|
tests := []struct {
|
|||
|
|
name string
|
|||
|
|
url string
|
|||
|
|
wantErr bool
|
|||
|
|
}{
|
|||
|
|
{
|
|||
|
|
name: "有效的PDF URL",
|
|||
|
|
url: "https://example.com/document.pdf",
|
|||
|
|
wantErr: false,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "有效的JPG URL",
|
|||
|
|
url: "https://example.com/image.jpg",
|
|||
|
|
wantErr: false,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "有效的JPEG URL",
|
|||
|
|
url: "https://example.com/image.jpeg",
|
|||
|
|
wantErr: false,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "有效的PNG URL",
|
|||
|
|
url: "https://example.com/image.png",
|
|||
|
|
wantErr: false,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "有效的BMP URL",
|
|||
|
|
url: "https://example.com/image.bmp",
|
|||
|
|
wantErr: false,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "HTTP协议的PDF URL",
|
|||
|
|
url: "http://example.com/document.pdf",
|
|||
|
|
wantErr: false,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "带查询参数的PDF URL",
|
|||
|
|
url: "https://example.com/document.pdf?version=1.0",
|
|||
|
|
wantErr: false,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "带路径的PDF URL",
|
|||
|
|
url: "https://example.com/files/documents/contract.pdf",
|
|||
|
|
wantErr: false,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "无效的URL格式",
|
|||
|
|
url: "not-a-url",
|
|||
|
|
wantErr: true,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "不支持的文件类型",
|
|||
|
|
url: "https://example.com/document.doc",
|
|||
|
|
wantErr: true,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "不支持的文件类型2",
|
|||
|
|
url: "https://example.com/document.txt",
|
|||
|
|
wantErr: true,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "没有文件扩展名",
|
|||
|
|
url: "https://example.com/document",
|
|||
|
|
wantErr: true,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "FTP协议（不支持）",
|
|||
|
|
url: "ftp://example.com/document.pdf",
|
|||
|
|
wantErr: true,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "空字符串",
|
|||
|
|
url: "",
|
|||
|
|
wantErr: true,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "大写扩展名",
|
|||
|
|
url: "https://example.com/document.PDF",
|
|||
|
|
wantErr: false,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "混合大小写扩展名",
|
|||
|
|
url: "https://example.com/document.JpG",
|
|||
|
|
wantErr: false,
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for _, tt := range tests {
|
|||
|
|
t.Run(tt.name, func(t *testing.T) {
|
|||
|
|
err := validate.Var(tt.url, "authorization_url")
|
|||
|
|
if (err != nil) != tt.wantErr {
|
|||
|
|
t.Errorf("validateAuthorizationURL() error = %v, wantErr %v", err, tt.wantErr)
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
}
|