github.com/cloudwego/kitex@v0.9.0/pkg/remote/trans/nphttp2/stream_test.go (about)

     1  /*
     2   * Copyright 2022 CloudWeGo Authors
     3   *
     4   * Licensed under the Apache License, Version 2.0 (the "License");
     5   * you may not use this file except in compliance with the License.
     6   * You may obtain a copy of the License at
     7   *
     8   *     http://www.apache.org/licenses/LICENSE-2.0
     9   *
    10   * Unless required by applicable law or agreed to in writing, software
    11   * distributed under the License is distributed on an "AS IS" BASIS,
    12   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    13   * See the License for the specific language governing permissions and
    14   * limitations under the License.
    15   */
    16  
    17  package nphttp2
    18  
    19  import (
    20  	"testing"
    21  
    22  	"github.com/cloudwego/kitex/internal/test"
    23  	"github.com/cloudwego/kitex/pkg/remote/trans/nphttp2/grpc"
    24  	"github.com/cloudwego/kitex/pkg/serviceinfo"
    25  )
    26  
    27  func TestStream(t *testing.T) {
    28  	// init
    29  	opt := newMockServerOption()
    30  	conn := newMockNpConn(mockAddr0)
    31  	conn.mockSettingFrame()
    32  	tr, err := newMockServerTransport(conn)
    33  	test.Assert(t, err == nil, err)
    34  	s := grpc.CreateStream(1, func(i int) {})
    35  	serverConn := newServerConn(tr, s)
    36  	defer serverConn.Close()
    37  
    38  	handler, err := NewSvrTransHandlerFactory().NewTransHandler(opt)
    39  	test.Assert(t, err == nil, err)
    40  	ctx := newMockCtxWithRPCInfo()
    41  
    42  	// test NewStream()
    43  	stream := NewStream(ctx, &serviceinfo.ServiceInfo{
    44  		PackageName:     "",
    45  		ServiceName:     "",
    46  		HandlerType:     nil,
    47  		Methods:         nil,
    48  		PayloadCodec:    0,
    49  		KiteXGenVersion: "",
    50  		Extra:           nil,
    51  		GenericMethod:   nil,
    52  	}, serverConn, handler)
    53  
    54  	// test Context()
    55  	strCtx := stream.Context()
    56  	test.Assert(t, strCtx == ctx)
    57  
    58  	// test recvMsg()
    59  	msg := newMockNewMessage().Data()
    60  	newMockStreamRecvHelloRequest(s)
    61  	err = stream.RecvMsg(msg)
    62  	test.Assert(t, err == nil, err)
    63  
    64  	// test SendMsg()
    65  	err = stream.SendMsg(msg)
    66  	test.Assert(t, err == nil, err)
    67  
    68  	// test Close()
    69  	err = stream.Close()
    70  	test.Assert(t, err == nil, err)
    71  }