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.
104 lines
2.4 KiB
104 lines
2.4 KiB
package proto
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
testData "github.com/go-kratos/kratos/v2/internal/testdata/encoding"
|
|
)
|
|
|
|
func TestName(t *testing.T) {
|
|
c := new(codec)
|
|
if !reflect.DeepEqual(c.Name(), "proto") {
|
|
t.Errorf("no expect float_key value: %v, but got: %v", c.Name(), "proto")
|
|
}
|
|
}
|
|
|
|
func TestCodec(t *testing.T) {
|
|
c := new(codec)
|
|
|
|
model := testData.TestModel{
|
|
Id: 1,
|
|
Name: "kratos",
|
|
Hobby: []string{"study", "eat", "play"},
|
|
}
|
|
|
|
m, err := c.Marshal(&model)
|
|
if err != nil {
|
|
t.Errorf("Marshal() should be nil, but got %s", err)
|
|
}
|
|
|
|
var res testData.TestModel
|
|
|
|
err = c.Unmarshal(m, &res)
|
|
if err != nil {
|
|
t.Errorf("Unmarshal() should be nil, but got %s", err)
|
|
}
|
|
if !reflect.DeepEqual(res.Id, model.Id) {
|
|
t.Errorf("ID should be %d, but got %d", res.Id, model.Id)
|
|
}
|
|
if !reflect.DeepEqual(res.Name, model.Name) {
|
|
t.Errorf("Name should be %s, but got %s", res.Name, model.Name)
|
|
}
|
|
if !reflect.DeepEqual(res.Hobby, model.Hobby) {
|
|
t.Errorf("Hobby should be %s, but got %s", res.Hobby, model.Hobby)
|
|
}
|
|
}
|
|
|
|
func TestCodec2(t *testing.T) {
|
|
c := new(codec)
|
|
|
|
model := testData.TestModel{
|
|
Id: 1,
|
|
Name: "kratos",
|
|
Hobby: []string{"study", "eat", "play"},
|
|
}
|
|
|
|
m, err := c.Marshal(&model)
|
|
if err != nil {
|
|
t.Errorf("Marshal() should be nil, but got %s", err)
|
|
}
|
|
|
|
var res testData.TestModel
|
|
rp := &res
|
|
|
|
err = c.Unmarshal(m, &rp)
|
|
if err != nil {
|
|
t.Errorf("Unmarshal() should be nil, but got %s", err)
|
|
}
|
|
if !reflect.DeepEqual(res.Id, model.Id) {
|
|
t.Errorf("ID should be %d, but got %d", res.Id, model.Id)
|
|
}
|
|
if !reflect.DeepEqual(res.Name, model.Name) {
|
|
t.Errorf("Name should be %s, but got %s", res.Name, model.Name)
|
|
}
|
|
if !reflect.DeepEqual(res.Hobby, model.Hobby) {
|
|
t.Errorf("Hobby should be %s, but got %s", res.Hobby, model.Hobby)
|
|
}
|
|
}
|
|
|
|
func Test_getProtoMessage(t *testing.T) {
|
|
p := &testData.TestModel{Id: 1}
|
|
type args struct {
|
|
v interface{}
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
wantErr bool
|
|
}{
|
|
{name: "test1", args: args{v: &testData.TestModel{}}, wantErr: false},
|
|
{name: "test2", args: args{v: testData.TestModel{}}, wantErr: true},
|
|
{name: "test3", args: args{v: &p}, wantErr: false},
|
|
{name: "test4", args: args{v: 1}, wantErr: true},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
_, err := getProtoMessage(tt.args.v)
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("getProtoMessage() error = %v, wantErr %v", err, tt.wantErr)
|
|
return
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|