github.com/hduhelp/go-zero@v1.4.3/gateway/internal/eventhandler_test.go (about)

     1  package internal
     2  
     3  import (
     4  	"io"
     5  	"testing"
     6  
     7  	"github.com/stretchr/testify/assert"
     8  	"google.golang.org/grpc/codes"
     9  	"google.golang.org/grpc/status"
    10  )
    11  
    12  func TestEventHandler(t *testing.T) {
    13  	h := NewEventHandler(io.Discard, nil)
    14  	h.OnResolveMethod(nil)
    15  	h.OnSendHeaders(nil)
    16  	h.OnReceiveHeaders(nil)
    17  	h.OnReceiveTrailers(status.New(codes.OK, ""), nil)
    18  	assert.Equal(t, codes.OK, h.Status.Code())
    19  	h.OnReceiveResponse(nil)
    20  }