git.zd.zone/hrpc/hrpc@v0.0.12/codec/message_test.go (about)

     1  package codec
     2  
     3  import (
     4  	"context"
     5  	"reflect"
     6  	"testing"
     7  	"time"
     8  
     9  	"google.golang.org/grpc/metadata"
    10  )
    11  
    12  func Test_message_Metadata(t *testing.T) {
    13  	type fields struct {
    14  		context   context.Context
    15  		TID       string
    16  		RTimeout  time.Duration
    17  		NameSpace string
    18  	}
    19  	tests := []struct {
    20  		name   string
    21  		fields fields
    22  		want   metadata.MD
    23  	}{
    24  		{
    25  			name: "Test1",
    26  			fields: fields{
    27  				context:   context.Background(),
    28  				TID:       "abcd",
    29  				RTimeout:  time.Second,
    30  				NameSpace: "Dev",
    31  			},
    32  		},
    33  	}
    34  	for _, tt := range tests {
    35  		t.Run(tt.name, func(t *testing.T) {
    36  			m := &message{
    37  				context:   tt.fields.context,
    38  				TID:       tt.fields.TID,
    39  				RTimeout:  tt.fields.RTimeout,
    40  				NameSpace: tt.fields.NameSpace,
    41  			}
    42  			got := m.Metadata()
    43  			t.Log(got.Get(MDMessage.String()))
    44  			if got := m.Metadata(); !reflect.DeepEqual(got, tt.want) {
    45  				t.Errorf("message.Metadata() = %v, want %v", got, tt.want)
    46  			}
    47  		})
    48  	}
    49  }
    50  
    51  func TestMessage(t *testing.T) {
    52  	type args struct {
    53  		ctx context.Context
    54  	}
    55  	tests := []struct {
    56  		name string
    57  		args args
    58  		want MSG
    59  	}{
    60  		{
    61  			name: "Test1",
    62  			args: args{
    63  				ctx: metadata.NewIncomingContext(
    64  					context.Background(), metadata.Pairs(MDMessage.String(), `{"trace_id":"abcd","request_timeout":1000000000,"namespace":"Dev"}`),
    65  				),
    66  			},
    67  		},
    68  	}
    69  	for _, tt := range tests {
    70  		t.Run(tt.name, func(t *testing.T) {
    71  			got := Message(tt.args.ctx)
    72  			t.Log(got.TraceID())
    73  			if got := Message(tt.args.ctx); !reflect.DeepEqual(got, tt.want) {
    74  				t.Errorf("Message() = %v, want %v", got, tt.want)
    75  			}
    76  		})
    77  	}
    78  }