kratos/transport/http/binding/bind_test.go

package binding

import (
	"github.com/go-kratos/kratos/v2/errors"
	"io"
	"net/http"
	"net/url"
	"reflect"
	"strings"
	"testing"
)

func TestBindQuery(t *testing.T) {
	type TestBind struct {
		Name string `json:"name"`
		URL  string `json:"url"`
	}

	type TestBind2 struct {
		Age int `json:"age"`
	}
	p1 := TestBind{}
	p2 := TestBind2{}
	type args struct {
		vars   url.Values
		target interface{}
	}
	tests := []struct {
		name    string
		args    args
		wantErr bool
		want    interface{}
	}{
		{
			name: "test",
			args: args{
				vars:   map[string][]string{"name": {"kratos"}, "url": {"https://go-kratos.dev/"}},
				target: &p1,
			},
			wantErr: false,
			want:    &TestBind{"kratos", "https://go-kratos.dev/"},
		},
		{
			name: "test1",
			args: args{
				vars:   map[string][]string{"age": {"kratos"}, "url": {"https://go-kratos.dev/"}},
				target: &p2,
			},
			wantErr: true,
			want:    errors.BadRequest("CODEC", ""),
		},
		{
			name: "test2",
			args: args{
				vars:   map[string][]string{"age": {"1"}, "url": {"https://go-kratos.dev/"}},
				target: &TestBind2{},
			},
			wantErr: false,
			want:    &TestBind2{Age: 1},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if err := BindQuery(tt.args.vars, tt.args.target); (err != nil) != tt.wantErr {
				t.Errorf("BindQuery() error = %v, wantErr %v", err, tt.wantErr)
			} else {
				t.Log(err)
			}
			if !tt.wantErr && !reflect.DeepEqual(tt.args.target, tt.want) {
				t.Errorf("BindQuery() target = %v, want %v", tt.args.target, tt.want)
			}
		})
	}
}

func TestBindForm(t *testing.T) {
	type TestBind struct {
		Name string `json:"name"`
		URL  string `json:"url"`
	}
	type TestBind2 struct {
		Age int `json:"age"`
	}
	p1 := TestBind{}
	type args struct {
		req    *http.Request
		target interface{}
	}
	tests := []struct {
		name    string
		args    args
		wantErr bool
		want    interface{}
	}{
		{
			name: "error not nil",
			args: args{
				req:    &http.Request{Method: "POST"},
				target: &p1,
			},
			wantErr: true,
			want:    nil,
		},
		{
			name: "error is nil",
			args: args{
				req: &http.Request{
					Method: "POST",
					Header: http.Header{"Content-Type": {"application/x-www-form-urlencoded; param=value"}},
					Body:   io.NopCloser(strings.NewReader("name=kratos&url=https://go-kratos.dev/")),
				},
				target: &p1,
			},
			wantErr: false,
			want:    &TestBind{"kratos", "https://go-kratos.dev/"},
		},
		{
			name: "error BadRequest",
			args: args{
				req: &http.Request{
					Method: "POST",
					Header: http.Header{"Content-Type": {"application/x-www-form-urlencoded; param=value"}},
					Body:   io.NopCloser(strings.NewReader("age=a")),
				},
				target: &TestBind2{},
			},
			wantErr: true,
			want:    nil,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			err := BindForm(tt.args.req, tt.args.target)
			if (err != nil) != tt.wantErr {
				t.Errorf("BindForm() error = %v, wantErr %v", err, tt.wantErr)
			}
			t.Log(err)
			if !tt.wantErr && !reflect.DeepEqual(tt.args.target, tt.want) {
				t.Errorf("BindForm() target = %v, want %v", tt.args.target, tt.want)
			}
		})
	}
}