github.com/pinpoint-apm/pinpoint-go-agent@v1.4.1-0.20240110120318-a50c2eb18c8c/protobuf/mock/mock_service.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: protobuf/Service.pb.go
     3  
     4  // Package mock_v1 is a generated GoMock package.
     5  package mock
     6  
     7  import (
     8  	context "context"
     9  	reflect "reflect"
    10  
    11  	gomock "github.com/golang/mock/gomock"
    12  	empty "github.com/golang/protobuf/ptypes/empty"
    13  	v1 "github.com/pinpoint-apm/pinpoint-go-agent/protobuf"
    14  	grpc "google.golang.org/grpc"
    15  	metadata "google.golang.org/grpc/metadata"
    16  )
    17  
    18  // MockSpanClient is a mock of SpanClient interface.
    19  type MockSpanClient struct {
    20  	ctrl     *gomock.Controller
    21  	recorder *MockSpanClientMockRecorder
    22  }
    23  
    24  // MockSpanClientMockRecorder is the mock recorder for MockSpanClient.
    25  type MockSpanClientMockRecorder struct {
    26  	mock *MockSpanClient
    27  }
    28  
    29  // NewMockSpanClient creates a new mock instance.
    30  func NewMockSpanClient(ctrl *gomock.Controller) *MockSpanClient {
    31  	mock := &MockSpanClient{ctrl: ctrl}
    32  	mock.recorder = &MockSpanClientMockRecorder{mock}
    33  	return mock
    34  }
    35  
    36  // EXPECT returns an object that allows the caller to indicate expected use.
    37  func (m *MockSpanClient) EXPECT() *MockSpanClientMockRecorder {
    38  	return m.recorder
    39  }
    40  
    41  // SendSpan mocks base method.
    42  func (m *MockSpanClient) SendSpan(ctx context.Context, opts ...grpc.CallOption) (v1.Span_SendSpanClient, error) {
    43  	m.ctrl.T.Helper()
    44  	varargs := []interface{}{ctx}
    45  	for _, a := range opts {
    46  		varargs = append(varargs, a)
    47  	}
    48  	ret := m.ctrl.Call(m, "SendSpan", varargs...)
    49  	ret0, _ := ret[0].(v1.Span_SendSpanClient)
    50  	ret1, _ := ret[1].(error)
    51  	return ret0, ret1
    52  }
    53  
    54  // SendSpan indicates an expected call of SendSpan.
    55  func (mr *MockSpanClientMockRecorder) SendSpan(ctx interface{}, opts ...interface{}) *gomock.Call {
    56  	mr.mock.ctrl.T.Helper()
    57  	varargs := append([]interface{}{ctx}, opts...)
    58  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendSpan", reflect.TypeOf((*MockSpanClient)(nil).SendSpan), varargs...)
    59  }
    60  
    61  // MockSpan_SendSpanClient is a mock of Span_SendSpanClient interface.
    62  type MockSpan_SendSpanClient struct {
    63  	ctrl     *gomock.Controller
    64  	recorder *MockSpan_SendSpanClientMockRecorder
    65  }
    66  
    67  // MockSpan_SendSpanClientMockRecorder is the mock recorder for MockSpan_SendSpanClient.
    68  type MockSpan_SendSpanClientMockRecorder struct {
    69  	mock *MockSpan_SendSpanClient
    70  }
    71  
    72  // NewMockSpan_SendSpanClient creates a new mock instance.
    73  func NewMockSpan_SendSpanClient(ctrl *gomock.Controller) *MockSpan_SendSpanClient {
    74  	mock := &MockSpan_SendSpanClient{ctrl: ctrl}
    75  	mock.recorder = &MockSpan_SendSpanClientMockRecorder{mock}
    76  	return mock
    77  }
    78  
    79  // EXPECT returns an object that allows the caller to indicate expected use.
    80  func (m *MockSpan_SendSpanClient) EXPECT() *MockSpan_SendSpanClientMockRecorder {
    81  	return m.recorder
    82  }
    83  
    84  // CloseAndRecv mocks base method.
    85  func (m *MockSpan_SendSpanClient) CloseAndRecv() (*empty.Empty, error) {
    86  	m.ctrl.T.Helper()
    87  	ret := m.ctrl.Call(m, "CloseAndRecv")
    88  	ret0, _ := ret[0].(*empty.Empty)
    89  	ret1, _ := ret[1].(error)
    90  	return ret0, ret1
    91  }
    92  
    93  // CloseAndRecv indicates an expected call of CloseAndRecv.
    94  func (mr *MockSpan_SendSpanClientMockRecorder) CloseAndRecv() *gomock.Call {
    95  	mr.mock.ctrl.T.Helper()
    96  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseAndRecv", reflect.TypeOf((*MockSpan_SendSpanClient)(nil).CloseAndRecv))
    97  }
    98  
    99  // CloseSend mocks base method.
   100  func (m *MockSpan_SendSpanClient) CloseSend() error {
   101  	m.ctrl.T.Helper()
   102  	ret := m.ctrl.Call(m, "CloseSend")
   103  	ret0, _ := ret[0].(error)
   104  	return ret0
   105  }
   106  
   107  // CloseSend indicates an expected call of CloseSend.
   108  func (mr *MockSpan_SendSpanClientMockRecorder) CloseSend() *gomock.Call {
   109  	mr.mock.ctrl.T.Helper()
   110  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseSend", reflect.TypeOf((*MockSpan_SendSpanClient)(nil).CloseSend))
   111  }
   112  
   113  // Context mocks base method.
   114  func (m *MockSpan_SendSpanClient) Context() context.Context {
   115  	m.ctrl.T.Helper()
   116  	ret := m.ctrl.Call(m, "Context")
   117  	ret0, _ := ret[0].(context.Context)
   118  	return ret0
   119  }
   120  
   121  // Context indicates an expected call of Context.
   122  func (mr *MockSpan_SendSpanClientMockRecorder) Context() *gomock.Call {
   123  	mr.mock.ctrl.T.Helper()
   124  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockSpan_SendSpanClient)(nil).Context))
   125  }
   126  
   127  // Header mocks base method.
   128  func (m *MockSpan_SendSpanClient) Header() (metadata.MD, error) {
   129  	m.ctrl.T.Helper()
   130  	ret := m.ctrl.Call(m, "Header")
   131  	ret0, _ := ret[0].(metadata.MD)
   132  	ret1, _ := ret[1].(error)
   133  	return ret0, ret1
   134  }
   135  
   136  // Header indicates an expected call of Header.
   137  func (mr *MockSpan_SendSpanClientMockRecorder) Header() *gomock.Call {
   138  	mr.mock.ctrl.T.Helper()
   139  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Header", reflect.TypeOf((*MockSpan_SendSpanClient)(nil).Header))
   140  }
   141  
   142  // RecvMsg mocks base method.
   143  func (m_2 *MockSpan_SendSpanClient) RecvMsg(m interface{}) error {
   144  	m_2.ctrl.T.Helper()
   145  	ret := m_2.ctrl.Call(m_2, "RecvMsg", m)
   146  	ret0, _ := ret[0].(error)
   147  	return ret0
   148  }
   149  
   150  // RecvMsg indicates an expected call of RecvMsg.
   151  func (mr *MockSpan_SendSpanClientMockRecorder) RecvMsg(m interface{}) *gomock.Call {
   152  	mr.mock.ctrl.T.Helper()
   153  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockSpan_SendSpanClient)(nil).RecvMsg), m)
   154  }
   155  
   156  // Send mocks base method.
   157  func (m *MockSpan_SendSpanClient) Send(arg0 *v1.PSpanMessage) error {
   158  	m.ctrl.T.Helper()
   159  	ret := m.ctrl.Call(m, "Send", arg0)
   160  	ret0, _ := ret[0].(error)
   161  	return ret0
   162  }
   163  
   164  // Send indicates an expected call of Send.
   165  func (mr *MockSpan_SendSpanClientMockRecorder) Send(arg0 interface{}) *gomock.Call {
   166  	mr.mock.ctrl.T.Helper()
   167  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockSpan_SendSpanClient)(nil).Send), arg0)
   168  }
   169  
   170  // SendMsg mocks base method.
   171  func (m_2 *MockSpan_SendSpanClient) SendMsg(m interface{}) error {
   172  	m_2.ctrl.T.Helper()
   173  	ret := m_2.ctrl.Call(m_2, "SendMsg", m)
   174  	ret0, _ := ret[0].(error)
   175  	return ret0
   176  }
   177  
   178  // SendMsg indicates an expected call of SendMsg.
   179  func (mr *MockSpan_SendSpanClientMockRecorder) SendMsg(m interface{}) *gomock.Call {
   180  	mr.mock.ctrl.T.Helper()
   181  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockSpan_SendSpanClient)(nil).SendMsg), m)
   182  }
   183  
   184  // Trailer mocks base method.
   185  func (m *MockSpan_SendSpanClient) Trailer() metadata.MD {
   186  	m.ctrl.T.Helper()
   187  	ret := m.ctrl.Call(m, "Trailer")
   188  	ret0, _ := ret[0].(metadata.MD)
   189  	return ret0
   190  }
   191  
   192  // Trailer indicates an expected call of Trailer.
   193  func (mr *MockSpan_SendSpanClientMockRecorder) Trailer() *gomock.Call {
   194  	mr.mock.ctrl.T.Helper()
   195  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Trailer", reflect.TypeOf((*MockSpan_SendSpanClient)(nil).Trailer))
   196  }
   197  
   198  // MockSpanServer is a mock of SpanServer interface.
   199  type MockSpanServer struct {
   200  	ctrl     *gomock.Controller
   201  	recorder *MockSpanServerMockRecorder
   202  }
   203  
   204  // MockSpanServerMockRecorder is the mock recorder for MockSpanServer.
   205  type MockSpanServerMockRecorder struct {
   206  	mock *MockSpanServer
   207  }
   208  
   209  // NewMockSpanServer creates a new mock instance.
   210  func NewMockSpanServer(ctrl *gomock.Controller) *MockSpanServer {
   211  	mock := &MockSpanServer{ctrl: ctrl}
   212  	mock.recorder = &MockSpanServerMockRecorder{mock}
   213  	return mock
   214  }
   215  
   216  // EXPECT returns an object that allows the caller to indicate expected use.
   217  func (m *MockSpanServer) EXPECT() *MockSpanServerMockRecorder {
   218  	return m.recorder
   219  }
   220  
   221  // SendSpan mocks base method.
   222  func (m *MockSpanServer) SendSpan(arg0 v1.Span_SendSpanServer) error {
   223  	m.ctrl.T.Helper()
   224  	ret := m.ctrl.Call(m, "SendSpan", arg0)
   225  	ret0, _ := ret[0].(error)
   226  	return ret0
   227  }
   228  
   229  // SendSpan indicates an expected call of SendSpan.
   230  func (mr *MockSpanServerMockRecorder) SendSpan(arg0 interface{}) *gomock.Call {
   231  	mr.mock.ctrl.T.Helper()
   232  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendSpan", reflect.TypeOf((*MockSpanServer)(nil).SendSpan), arg0)
   233  }
   234  
   235  // MockSpan_SendSpanServer is a mock of Span_SendSpanServer interface.
   236  type MockSpan_SendSpanServer struct {
   237  	ctrl     *gomock.Controller
   238  	recorder *MockSpan_SendSpanServerMockRecorder
   239  }
   240  
   241  // MockSpan_SendSpanServerMockRecorder is the mock recorder for MockSpan_SendSpanServer.
   242  type MockSpan_SendSpanServerMockRecorder struct {
   243  	mock *MockSpan_SendSpanServer
   244  }
   245  
   246  // NewMockSpan_SendSpanServer creates a new mock instance.
   247  func NewMockSpan_SendSpanServer(ctrl *gomock.Controller) *MockSpan_SendSpanServer {
   248  	mock := &MockSpan_SendSpanServer{ctrl: ctrl}
   249  	mock.recorder = &MockSpan_SendSpanServerMockRecorder{mock}
   250  	return mock
   251  }
   252  
   253  // EXPECT returns an object that allows the caller to indicate expected use.
   254  func (m *MockSpan_SendSpanServer) EXPECT() *MockSpan_SendSpanServerMockRecorder {
   255  	return m.recorder
   256  }
   257  
   258  // Context mocks base method.
   259  func (m *MockSpan_SendSpanServer) Context() context.Context {
   260  	m.ctrl.T.Helper()
   261  	ret := m.ctrl.Call(m, "Context")
   262  	ret0, _ := ret[0].(context.Context)
   263  	return ret0
   264  }
   265  
   266  // Context indicates an expected call of Context.
   267  func (mr *MockSpan_SendSpanServerMockRecorder) Context() *gomock.Call {
   268  	mr.mock.ctrl.T.Helper()
   269  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockSpan_SendSpanServer)(nil).Context))
   270  }
   271  
   272  // Recv mocks base method.
   273  func (m *MockSpan_SendSpanServer) Recv() (*v1.PSpanMessage, error) {
   274  	m.ctrl.T.Helper()
   275  	ret := m.ctrl.Call(m, "Recv")
   276  	ret0, _ := ret[0].(*v1.PSpanMessage)
   277  	ret1, _ := ret[1].(error)
   278  	return ret0, ret1
   279  }
   280  
   281  // Recv indicates an expected call of Recv.
   282  func (mr *MockSpan_SendSpanServerMockRecorder) Recv() *gomock.Call {
   283  	mr.mock.ctrl.T.Helper()
   284  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Recv", reflect.TypeOf((*MockSpan_SendSpanServer)(nil).Recv))
   285  }
   286  
   287  // RecvMsg mocks base method.
   288  func (m_2 *MockSpan_SendSpanServer) RecvMsg(m interface{}) error {
   289  	m_2.ctrl.T.Helper()
   290  	ret := m_2.ctrl.Call(m_2, "RecvMsg", m)
   291  	ret0, _ := ret[0].(error)
   292  	return ret0
   293  }
   294  
   295  // RecvMsg indicates an expected call of RecvMsg.
   296  func (mr *MockSpan_SendSpanServerMockRecorder) RecvMsg(m interface{}) *gomock.Call {
   297  	mr.mock.ctrl.T.Helper()
   298  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockSpan_SendSpanServer)(nil).RecvMsg), m)
   299  }
   300  
   301  // SendAndClose mocks base method.
   302  func (m *MockSpan_SendSpanServer) SendAndClose(arg0 *empty.Empty) error {
   303  	m.ctrl.T.Helper()
   304  	ret := m.ctrl.Call(m, "SendAndClose", arg0)
   305  	ret0, _ := ret[0].(error)
   306  	return ret0
   307  }
   308  
   309  // SendAndClose indicates an expected call of SendAndClose.
   310  func (mr *MockSpan_SendSpanServerMockRecorder) SendAndClose(arg0 interface{}) *gomock.Call {
   311  	mr.mock.ctrl.T.Helper()
   312  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendAndClose", reflect.TypeOf((*MockSpan_SendSpanServer)(nil).SendAndClose), arg0)
   313  }
   314  
   315  // SendHeader mocks base method.
   316  func (m *MockSpan_SendSpanServer) SendHeader(arg0 metadata.MD) error {
   317  	m.ctrl.T.Helper()
   318  	ret := m.ctrl.Call(m, "SendHeader", arg0)
   319  	ret0, _ := ret[0].(error)
   320  	return ret0
   321  }
   322  
   323  // SendHeader indicates an expected call of SendHeader.
   324  func (mr *MockSpan_SendSpanServerMockRecorder) SendHeader(arg0 interface{}) *gomock.Call {
   325  	mr.mock.ctrl.T.Helper()
   326  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendHeader", reflect.TypeOf((*MockSpan_SendSpanServer)(nil).SendHeader), arg0)
   327  }
   328  
   329  // SendMsg mocks base method.
   330  func (m_2 *MockSpan_SendSpanServer) SendMsg(m interface{}) error {
   331  	m_2.ctrl.T.Helper()
   332  	ret := m_2.ctrl.Call(m_2, "SendMsg", m)
   333  	ret0, _ := ret[0].(error)
   334  	return ret0
   335  }
   336  
   337  // SendMsg indicates an expected call of SendMsg.
   338  func (mr *MockSpan_SendSpanServerMockRecorder) SendMsg(m interface{}) *gomock.Call {
   339  	mr.mock.ctrl.T.Helper()
   340  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockSpan_SendSpanServer)(nil).SendMsg), m)
   341  }
   342  
   343  // SetHeader mocks base method.
   344  func (m *MockSpan_SendSpanServer) SetHeader(arg0 metadata.MD) error {
   345  	m.ctrl.T.Helper()
   346  	ret := m.ctrl.Call(m, "SetHeader", arg0)
   347  	ret0, _ := ret[0].(error)
   348  	return ret0
   349  }
   350  
   351  // SetHeader indicates an expected call of SetHeader.
   352  func (mr *MockSpan_SendSpanServerMockRecorder) SetHeader(arg0 interface{}) *gomock.Call {
   353  	mr.mock.ctrl.T.Helper()
   354  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetHeader", reflect.TypeOf((*MockSpan_SendSpanServer)(nil).SetHeader), arg0)
   355  }
   356  
   357  // SetTrailer mocks base method.
   358  func (m *MockSpan_SendSpanServer) SetTrailer(arg0 metadata.MD) {
   359  	m.ctrl.T.Helper()
   360  	m.ctrl.Call(m, "SetTrailer", arg0)
   361  }
   362  
   363  // SetTrailer indicates an expected call of SetTrailer.
   364  func (mr *MockSpan_SendSpanServerMockRecorder) SetTrailer(arg0 interface{}) *gomock.Call {
   365  	mr.mock.ctrl.T.Helper()
   366  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTrailer", reflect.TypeOf((*MockSpan_SendSpanServer)(nil).SetTrailer), arg0)
   367  }
   368  
   369  // MockAgentClient is a mock of AgentClient interface.
   370  type MockAgentClient struct {
   371  	ctrl     *gomock.Controller
   372  	recorder *MockAgentClientMockRecorder
   373  }
   374  
   375  // MockAgentClientMockRecorder is the mock recorder for MockAgentClient.
   376  type MockAgentClientMockRecorder struct {
   377  	mock *MockAgentClient
   378  }
   379  
   380  // NewMockAgentClient creates a new mock instance.
   381  func NewMockAgentClient(ctrl *gomock.Controller) *MockAgentClient {
   382  	mock := &MockAgentClient{ctrl: ctrl}
   383  	mock.recorder = &MockAgentClientMockRecorder{mock}
   384  	return mock
   385  }
   386  
   387  // EXPECT returns an object that allows the caller to indicate expected use.
   388  func (m *MockAgentClient) EXPECT() *MockAgentClientMockRecorder {
   389  	return m.recorder
   390  }
   391  
   392  // PingSession mocks base method.
   393  func (m *MockAgentClient) PingSession(ctx context.Context, opts ...grpc.CallOption) (v1.Agent_PingSessionClient, error) {
   394  	m.ctrl.T.Helper()
   395  	varargs := []interface{}{ctx}
   396  	for _, a := range opts {
   397  		varargs = append(varargs, a)
   398  	}
   399  	ret := m.ctrl.Call(m, "PingSession", varargs...)
   400  	ret0, _ := ret[0].(v1.Agent_PingSessionClient)
   401  	ret1, _ := ret[1].(error)
   402  	return ret0, ret1
   403  }
   404  
   405  // PingSession indicates an expected call of PingSession.
   406  func (mr *MockAgentClientMockRecorder) PingSession(ctx interface{}, opts ...interface{}) *gomock.Call {
   407  	mr.mock.ctrl.T.Helper()
   408  	varargs := append([]interface{}{ctx}, opts...)
   409  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PingSession", reflect.TypeOf((*MockAgentClient)(nil).PingSession), varargs...)
   410  }
   411  
   412  // RequestAgentInfo mocks base method.
   413  func (m *MockAgentClient) RequestAgentInfo(ctx context.Context, in *v1.PAgentInfo, opts ...grpc.CallOption) (*v1.PResult, error) {
   414  	m.ctrl.T.Helper()
   415  	varargs := []interface{}{ctx, in}
   416  	for _, a := range opts {
   417  		varargs = append(varargs, a)
   418  	}
   419  	ret := m.ctrl.Call(m, "RequestAgentInfo", varargs...)
   420  	ret0, _ := ret[0].(*v1.PResult)
   421  	ret1, _ := ret[1].(error)
   422  	return ret0, ret1
   423  }
   424  
   425  // RequestAgentInfo indicates an expected call of RequestAgentInfo.
   426  func (mr *MockAgentClientMockRecorder) RequestAgentInfo(ctx, in interface{}, opts ...interface{}) *gomock.Call {
   427  	mr.mock.ctrl.T.Helper()
   428  	varargs := append([]interface{}{ctx, in}, opts...)
   429  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RequestAgentInfo", reflect.TypeOf((*MockAgentClient)(nil).RequestAgentInfo), varargs...)
   430  }
   431  
   432  // MockAgent_PingSessionClient is a mock of Agent_PingSessionClient interface.
   433  type MockAgent_PingSessionClient struct {
   434  	ctrl     *gomock.Controller
   435  	recorder *MockAgent_PingSessionClientMockRecorder
   436  }
   437  
   438  // MockAgent_PingSessionClientMockRecorder is the mock recorder for MockAgent_PingSessionClient.
   439  type MockAgent_PingSessionClientMockRecorder struct {
   440  	mock *MockAgent_PingSessionClient
   441  }
   442  
   443  // NewMockAgent_PingSessionClient creates a new mock instance.
   444  func NewMockAgent_PingSessionClient(ctrl *gomock.Controller) *MockAgent_PingSessionClient {
   445  	mock := &MockAgent_PingSessionClient{ctrl: ctrl}
   446  	mock.recorder = &MockAgent_PingSessionClientMockRecorder{mock}
   447  	return mock
   448  }
   449  
   450  // EXPECT returns an object that allows the caller to indicate expected use.
   451  func (m *MockAgent_PingSessionClient) EXPECT() *MockAgent_PingSessionClientMockRecorder {
   452  	return m.recorder
   453  }
   454  
   455  // CloseSend mocks base method.
   456  func (m *MockAgent_PingSessionClient) CloseSend() error {
   457  	m.ctrl.T.Helper()
   458  	ret := m.ctrl.Call(m, "CloseSend")
   459  	ret0, _ := ret[0].(error)
   460  	return ret0
   461  }
   462  
   463  // CloseSend indicates an expected call of CloseSend.
   464  func (mr *MockAgent_PingSessionClientMockRecorder) CloseSend() *gomock.Call {
   465  	mr.mock.ctrl.T.Helper()
   466  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseSend", reflect.TypeOf((*MockAgent_PingSessionClient)(nil).CloseSend))
   467  }
   468  
   469  // Context mocks base method.
   470  func (m *MockAgent_PingSessionClient) Context() context.Context {
   471  	m.ctrl.T.Helper()
   472  	ret := m.ctrl.Call(m, "Context")
   473  	ret0, _ := ret[0].(context.Context)
   474  	return ret0
   475  }
   476  
   477  // Context indicates an expected call of Context.
   478  func (mr *MockAgent_PingSessionClientMockRecorder) Context() *gomock.Call {
   479  	mr.mock.ctrl.T.Helper()
   480  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockAgent_PingSessionClient)(nil).Context))
   481  }
   482  
   483  // Header mocks base method.
   484  func (m *MockAgent_PingSessionClient) Header() (metadata.MD, error) {
   485  	m.ctrl.T.Helper()
   486  	ret := m.ctrl.Call(m, "Header")
   487  	ret0, _ := ret[0].(metadata.MD)
   488  	ret1, _ := ret[1].(error)
   489  	return ret0, ret1
   490  }
   491  
   492  // Header indicates an expected call of Header.
   493  func (mr *MockAgent_PingSessionClientMockRecorder) Header() *gomock.Call {
   494  	mr.mock.ctrl.T.Helper()
   495  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Header", reflect.TypeOf((*MockAgent_PingSessionClient)(nil).Header))
   496  }
   497  
   498  // Recv mocks base method.
   499  func (m *MockAgent_PingSessionClient) Recv() (*v1.PPing, error) {
   500  	m.ctrl.T.Helper()
   501  	ret := m.ctrl.Call(m, "Recv")
   502  	ret0, _ := ret[0].(*v1.PPing)
   503  	ret1, _ := ret[1].(error)
   504  	return ret0, ret1
   505  }
   506  
   507  // Recv indicates an expected call of Recv.
   508  func (mr *MockAgent_PingSessionClientMockRecorder) Recv() *gomock.Call {
   509  	mr.mock.ctrl.T.Helper()
   510  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Recv", reflect.TypeOf((*MockAgent_PingSessionClient)(nil).Recv))
   511  }
   512  
   513  // RecvMsg mocks base method.
   514  func (m_2 *MockAgent_PingSessionClient) RecvMsg(m interface{}) error {
   515  	m_2.ctrl.T.Helper()
   516  	ret := m_2.ctrl.Call(m_2, "RecvMsg", m)
   517  	ret0, _ := ret[0].(error)
   518  	return ret0
   519  }
   520  
   521  // RecvMsg indicates an expected call of RecvMsg.
   522  func (mr *MockAgent_PingSessionClientMockRecorder) RecvMsg(m interface{}) *gomock.Call {
   523  	mr.mock.ctrl.T.Helper()
   524  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockAgent_PingSessionClient)(nil).RecvMsg), m)
   525  }
   526  
   527  // Send mocks base method.
   528  func (m *MockAgent_PingSessionClient) Send(arg0 *v1.PPing) error {
   529  	m.ctrl.T.Helper()
   530  	ret := m.ctrl.Call(m, "Send", arg0)
   531  	ret0, _ := ret[0].(error)
   532  	return ret0
   533  }
   534  
   535  // Send indicates an expected call of Send.
   536  func (mr *MockAgent_PingSessionClientMockRecorder) Send(arg0 interface{}) *gomock.Call {
   537  	mr.mock.ctrl.T.Helper()
   538  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockAgent_PingSessionClient)(nil).Send), arg0)
   539  }
   540  
   541  // SendMsg mocks base method.
   542  func (m_2 *MockAgent_PingSessionClient) SendMsg(m interface{}) error {
   543  	m_2.ctrl.T.Helper()
   544  	ret := m_2.ctrl.Call(m_2, "SendMsg", m)
   545  	ret0, _ := ret[0].(error)
   546  	return ret0
   547  }
   548  
   549  // SendMsg indicates an expected call of SendMsg.
   550  func (mr *MockAgent_PingSessionClientMockRecorder) SendMsg(m interface{}) *gomock.Call {
   551  	mr.mock.ctrl.T.Helper()
   552  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockAgent_PingSessionClient)(nil).SendMsg), m)
   553  }
   554  
   555  // Trailer mocks base method.
   556  func (m *MockAgent_PingSessionClient) Trailer() metadata.MD {
   557  	m.ctrl.T.Helper()
   558  	ret := m.ctrl.Call(m, "Trailer")
   559  	ret0, _ := ret[0].(metadata.MD)
   560  	return ret0
   561  }
   562  
   563  // Trailer indicates an expected call of Trailer.
   564  func (mr *MockAgent_PingSessionClientMockRecorder) Trailer() *gomock.Call {
   565  	mr.mock.ctrl.T.Helper()
   566  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Trailer", reflect.TypeOf((*MockAgent_PingSessionClient)(nil).Trailer))
   567  }
   568  
   569  // MockAgentServer is a mock of AgentServer interface.
   570  type MockAgentServer struct {
   571  	ctrl     *gomock.Controller
   572  	recorder *MockAgentServerMockRecorder
   573  }
   574  
   575  // MockAgentServerMockRecorder is the mock recorder for MockAgentServer.
   576  type MockAgentServerMockRecorder struct {
   577  	mock *MockAgentServer
   578  }
   579  
   580  // NewMockAgentServer creates a new mock instance.
   581  func NewMockAgentServer(ctrl *gomock.Controller) *MockAgentServer {
   582  	mock := &MockAgentServer{ctrl: ctrl}
   583  	mock.recorder = &MockAgentServerMockRecorder{mock}
   584  	return mock
   585  }
   586  
   587  // EXPECT returns an object that allows the caller to indicate expected use.
   588  func (m *MockAgentServer) EXPECT() *MockAgentServerMockRecorder {
   589  	return m.recorder
   590  }
   591  
   592  // PingSession mocks base method.
   593  func (m *MockAgentServer) PingSession(arg0 v1.Agent_PingSessionServer) error {
   594  	m.ctrl.T.Helper()
   595  	ret := m.ctrl.Call(m, "PingSession", arg0)
   596  	ret0, _ := ret[0].(error)
   597  	return ret0
   598  }
   599  
   600  // PingSession indicates an expected call of PingSession.
   601  func (mr *MockAgentServerMockRecorder) PingSession(arg0 interface{}) *gomock.Call {
   602  	mr.mock.ctrl.T.Helper()
   603  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PingSession", reflect.TypeOf((*MockAgentServer)(nil).PingSession), arg0)
   604  }
   605  
   606  // RequestAgentInfo mocks base method.
   607  func (m *MockAgentServer) RequestAgentInfo(arg0 context.Context, arg1 *v1.PAgentInfo) (*v1.PResult, error) {
   608  	m.ctrl.T.Helper()
   609  	ret := m.ctrl.Call(m, "RequestAgentInfo", arg0, arg1)
   610  	ret0, _ := ret[0].(*v1.PResult)
   611  	ret1, _ := ret[1].(error)
   612  	return ret0, ret1
   613  }
   614  
   615  // RequestAgentInfo indicates an expected call of RequestAgentInfo.
   616  func (mr *MockAgentServerMockRecorder) RequestAgentInfo(arg0, arg1 interface{}) *gomock.Call {
   617  	mr.mock.ctrl.T.Helper()
   618  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RequestAgentInfo", reflect.TypeOf((*MockAgentServer)(nil).RequestAgentInfo), arg0, arg1)
   619  }
   620  
   621  // MockAgent_PingSessionServer is a mock of Agent_PingSessionServer interface.
   622  type MockAgent_PingSessionServer struct {
   623  	ctrl     *gomock.Controller
   624  	recorder *MockAgent_PingSessionServerMockRecorder
   625  }
   626  
   627  // MockAgent_PingSessionServerMockRecorder is the mock recorder for MockAgent_PingSessionServer.
   628  type MockAgent_PingSessionServerMockRecorder struct {
   629  	mock *MockAgent_PingSessionServer
   630  }
   631  
   632  // NewMockAgent_PingSessionServer creates a new mock instance.
   633  func NewMockAgent_PingSessionServer(ctrl *gomock.Controller) *MockAgent_PingSessionServer {
   634  	mock := &MockAgent_PingSessionServer{ctrl: ctrl}
   635  	mock.recorder = &MockAgent_PingSessionServerMockRecorder{mock}
   636  	return mock
   637  }
   638  
   639  // EXPECT returns an object that allows the caller to indicate expected use.
   640  func (m *MockAgent_PingSessionServer) EXPECT() *MockAgent_PingSessionServerMockRecorder {
   641  	return m.recorder
   642  }
   643  
   644  // Context mocks base method.
   645  func (m *MockAgent_PingSessionServer) Context() context.Context {
   646  	m.ctrl.T.Helper()
   647  	ret := m.ctrl.Call(m, "Context")
   648  	ret0, _ := ret[0].(context.Context)
   649  	return ret0
   650  }
   651  
   652  // Context indicates an expected call of Context.
   653  func (mr *MockAgent_PingSessionServerMockRecorder) Context() *gomock.Call {
   654  	mr.mock.ctrl.T.Helper()
   655  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockAgent_PingSessionServer)(nil).Context))
   656  }
   657  
   658  // Recv mocks base method.
   659  func (m *MockAgent_PingSessionServer) Recv() (*v1.PPing, error) {
   660  	m.ctrl.T.Helper()
   661  	ret := m.ctrl.Call(m, "Recv")
   662  	ret0, _ := ret[0].(*v1.PPing)
   663  	ret1, _ := ret[1].(error)
   664  	return ret0, ret1
   665  }
   666  
   667  // Recv indicates an expected call of Recv.
   668  func (mr *MockAgent_PingSessionServerMockRecorder) Recv() *gomock.Call {
   669  	mr.mock.ctrl.T.Helper()
   670  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Recv", reflect.TypeOf((*MockAgent_PingSessionServer)(nil).Recv))
   671  }
   672  
   673  // RecvMsg mocks base method.
   674  func (m_2 *MockAgent_PingSessionServer) RecvMsg(m interface{}) error {
   675  	m_2.ctrl.T.Helper()
   676  	ret := m_2.ctrl.Call(m_2, "RecvMsg", m)
   677  	ret0, _ := ret[0].(error)
   678  	return ret0
   679  }
   680  
   681  // RecvMsg indicates an expected call of RecvMsg.
   682  func (mr *MockAgent_PingSessionServerMockRecorder) RecvMsg(m interface{}) *gomock.Call {
   683  	mr.mock.ctrl.T.Helper()
   684  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockAgent_PingSessionServer)(nil).RecvMsg), m)
   685  }
   686  
   687  // Send mocks base method.
   688  func (m *MockAgent_PingSessionServer) Send(arg0 *v1.PPing) error {
   689  	m.ctrl.T.Helper()
   690  	ret := m.ctrl.Call(m, "Send", arg0)
   691  	ret0, _ := ret[0].(error)
   692  	return ret0
   693  }
   694  
   695  // Send indicates an expected call of Send.
   696  func (mr *MockAgent_PingSessionServerMockRecorder) Send(arg0 interface{}) *gomock.Call {
   697  	mr.mock.ctrl.T.Helper()
   698  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockAgent_PingSessionServer)(nil).Send), arg0)
   699  }
   700  
   701  // SendHeader mocks base method.
   702  func (m *MockAgent_PingSessionServer) SendHeader(arg0 metadata.MD) error {
   703  	m.ctrl.T.Helper()
   704  	ret := m.ctrl.Call(m, "SendHeader", arg0)
   705  	ret0, _ := ret[0].(error)
   706  	return ret0
   707  }
   708  
   709  // SendHeader indicates an expected call of SendHeader.
   710  func (mr *MockAgent_PingSessionServerMockRecorder) SendHeader(arg0 interface{}) *gomock.Call {
   711  	mr.mock.ctrl.T.Helper()
   712  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendHeader", reflect.TypeOf((*MockAgent_PingSessionServer)(nil).SendHeader), arg0)
   713  }
   714  
   715  // SendMsg mocks base method.
   716  func (m_2 *MockAgent_PingSessionServer) SendMsg(m interface{}) error {
   717  	m_2.ctrl.T.Helper()
   718  	ret := m_2.ctrl.Call(m_2, "SendMsg", m)
   719  	ret0, _ := ret[0].(error)
   720  	return ret0
   721  }
   722  
   723  // SendMsg indicates an expected call of SendMsg.
   724  func (mr *MockAgent_PingSessionServerMockRecorder) SendMsg(m interface{}) *gomock.Call {
   725  	mr.mock.ctrl.T.Helper()
   726  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockAgent_PingSessionServer)(nil).SendMsg), m)
   727  }
   728  
   729  // SetHeader mocks base method.
   730  func (m *MockAgent_PingSessionServer) SetHeader(arg0 metadata.MD) error {
   731  	m.ctrl.T.Helper()
   732  	ret := m.ctrl.Call(m, "SetHeader", arg0)
   733  	ret0, _ := ret[0].(error)
   734  	return ret0
   735  }
   736  
   737  // SetHeader indicates an expected call of SetHeader.
   738  func (mr *MockAgent_PingSessionServerMockRecorder) SetHeader(arg0 interface{}) *gomock.Call {
   739  	mr.mock.ctrl.T.Helper()
   740  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetHeader", reflect.TypeOf((*MockAgent_PingSessionServer)(nil).SetHeader), arg0)
   741  }
   742  
   743  // SetTrailer mocks base method.
   744  func (m *MockAgent_PingSessionServer) SetTrailer(arg0 metadata.MD) {
   745  	m.ctrl.T.Helper()
   746  	m.ctrl.Call(m, "SetTrailer", arg0)
   747  }
   748  
   749  // SetTrailer indicates an expected call of SetTrailer.
   750  func (mr *MockAgent_PingSessionServerMockRecorder) SetTrailer(arg0 interface{}) *gomock.Call {
   751  	mr.mock.ctrl.T.Helper()
   752  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTrailer", reflect.TypeOf((*MockAgent_PingSessionServer)(nil).SetTrailer), arg0)
   753  }
   754  
   755  // MockMetadataClient is a mock of MetadataClient interface.
   756  type MockMetadataClient struct {
   757  	ctrl     *gomock.Controller
   758  	recorder *MockMetadataClientMockRecorder
   759  }
   760  
   761  // MockMetadataClientMockRecorder is the mock recorder for MockMetadataClient.
   762  type MockMetadataClientMockRecorder struct {
   763  	mock *MockMetadataClient
   764  }
   765  
   766  // NewMockMetadataClient creates a new mock instance.
   767  func NewMockMetadataClient(ctrl *gomock.Controller) *MockMetadataClient {
   768  	mock := &MockMetadataClient{ctrl: ctrl}
   769  	mock.recorder = &MockMetadataClientMockRecorder{mock}
   770  	return mock
   771  }
   772  
   773  // EXPECT returns an object that allows the caller to indicate expected use.
   774  func (m *MockMetadataClient) EXPECT() *MockMetadataClientMockRecorder {
   775  	return m.recorder
   776  }
   777  
   778  // RequestApiMetaData mocks base method.
   779  func (m *MockMetadataClient) RequestApiMetaData(ctx context.Context, in *v1.PApiMetaData, opts ...grpc.CallOption) (*v1.PResult, error) {
   780  	m.ctrl.T.Helper()
   781  	varargs := []interface{}{ctx, in}
   782  	for _, a := range opts {
   783  		varargs = append(varargs, a)
   784  	}
   785  	ret := m.ctrl.Call(m, "RequestApiMetaData", varargs...)
   786  	ret0, _ := ret[0].(*v1.PResult)
   787  	ret1, _ := ret[1].(error)
   788  	return ret0, ret1
   789  }
   790  
   791  // RequestApiMetaData indicates an expected call of RequestApiMetaData.
   792  func (mr *MockMetadataClientMockRecorder) RequestApiMetaData(ctx, in interface{}, opts ...interface{}) *gomock.Call {
   793  	mr.mock.ctrl.T.Helper()
   794  	varargs := append([]interface{}{ctx, in}, opts...)
   795  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RequestApiMetaData", reflect.TypeOf((*MockMetadataClient)(nil).RequestApiMetaData), varargs...)
   796  }
   797  
   798  // RequestSqlMetaData mocks base method.
   799  func (m *MockMetadataClient) RequestSqlMetaData(ctx context.Context, in *v1.PSqlMetaData, opts ...grpc.CallOption) (*v1.PResult, error) {
   800  	m.ctrl.T.Helper()
   801  	varargs := []interface{}{ctx, in}
   802  	for _, a := range opts {
   803  		varargs = append(varargs, a)
   804  	}
   805  	ret := m.ctrl.Call(m, "RequestSqlMetaData", varargs...)
   806  	ret0, _ := ret[0].(*v1.PResult)
   807  	ret1, _ := ret[1].(error)
   808  	return ret0, ret1
   809  }
   810  
   811  // RequestSqlMetaData indicates an expected call of RequestSqlMetaData.
   812  func (mr *MockMetadataClientMockRecorder) RequestSqlMetaData(ctx, in interface{}, opts ...interface{}) *gomock.Call {
   813  	mr.mock.ctrl.T.Helper()
   814  	varargs := append([]interface{}{ctx, in}, opts...)
   815  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RequestSqlMetaData", reflect.TypeOf((*MockMetadataClient)(nil).RequestSqlMetaData), varargs...)
   816  }
   817  
   818  // RequestStringMetaData mocks base method.
   819  func (m *MockMetadataClient) RequestStringMetaData(ctx context.Context, in *v1.PStringMetaData, opts ...grpc.CallOption) (*v1.PResult, error) {
   820  	m.ctrl.T.Helper()
   821  	varargs := []interface{}{ctx, in}
   822  	for _, a := range opts {
   823  		varargs = append(varargs, a)
   824  	}
   825  	ret := m.ctrl.Call(m, "RequestStringMetaData", varargs...)
   826  	ret0, _ := ret[0].(*v1.PResult)
   827  	ret1, _ := ret[1].(error)
   828  	return ret0, ret1
   829  }
   830  
   831  // RequestStringMetaData indicates an expected call of RequestStringMetaData.
   832  func (mr *MockMetadataClientMockRecorder) RequestStringMetaData(ctx, in interface{}, opts ...interface{}) *gomock.Call {
   833  	mr.mock.ctrl.T.Helper()
   834  	varargs := append([]interface{}{ctx, in}, opts...)
   835  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RequestStringMetaData", reflect.TypeOf((*MockMetadataClient)(nil).RequestStringMetaData), varargs...)
   836  }
   837  
   838  // MockMetadataServer is a mock of MetadataServer interface.
   839  type MockMetadataServer struct {
   840  	ctrl     *gomock.Controller
   841  	recorder *MockMetadataServerMockRecorder
   842  }
   843  
   844  // MockMetadataServerMockRecorder is the mock recorder for MockMetadataServer.
   845  type MockMetadataServerMockRecorder struct {
   846  	mock *MockMetadataServer
   847  }
   848  
   849  // NewMockMetadataServer creates a new mock instance.
   850  func NewMockMetadataServer(ctrl *gomock.Controller) *MockMetadataServer {
   851  	mock := &MockMetadataServer{ctrl: ctrl}
   852  	mock.recorder = &MockMetadataServerMockRecorder{mock}
   853  	return mock
   854  }
   855  
   856  // EXPECT returns an object that allows the caller to indicate expected use.
   857  func (m *MockMetadataServer) EXPECT() *MockMetadataServerMockRecorder {
   858  	return m.recorder
   859  }
   860  
   861  // RequestApiMetaData mocks base method.
   862  func (m *MockMetadataServer) RequestApiMetaData(arg0 context.Context, arg1 *v1.PApiMetaData) (*v1.PResult, error) {
   863  	m.ctrl.T.Helper()
   864  	ret := m.ctrl.Call(m, "RequestApiMetaData", arg0, arg1)
   865  	ret0, _ := ret[0].(*v1.PResult)
   866  	ret1, _ := ret[1].(error)
   867  	return ret0, ret1
   868  }
   869  
   870  // RequestApiMetaData indicates an expected call of RequestApiMetaData.
   871  func (mr *MockMetadataServerMockRecorder) RequestApiMetaData(arg0, arg1 interface{}) *gomock.Call {
   872  	mr.mock.ctrl.T.Helper()
   873  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RequestApiMetaData", reflect.TypeOf((*MockMetadataServer)(nil).RequestApiMetaData), arg0, arg1)
   874  }
   875  
   876  // RequestSqlMetaData mocks base method.
   877  func (m *MockMetadataServer) RequestSqlMetaData(arg0 context.Context, arg1 *v1.PSqlMetaData) (*v1.PResult, error) {
   878  	m.ctrl.T.Helper()
   879  	ret := m.ctrl.Call(m, "RequestSqlMetaData", arg0, arg1)
   880  	ret0, _ := ret[0].(*v1.PResult)
   881  	ret1, _ := ret[1].(error)
   882  	return ret0, ret1
   883  }
   884  
   885  // RequestSqlMetaData indicates an expected call of RequestSqlMetaData.
   886  func (mr *MockMetadataServerMockRecorder) RequestSqlMetaData(arg0, arg1 interface{}) *gomock.Call {
   887  	mr.mock.ctrl.T.Helper()
   888  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RequestSqlMetaData", reflect.TypeOf((*MockMetadataServer)(nil).RequestSqlMetaData), arg0, arg1)
   889  }
   890  
   891  // RequestStringMetaData mocks base method.
   892  func (m *MockMetadataServer) RequestStringMetaData(arg0 context.Context, arg1 *v1.PStringMetaData) (*v1.PResult, error) {
   893  	m.ctrl.T.Helper()
   894  	ret := m.ctrl.Call(m, "RequestStringMetaData", arg0, arg1)
   895  	ret0, _ := ret[0].(*v1.PResult)
   896  	ret1, _ := ret[1].(error)
   897  	return ret0, ret1
   898  }
   899  
   900  // RequestStringMetaData indicates an expected call of RequestStringMetaData.
   901  func (mr *MockMetadataServerMockRecorder) RequestStringMetaData(arg0, arg1 interface{}) *gomock.Call {
   902  	mr.mock.ctrl.T.Helper()
   903  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RequestStringMetaData", reflect.TypeOf((*MockMetadataServer)(nil).RequestStringMetaData), arg0, arg1)
   904  }
   905  
   906  // MockStatClient is a mock of StatClient interface.
   907  type MockStatClient struct {
   908  	ctrl     *gomock.Controller
   909  	recorder *MockStatClientMockRecorder
   910  }
   911  
   912  // MockStatClientMockRecorder is the mock recorder for MockStatClient.
   913  type MockStatClientMockRecorder struct {
   914  	mock *MockStatClient
   915  }
   916  
   917  // NewMockStatClient creates a new mock instance.
   918  func NewMockStatClient(ctrl *gomock.Controller) *MockStatClient {
   919  	mock := &MockStatClient{ctrl: ctrl}
   920  	mock.recorder = &MockStatClientMockRecorder{mock}
   921  	return mock
   922  }
   923  
   924  // EXPECT returns an object that allows the caller to indicate expected use.
   925  func (m *MockStatClient) EXPECT() *MockStatClientMockRecorder {
   926  	return m.recorder
   927  }
   928  
   929  // SendAgentStat mocks base method.
   930  func (m *MockStatClient) SendAgentStat(ctx context.Context, opts ...grpc.CallOption) (v1.Stat_SendAgentStatClient, error) {
   931  	m.ctrl.T.Helper()
   932  	varargs := []interface{}{ctx}
   933  	for _, a := range opts {
   934  		varargs = append(varargs, a)
   935  	}
   936  	ret := m.ctrl.Call(m, "SendAgentStat", varargs...)
   937  	ret0, _ := ret[0].(v1.Stat_SendAgentStatClient)
   938  	ret1, _ := ret[1].(error)
   939  	return ret0, ret1
   940  }
   941  
   942  // SendAgentStat indicates an expected call of SendAgentStat.
   943  func (mr *MockStatClientMockRecorder) SendAgentStat(ctx interface{}, opts ...interface{}) *gomock.Call {
   944  	mr.mock.ctrl.T.Helper()
   945  	varargs := append([]interface{}{ctx}, opts...)
   946  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendAgentStat", reflect.TypeOf((*MockStatClient)(nil).SendAgentStat), varargs...)
   947  }
   948  
   949  // MockStat_SendAgentStatClient is a mock of Stat_SendAgentStatClient interface.
   950  type MockStat_SendAgentStatClient struct {
   951  	ctrl     *gomock.Controller
   952  	recorder *MockStat_SendAgentStatClientMockRecorder
   953  }
   954  
   955  // MockStat_SendAgentStatClientMockRecorder is the mock recorder for MockStat_SendAgentStatClient.
   956  type MockStat_SendAgentStatClientMockRecorder struct {
   957  	mock *MockStat_SendAgentStatClient
   958  }
   959  
   960  // NewMockStat_SendAgentStatClient creates a new mock instance.
   961  func NewMockStat_SendAgentStatClient(ctrl *gomock.Controller) *MockStat_SendAgentStatClient {
   962  	mock := &MockStat_SendAgentStatClient{ctrl: ctrl}
   963  	mock.recorder = &MockStat_SendAgentStatClientMockRecorder{mock}
   964  	return mock
   965  }
   966  
   967  // EXPECT returns an object that allows the caller to indicate expected use.
   968  func (m *MockStat_SendAgentStatClient) EXPECT() *MockStat_SendAgentStatClientMockRecorder {
   969  	return m.recorder
   970  }
   971  
   972  // CloseAndRecv mocks base method.
   973  func (m *MockStat_SendAgentStatClient) CloseAndRecv() (*empty.Empty, error) {
   974  	m.ctrl.T.Helper()
   975  	ret := m.ctrl.Call(m, "CloseAndRecv")
   976  	ret0, _ := ret[0].(*empty.Empty)
   977  	ret1, _ := ret[1].(error)
   978  	return ret0, ret1
   979  }
   980  
   981  // CloseAndRecv indicates an expected call of CloseAndRecv.
   982  func (mr *MockStat_SendAgentStatClientMockRecorder) CloseAndRecv() *gomock.Call {
   983  	mr.mock.ctrl.T.Helper()
   984  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseAndRecv", reflect.TypeOf((*MockStat_SendAgentStatClient)(nil).CloseAndRecv))
   985  }
   986  
   987  // CloseSend mocks base method.
   988  func (m *MockStat_SendAgentStatClient) CloseSend() error {
   989  	m.ctrl.T.Helper()
   990  	ret := m.ctrl.Call(m, "CloseSend")
   991  	ret0, _ := ret[0].(error)
   992  	return ret0
   993  }
   994  
   995  // CloseSend indicates an expected call of CloseSend.
   996  func (mr *MockStat_SendAgentStatClientMockRecorder) CloseSend() *gomock.Call {
   997  	mr.mock.ctrl.T.Helper()
   998  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseSend", reflect.TypeOf((*MockStat_SendAgentStatClient)(nil).CloseSend))
   999  }
  1000  
  1001  // Context mocks base method.
  1002  func (m *MockStat_SendAgentStatClient) Context() context.Context {
  1003  	m.ctrl.T.Helper()
  1004  	ret := m.ctrl.Call(m, "Context")
  1005  	ret0, _ := ret[0].(context.Context)
  1006  	return ret0
  1007  }
  1008  
  1009  // Context indicates an expected call of Context.
  1010  func (mr *MockStat_SendAgentStatClientMockRecorder) Context() *gomock.Call {
  1011  	mr.mock.ctrl.T.Helper()
  1012  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockStat_SendAgentStatClient)(nil).Context))
  1013  }
  1014  
  1015  // Header mocks base method.
  1016  func (m *MockStat_SendAgentStatClient) Header() (metadata.MD, error) {
  1017  	m.ctrl.T.Helper()
  1018  	ret := m.ctrl.Call(m, "Header")
  1019  	ret0, _ := ret[0].(metadata.MD)
  1020  	ret1, _ := ret[1].(error)
  1021  	return ret0, ret1
  1022  }
  1023  
  1024  // Header indicates an expected call of Header.
  1025  func (mr *MockStat_SendAgentStatClientMockRecorder) Header() *gomock.Call {
  1026  	mr.mock.ctrl.T.Helper()
  1027  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Header", reflect.TypeOf((*MockStat_SendAgentStatClient)(nil).Header))
  1028  }
  1029  
  1030  // RecvMsg mocks base method.
  1031  func (m_2 *MockStat_SendAgentStatClient) RecvMsg(m interface{}) error {
  1032  	m_2.ctrl.T.Helper()
  1033  	ret := m_2.ctrl.Call(m_2, "RecvMsg", m)
  1034  	ret0, _ := ret[0].(error)
  1035  	return ret0
  1036  }
  1037  
  1038  // RecvMsg indicates an expected call of RecvMsg.
  1039  func (mr *MockStat_SendAgentStatClientMockRecorder) RecvMsg(m interface{}) *gomock.Call {
  1040  	mr.mock.ctrl.T.Helper()
  1041  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockStat_SendAgentStatClient)(nil).RecvMsg), m)
  1042  }
  1043  
  1044  // Send mocks base method.
  1045  func (m *MockStat_SendAgentStatClient) Send(arg0 *v1.PStatMessage) error {
  1046  	m.ctrl.T.Helper()
  1047  	ret := m.ctrl.Call(m, "Send", arg0)
  1048  	ret0, _ := ret[0].(error)
  1049  	return ret0
  1050  }
  1051  
  1052  // Send indicates an expected call of Send.
  1053  func (mr *MockStat_SendAgentStatClientMockRecorder) Send(arg0 interface{}) *gomock.Call {
  1054  	mr.mock.ctrl.T.Helper()
  1055  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockStat_SendAgentStatClient)(nil).Send), arg0)
  1056  }
  1057  
  1058  // SendMsg mocks base method.
  1059  func (m_2 *MockStat_SendAgentStatClient) SendMsg(m interface{}) error {
  1060  	m_2.ctrl.T.Helper()
  1061  	ret := m_2.ctrl.Call(m_2, "SendMsg", m)
  1062  	ret0, _ := ret[0].(error)
  1063  	return ret0
  1064  }
  1065  
  1066  // SendMsg indicates an expected call of SendMsg.
  1067  func (mr *MockStat_SendAgentStatClientMockRecorder) SendMsg(m interface{}) *gomock.Call {
  1068  	mr.mock.ctrl.T.Helper()
  1069  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockStat_SendAgentStatClient)(nil).SendMsg), m)
  1070  }
  1071  
  1072  // Trailer mocks base method.
  1073  func (m *MockStat_SendAgentStatClient) Trailer() metadata.MD {
  1074  	m.ctrl.T.Helper()
  1075  	ret := m.ctrl.Call(m, "Trailer")
  1076  	ret0, _ := ret[0].(metadata.MD)
  1077  	return ret0
  1078  }
  1079  
  1080  // Trailer indicates an expected call of Trailer.
  1081  func (mr *MockStat_SendAgentStatClientMockRecorder) Trailer() *gomock.Call {
  1082  	mr.mock.ctrl.T.Helper()
  1083  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Trailer", reflect.TypeOf((*MockStat_SendAgentStatClient)(nil).Trailer))
  1084  }
  1085  
  1086  // MockStatServer is a mock of StatServer interface.
  1087  type MockStatServer struct {
  1088  	ctrl     *gomock.Controller
  1089  	recorder *MockStatServerMockRecorder
  1090  }
  1091  
  1092  // MockStatServerMockRecorder is the mock recorder for MockStatServer.
  1093  type MockStatServerMockRecorder struct {
  1094  	mock *MockStatServer
  1095  }
  1096  
  1097  // NewMockStatServer creates a new mock instance.
  1098  func NewMockStatServer(ctrl *gomock.Controller) *MockStatServer {
  1099  	mock := &MockStatServer{ctrl: ctrl}
  1100  	mock.recorder = &MockStatServerMockRecorder{mock}
  1101  	return mock
  1102  }
  1103  
  1104  // EXPECT returns an object that allows the caller to indicate expected use.
  1105  func (m *MockStatServer) EXPECT() *MockStatServerMockRecorder {
  1106  	return m.recorder
  1107  }
  1108  
  1109  // SendAgentStat mocks base method.
  1110  func (m *MockStatServer) SendAgentStat(arg0 v1.Stat_SendAgentStatServer) error {
  1111  	m.ctrl.T.Helper()
  1112  	ret := m.ctrl.Call(m, "SendAgentStat", arg0)
  1113  	ret0, _ := ret[0].(error)
  1114  	return ret0
  1115  }
  1116  
  1117  // SendAgentStat indicates an expected call of SendAgentStat.
  1118  func (mr *MockStatServerMockRecorder) SendAgentStat(arg0 interface{}) *gomock.Call {
  1119  	mr.mock.ctrl.T.Helper()
  1120  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendAgentStat", reflect.TypeOf((*MockStatServer)(nil).SendAgentStat), arg0)
  1121  }
  1122  
  1123  // MockStat_SendAgentStatServer is a mock of Stat_SendAgentStatServer interface.
  1124  type MockStat_SendAgentStatServer struct {
  1125  	ctrl     *gomock.Controller
  1126  	recorder *MockStat_SendAgentStatServerMockRecorder
  1127  }
  1128  
  1129  // MockStat_SendAgentStatServerMockRecorder is the mock recorder for MockStat_SendAgentStatServer.
  1130  type MockStat_SendAgentStatServerMockRecorder struct {
  1131  	mock *MockStat_SendAgentStatServer
  1132  }
  1133  
  1134  // NewMockStat_SendAgentStatServer creates a new mock instance.
  1135  func NewMockStat_SendAgentStatServer(ctrl *gomock.Controller) *MockStat_SendAgentStatServer {
  1136  	mock := &MockStat_SendAgentStatServer{ctrl: ctrl}
  1137  	mock.recorder = &MockStat_SendAgentStatServerMockRecorder{mock}
  1138  	return mock
  1139  }
  1140  
  1141  // EXPECT returns an object that allows the caller to indicate expected use.
  1142  func (m *MockStat_SendAgentStatServer) EXPECT() *MockStat_SendAgentStatServerMockRecorder {
  1143  	return m.recorder
  1144  }
  1145  
  1146  // Context mocks base method.
  1147  func (m *MockStat_SendAgentStatServer) Context() context.Context {
  1148  	m.ctrl.T.Helper()
  1149  	ret := m.ctrl.Call(m, "Context")
  1150  	ret0, _ := ret[0].(context.Context)
  1151  	return ret0
  1152  }
  1153  
  1154  // Context indicates an expected call of Context.
  1155  func (mr *MockStat_SendAgentStatServerMockRecorder) Context() *gomock.Call {
  1156  	mr.mock.ctrl.T.Helper()
  1157  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockStat_SendAgentStatServer)(nil).Context))
  1158  }
  1159  
  1160  // Recv mocks base method.
  1161  func (m *MockStat_SendAgentStatServer) Recv() (*v1.PStatMessage, error) {
  1162  	m.ctrl.T.Helper()
  1163  	ret := m.ctrl.Call(m, "Recv")
  1164  	ret0, _ := ret[0].(*v1.PStatMessage)
  1165  	ret1, _ := ret[1].(error)
  1166  	return ret0, ret1
  1167  }
  1168  
  1169  // Recv indicates an expected call of Recv.
  1170  func (mr *MockStat_SendAgentStatServerMockRecorder) Recv() *gomock.Call {
  1171  	mr.mock.ctrl.T.Helper()
  1172  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Recv", reflect.TypeOf((*MockStat_SendAgentStatServer)(nil).Recv))
  1173  }
  1174  
  1175  // RecvMsg mocks base method.
  1176  func (m_2 *MockStat_SendAgentStatServer) RecvMsg(m interface{}) error {
  1177  	m_2.ctrl.T.Helper()
  1178  	ret := m_2.ctrl.Call(m_2, "RecvMsg", m)
  1179  	ret0, _ := ret[0].(error)
  1180  	return ret0
  1181  }
  1182  
  1183  // RecvMsg indicates an expected call of RecvMsg.
  1184  func (mr *MockStat_SendAgentStatServerMockRecorder) RecvMsg(m interface{}) *gomock.Call {
  1185  	mr.mock.ctrl.T.Helper()
  1186  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockStat_SendAgentStatServer)(nil).RecvMsg), m)
  1187  }
  1188  
  1189  // SendAndClose mocks base method.
  1190  func (m *MockStat_SendAgentStatServer) SendAndClose(arg0 *empty.Empty) error {
  1191  	m.ctrl.T.Helper()
  1192  	ret := m.ctrl.Call(m, "SendAndClose", arg0)
  1193  	ret0, _ := ret[0].(error)
  1194  	return ret0
  1195  }
  1196  
  1197  // SendAndClose indicates an expected call of SendAndClose.
  1198  func (mr *MockStat_SendAgentStatServerMockRecorder) SendAndClose(arg0 interface{}) *gomock.Call {
  1199  	mr.mock.ctrl.T.Helper()
  1200  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendAndClose", reflect.TypeOf((*MockStat_SendAgentStatServer)(nil).SendAndClose), arg0)
  1201  }
  1202  
  1203  // SendHeader mocks base method.
  1204  func (m *MockStat_SendAgentStatServer) SendHeader(arg0 metadata.MD) error {
  1205  	m.ctrl.T.Helper()
  1206  	ret := m.ctrl.Call(m, "SendHeader", arg0)
  1207  	ret0, _ := ret[0].(error)
  1208  	return ret0
  1209  }
  1210  
  1211  // SendHeader indicates an expected call of SendHeader.
  1212  func (mr *MockStat_SendAgentStatServerMockRecorder) SendHeader(arg0 interface{}) *gomock.Call {
  1213  	mr.mock.ctrl.T.Helper()
  1214  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendHeader", reflect.TypeOf((*MockStat_SendAgentStatServer)(nil).SendHeader), arg0)
  1215  }
  1216  
  1217  // SendMsg mocks base method.
  1218  func (m_2 *MockStat_SendAgentStatServer) SendMsg(m interface{}) error {
  1219  	m_2.ctrl.T.Helper()
  1220  	ret := m_2.ctrl.Call(m_2, "SendMsg", m)
  1221  	ret0, _ := ret[0].(error)
  1222  	return ret0
  1223  }
  1224  
  1225  // SendMsg indicates an expected call of SendMsg.
  1226  func (mr *MockStat_SendAgentStatServerMockRecorder) SendMsg(m interface{}) *gomock.Call {
  1227  	mr.mock.ctrl.T.Helper()
  1228  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockStat_SendAgentStatServer)(nil).SendMsg), m)
  1229  }
  1230  
  1231  // SetHeader mocks base method.
  1232  func (m *MockStat_SendAgentStatServer) SetHeader(arg0 metadata.MD) error {
  1233  	m.ctrl.T.Helper()
  1234  	ret := m.ctrl.Call(m, "SetHeader", arg0)
  1235  	ret0, _ := ret[0].(error)
  1236  	return ret0
  1237  }
  1238  
  1239  // SetHeader indicates an expected call of SetHeader.
  1240  func (mr *MockStat_SendAgentStatServerMockRecorder) SetHeader(arg0 interface{}) *gomock.Call {
  1241  	mr.mock.ctrl.T.Helper()
  1242  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetHeader", reflect.TypeOf((*MockStat_SendAgentStatServer)(nil).SetHeader), arg0)
  1243  }
  1244  
  1245  // SetTrailer mocks base method.
  1246  func (m *MockStat_SendAgentStatServer) SetTrailer(arg0 metadata.MD) {
  1247  	m.ctrl.T.Helper()
  1248  	m.ctrl.Call(m, "SetTrailer", arg0)
  1249  }
  1250  
  1251  // SetTrailer indicates an expected call of SetTrailer.
  1252  func (mr *MockStat_SendAgentStatServerMockRecorder) SetTrailer(arg0 interface{}) *gomock.Call {
  1253  	mr.mock.ctrl.T.Helper()
  1254  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTrailer", reflect.TypeOf((*MockStat_SendAgentStatServer)(nil).SetTrailer), arg0)
  1255  }
  1256  
  1257  // MockProfilerCommandServiceClient is a mock of ProfilerCommandServiceClient interface.
  1258  type MockProfilerCommandServiceClient struct {
  1259  	ctrl     *gomock.Controller
  1260  	recorder *MockProfilerCommandServiceClientMockRecorder
  1261  }
  1262  
  1263  // MockProfilerCommandServiceClientMockRecorder is the mock recorder for MockProfilerCommandServiceClient.
  1264  type MockProfilerCommandServiceClientMockRecorder struct {
  1265  	mock *MockProfilerCommandServiceClient
  1266  }
  1267  
  1268  // NewMockProfilerCommandServiceClient creates a new mock instance.
  1269  func NewMockProfilerCommandServiceClient(ctrl *gomock.Controller) *MockProfilerCommandServiceClient {
  1270  	mock := &MockProfilerCommandServiceClient{ctrl: ctrl}
  1271  	mock.recorder = &MockProfilerCommandServiceClientMockRecorder{mock}
  1272  	return mock
  1273  }
  1274  
  1275  // EXPECT returns an object that allows the caller to indicate expected use.
  1276  func (m *MockProfilerCommandServiceClient) EXPECT() *MockProfilerCommandServiceClientMockRecorder {
  1277  	return m.recorder
  1278  }
  1279  
  1280  // CommandActiveThreadDump mocks base method.
  1281  func (m *MockProfilerCommandServiceClient) CommandActiveThreadDump(ctx context.Context, in *v1.PCmdActiveThreadDumpRes, opts ...grpc.CallOption) (*empty.Empty, error) {
  1282  	m.ctrl.T.Helper()
  1283  	varargs := []interface{}{ctx, in}
  1284  	for _, a := range opts {
  1285  		varargs = append(varargs, a)
  1286  	}
  1287  	ret := m.ctrl.Call(m, "CommandActiveThreadDump", varargs...)
  1288  	ret0, _ := ret[0].(*empty.Empty)
  1289  	ret1, _ := ret[1].(error)
  1290  	return ret0, ret1
  1291  }
  1292  
  1293  // CommandActiveThreadDump indicates an expected call of CommandActiveThreadDump.
  1294  func (mr *MockProfilerCommandServiceClientMockRecorder) CommandActiveThreadDump(ctx, in interface{}, opts ...interface{}) *gomock.Call {
  1295  	mr.mock.ctrl.T.Helper()
  1296  	varargs := append([]interface{}{ctx, in}, opts...)
  1297  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CommandActiveThreadDump", reflect.TypeOf((*MockProfilerCommandServiceClient)(nil).CommandActiveThreadDump), varargs...)
  1298  }
  1299  
  1300  // CommandActiveThreadLightDump mocks base method.
  1301  func (m *MockProfilerCommandServiceClient) CommandActiveThreadLightDump(ctx context.Context, in *v1.PCmdActiveThreadLightDumpRes, opts ...grpc.CallOption) (*empty.Empty, error) {
  1302  	m.ctrl.T.Helper()
  1303  	varargs := []interface{}{ctx, in}
  1304  	for _, a := range opts {
  1305  		varargs = append(varargs, a)
  1306  	}
  1307  	ret := m.ctrl.Call(m, "CommandActiveThreadLightDump", varargs...)
  1308  	ret0, _ := ret[0].(*empty.Empty)
  1309  	ret1, _ := ret[1].(error)
  1310  	return ret0, ret1
  1311  }
  1312  
  1313  // CommandActiveThreadLightDump indicates an expected call of CommandActiveThreadLightDump.
  1314  func (mr *MockProfilerCommandServiceClientMockRecorder) CommandActiveThreadLightDump(ctx, in interface{}, opts ...interface{}) *gomock.Call {
  1315  	mr.mock.ctrl.T.Helper()
  1316  	varargs := append([]interface{}{ctx, in}, opts...)
  1317  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CommandActiveThreadLightDump", reflect.TypeOf((*MockProfilerCommandServiceClient)(nil).CommandActiveThreadLightDump), varargs...)
  1318  }
  1319  
  1320  // CommandEcho mocks base method.
  1321  func (m *MockProfilerCommandServiceClient) CommandEcho(ctx context.Context, in *v1.PCmdEchoResponse, opts ...grpc.CallOption) (*empty.Empty, error) {
  1322  	m.ctrl.T.Helper()
  1323  	varargs := []interface{}{ctx, in}
  1324  	for _, a := range opts {
  1325  		varargs = append(varargs, a)
  1326  	}
  1327  	ret := m.ctrl.Call(m, "CommandEcho", varargs...)
  1328  	ret0, _ := ret[0].(*empty.Empty)
  1329  	ret1, _ := ret[1].(error)
  1330  	return ret0, ret1
  1331  }
  1332  
  1333  // CommandEcho indicates an expected call of CommandEcho.
  1334  func (mr *MockProfilerCommandServiceClientMockRecorder) CommandEcho(ctx, in interface{}, opts ...interface{}) *gomock.Call {
  1335  	mr.mock.ctrl.T.Helper()
  1336  	varargs := append([]interface{}{ctx, in}, opts...)
  1337  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CommandEcho", reflect.TypeOf((*MockProfilerCommandServiceClient)(nil).CommandEcho), varargs...)
  1338  }
  1339  
  1340  // CommandStreamActiveThreadCount mocks base method.
  1341  func (m *MockProfilerCommandServiceClient) CommandStreamActiveThreadCount(ctx context.Context, opts ...grpc.CallOption) (v1.ProfilerCommandService_CommandStreamActiveThreadCountClient, error) {
  1342  	m.ctrl.T.Helper()
  1343  	varargs := []interface{}{ctx}
  1344  	for _, a := range opts {
  1345  		varargs = append(varargs, a)
  1346  	}
  1347  	ret := m.ctrl.Call(m, "CommandStreamActiveThreadCount", varargs...)
  1348  	ret0, _ := ret[0].(v1.ProfilerCommandService_CommandStreamActiveThreadCountClient)
  1349  	ret1, _ := ret[1].(error)
  1350  	return ret0, ret1
  1351  }
  1352  
  1353  // CommandStreamActiveThreadCount indicates an expected call of CommandStreamActiveThreadCount.
  1354  func (mr *MockProfilerCommandServiceClientMockRecorder) CommandStreamActiveThreadCount(ctx interface{}, opts ...interface{}) *gomock.Call {
  1355  	mr.mock.ctrl.T.Helper()
  1356  	varargs := append([]interface{}{ctx}, opts...)
  1357  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CommandStreamActiveThreadCount", reflect.TypeOf((*MockProfilerCommandServiceClient)(nil).CommandStreamActiveThreadCount), varargs...)
  1358  }
  1359  
  1360  // HandleCommand mocks base method.
  1361  func (m *MockProfilerCommandServiceClient) HandleCommand(ctx context.Context, opts ...grpc.CallOption) (v1.ProfilerCommandService_HandleCommandClient, error) {
  1362  	m.ctrl.T.Helper()
  1363  	varargs := []interface{}{ctx}
  1364  	for _, a := range opts {
  1365  		varargs = append(varargs, a)
  1366  	}
  1367  	ret := m.ctrl.Call(m, "HandleCommand", varargs...)
  1368  	ret0, _ := ret[0].(v1.ProfilerCommandService_HandleCommandClient)
  1369  	ret1, _ := ret[1].(error)
  1370  	return ret0, ret1
  1371  }
  1372  
  1373  // HandleCommand indicates an expected call of HandleCommand.
  1374  func (mr *MockProfilerCommandServiceClientMockRecorder) HandleCommand(ctx interface{}, opts ...interface{}) *gomock.Call {
  1375  	mr.mock.ctrl.T.Helper()
  1376  	varargs := append([]interface{}{ctx}, opts...)
  1377  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HandleCommand", reflect.TypeOf((*MockProfilerCommandServiceClient)(nil).HandleCommand), varargs...)
  1378  }
  1379  
  1380  // MockProfilerCommandService_HandleCommandClient is a mock of ProfilerCommandService_HandleCommandClient interface.
  1381  type MockProfilerCommandService_HandleCommandClient struct {
  1382  	ctrl     *gomock.Controller
  1383  	recorder *MockProfilerCommandService_HandleCommandClientMockRecorder
  1384  }
  1385  
  1386  // MockProfilerCommandService_HandleCommandClientMockRecorder is the mock recorder for MockProfilerCommandService_HandleCommandClient.
  1387  type MockProfilerCommandService_HandleCommandClientMockRecorder struct {
  1388  	mock *MockProfilerCommandService_HandleCommandClient
  1389  }
  1390  
  1391  // NewMockProfilerCommandService_HandleCommandClient creates a new mock instance.
  1392  func NewMockProfilerCommandService_HandleCommandClient(ctrl *gomock.Controller) *MockProfilerCommandService_HandleCommandClient {
  1393  	mock := &MockProfilerCommandService_HandleCommandClient{ctrl: ctrl}
  1394  	mock.recorder = &MockProfilerCommandService_HandleCommandClientMockRecorder{mock}
  1395  	return mock
  1396  }
  1397  
  1398  // EXPECT returns an object that allows the caller to indicate expected use.
  1399  func (m *MockProfilerCommandService_HandleCommandClient) EXPECT() *MockProfilerCommandService_HandleCommandClientMockRecorder {
  1400  	return m.recorder
  1401  }
  1402  
  1403  // CloseSend mocks base method.
  1404  func (m *MockProfilerCommandService_HandleCommandClient) CloseSend() error {
  1405  	m.ctrl.T.Helper()
  1406  	ret := m.ctrl.Call(m, "CloseSend")
  1407  	ret0, _ := ret[0].(error)
  1408  	return ret0
  1409  }
  1410  
  1411  // CloseSend indicates an expected call of CloseSend.
  1412  func (mr *MockProfilerCommandService_HandleCommandClientMockRecorder) CloseSend() *gomock.Call {
  1413  	mr.mock.ctrl.T.Helper()
  1414  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseSend", reflect.TypeOf((*MockProfilerCommandService_HandleCommandClient)(nil).CloseSend))
  1415  }
  1416  
  1417  // Context mocks base method.
  1418  func (m *MockProfilerCommandService_HandleCommandClient) Context() context.Context {
  1419  	m.ctrl.T.Helper()
  1420  	ret := m.ctrl.Call(m, "Context")
  1421  	ret0, _ := ret[0].(context.Context)
  1422  	return ret0
  1423  }
  1424  
  1425  // Context indicates an expected call of Context.
  1426  func (mr *MockProfilerCommandService_HandleCommandClientMockRecorder) Context() *gomock.Call {
  1427  	mr.mock.ctrl.T.Helper()
  1428  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockProfilerCommandService_HandleCommandClient)(nil).Context))
  1429  }
  1430  
  1431  // Header mocks base method.
  1432  func (m *MockProfilerCommandService_HandleCommandClient) Header() (metadata.MD, error) {
  1433  	m.ctrl.T.Helper()
  1434  	ret := m.ctrl.Call(m, "Header")
  1435  	ret0, _ := ret[0].(metadata.MD)
  1436  	ret1, _ := ret[1].(error)
  1437  	return ret0, ret1
  1438  }
  1439  
  1440  // Header indicates an expected call of Header.
  1441  func (mr *MockProfilerCommandService_HandleCommandClientMockRecorder) Header() *gomock.Call {
  1442  	mr.mock.ctrl.T.Helper()
  1443  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Header", reflect.TypeOf((*MockProfilerCommandService_HandleCommandClient)(nil).Header))
  1444  }
  1445  
  1446  // Recv mocks base method.
  1447  func (m *MockProfilerCommandService_HandleCommandClient) Recv() (*v1.PCmdRequest, error) {
  1448  	m.ctrl.T.Helper()
  1449  	ret := m.ctrl.Call(m, "Recv")
  1450  	ret0, _ := ret[0].(*v1.PCmdRequest)
  1451  	ret1, _ := ret[1].(error)
  1452  	return ret0, ret1
  1453  }
  1454  
  1455  // Recv indicates an expected call of Recv.
  1456  func (mr *MockProfilerCommandService_HandleCommandClientMockRecorder) Recv() *gomock.Call {
  1457  	mr.mock.ctrl.T.Helper()
  1458  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Recv", reflect.TypeOf((*MockProfilerCommandService_HandleCommandClient)(nil).Recv))
  1459  }
  1460  
  1461  // RecvMsg mocks base method.
  1462  func (m_2 *MockProfilerCommandService_HandleCommandClient) RecvMsg(m interface{}) error {
  1463  	m_2.ctrl.T.Helper()
  1464  	ret := m_2.ctrl.Call(m_2, "RecvMsg", m)
  1465  	ret0, _ := ret[0].(error)
  1466  	return ret0
  1467  }
  1468  
  1469  // RecvMsg indicates an expected call of RecvMsg.
  1470  func (mr *MockProfilerCommandService_HandleCommandClientMockRecorder) RecvMsg(m interface{}) *gomock.Call {
  1471  	mr.mock.ctrl.T.Helper()
  1472  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockProfilerCommandService_HandleCommandClient)(nil).RecvMsg), m)
  1473  }
  1474  
  1475  // Send mocks base method.
  1476  func (m *MockProfilerCommandService_HandleCommandClient) Send(arg0 *v1.PCmdMessage) error {
  1477  	m.ctrl.T.Helper()
  1478  	ret := m.ctrl.Call(m, "Send", arg0)
  1479  	ret0, _ := ret[0].(error)
  1480  	return ret0
  1481  }
  1482  
  1483  // Send indicates an expected call of Send.
  1484  func (mr *MockProfilerCommandService_HandleCommandClientMockRecorder) Send(arg0 interface{}) *gomock.Call {
  1485  	mr.mock.ctrl.T.Helper()
  1486  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockProfilerCommandService_HandleCommandClient)(nil).Send), arg0)
  1487  }
  1488  
  1489  // SendMsg mocks base method.
  1490  func (m_2 *MockProfilerCommandService_HandleCommandClient) SendMsg(m interface{}) error {
  1491  	m_2.ctrl.T.Helper()
  1492  	ret := m_2.ctrl.Call(m_2, "SendMsg", m)
  1493  	ret0, _ := ret[0].(error)
  1494  	return ret0
  1495  }
  1496  
  1497  // SendMsg indicates an expected call of SendMsg.
  1498  func (mr *MockProfilerCommandService_HandleCommandClientMockRecorder) SendMsg(m interface{}) *gomock.Call {
  1499  	mr.mock.ctrl.T.Helper()
  1500  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockProfilerCommandService_HandleCommandClient)(nil).SendMsg), m)
  1501  }
  1502  
  1503  // Trailer mocks base method.
  1504  func (m *MockProfilerCommandService_HandleCommandClient) Trailer() metadata.MD {
  1505  	m.ctrl.T.Helper()
  1506  	ret := m.ctrl.Call(m, "Trailer")
  1507  	ret0, _ := ret[0].(metadata.MD)
  1508  	return ret0
  1509  }
  1510  
  1511  // Trailer indicates an expected call of Trailer.
  1512  func (mr *MockProfilerCommandService_HandleCommandClientMockRecorder) Trailer() *gomock.Call {
  1513  	mr.mock.ctrl.T.Helper()
  1514  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Trailer", reflect.TypeOf((*MockProfilerCommandService_HandleCommandClient)(nil).Trailer))
  1515  }
  1516  
  1517  // MockProfilerCommandService_CommandStreamActiveThreadCountClient is a mock of ProfilerCommandService_CommandStreamActiveThreadCountClient interface.
  1518  type MockProfilerCommandService_CommandStreamActiveThreadCountClient struct {
  1519  	ctrl     *gomock.Controller
  1520  	recorder *MockProfilerCommandService_CommandStreamActiveThreadCountClientMockRecorder
  1521  }
  1522  
  1523  // MockProfilerCommandService_CommandStreamActiveThreadCountClientMockRecorder is the mock recorder for MockProfilerCommandService_CommandStreamActiveThreadCountClient.
  1524  type MockProfilerCommandService_CommandStreamActiveThreadCountClientMockRecorder struct {
  1525  	mock *MockProfilerCommandService_CommandStreamActiveThreadCountClient
  1526  }
  1527  
  1528  // NewMockProfilerCommandService_CommandStreamActiveThreadCountClient creates a new mock instance.
  1529  func NewMockProfilerCommandService_CommandStreamActiveThreadCountClient(ctrl *gomock.Controller) *MockProfilerCommandService_CommandStreamActiveThreadCountClient {
  1530  	mock := &MockProfilerCommandService_CommandStreamActiveThreadCountClient{ctrl: ctrl}
  1531  	mock.recorder = &MockProfilerCommandService_CommandStreamActiveThreadCountClientMockRecorder{mock}
  1532  	return mock
  1533  }
  1534  
  1535  // EXPECT returns an object that allows the caller to indicate expected use.
  1536  func (m *MockProfilerCommandService_CommandStreamActiveThreadCountClient) EXPECT() *MockProfilerCommandService_CommandStreamActiveThreadCountClientMockRecorder {
  1537  	return m.recorder
  1538  }
  1539  
  1540  // CloseAndRecv mocks base method.
  1541  func (m *MockProfilerCommandService_CommandStreamActiveThreadCountClient) CloseAndRecv() (*empty.Empty, error) {
  1542  	m.ctrl.T.Helper()
  1543  	ret := m.ctrl.Call(m, "CloseAndRecv")
  1544  	ret0, _ := ret[0].(*empty.Empty)
  1545  	ret1, _ := ret[1].(error)
  1546  	return ret0, ret1
  1547  }
  1548  
  1549  // CloseAndRecv indicates an expected call of CloseAndRecv.
  1550  func (mr *MockProfilerCommandService_CommandStreamActiveThreadCountClientMockRecorder) CloseAndRecv() *gomock.Call {
  1551  	mr.mock.ctrl.T.Helper()
  1552  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseAndRecv", reflect.TypeOf((*MockProfilerCommandService_CommandStreamActiveThreadCountClient)(nil).CloseAndRecv))
  1553  }
  1554  
  1555  // CloseSend mocks base method.
  1556  func (m *MockProfilerCommandService_CommandStreamActiveThreadCountClient) CloseSend() error {
  1557  	m.ctrl.T.Helper()
  1558  	ret := m.ctrl.Call(m, "CloseSend")
  1559  	ret0, _ := ret[0].(error)
  1560  	return ret0
  1561  }
  1562  
  1563  // CloseSend indicates an expected call of CloseSend.
  1564  func (mr *MockProfilerCommandService_CommandStreamActiveThreadCountClientMockRecorder) CloseSend() *gomock.Call {
  1565  	mr.mock.ctrl.T.Helper()
  1566  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseSend", reflect.TypeOf((*MockProfilerCommandService_CommandStreamActiveThreadCountClient)(nil).CloseSend))
  1567  }
  1568  
  1569  // Context mocks base method.
  1570  func (m *MockProfilerCommandService_CommandStreamActiveThreadCountClient) Context() context.Context {
  1571  	m.ctrl.T.Helper()
  1572  	ret := m.ctrl.Call(m, "Context")
  1573  	ret0, _ := ret[0].(context.Context)
  1574  	return ret0
  1575  }
  1576  
  1577  // Context indicates an expected call of Context.
  1578  func (mr *MockProfilerCommandService_CommandStreamActiveThreadCountClientMockRecorder) Context() *gomock.Call {
  1579  	mr.mock.ctrl.T.Helper()
  1580  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockProfilerCommandService_CommandStreamActiveThreadCountClient)(nil).Context))
  1581  }
  1582  
  1583  // Header mocks base method.
  1584  func (m *MockProfilerCommandService_CommandStreamActiveThreadCountClient) Header() (metadata.MD, error) {
  1585  	m.ctrl.T.Helper()
  1586  	ret := m.ctrl.Call(m, "Header")
  1587  	ret0, _ := ret[0].(metadata.MD)
  1588  	ret1, _ := ret[1].(error)
  1589  	return ret0, ret1
  1590  }
  1591  
  1592  // Header indicates an expected call of Header.
  1593  func (mr *MockProfilerCommandService_CommandStreamActiveThreadCountClientMockRecorder) Header() *gomock.Call {
  1594  	mr.mock.ctrl.T.Helper()
  1595  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Header", reflect.TypeOf((*MockProfilerCommandService_CommandStreamActiveThreadCountClient)(nil).Header))
  1596  }
  1597  
  1598  // RecvMsg mocks base method.
  1599  func (m_2 *MockProfilerCommandService_CommandStreamActiveThreadCountClient) RecvMsg(m interface{}) error {
  1600  	m_2.ctrl.T.Helper()
  1601  	ret := m_2.ctrl.Call(m_2, "RecvMsg", m)
  1602  	ret0, _ := ret[0].(error)
  1603  	return ret0
  1604  }
  1605  
  1606  // RecvMsg indicates an expected call of RecvMsg.
  1607  func (mr *MockProfilerCommandService_CommandStreamActiveThreadCountClientMockRecorder) RecvMsg(m interface{}) *gomock.Call {
  1608  	mr.mock.ctrl.T.Helper()
  1609  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockProfilerCommandService_CommandStreamActiveThreadCountClient)(nil).RecvMsg), m)
  1610  }
  1611  
  1612  // Send mocks base method.
  1613  func (m *MockProfilerCommandService_CommandStreamActiveThreadCountClient) Send(arg0 *v1.PCmdActiveThreadCountRes) error {
  1614  	m.ctrl.T.Helper()
  1615  	ret := m.ctrl.Call(m, "Send", arg0)
  1616  	ret0, _ := ret[0].(error)
  1617  	return ret0
  1618  }
  1619  
  1620  // Send indicates an expected call of Send.
  1621  func (mr *MockProfilerCommandService_CommandStreamActiveThreadCountClientMockRecorder) Send(arg0 interface{}) *gomock.Call {
  1622  	mr.mock.ctrl.T.Helper()
  1623  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockProfilerCommandService_CommandStreamActiveThreadCountClient)(nil).Send), arg0)
  1624  }
  1625  
  1626  // SendMsg mocks base method.
  1627  func (m_2 *MockProfilerCommandService_CommandStreamActiveThreadCountClient) SendMsg(m interface{}) error {
  1628  	m_2.ctrl.T.Helper()
  1629  	ret := m_2.ctrl.Call(m_2, "SendMsg", m)
  1630  	ret0, _ := ret[0].(error)
  1631  	return ret0
  1632  }
  1633  
  1634  // SendMsg indicates an expected call of SendMsg.
  1635  func (mr *MockProfilerCommandService_CommandStreamActiveThreadCountClientMockRecorder) SendMsg(m interface{}) *gomock.Call {
  1636  	mr.mock.ctrl.T.Helper()
  1637  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockProfilerCommandService_CommandStreamActiveThreadCountClient)(nil).SendMsg), m)
  1638  }
  1639  
  1640  // Trailer mocks base method.
  1641  func (m *MockProfilerCommandService_CommandStreamActiveThreadCountClient) Trailer() metadata.MD {
  1642  	m.ctrl.T.Helper()
  1643  	ret := m.ctrl.Call(m, "Trailer")
  1644  	ret0, _ := ret[0].(metadata.MD)
  1645  	return ret0
  1646  }
  1647  
  1648  // Trailer indicates an expected call of Trailer.
  1649  func (mr *MockProfilerCommandService_CommandStreamActiveThreadCountClientMockRecorder) Trailer() *gomock.Call {
  1650  	mr.mock.ctrl.T.Helper()
  1651  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Trailer", reflect.TypeOf((*MockProfilerCommandService_CommandStreamActiveThreadCountClient)(nil).Trailer))
  1652  }
  1653  
  1654  // MockProfilerCommandServiceServer is a mock of ProfilerCommandServiceServer interface.
  1655  type MockProfilerCommandServiceServer struct {
  1656  	ctrl     *gomock.Controller
  1657  	recorder *MockProfilerCommandServiceServerMockRecorder
  1658  }
  1659  
  1660  // MockProfilerCommandServiceServerMockRecorder is the mock recorder for MockProfilerCommandServiceServer.
  1661  type MockProfilerCommandServiceServerMockRecorder struct {
  1662  	mock *MockProfilerCommandServiceServer
  1663  }
  1664  
  1665  // NewMockProfilerCommandServiceServer creates a new mock instance.
  1666  func NewMockProfilerCommandServiceServer(ctrl *gomock.Controller) *MockProfilerCommandServiceServer {
  1667  	mock := &MockProfilerCommandServiceServer{ctrl: ctrl}
  1668  	mock.recorder = &MockProfilerCommandServiceServerMockRecorder{mock}
  1669  	return mock
  1670  }
  1671  
  1672  // EXPECT returns an object that allows the caller to indicate expected use.
  1673  func (m *MockProfilerCommandServiceServer) EXPECT() *MockProfilerCommandServiceServerMockRecorder {
  1674  	return m.recorder
  1675  }
  1676  
  1677  // CommandActiveThreadDump mocks base method.
  1678  func (m *MockProfilerCommandServiceServer) CommandActiveThreadDump(arg0 context.Context, arg1 *v1.PCmdActiveThreadDumpRes) (*empty.Empty, error) {
  1679  	m.ctrl.T.Helper()
  1680  	ret := m.ctrl.Call(m, "CommandActiveThreadDump", arg0, arg1)
  1681  	ret0, _ := ret[0].(*empty.Empty)
  1682  	ret1, _ := ret[1].(error)
  1683  	return ret0, ret1
  1684  }
  1685  
  1686  // CommandActiveThreadDump indicates an expected call of CommandActiveThreadDump.
  1687  func (mr *MockProfilerCommandServiceServerMockRecorder) CommandActiveThreadDump(arg0, arg1 interface{}) *gomock.Call {
  1688  	mr.mock.ctrl.T.Helper()
  1689  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CommandActiveThreadDump", reflect.TypeOf((*MockProfilerCommandServiceServer)(nil).CommandActiveThreadDump), arg0, arg1)
  1690  }
  1691  
  1692  // CommandActiveThreadLightDump mocks base method.
  1693  func (m *MockProfilerCommandServiceServer) CommandActiveThreadLightDump(arg0 context.Context, arg1 *v1.PCmdActiveThreadLightDumpRes) (*empty.Empty, error) {
  1694  	m.ctrl.T.Helper()
  1695  	ret := m.ctrl.Call(m, "CommandActiveThreadLightDump", arg0, arg1)
  1696  	ret0, _ := ret[0].(*empty.Empty)
  1697  	ret1, _ := ret[1].(error)
  1698  	return ret0, ret1
  1699  }
  1700  
  1701  // CommandActiveThreadLightDump indicates an expected call of CommandActiveThreadLightDump.
  1702  func (mr *MockProfilerCommandServiceServerMockRecorder) CommandActiveThreadLightDump(arg0, arg1 interface{}) *gomock.Call {
  1703  	mr.mock.ctrl.T.Helper()
  1704  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CommandActiveThreadLightDump", reflect.TypeOf((*MockProfilerCommandServiceServer)(nil).CommandActiveThreadLightDump), arg0, arg1)
  1705  }
  1706  
  1707  // CommandEcho mocks base method.
  1708  func (m *MockProfilerCommandServiceServer) CommandEcho(arg0 context.Context, arg1 *v1.PCmdEchoResponse) (*empty.Empty, error) {
  1709  	m.ctrl.T.Helper()
  1710  	ret := m.ctrl.Call(m, "CommandEcho", arg0, arg1)
  1711  	ret0, _ := ret[0].(*empty.Empty)
  1712  	ret1, _ := ret[1].(error)
  1713  	return ret0, ret1
  1714  }
  1715  
  1716  // CommandEcho indicates an expected call of CommandEcho.
  1717  func (mr *MockProfilerCommandServiceServerMockRecorder) CommandEcho(arg0, arg1 interface{}) *gomock.Call {
  1718  	mr.mock.ctrl.T.Helper()
  1719  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CommandEcho", reflect.TypeOf((*MockProfilerCommandServiceServer)(nil).CommandEcho), arg0, arg1)
  1720  }
  1721  
  1722  // CommandStreamActiveThreadCount mocks base method.
  1723  func (m *MockProfilerCommandServiceServer) CommandStreamActiveThreadCount(arg0 v1.ProfilerCommandService_CommandStreamActiveThreadCountServer) error {
  1724  	m.ctrl.T.Helper()
  1725  	ret := m.ctrl.Call(m, "CommandStreamActiveThreadCount", arg0)
  1726  	ret0, _ := ret[0].(error)
  1727  	return ret0
  1728  }
  1729  
  1730  // CommandStreamActiveThreadCount indicates an expected call of CommandStreamActiveThreadCount.
  1731  func (mr *MockProfilerCommandServiceServerMockRecorder) CommandStreamActiveThreadCount(arg0 interface{}) *gomock.Call {
  1732  	mr.mock.ctrl.T.Helper()
  1733  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CommandStreamActiveThreadCount", reflect.TypeOf((*MockProfilerCommandServiceServer)(nil).CommandStreamActiveThreadCount), arg0)
  1734  }
  1735  
  1736  // HandleCommand mocks base method.
  1737  func (m *MockProfilerCommandServiceServer) HandleCommand(arg0 v1.ProfilerCommandService_HandleCommandServer) error {
  1738  	m.ctrl.T.Helper()
  1739  	ret := m.ctrl.Call(m, "HandleCommand", arg0)
  1740  	ret0, _ := ret[0].(error)
  1741  	return ret0
  1742  }
  1743  
  1744  // HandleCommand indicates an expected call of HandleCommand.
  1745  func (mr *MockProfilerCommandServiceServerMockRecorder) HandleCommand(arg0 interface{}) *gomock.Call {
  1746  	mr.mock.ctrl.T.Helper()
  1747  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HandleCommand", reflect.TypeOf((*MockProfilerCommandServiceServer)(nil).HandleCommand), arg0)
  1748  }
  1749  
  1750  // MockProfilerCommandService_HandleCommandServer is a mock of ProfilerCommandService_HandleCommandServer interface.
  1751  type MockProfilerCommandService_HandleCommandServer struct {
  1752  	ctrl     *gomock.Controller
  1753  	recorder *MockProfilerCommandService_HandleCommandServerMockRecorder
  1754  }
  1755  
  1756  // MockProfilerCommandService_HandleCommandServerMockRecorder is the mock recorder for MockProfilerCommandService_HandleCommandServer.
  1757  type MockProfilerCommandService_HandleCommandServerMockRecorder struct {
  1758  	mock *MockProfilerCommandService_HandleCommandServer
  1759  }
  1760  
  1761  // NewMockProfilerCommandService_HandleCommandServer creates a new mock instance.
  1762  func NewMockProfilerCommandService_HandleCommandServer(ctrl *gomock.Controller) *MockProfilerCommandService_HandleCommandServer {
  1763  	mock := &MockProfilerCommandService_HandleCommandServer{ctrl: ctrl}
  1764  	mock.recorder = &MockProfilerCommandService_HandleCommandServerMockRecorder{mock}
  1765  	return mock
  1766  }
  1767  
  1768  // EXPECT returns an object that allows the caller to indicate expected use.
  1769  func (m *MockProfilerCommandService_HandleCommandServer) EXPECT() *MockProfilerCommandService_HandleCommandServerMockRecorder {
  1770  	return m.recorder
  1771  }
  1772  
  1773  // Context mocks base method.
  1774  func (m *MockProfilerCommandService_HandleCommandServer) Context() context.Context {
  1775  	m.ctrl.T.Helper()
  1776  	ret := m.ctrl.Call(m, "Context")
  1777  	ret0, _ := ret[0].(context.Context)
  1778  	return ret0
  1779  }
  1780  
  1781  // Context indicates an expected call of Context.
  1782  func (mr *MockProfilerCommandService_HandleCommandServerMockRecorder) Context() *gomock.Call {
  1783  	mr.mock.ctrl.T.Helper()
  1784  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockProfilerCommandService_HandleCommandServer)(nil).Context))
  1785  }
  1786  
  1787  // Recv mocks base method.
  1788  func (m *MockProfilerCommandService_HandleCommandServer) Recv() (*v1.PCmdMessage, error) {
  1789  	m.ctrl.T.Helper()
  1790  	ret := m.ctrl.Call(m, "Recv")
  1791  	ret0, _ := ret[0].(*v1.PCmdMessage)
  1792  	ret1, _ := ret[1].(error)
  1793  	return ret0, ret1
  1794  }
  1795  
  1796  // Recv indicates an expected call of Recv.
  1797  func (mr *MockProfilerCommandService_HandleCommandServerMockRecorder) Recv() *gomock.Call {
  1798  	mr.mock.ctrl.T.Helper()
  1799  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Recv", reflect.TypeOf((*MockProfilerCommandService_HandleCommandServer)(nil).Recv))
  1800  }
  1801  
  1802  // RecvMsg mocks base method.
  1803  func (m_2 *MockProfilerCommandService_HandleCommandServer) RecvMsg(m interface{}) error {
  1804  	m_2.ctrl.T.Helper()
  1805  	ret := m_2.ctrl.Call(m_2, "RecvMsg", m)
  1806  	ret0, _ := ret[0].(error)
  1807  	return ret0
  1808  }
  1809  
  1810  // RecvMsg indicates an expected call of RecvMsg.
  1811  func (mr *MockProfilerCommandService_HandleCommandServerMockRecorder) RecvMsg(m interface{}) *gomock.Call {
  1812  	mr.mock.ctrl.T.Helper()
  1813  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockProfilerCommandService_HandleCommandServer)(nil).RecvMsg), m)
  1814  }
  1815  
  1816  // Send mocks base method.
  1817  func (m *MockProfilerCommandService_HandleCommandServer) Send(arg0 *v1.PCmdRequest) error {
  1818  	m.ctrl.T.Helper()
  1819  	ret := m.ctrl.Call(m, "Send", arg0)
  1820  	ret0, _ := ret[0].(error)
  1821  	return ret0
  1822  }
  1823  
  1824  // Send indicates an expected call of Send.
  1825  func (mr *MockProfilerCommandService_HandleCommandServerMockRecorder) Send(arg0 interface{}) *gomock.Call {
  1826  	mr.mock.ctrl.T.Helper()
  1827  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockProfilerCommandService_HandleCommandServer)(nil).Send), arg0)
  1828  }
  1829  
  1830  // SendHeader mocks base method.
  1831  func (m *MockProfilerCommandService_HandleCommandServer) SendHeader(arg0 metadata.MD) error {
  1832  	m.ctrl.T.Helper()
  1833  	ret := m.ctrl.Call(m, "SendHeader", arg0)
  1834  	ret0, _ := ret[0].(error)
  1835  	return ret0
  1836  }
  1837  
  1838  // SendHeader indicates an expected call of SendHeader.
  1839  func (mr *MockProfilerCommandService_HandleCommandServerMockRecorder) SendHeader(arg0 interface{}) *gomock.Call {
  1840  	mr.mock.ctrl.T.Helper()
  1841  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendHeader", reflect.TypeOf((*MockProfilerCommandService_HandleCommandServer)(nil).SendHeader), arg0)
  1842  }
  1843  
  1844  // SendMsg mocks base method.
  1845  func (m_2 *MockProfilerCommandService_HandleCommandServer) SendMsg(m interface{}) error {
  1846  	m_2.ctrl.T.Helper()
  1847  	ret := m_2.ctrl.Call(m_2, "SendMsg", m)
  1848  	ret0, _ := ret[0].(error)
  1849  	return ret0
  1850  }
  1851  
  1852  // SendMsg indicates an expected call of SendMsg.
  1853  func (mr *MockProfilerCommandService_HandleCommandServerMockRecorder) SendMsg(m interface{}) *gomock.Call {
  1854  	mr.mock.ctrl.T.Helper()
  1855  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockProfilerCommandService_HandleCommandServer)(nil).SendMsg), m)
  1856  }
  1857  
  1858  // SetHeader mocks base method.
  1859  func (m *MockProfilerCommandService_HandleCommandServer) SetHeader(arg0 metadata.MD) error {
  1860  	m.ctrl.T.Helper()
  1861  	ret := m.ctrl.Call(m, "SetHeader", arg0)
  1862  	ret0, _ := ret[0].(error)
  1863  	return ret0
  1864  }
  1865  
  1866  // SetHeader indicates an expected call of SetHeader.
  1867  func (mr *MockProfilerCommandService_HandleCommandServerMockRecorder) SetHeader(arg0 interface{}) *gomock.Call {
  1868  	mr.mock.ctrl.T.Helper()
  1869  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetHeader", reflect.TypeOf((*MockProfilerCommandService_HandleCommandServer)(nil).SetHeader), arg0)
  1870  }
  1871  
  1872  // SetTrailer mocks base method.
  1873  func (m *MockProfilerCommandService_HandleCommandServer) SetTrailer(arg0 metadata.MD) {
  1874  	m.ctrl.T.Helper()
  1875  	m.ctrl.Call(m, "SetTrailer", arg0)
  1876  }
  1877  
  1878  // SetTrailer indicates an expected call of SetTrailer.
  1879  func (mr *MockProfilerCommandService_HandleCommandServerMockRecorder) SetTrailer(arg0 interface{}) *gomock.Call {
  1880  	mr.mock.ctrl.T.Helper()
  1881  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTrailer", reflect.TypeOf((*MockProfilerCommandService_HandleCommandServer)(nil).SetTrailer), arg0)
  1882  }
  1883  
  1884  // MockProfilerCommandService_CommandStreamActiveThreadCountServer is a mock of ProfilerCommandService_CommandStreamActiveThreadCountServer interface.
  1885  type MockProfilerCommandService_CommandStreamActiveThreadCountServer struct {
  1886  	ctrl     *gomock.Controller
  1887  	recorder *MockProfilerCommandService_CommandStreamActiveThreadCountServerMockRecorder
  1888  }
  1889  
  1890  // MockProfilerCommandService_CommandStreamActiveThreadCountServerMockRecorder is the mock recorder for MockProfilerCommandService_CommandStreamActiveThreadCountServer.
  1891  type MockProfilerCommandService_CommandStreamActiveThreadCountServerMockRecorder struct {
  1892  	mock *MockProfilerCommandService_CommandStreamActiveThreadCountServer
  1893  }
  1894  
  1895  // NewMockProfilerCommandService_CommandStreamActiveThreadCountServer creates a new mock instance.
  1896  func NewMockProfilerCommandService_CommandStreamActiveThreadCountServer(ctrl *gomock.Controller) *MockProfilerCommandService_CommandStreamActiveThreadCountServer {
  1897  	mock := &MockProfilerCommandService_CommandStreamActiveThreadCountServer{ctrl: ctrl}
  1898  	mock.recorder = &MockProfilerCommandService_CommandStreamActiveThreadCountServerMockRecorder{mock}
  1899  	return mock
  1900  }
  1901  
  1902  // EXPECT returns an object that allows the caller to indicate expected use.
  1903  func (m *MockProfilerCommandService_CommandStreamActiveThreadCountServer) EXPECT() *MockProfilerCommandService_CommandStreamActiveThreadCountServerMockRecorder {
  1904  	return m.recorder
  1905  }
  1906  
  1907  // Context mocks base method.
  1908  func (m *MockProfilerCommandService_CommandStreamActiveThreadCountServer) Context() context.Context {
  1909  	m.ctrl.T.Helper()
  1910  	ret := m.ctrl.Call(m, "Context")
  1911  	ret0, _ := ret[0].(context.Context)
  1912  	return ret0
  1913  }
  1914  
  1915  // Context indicates an expected call of Context.
  1916  func (mr *MockProfilerCommandService_CommandStreamActiveThreadCountServerMockRecorder) Context() *gomock.Call {
  1917  	mr.mock.ctrl.T.Helper()
  1918  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockProfilerCommandService_CommandStreamActiveThreadCountServer)(nil).Context))
  1919  }
  1920  
  1921  // Recv mocks base method.
  1922  func (m *MockProfilerCommandService_CommandStreamActiveThreadCountServer) Recv() (*v1.PCmdActiveThreadCountRes, error) {
  1923  	m.ctrl.T.Helper()
  1924  	ret := m.ctrl.Call(m, "Recv")
  1925  	ret0, _ := ret[0].(*v1.PCmdActiveThreadCountRes)
  1926  	ret1, _ := ret[1].(error)
  1927  	return ret0, ret1
  1928  }
  1929  
  1930  // Recv indicates an expected call of Recv.
  1931  func (mr *MockProfilerCommandService_CommandStreamActiveThreadCountServerMockRecorder) Recv() *gomock.Call {
  1932  	mr.mock.ctrl.T.Helper()
  1933  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Recv", reflect.TypeOf((*MockProfilerCommandService_CommandStreamActiveThreadCountServer)(nil).Recv))
  1934  }
  1935  
  1936  // RecvMsg mocks base method.
  1937  func (m_2 *MockProfilerCommandService_CommandStreamActiveThreadCountServer) RecvMsg(m interface{}) error {
  1938  	m_2.ctrl.T.Helper()
  1939  	ret := m_2.ctrl.Call(m_2, "RecvMsg", m)
  1940  	ret0, _ := ret[0].(error)
  1941  	return ret0
  1942  }
  1943  
  1944  // RecvMsg indicates an expected call of RecvMsg.
  1945  func (mr *MockProfilerCommandService_CommandStreamActiveThreadCountServerMockRecorder) RecvMsg(m interface{}) *gomock.Call {
  1946  	mr.mock.ctrl.T.Helper()
  1947  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockProfilerCommandService_CommandStreamActiveThreadCountServer)(nil).RecvMsg), m)
  1948  }
  1949  
  1950  // SendAndClose mocks base method.
  1951  func (m *MockProfilerCommandService_CommandStreamActiveThreadCountServer) SendAndClose(arg0 *empty.Empty) error {
  1952  	m.ctrl.T.Helper()
  1953  	ret := m.ctrl.Call(m, "SendAndClose", arg0)
  1954  	ret0, _ := ret[0].(error)
  1955  	return ret0
  1956  }
  1957  
  1958  // SendAndClose indicates an expected call of SendAndClose.
  1959  func (mr *MockProfilerCommandService_CommandStreamActiveThreadCountServerMockRecorder) SendAndClose(arg0 interface{}) *gomock.Call {
  1960  	mr.mock.ctrl.T.Helper()
  1961  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendAndClose", reflect.TypeOf((*MockProfilerCommandService_CommandStreamActiveThreadCountServer)(nil).SendAndClose), arg0)
  1962  }
  1963  
  1964  // SendHeader mocks base method.
  1965  func (m *MockProfilerCommandService_CommandStreamActiveThreadCountServer) SendHeader(arg0 metadata.MD) error {
  1966  	m.ctrl.T.Helper()
  1967  	ret := m.ctrl.Call(m, "SendHeader", arg0)
  1968  	ret0, _ := ret[0].(error)
  1969  	return ret0
  1970  }
  1971  
  1972  // SendHeader indicates an expected call of SendHeader.
  1973  func (mr *MockProfilerCommandService_CommandStreamActiveThreadCountServerMockRecorder) SendHeader(arg0 interface{}) *gomock.Call {
  1974  	mr.mock.ctrl.T.Helper()
  1975  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendHeader", reflect.TypeOf((*MockProfilerCommandService_CommandStreamActiveThreadCountServer)(nil).SendHeader), arg0)
  1976  }
  1977  
  1978  // SendMsg mocks base method.
  1979  func (m_2 *MockProfilerCommandService_CommandStreamActiveThreadCountServer) SendMsg(m interface{}) error {
  1980  	m_2.ctrl.T.Helper()
  1981  	ret := m_2.ctrl.Call(m_2, "SendMsg", m)
  1982  	ret0, _ := ret[0].(error)
  1983  	return ret0
  1984  }
  1985  
  1986  // SendMsg indicates an expected call of SendMsg.
  1987  func (mr *MockProfilerCommandService_CommandStreamActiveThreadCountServerMockRecorder) SendMsg(m interface{}) *gomock.Call {
  1988  	mr.mock.ctrl.T.Helper()
  1989  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockProfilerCommandService_CommandStreamActiveThreadCountServer)(nil).SendMsg), m)
  1990  }
  1991  
  1992  // SetHeader mocks base method.
  1993  func (m *MockProfilerCommandService_CommandStreamActiveThreadCountServer) SetHeader(arg0 metadata.MD) error {
  1994  	m.ctrl.T.Helper()
  1995  	ret := m.ctrl.Call(m, "SetHeader", arg0)
  1996  	ret0, _ := ret[0].(error)
  1997  	return ret0
  1998  }
  1999  
  2000  // SetHeader indicates an expected call of SetHeader.
  2001  func (mr *MockProfilerCommandService_CommandStreamActiveThreadCountServerMockRecorder) SetHeader(arg0 interface{}) *gomock.Call {
  2002  	mr.mock.ctrl.T.Helper()
  2003  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetHeader", reflect.TypeOf((*MockProfilerCommandService_CommandStreamActiveThreadCountServer)(nil).SetHeader), arg0)
  2004  }
  2005  
  2006  // SetTrailer mocks base method.
  2007  func (m *MockProfilerCommandService_CommandStreamActiveThreadCountServer) SetTrailer(arg0 metadata.MD) {
  2008  	m.ctrl.T.Helper()
  2009  	m.ctrl.Call(m, "SetTrailer", arg0)
  2010  }
  2011  
  2012  // SetTrailer indicates an expected call of SetTrailer.
  2013  func (mr *MockProfilerCommandService_CommandStreamActiveThreadCountServerMockRecorder) SetTrailer(arg0 interface{}) *gomock.Call {
  2014  	mr.mock.ctrl.T.Helper()
  2015  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTrailer", reflect.TypeOf((*MockProfilerCommandService_CommandStreamActiveThreadCountServer)(nil).SetTrailer), arg0)
  2016  }