add middle/validate test (#1002)
parent
48851a1ffd
commit
97946ddcbd
@ -0,0 +1,43 @@ |
||||
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) |
||||
} |
||||
}) |
||||
} |
||||
} |
Loading…
Reference in new issue