You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
893 B
43 lines
893 B
package validate |
|
|
|
import ( |
|
"context" |
|
"fmt" |
|
"testing" |
|
|
|
"github.com/go-kratos/kratos/v2/errors" |
|
"github.com/go-kratos/kratos/v2/middleware" |
|
) |
|
|
|
// protoVali implement validate.validator |
|
type protoVali struct { |
|
name string |
|
age int |
|
isErr bool |
|
} |
|
|
|
func (v protoVali) Validate() error { |
|
if v.name == "" || v.age < 0 { |
|
return fmt.Errorf("err") |
|
} |
|
return nil |
|
} |
|
|
|
func TestTable(t *testing.T) { |
|
var mock middleware.Handler = func(ctx context.Context, req interface{}) (interface{}, error) { return nil, nil } |
|
|
|
tests := []protoVali{ |
|
{"v1", 365, false}, |
|
{"v2", -1, true}, |
|
{"", 365, true}, |
|
} |
|
for _, test := range tests { |
|
t.Run(test.name, func(t *testing.T) { |
|
v := Validator()(mock) |
|
_, err := v(context.Background(), test) |
|
if want, have := test.isErr, errors.IsBadRequest(err); want != have { |
|
t.Errorf("fail data %v, want %v, have %v", test, want, have) |
|
} |
|
}) |
|
} |
|
}
|
|
|