github.com/kisexp/xdchain@v0.0.0-20211206025815-490d6b732aa7/plugin/initializer/gateway_test.go (about)

     1  package initializer
     2  
     3  import (
     4  	"context"
     5  	"testing"
     6  
     7  	"github.com/kisexp/xdchain/plugin/gen/proto_common"
     8  	"github.com/golang/mock/gomock"
     9  	"github.com/stretchr/testify/assert"
    10  )
    11  
    12  func TestPluginGateway_Init(t *testing.T) {
    13  	ctrl := gomock.NewController(t)
    14  	defer ctrl.Finish()
    15  
    16  	req := &proto_common.PluginInitialization_Request{
    17  		HostIdentity:     "arbitraryName",
    18  		RawConfiguration: []byte("arbitrary config"),
    19  	}
    20  
    21  	mockClient := proto_common.NewMockPluginInitializerClient(ctrl)
    22  	mockClient.
    23  		EXPECT().
    24  		Init(gomock.Any(), gomock.Eq(req)).
    25  		Return(&proto_common.PluginInitialization_Response{}, nil)
    26  
    27  	testObject := &PluginGateway{client: mockClient}
    28  
    29  	err := testObject.Init(context.Background(), req.HostIdentity, req.RawConfiguration)
    30  
    31  	assert.NoError(t, err)
    32  }