From 7fda7f0b94cdaedbf445ed580868b912d1805e55 Mon Sep 17 00:00:00 2001 From: Yashvi Jain <91940132+YashviJain01@users.noreply.github.com> Date: Wed, 26 Jan 2022 20:50:33 +0530 Subject: [PATCH] Migration Data service creation and created a rpc method for upload zip file (#6612) * Changes for stream api for upload Signed-off-by: Yashvi Jain * changes Signed-off-by: Yashvi Jain * added some changes for the protofile Signed-off-by: Vinay Sharma * added function changes * added changes for the proto * added changes to devide the chunks in bytes * added changes for the validation Signed-off-by: Vinay Sharma * adding status in database Signed-off-by: Yashvi Jain * adding the request for server id and filename Signed-off-by: Yashvi Jain * Adding loggers and saving the failed migration phase Signed-off-by: Yashvi Jain * optimising the code Signed-off-by: Yashvi Jain * Changes for content-type Signed-off-by: Yashvi Jain * added changes for the validation Signed-off-by: Vinay Sharma * Upload to the hab svc data folder Signed-off-by: Kallol Roy * Enhance log and modify migration phases Signed-off-by: Kallol Roy * Granularize the migration stages Signed-off-by: Kallol Roy * Granularize the migration stages Signed-off-by: Kallol Roy Co-authored-by: Vinay Sharma Co-authored-by: Kallol Roy --- .../infra_proxy/infra_proxy.pb.client_mock.go | 1709 ++++++++--------- .../migrations/request/migrations.pb.go | 287 ++- .../migrations/request/migrations.proto | 24 + .../migrations/response/migrations.pb.go | 120 +- .../migrations/response/migrations.proto | 10 + .../secrets/secrets.pb.client_mock.go | 139 +- .../authn/authenticate.pb.client_mock.go | 27 +- .../authz/authz.pb.client_mock.go | 115 +- .../authz/project.pb.client_mock.go | 447 +++-- .../cfgmgmt/service/cfgmgmt.pb.client_mock.go | 387 ++-- .../ingest/compliance.pb.client_mock.go | 83 +- .../event/event.pb.client_mock.go | 107 +- .../event_feed/event_feed.pb.client_mock.go | 131 +- .../migrations/request/migrations.pb.go | 287 ++- .../migrations/request/migrations.proto | 24 + .../migrations/response/migrations.pb.go | 120 +- .../migrations/response/migrations.proto | 9 + .../migrations/service/migrations.pb.go | 166 +- .../migrations/service/migrations.proto | 10 + .../ingest/chef.pb.client_mock.go | 149 +- .../manager/manager.pb.client_mock.go | 423 ++-- .../nodemanager/nodes/nodes.pb.client_mock.go | 271 ++- .../api/migrations.pb.swagger.go | 54 - .../automate-gateway/gateway/clients.go | 10 + components/automate-gateway/gateway/server.go | 3 + .../automate-gateway/gateway/services.go | 119 +- .../mock_gateway/clients_mock.go | 22 +- .../handler/infra_proxy/infra_proxy.go | 9 +- .../constants/constants.go | 32 +- .../infra-proxy-service/server/migrations.go | 133 ++ .../infra-proxy-service/server/server.go | 2 + .../migration/sql/07_migration.up.sql | 32 +- .../storage/postgres/migrations.go | 143 +- .../infra-proxy-service/storage/storage.go | 27 +- ec2/Vagrantfile | 8 +- 35 files changed, 3435 insertions(+), 2204 deletions(-) delete mode 100644 components/automate-gateway/api/migrations.pb.swagger.go create mode 100644 components/infra-proxy-service/server/migrations.go diff --git a/api/external/infra_proxy/infra_proxy.pb.client_mock.go b/api/external/infra_proxy/infra_proxy.pb.client_mock.go index 2087a233820..0e5c28472c6 100644 --- a/api/external/infra_proxy/infra_proxy.pb.client_mock.go +++ b/api/external/infra_proxy/infra_proxy.pb.client_mock.go @@ -6,278 +6,277 @@ package infra_proxy import ( context "context" - reflect "reflect" - request "github.com/chef/automate/api/external/infra_proxy/request" response "github.com/chef/automate/api/external/infra_proxy/response" gomock "github.com/golang/mock/gomock" grpc "google.golang.org/grpc" + reflect "reflect" ) -// MockInfraProxyClient is a mock of InfraProxyClient interface. +// MockInfraProxyClient is a mock of InfraProxyClient interface type MockInfraProxyClient struct { ctrl *gomock.Controller recorder *MockInfraProxyClientMockRecorder } -// MockInfraProxyClientMockRecorder is the mock recorder for MockInfraProxyClient. +// MockInfraProxyClientMockRecorder is the mock recorder for MockInfraProxyClient type MockInfraProxyClientMockRecorder struct { mock *MockInfraProxyClient } -// NewMockInfraProxyClient creates a new mock instance. +// NewMockInfraProxyClient creates a new mock instance func NewMockInfraProxyClient(ctrl *gomock.Controller) *MockInfraProxyClient { mock := &MockInfraProxyClient{ctrl: ctrl} mock.recorder = &MockInfraProxyClientMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockInfraProxyClient) EXPECT() *MockInfraProxyClientMockRecorder { return m.recorder } -// CreateClient mocks base method. -func (m *MockInfraProxyClient) CreateClient(ctx context.Context, in *request.CreateClient, opts ...grpc.CallOption) (*response.CreateClient, error) { +// GetServers mocks base method +func (m *MockInfraProxyClient) GetServers(ctx context.Context, in *request.GetServers, opts ...grpc.CallOption) (*response.GetServers, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "CreateClient", varargs...) - ret0, _ := ret[0].(*response.CreateClient) + ret := m.ctrl.Call(m, "GetServers", varargs...) + ret0, _ := ret[0].(*response.GetServers) ret1, _ := ret[1].(error) return ret0, ret1 } -// CreateClient indicates an expected call of CreateClient. -func (mr *MockInfraProxyClientMockRecorder) CreateClient(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetServers indicates an expected call of GetServers +func (mr *MockInfraProxyClientMockRecorder) GetServers(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateClient", reflect.TypeOf((*MockInfraProxyClient)(nil).CreateClient), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetServers", reflect.TypeOf((*MockInfraProxyClient)(nil).GetServers), varargs...) } -// CreateDataBag mocks base method. -func (m *MockInfraProxyClient) CreateDataBag(ctx context.Context, in *request.CreateDataBag, opts ...grpc.CallOption) (*response.CreateDataBag, error) { +// GetServerStatus mocks base method +func (m *MockInfraProxyClient) GetServerStatus(ctx context.Context, in *request.GetServerStatus, opts ...grpc.CallOption) (*response.GetServerStatus, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "CreateDataBag", varargs...) - ret0, _ := ret[0].(*response.CreateDataBag) + ret := m.ctrl.Call(m, "GetServerStatus", varargs...) + ret0, _ := ret[0].(*response.GetServerStatus) ret1, _ := ret[1].(error) return ret0, ret1 } -// CreateDataBag indicates an expected call of CreateDataBag. -func (mr *MockInfraProxyClientMockRecorder) CreateDataBag(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetServerStatus indicates an expected call of GetServerStatus +func (mr *MockInfraProxyClientMockRecorder) GetServerStatus(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateDataBag", reflect.TypeOf((*MockInfraProxyClient)(nil).CreateDataBag), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetServerStatus", reflect.TypeOf((*MockInfraProxyClient)(nil).GetServerStatus), varargs...) } -// CreateDataBagItem mocks base method. -func (m *MockInfraProxyClient) CreateDataBagItem(ctx context.Context, in *request.CreateDataBagItem, opts ...grpc.CallOption) (*response.CreateDataBagItem, error) { +// ValidateWebuiKey mocks base method +func (m *MockInfraProxyClient) ValidateWebuiKey(ctx context.Context, in *request.ValidateWebuiKey, opts ...grpc.CallOption) (*response.ValidateWebuiKey, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "CreateDataBagItem", varargs...) - ret0, _ := ret[0].(*response.CreateDataBagItem) + ret := m.ctrl.Call(m, "ValidateWebuiKey", varargs...) + ret0, _ := ret[0].(*response.ValidateWebuiKey) ret1, _ := ret[1].(error) return ret0, ret1 } -// CreateDataBagItem indicates an expected call of CreateDataBagItem. -func (mr *MockInfraProxyClientMockRecorder) CreateDataBagItem(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// ValidateWebuiKey indicates an expected call of ValidateWebuiKey +func (mr *MockInfraProxyClientMockRecorder) ValidateWebuiKey(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateDataBagItem", reflect.TypeOf((*MockInfraProxyClient)(nil).CreateDataBagItem), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ValidateWebuiKey", reflect.TypeOf((*MockInfraProxyClient)(nil).ValidateWebuiKey), varargs...) } -// CreateEnvironment mocks base method. -func (m *MockInfraProxyClient) CreateEnvironment(ctx context.Context, in *request.CreateEnvironment, opts ...grpc.CallOption) (*response.Environment, error) { +// UpdateWebuiKey mocks base method +func (m *MockInfraProxyClient) UpdateWebuiKey(ctx context.Context, in *request.UpdateWebuiKey, opts ...grpc.CallOption) (*response.UpdateWebuiKey, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "CreateEnvironment", varargs...) - ret0, _ := ret[0].(*response.Environment) + ret := m.ctrl.Call(m, "UpdateWebuiKey", varargs...) + ret0, _ := ret[0].(*response.UpdateWebuiKey) ret1, _ := ret[1].(error) return ret0, ret1 } -// CreateEnvironment indicates an expected call of CreateEnvironment. -func (mr *MockInfraProxyClientMockRecorder) CreateEnvironment(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// UpdateWebuiKey indicates an expected call of UpdateWebuiKey +func (mr *MockInfraProxyClientMockRecorder) UpdateWebuiKey(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateEnvironment", reflect.TypeOf((*MockInfraProxyClient)(nil).CreateEnvironment), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWebuiKey", reflect.TypeOf((*MockInfraProxyClient)(nil).UpdateWebuiKey), varargs...) } -// CreateOrg mocks base method. -func (m *MockInfraProxyClient) CreateOrg(ctx context.Context, in *request.CreateOrg, opts ...grpc.CallOption) (*response.CreateOrg, error) { +// GetServer mocks base method +func (m *MockInfraProxyClient) GetServer(ctx context.Context, in *request.GetServer, opts ...grpc.CallOption) (*response.GetServer, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "CreateOrg", varargs...) - ret0, _ := ret[0].(*response.CreateOrg) + ret := m.ctrl.Call(m, "GetServer", varargs...) + ret0, _ := ret[0].(*response.GetServer) ret1, _ := ret[1].(error) return ret0, ret1 } -// CreateOrg indicates an expected call of CreateOrg. -func (mr *MockInfraProxyClientMockRecorder) CreateOrg(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetServer indicates an expected call of GetServer +func (mr *MockInfraProxyClientMockRecorder) GetServer(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateOrg", reflect.TypeOf((*MockInfraProxyClient)(nil).CreateOrg), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetServer", reflect.TypeOf((*MockInfraProxyClient)(nil).GetServer), varargs...) } -// CreateRole mocks base method. -func (m *MockInfraProxyClient) CreateRole(ctx context.Context, in *request.CreateRole, opts ...grpc.CallOption) (*response.Role, error) { +// CreateServer mocks base method +func (m *MockInfraProxyClient) CreateServer(ctx context.Context, in *request.CreateServer, opts ...grpc.CallOption) (*response.CreateServer, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "CreateRole", varargs...) - ret0, _ := ret[0].(*response.Role) + ret := m.ctrl.Call(m, "CreateServer", varargs...) + ret0, _ := ret[0].(*response.CreateServer) ret1, _ := ret[1].(error) return ret0, ret1 } -// CreateRole indicates an expected call of CreateRole. -func (mr *MockInfraProxyClientMockRecorder) CreateRole(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// CreateServer indicates an expected call of CreateServer +func (mr *MockInfraProxyClientMockRecorder) CreateServer(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateRole", reflect.TypeOf((*MockInfraProxyClient)(nil).CreateRole), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateServer", reflect.TypeOf((*MockInfraProxyClient)(nil).CreateServer), varargs...) } -// CreateServer mocks base method. -func (m *MockInfraProxyClient) CreateServer(ctx context.Context, in *request.CreateServer, opts ...grpc.CallOption) (*response.CreateServer, error) { +// UpdateServer mocks base method +func (m *MockInfraProxyClient) UpdateServer(ctx context.Context, in *request.UpdateServer, opts ...grpc.CallOption) (*response.UpdateServer, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "CreateServer", varargs...) - ret0, _ := ret[0].(*response.CreateServer) + ret := m.ctrl.Call(m, "UpdateServer", varargs...) + ret0, _ := ret[0].(*response.UpdateServer) ret1, _ := ret[1].(error) return ret0, ret1 } -// CreateServer indicates an expected call of CreateServer. -func (mr *MockInfraProxyClientMockRecorder) CreateServer(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// UpdateServer indicates an expected call of UpdateServer +func (mr *MockInfraProxyClientMockRecorder) UpdateServer(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateServer", reflect.TypeOf((*MockInfraProxyClient)(nil).CreateServer), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateServer", reflect.TypeOf((*MockInfraProxyClient)(nil).UpdateServer), varargs...) } -// DeleteClient mocks base method. -func (m *MockInfraProxyClient) DeleteClient(ctx context.Context, in *request.Client, opts ...grpc.CallOption) (*response.Client, error) { +// DeleteServer mocks base method +func (m *MockInfraProxyClient) DeleteServer(ctx context.Context, in *request.DeleteServer, opts ...grpc.CallOption) (*response.DeleteServer, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "DeleteClient", varargs...) - ret0, _ := ret[0].(*response.Client) + ret := m.ctrl.Call(m, "DeleteServer", varargs...) + ret0, _ := ret[0].(*response.DeleteServer) ret1, _ := ret[1].(error) return ret0, ret1 } -// DeleteClient indicates an expected call of DeleteClient. -func (mr *MockInfraProxyClientMockRecorder) DeleteClient(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// DeleteServer indicates an expected call of DeleteServer +func (mr *MockInfraProxyClientMockRecorder) DeleteServer(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteClient", reflect.TypeOf((*MockInfraProxyClient)(nil).DeleteClient), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteServer", reflect.TypeOf((*MockInfraProxyClient)(nil).DeleteServer), varargs...) } -// DeleteDataBag mocks base method. -func (m *MockInfraProxyClient) DeleteDataBag(ctx context.Context, in *request.DataBag, opts ...grpc.CallOption) (*response.DataBag, error) { +// GetOrgs mocks base method +func (m *MockInfraProxyClient) GetOrgs(ctx context.Context, in *request.GetOrgs, opts ...grpc.CallOption) (*response.GetOrgs, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "DeleteDataBag", varargs...) - ret0, _ := ret[0].(*response.DataBag) + ret := m.ctrl.Call(m, "GetOrgs", varargs...) + ret0, _ := ret[0].(*response.GetOrgs) ret1, _ := ret[1].(error) return ret0, ret1 } -// DeleteDataBag indicates an expected call of DeleteDataBag. -func (mr *MockInfraProxyClientMockRecorder) DeleteDataBag(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetOrgs indicates an expected call of GetOrgs +func (mr *MockInfraProxyClientMockRecorder) GetOrgs(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteDataBag", reflect.TypeOf((*MockInfraProxyClient)(nil).DeleteDataBag), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrgs", reflect.TypeOf((*MockInfraProxyClient)(nil).GetOrgs), varargs...) } -// DeleteDataBagItem mocks base method. -func (m *MockInfraProxyClient) DeleteDataBagItem(ctx context.Context, in *request.DataBagItem, opts ...grpc.CallOption) (*response.DataBagItem, error) { +// GetOrg mocks base method +func (m *MockInfraProxyClient) GetOrg(ctx context.Context, in *request.GetOrg, opts ...grpc.CallOption) (*response.GetOrg, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "DeleteDataBagItem", varargs...) - ret0, _ := ret[0].(*response.DataBagItem) + ret := m.ctrl.Call(m, "GetOrg", varargs...) + ret0, _ := ret[0].(*response.GetOrg) ret1, _ := ret[1].(error) return ret0, ret1 } -// DeleteDataBagItem indicates an expected call of DeleteDataBagItem. -func (mr *MockInfraProxyClientMockRecorder) DeleteDataBagItem(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetOrg indicates an expected call of GetOrg +func (mr *MockInfraProxyClientMockRecorder) GetOrg(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteDataBagItem", reflect.TypeOf((*MockInfraProxyClient)(nil).DeleteDataBagItem), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrg", reflect.TypeOf((*MockInfraProxyClient)(nil).GetOrg), varargs...) } -// DeleteEnvironment mocks base method. -func (m *MockInfraProxyClient) DeleteEnvironment(ctx context.Context, in *request.Environment, opts ...grpc.CallOption) (*response.Environment, error) { +// CreateOrg mocks base method +func (m *MockInfraProxyClient) CreateOrg(ctx context.Context, in *request.CreateOrg, opts ...grpc.CallOption) (*response.CreateOrg, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "DeleteEnvironment", varargs...) - ret0, _ := ret[0].(*response.Environment) + ret := m.ctrl.Call(m, "CreateOrg", varargs...) + ret0, _ := ret[0].(*response.CreateOrg) ret1, _ := ret[1].(error) return ret0, ret1 } -// DeleteEnvironment indicates an expected call of DeleteEnvironment. -func (mr *MockInfraProxyClientMockRecorder) DeleteEnvironment(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// CreateOrg indicates an expected call of CreateOrg +func (mr *MockInfraProxyClientMockRecorder) CreateOrg(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteEnvironment", reflect.TypeOf((*MockInfraProxyClient)(nil).DeleteEnvironment), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateOrg", reflect.TypeOf((*MockInfraProxyClient)(nil).CreateOrg), varargs...) } -// DeleteNode mocks base method. -func (m *MockInfraProxyClient) DeleteNode(ctx context.Context, in *request.Node, opts ...grpc.CallOption) (*response.DeleteNode, error) { +// UpdateOrg mocks base method +func (m *MockInfraProxyClient) UpdateOrg(ctx context.Context, in *request.UpdateOrg, opts ...grpc.CallOption) (*response.UpdateOrg, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "DeleteNode", varargs...) - ret0, _ := ret[0].(*response.DeleteNode) + ret := m.ctrl.Call(m, "UpdateOrg", varargs...) + ret0, _ := ret[0].(*response.UpdateOrg) ret1, _ := ret[1].(error) return ret0, ret1 } -// DeleteNode indicates an expected call of DeleteNode. -func (mr *MockInfraProxyClientMockRecorder) DeleteNode(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// UpdateOrg indicates an expected call of UpdateOrg +func (mr *MockInfraProxyClientMockRecorder) UpdateOrg(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteNode", reflect.TypeOf((*MockInfraProxyClient)(nil).DeleteNode), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateOrg", reflect.TypeOf((*MockInfraProxyClient)(nil).UpdateOrg), varargs...) } -// DeleteOrg mocks base method. +// DeleteOrg mocks base method func (m *MockInfraProxyClient) DeleteOrg(ctx context.Context, in *request.DeleteOrg, opts ...grpc.CallOption) (*response.DeleteOrg, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} @@ -290,1177 +289,1177 @@ func (m *MockInfraProxyClient) DeleteOrg(ctx context.Context, in *request.Delete return ret0, ret1 } -// DeleteOrg indicates an expected call of DeleteOrg. +// DeleteOrg indicates an expected call of DeleteOrg func (mr *MockInfraProxyClientMockRecorder) DeleteOrg(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOrg", reflect.TypeOf((*MockInfraProxyClient)(nil).DeleteOrg), varargs...) } -// DeletePolicyfile mocks base method. -func (m *MockInfraProxyClient) DeletePolicyfile(ctx context.Context, in *request.DeletePolicyfile, opts ...grpc.CallOption) (*response.DeletePolicyfile, error) { +// ResetOrgAdminKey mocks base method +func (m *MockInfraProxyClient) ResetOrgAdminKey(ctx context.Context, in *request.ResetOrgAdminKey, opts ...grpc.CallOption) (*response.ResetOrgAdminKey, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "DeletePolicyfile", varargs...) - ret0, _ := ret[0].(*response.DeletePolicyfile) + ret := m.ctrl.Call(m, "ResetOrgAdminKey", varargs...) + ret0, _ := ret[0].(*response.ResetOrgAdminKey) ret1, _ := ret[1].(error) return ret0, ret1 } -// DeletePolicyfile indicates an expected call of DeletePolicyfile. -func (mr *MockInfraProxyClientMockRecorder) DeletePolicyfile(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// ResetOrgAdminKey indicates an expected call of ResetOrgAdminKey +func (mr *MockInfraProxyClientMockRecorder) ResetOrgAdminKey(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeletePolicyfile", reflect.TypeOf((*MockInfraProxyClient)(nil).DeletePolicyfile), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ResetOrgAdminKey", reflect.TypeOf((*MockInfraProxyClient)(nil).ResetOrgAdminKey), varargs...) } -// DeleteRole mocks base method. -func (m *MockInfraProxyClient) DeleteRole(ctx context.Context, in *request.Role, opts ...grpc.CallOption) (*response.Role, error) { +// GetInfraServerOrgs mocks base method +func (m *MockInfraProxyClient) GetInfraServerOrgs(ctx context.Context, in *request.GetInfraServerOrgs, opts ...grpc.CallOption) (*response.GetInfraServerOrgs, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "DeleteRole", varargs...) - ret0, _ := ret[0].(*response.Role) + ret := m.ctrl.Call(m, "GetInfraServerOrgs", varargs...) + ret0, _ := ret[0].(*response.GetInfraServerOrgs) ret1, _ := ret[1].(error) return ret0, ret1 } -// DeleteRole indicates an expected call of DeleteRole. -func (mr *MockInfraProxyClientMockRecorder) DeleteRole(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetInfraServerOrgs indicates an expected call of GetInfraServerOrgs +func (mr *MockInfraProxyClientMockRecorder) GetInfraServerOrgs(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteRole", reflect.TypeOf((*MockInfraProxyClient)(nil).DeleteRole), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetInfraServerOrgs", reflect.TypeOf((*MockInfraProxyClient)(nil).GetInfraServerOrgs), varargs...) } -// DeleteServer mocks base method. -func (m *MockInfraProxyClient) DeleteServer(ctx context.Context, in *request.DeleteServer, opts ...grpc.CallOption) (*response.DeleteServer, error) { +// GetCookbooks mocks base method +func (m *MockInfraProxyClient) GetCookbooks(ctx context.Context, in *request.Cookbooks, opts ...grpc.CallOption) (*response.Cookbooks, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "DeleteServer", varargs...) - ret0, _ := ret[0].(*response.DeleteServer) + ret := m.ctrl.Call(m, "GetCookbooks", varargs...) + ret0, _ := ret[0].(*response.Cookbooks) ret1, _ := ret[1].(error) return ret0, ret1 } -// DeleteServer indicates an expected call of DeleteServer. -func (mr *MockInfraProxyClientMockRecorder) DeleteServer(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetCookbooks indicates an expected call of GetCookbooks +func (mr *MockInfraProxyClientMockRecorder) GetCookbooks(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteServer", reflect.TypeOf((*MockInfraProxyClient)(nil).DeleteServer), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCookbooks", reflect.TypeOf((*MockInfraProxyClient)(nil).GetCookbooks), varargs...) } -// GetAutomateInfraServerUsersList mocks base method. -func (m *MockInfraProxyClient) GetAutomateInfraServerUsersList(ctx context.Context, in *request.AutomateInfraServerUsers, opts ...grpc.CallOption) (*response.AutomateInfraServerUsers, error) { +// GetCookbookVersions mocks base method +func (m *MockInfraProxyClient) GetCookbookVersions(ctx context.Context, in *request.CookbookVersions, opts ...grpc.CallOption) (*response.CookbookVersions, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetAutomateInfraServerUsersList", varargs...) - ret0, _ := ret[0].(*response.AutomateInfraServerUsers) + ret := m.ctrl.Call(m, "GetCookbookVersions", varargs...) + ret0, _ := ret[0].(*response.CookbookVersions) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetAutomateInfraServerUsersList indicates an expected call of GetAutomateInfraServerUsersList. -func (mr *MockInfraProxyClientMockRecorder) GetAutomateInfraServerUsersList(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetCookbookVersions indicates an expected call of GetCookbookVersions +func (mr *MockInfraProxyClientMockRecorder) GetCookbookVersions(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAutomateInfraServerUsersList", reflect.TypeOf((*MockInfraProxyClient)(nil).GetAutomateInfraServerUsersList), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCookbookVersions", reflect.TypeOf((*MockInfraProxyClient)(nil).GetCookbookVersions), varargs...) } -// GetClient mocks base method. -func (m *MockInfraProxyClient) GetClient(ctx context.Context, in *request.Client, opts ...grpc.CallOption) (*response.Client, error) { +// GetCookbook mocks base method +func (m *MockInfraProxyClient) GetCookbook(ctx context.Context, in *request.Cookbook, opts ...grpc.CallOption) (*response.Cookbook, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetClient", varargs...) - ret0, _ := ret[0].(*response.Client) + ret := m.ctrl.Call(m, "GetCookbook", varargs...) + ret0, _ := ret[0].(*response.Cookbook) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetClient indicates an expected call of GetClient. -func (mr *MockInfraProxyClientMockRecorder) GetClient(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetCookbook indicates an expected call of GetCookbook +func (mr *MockInfraProxyClientMockRecorder) GetCookbook(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetClient", reflect.TypeOf((*MockInfraProxyClient)(nil).GetClient), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCookbook", reflect.TypeOf((*MockInfraProxyClient)(nil).GetCookbook), varargs...) } -// GetClients mocks base method. -func (m *MockInfraProxyClient) GetClients(ctx context.Context, in *request.Clients, opts ...grpc.CallOption) (*response.Clients, error) { +// GetCookbookFileContent mocks base method +func (m *MockInfraProxyClient) GetCookbookFileContent(ctx context.Context, in *request.CookbookFileContent, opts ...grpc.CallOption) (*response.CookbookFileContent, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetClients", varargs...) - ret0, _ := ret[0].(*response.Clients) + ret := m.ctrl.Call(m, "GetCookbookFileContent", varargs...) + ret0, _ := ret[0].(*response.CookbookFileContent) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetClients indicates an expected call of GetClients. -func (mr *MockInfraProxyClientMockRecorder) GetClients(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetCookbookFileContent indicates an expected call of GetCookbookFileContent +func (mr *MockInfraProxyClientMockRecorder) GetCookbookFileContent(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetClients", reflect.TypeOf((*MockInfraProxyClient)(nil).GetClients), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCookbookFileContent", reflect.TypeOf((*MockInfraProxyClient)(nil).GetCookbookFileContent), varargs...) } -// GetCookbook mocks base method. -func (m *MockInfraProxyClient) GetCookbook(ctx context.Context, in *request.Cookbook, opts ...grpc.CallOption) (*response.Cookbook, error) { +// GetRoles mocks base method +func (m *MockInfraProxyClient) GetRoles(ctx context.Context, in *request.Roles, opts ...grpc.CallOption) (*response.Roles, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetCookbook", varargs...) - ret0, _ := ret[0].(*response.Cookbook) + ret := m.ctrl.Call(m, "GetRoles", varargs...) + ret0, _ := ret[0].(*response.Roles) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetCookbook indicates an expected call of GetCookbook. -func (mr *MockInfraProxyClientMockRecorder) GetCookbook(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetRoles indicates an expected call of GetRoles +func (mr *MockInfraProxyClientMockRecorder) GetRoles(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCookbook", reflect.TypeOf((*MockInfraProxyClient)(nil).GetCookbook), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRoles", reflect.TypeOf((*MockInfraProxyClient)(nil).GetRoles), varargs...) } -// GetCookbookFileContent mocks base method. -func (m *MockInfraProxyClient) GetCookbookFileContent(ctx context.Context, in *request.CookbookFileContent, opts ...grpc.CallOption) (*response.CookbookFileContent, error) { +// GetRole mocks base method +func (m *MockInfraProxyClient) GetRole(ctx context.Context, in *request.Role, opts ...grpc.CallOption) (*response.Role, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetCookbookFileContent", varargs...) - ret0, _ := ret[0].(*response.CookbookFileContent) + ret := m.ctrl.Call(m, "GetRole", varargs...) + ret0, _ := ret[0].(*response.Role) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetCookbookFileContent indicates an expected call of GetCookbookFileContent. -func (mr *MockInfraProxyClientMockRecorder) GetCookbookFileContent(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetRole indicates an expected call of GetRole +func (mr *MockInfraProxyClientMockRecorder) GetRole(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCookbookFileContent", reflect.TypeOf((*MockInfraProxyClient)(nil).GetCookbookFileContent), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRole", reflect.TypeOf((*MockInfraProxyClient)(nil).GetRole), varargs...) } -// GetCookbookVersions mocks base method. -func (m *MockInfraProxyClient) GetCookbookVersions(ctx context.Context, in *request.CookbookVersions, opts ...grpc.CallOption) (*response.CookbookVersions, error) { +// GetRoleExpandedRunList mocks base method +func (m *MockInfraProxyClient) GetRoleExpandedRunList(ctx context.Context, in *request.ExpandedRunList, opts ...grpc.CallOption) (*response.ExpandedRunList, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetCookbookVersions", varargs...) - ret0, _ := ret[0].(*response.CookbookVersions) + ret := m.ctrl.Call(m, "GetRoleExpandedRunList", varargs...) + ret0, _ := ret[0].(*response.ExpandedRunList) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetCookbookVersions indicates an expected call of GetCookbookVersions. -func (mr *MockInfraProxyClientMockRecorder) GetCookbookVersions(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetRoleExpandedRunList indicates an expected call of GetRoleExpandedRunList +func (mr *MockInfraProxyClientMockRecorder) GetRoleExpandedRunList(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCookbookVersions", reflect.TypeOf((*MockInfraProxyClient)(nil).GetCookbookVersions), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRoleExpandedRunList", reflect.TypeOf((*MockInfraProxyClient)(nil).GetRoleExpandedRunList), varargs...) } -// GetCookbooks mocks base method. -func (m *MockInfraProxyClient) GetCookbooks(ctx context.Context, in *request.Cookbooks, opts ...grpc.CallOption) (*response.Cookbooks, error) { +// GetRoleEnvironments mocks base method +func (m *MockInfraProxyClient) GetRoleEnvironments(ctx context.Context, in *request.Role, opts ...grpc.CallOption) (*response.RoleEnvironments, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetCookbooks", varargs...) - ret0, _ := ret[0].(*response.Cookbooks) + ret := m.ctrl.Call(m, "GetRoleEnvironments", varargs...) + ret0, _ := ret[0].(*response.RoleEnvironments) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetCookbooks indicates an expected call of GetCookbooks. -func (mr *MockInfraProxyClientMockRecorder) GetCookbooks(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetRoleEnvironments indicates an expected call of GetRoleEnvironments +func (mr *MockInfraProxyClientMockRecorder) GetRoleEnvironments(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCookbooks", reflect.TypeOf((*MockInfraProxyClient)(nil).GetCookbooks), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRoleEnvironments", reflect.TypeOf((*MockInfraProxyClient)(nil).GetRoleEnvironments), varargs...) } -// GetDataBagItem mocks base method. -func (m *MockInfraProxyClient) GetDataBagItem(ctx context.Context, in *request.DataBagItem, opts ...grpc.CallOption) (*response.DataBagItem, error) { +// CreateRole mocks base method +func (m *MockInfraProxyClient) CreateRole(ctx context.Context, in *request.CreateRole, opts ...grpc.CallOption) (*response.Role, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetDataBagItem", varargs...) - ret0, _ := ret[0].(*response.DataBagItem) + ret := m.ctrl.Call(m, "CreateRole", varargs...) + ret0, _ := ret[0].(*response.Role) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetDataBagItem indicates an expected call of GetDataBagItem. -func (mr *MockInfraProxyClientMockRecorder) GetDataBagItem(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// CreateRole indicates an expected call of CreateRole +func (mr *MockInfraProxyClientMockRecorder) CreateRole(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDataBagItem", reflect.TypeOf((*MockInfraProxyClient)(nil).GetDataBagItem), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateRole", reflect.TypeOf((*MockInfraProxyClient)(nil).CreateRole), varargs...) } -// GetDataBagItems mocks base method. -func (m *MockInfraProxyClient) GetDataBagItems(ctx context.Context, in *request.DataBagItems, opts ...grpc.CallOption) (*response.DataBagItems, error) { +// DeleteRole mocks base method +func (m *MockInfraProxyClient) DeleteRole(ctx context.Context, in *request.Role, opts ...grpc.CallOption) (*response.Role, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetDataBagItems", varargs...) - ret0, _ := ret[0].(*response.DataBagItems) + ret := m.ctrl.Call(m, "DeleteRole", varargs...) + ret0, _ := ret[0].(*response.Role) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetDataBagItems indicates an expected call of GetDataBagItems. -func (mr *MockInfraProxyClientMockRecorder) GetDataBagItems(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// DeleteRole indicates an expected call of DeleteRole +func (mr *MockInfraProxyClientMockRecorder) DeleteRole(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDataBagItems", reflect.TypeOf((*MockInfraProxyClient)(nil).GetDataBagItems), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteRole", reflect.TypeOf((*MockInfraProxyClient)(nil).DeleteRole), varargs...) } -// GetDataBags mocks base method. -func (m *MockInfraProxyClient) GetDataBags(ctx context.Context, in *request.DataBags, opts ...grpc.CallOption) (*response.DataBags, error) { +// UpdateRole mocks base method +func (m *MockInfraProxyClient) UpdateRole(ctx context.Context, in *request.UpdateRole, opts ...grpc.CallOption) (*response.Role, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetDataBags", varargs...) - ret0, _ := ret[0].(*response.DataBags) + ret := m.ctrl.Call(m, "UpdateRole", varargs...) + ret0, _ := ret[0].(*response.Role) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetDataBags indicates an expected call of GetDataBags. -func (mr *MockInfraProxyClientMockRecorder) GetDataBags(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// UpdateRole indicates an expected call of UpdateRole +func (mr *MockInfraProxyClientMockRecorder) UpdateRole(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDataBags", reflect.TypeOf((*MockInfraProxyClient)(nil).GetDataBags), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateRole", reflect.TypeOf((*MockInfraProxyClient)(nil).UpdateRole), varargs...) } -// GetEnvironment mocks base method. -func (m *MockInfraProxyClient) GetEnvironment(ctx context.Context, in *request.Environment, opts ...grpc.CallOption) (*response.Environment, error) { +// GetClients mocks base method +func (m *MockInfraProxyClient) GetClients(ctx context.Context, in *request.Clients, opts ...grpc.CallOption) (*response.Clients, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetEnvironment", varargs...) - ret0, _ := ret[0].(*response.Environment) + ret := m.ctrl.Call(m, "GetClients", varargs...) + ret0, _ := ret[0].(*response.Clients) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetEnvironment indicates an expected call of GetEnvironment. -func (mr *MockInfraProxyClientMockRecorder) GetEnvironment(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetClients indicates an expected call of GetClients +func (mr *MockInfraProxyClientMockRecorder) GetClients(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetEnvironment", reflect.TypeOf((*MockInfraProxyClient)(nil).GetEnvironment), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetClients", reflect.TypeOf((*MockInfraProxyClient)(nil).GetClients), varargs...) } -// GetEnvironmentRecipes mocks base method. -func (m *MockInfraProxyClient) GetEnvironmentRecipes(ctx context.Context, in *request.Environment, opts ...grpc.CallOption) (*response.EnvironmentRecipesList, error) { +// GetClient mocks base method +func (m *MockInfraProxyClient) GetClient(ctx context.Context, in *request.Client, opts ...grpc.CallOption) (*response.Client, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetEnvironmentRecipes", varargs...) - ret0, _ := ret[0].(*response.EnvironmentRecipesList) + ret := m.ctrl.Call(m, "GetClient", varargs...) + ret0, _ := ret[0].(*response.Client) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetEnvironmentRecipes indicates an expected call of GetEnvironmentRecipes. -func (mr *MockInfraProxyClientMockRecorder) GetEnvironmentRecipes(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetClient indicates an expected call of GetClient +func (mr *MockInfraProxyClientMockRecorder) GetClient(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetEnvironmentRecipes", reflect.TypeOf((*MockInfraProxyClient)(nil).GetEnvironmentRecipes), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetClient", reflect.TypeOf((*MockInfraProxyClient)(nil).GetClient), varargs...) } -// GetEnvironments mocks base method. -func (m *MockInfraProxyClient) GetEnvironments(ctx context.Context, in *request.Environments, opts ...grpc.CallOption) (*response.Environments, error) { +// CreateClient mocks base method +func (m *MockInfraProxyClient) CreateClient(ctx context.Context, in *request.CreateClient, opts ...grpc.CallOption) (*response.CreateClient, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetEnvironments", varargs...) - ret0, _ := ret[0].(*response.Environments) + ret := m.ctrl.Call(m, "CreateClient", varargs...) + ret0, _ := ret[0].(*response.CreateClient) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetEnvironments indicates an expected call of GetEnvironments. -func (mr *MockInfraProxyClientMockRecorder) GetEnvironments(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// CreateClient indicates an expected call of CreateClient +func (mr *MockInfraProxyClientMockRecorder) CreateClient(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetEnvironments", reflect.TypeOf((*MockInfraProxyClient)(nil).GetEnvironments), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateClient", reflect.TypeOf((*MockInfraProxyClient)(nil).CreateClient), varargs...) } -// GetInfraServerOrgs mocks base method. -func (m *MockInfraProxyClient) GetInfraServerOrgs(ctx context.Context, in *request.GetInfraServerOrgs, opts ...grpc.CallOption) (*response.GetInfraServerOrgs, error) { +// DeleteClient mocks base method +func (m *MockInfraProxyClient) DeleteClient(ctx context.Context, in *request.Client, opts ...grpc.CallOption) (*response.Client, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetInfraServerOrgs", varargs...) - ret0, _ := ret[0].(*response.GetInfraServerOrgs) + ret := m.ctrl.Call(m, "DeleteClient", varargs...) + ret0, _ := ret[0].(*response.Client) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetInfraServerOrgs indicates an expected call of GetInfraServerOrgs. -func (mr *MockInfraProxyClientMockRecorder) GetInfraServerOrgs(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// DeleteClient indicates an expected call of DeleteClient +func (mr *MockInfraProxyClientMockRecorder) DeleteClient(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetInfraServerOrgs", reflect.TypeOf((*MockInfraProxyClient)(nil).GetInfraServerOrgs), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteClient", reflect.TypeOf((*MockInfraProxyClient)(nil).DeleteClient), varargs...) } -// GetNode mocks base method. -func (m *MockInfraProxyClient) GetNode(ctx context.Context, in *request.Node, opts ...grpc.CallOption) (*response.Node, error) { +// ResetClientKey mocks base method +func (m *MockInfraProxyClient) ResetClientKey(ctx context.Context, in *request.ClientKey, opts ...grpc.CallOption) (*response.ResetClient, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetNode", varargs...) - ret0, _ := ret[0].(*response.Node) + ret := m.ctrl.Call(m, "ResetClientKey", varargs...) + ret0, _ := ret[0].(*response.ResetClient) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetNode indicates an expected call of GetNode. -func (mr *MockInfraProxyClientMockRecorder) GetNode(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// ResetClientKey indicates an expected call of ResetClientKey +func (mr *MockInfraProxyClientMockRecorder) ResetClientKey(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNode", reflect.TypeOf((*MockInfraProxyClient)(nil).GetNode), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ResetClientKey", reflect.TypeOf((*MockInfraProxyClient)(nil).ResetClientKey), varargs...) } -// GetNodeExpandedRunList mocks base method. -func (m *MockInfraProxyClient) GetNodeExpandedRunList(ctx context.Context, in *request.NodeExpandedRunList, opts ...grpc.CallOption) (*response.NodeExpandedRunList, error) { +// GetDataBags mocks base method +func (m *MockInfraProxyClient) GetDataBags(ctx context.Context, in *request.DataBags, opts ...grpc.CallOption) (*response.DataBags, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetNodeExpandedRunList", varargs...) - ret0, _ := ret[0].(*response.NodeExpandedRunList) + ret := m.ctrl.Call(m, "GetDataBags", varargs...) + ret0, _ := ret[0].(*response.DataBags) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetNodeExpandedRunList indicates an expected call of GetNodeExpandedRunList. -func (mr *MockInfraProxyClientMockRecorder) GetNodeExpandedRunList(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetDataBags indicates an expected call of GetDataBags +func (mr *MockInfraProxyClientMockRecorder) GetDataBags(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodeExpandedRunList", reflect.TypeOf((*MockInfraProxyClient)(nil).GetNodeExpandedRunList), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDataBags", reflect.TypeOf((*MockInfraProxyClient)(nil).GetDataBags), varargs...) } -// GetNodes mocks base method. -func (m *MockInfraProxyClient) GetNodes(ctx context.Context, in *request.Nodes, opts ...grpc.CallOption) (*response.Nodes, error) { +// GetDataBagItems mocks base method +func (m *MockInfraProxyClient) GetDataBagItems(ctx context.Context, in *request.DataBagItems, opts ...grpc.CallOption) (*response.DataBagItems, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetNodes", varargs...) - ret0, _ := ret[0].(*response.Nodes) + ret := m.ctrl.Call(m, "GetDataBagItems", varargs...) + ret0, _ := ret[0].(*response.DataBagItems) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetNodes indicates an expected call of GetNodes. -func (mr *MockInfraProxyClientMockRecorder) GetNodes(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetDataBagItems indicates an expected call of GetDataBagItems +func (mr *MockInfraProxyClientMockRecorder) GetDataBagItems(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodes", reflect.TypeOf((*MockInfraProxyClient)(nil).GetNodes), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDataBagItems", reflect.TypeOf((*MockInfraProxyClient)(nil).GetDataBagItems), varargs...) } -// GetOrg mocks base method. -func (m *MockInfraProxyClient) GetOrg(ctx context.Context, in *request.GetOrg, opts ...grpc.CallOption) (*response.GetOrg, error) { +// GetDataBagItem mocks base method +func (m *MockInfraProxyClient) GetDataBagItem(ctx context.Context, in *request.DataBagItem, opts ...grpc.CallOption) (*response.DataBagItem, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetOrg", varargs...) - ret0, _ := ret[0].(*response.GetOrg) + ret := m.ctrl.Call(m, "GetDataBagItem", varargs...) + ret0, _ := ret[0].(*response.DataBagItem) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetOrg indicates an expected call of GetOrg. -func (mr *MockInfraProxyClientMockRecorder) GetOrg(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetDataBagItem indicates an expected call of GetDataBagItem +func (mr *MockInfraProxyClientMockRecorder) GetDataBagItem(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrg", reflect.TypeOf((*MockInfraProxyClient)(nil).GetOrg), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDataBagItem", reflect.TypeOf((*MockInfraProxyClient)(nil).GetDataBagItem), varargs...) } -// GetOrgUsersList mocks base method. -func (m *MockInfraProxyClient) GetOrgUsersList(ctx context.Context, in *request.OrgUsers, opts ...grpc.CallOption) (*response.OrgUsers, error) { +// CreateDataBag mocks base method +func (m *MockInfraProxyClient) CreateDataBag(ctx context.Context, in *request.CreateDataBag, opts ...grpc.CallOption) (*response.CreateDataBag, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetOrgUsersList", varargs...) - ret0, _ := ret[0].(*response.OrgUsers) + ret := m.ctrl.Call(m, "CreateDataBag", varargs...) + ret0, _ := ret[0].(*response.CreateDataBag) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetOrgUsersList indicates an expected call of GetOrgUsersList. -func (mr *MockInfraProxyClientMockRecorder) GetOrgUsersList(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// CreateDataBag indicates an expected call of CreateDataBag +func (mr *MockInfraProxyClientMockRecorder) CreateDataBag(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrgUsersList", reflect.TypeOf((*MockInfraProxyClient)(nil).GetOrgUsersList), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateDataBag", reflect.TypeOf((*MockInfraProxyClient)(nil).CreateDataBag), varargs...) } -// GetOrgs mocks base method. -func (m *MockInfraProxyClient) GetOrgs(ctx context.Context, in *request.GetOrgs, opts ...grpc.CallOption) (*response.GetOrgs, error) { +// CreateDataBagItem mocks base method +func (m *MockInfraProxyClient) CreateDataBagItem(ctx context.Context, in *request.CreateDataBagItem, opts ...grpc.CallOption) (*response.CreateDataBagItem, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetOrgs", varargs...) - ret0, _ := ret[0].(*response.GetOrgs) + ret := m.ctrl.Call(m, "CreateDataBagItem", varargs...) + ret0, _ := ret[0].(*response.CreateDataBagItem) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetOrgs indicates an expected call of GetOrgs. -func (mr *MockInfraProxyClientMockRecorder) GetOrgs(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// CreateDataBagItem indicates an expected call of CreateDataBagItem +func (mr *MockInfraProxyClientMockRecorder) CreateDataBagItem(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrgs", reflect.TypeOf((*MockInfraProxyClient)(nil).GetOrgs), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateDataBagItem", reflect.TypeOf((*MockInfraProxyClient)(nil).CreateDataBagItem), varargs...) } -// GetPolicyfile mocks base method. -func (m *MockInfraProxyClient) GetPolicyfile(ctx context.Context, in *request.Policyfile, opts ...grpc.CallOption) (*response.Policyfile, error) { +// DeleteDataBag mocks base method +func (m *MockInfraProxyClient) DeleteDataBag(ctx context.Context, in *request.DataBag, opts ...grpc.CallOption) (*response.DataBag, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetPolicyfile", varargs...) - ret0, _ := ret[0].(*response.Policyfile) + ret := m.ctrl.Call(m, "DeleteDataBag", varargs...) + ret0, _ := ret[0].(*response.DataBag) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetPolicyfile indicates an expected call of GetPolicyfile. -func (mr *MockInfraProxyClientMockRecorder) GetPolicyfile(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// DeleteDataBag indicates an expected call of DeleteDataBag +func (mr *MockInfraProxyClientMockRecorder) DeleteDataBag(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPolicyfile", reflect.TypeOf((*MockInfraProxyClient)(nil).GetPolicyfile), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteDataBag", reflect.TypeOf((*MockInfraProxyClient)(nil).DeleteDataBag), varargs...) } -// GetPolicyfileRevisions mocks base method. -func (m *MockInfraProxyClient) GetPolicyfileRevisions(ctx context.Context, in *request.PolicyfileRevisions, opts ...grpc.CallOption) (*response.PolicyfileRevisions, error) { +// DeleteDataBagItem mocks base method +func (m *MockInfraProxyClient) DeleteDataBagItem(ctx context.Context, in *request.DataBagItem, opts ...grpc.CallOption) (*response.DataBagItem, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetPolicyfileRevisions", varargs...) - ret0, _ := ret[0].(*response.PolicyfileRevisions) + ret := m.ctrl.Call(m, "DeleteDataBagItem", varargs...) + ret0, _ := ret[0].(*response.DataBagItem) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetPolicyfileRevisions indicates an expected call of GetPolicyfileRevisions. -func (mr *MockInfraProxyClientMockRecorder) GetPolicyfileRevisions(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// DeleteDataBagItem indicates an expected call of DeleteDataBagItem +func (mr *MockInfraProxyClientMockRecorder) DeleteDataBagItem(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPolicyfileRevisions", reflect.TypeOf((*MockInfraProxyClient)(nil).GetPolicyfileRevisions), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteDataBagItem", reflect.TypeOf((*MockInfraProxyClient)(nil).DeleteDataBagItem), varargs...) } -// GetPolicyfiles mocks base method. -func (m *MockInfraProxyClient) GetPolicyfiles(ctx context.Context, in *request.Policyfiles, opts ...grpc.CallOption) (*response.Policyfiles, error) { +// UpdateDataBagItem mocks base method +func (m *MockInfraProxyClient) UpdateDataBagItem(ctx context.Context, in *request.UpdateDataBagItem, opts ...grpc.CallOption) (*response.UpdateDataBagItem, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetPolicyfiles", varargs...) - ret0, _ := ret[0].(*response.Policyfiles) + ret := m.ctrl.Call(m, "UpdateDataBagItem", varargs...) + ret0, _ := ret[0].(*response.UpdateDataBagItem) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetPolicyfiles indicates an expected call of GetPolicyfiles. -func (mr *MockInfraProxyClientMockRecorder) GetPolicyfiles(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// UpdateDataBagItem indicates an expected call of UpdateDataBagItem +func (mr *MockInfraProxyClientMockRecorder) UpdateDataBagItem(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPolicyfiles", reflect.TypeOf((*MockInfraProxyClient)(nil).GetPolicyfiles), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateDataBagItem", reflect.TypeOf((*MockInfraProxyClient)(nil).UpdateDataBagItem), varargs...) } -// GetPolicygroup mocks base method. -func (m *MockInfraProxyClient) GetPolicygroup(ctx context.Context, in *request.Policygroup, opts ...grpc.CallOption) (*response.Policygroup, error) { +// GetEnvironments mocks base method +func (m *MockInfraProxyClient) GetEnvironments(ctx context.Context, in *request.Environments, opts ...grpc.CallOption) (*response.Environments, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetPolicygroup", varargs...) - ret0, _ := ret[0].(*response.Policygroup) + ret := m.ctrl.Call(m, "GetEnvironments", varargs...) + ret0, _ := ret[0].(*response.Environments) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetPolicygroup indicates an expected call of GetPolicygroup. -func (mr *MockInfraProxyClientMockRecorder) GetPolicygroup(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetEnvironments indicates an expected call of GetEnvironments +func (mr *MockInfraProxyClientMockRecorder) GetEnvironments(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPolicygroup", reflect.TypeOf((*MockInfraProxyClient)(nil).GetPolicygroup), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetEnvironments", reflect.TypeOf((*MockInfraProxyClient)(nil).GetEnvironments), varargs...) } -// GetRole mocks base method. -func (m *MockInfraProxyClient) GetRole(ctx context.Context, in *request.Role, opts ...grpc.CallOption) (*response.Role, error) { +// GetEnvironment mocks base method +func (m *MockInfraProxyClient) GetEnvironment(ctx context.Context, in *request.Environment, opts ...grpc.CallOption) (*response.Environment, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetRole", varargs...) - ret0, _ := ret[0].(*response.Role) + ret := m.ctrl.Call(m, "GetEnvironment", varargs...) + ret0, _ := ret[0].(*response.Environment) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetRole indicates an expected call of GetRole. -func (mr *MockInfraProxyClientMockRecorder) GetRole(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetEnvironment indicates an expected call of GetEnvironment +func (mr *MockInfraProxyClientMockRecorder) GetEnvironment(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRole", reflect.TypeOf((*MockInfraProxyClient)(nil).GetRole), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetEnvironment", reflect.TypeOf((*MockInfraProxyClient)(nil).GetEnvironment), varargs...) } -// GetRoleEnvironments mocks base method. -func (m *MockInfraProxyClient) GetRoleEnvironments(ctx context.Context, in *request.Role, opts ...grpc.CallOption) (*response.RoleEnvironments, error) { +// CreateEnvironment mocks base method +func (m *MockInfraProxyClient) CreateEnvironment(ctx context.Context, in *request.CreateEnvironment, opts ...grpc.CallOption) (*response.Environment, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetRoleEnvironments", varargs...) - ret0, _ := ret[0].(*response.RoleEnvironments) + ret := m.ctrl.Call(m, "CreateEnvironment", varargs...) + ret0, _ := ret[0].(*response.Environment) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetRoleEnvironments indicates an expected call of GetRoleEnvironments. -func (mr *MockInfraProxyClientMockRecorder) GetRoleEnvironments(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// CreateEnvironment indicates an expected call of CreateEnvironment +func (mr *MockInfraProxyClientMockRecorder) CreateEnvironment(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRoleEnvironments", reflect.TypeOf((*MockInfraProxyClient)(nil).GetRoleEnvironments), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateEnvironment", reflect.TypeOf((*MockInfraProxyClient)(nil).CreateEnvironment), varargs...) } -// GetRoleExpandedRunList mocks base method. -func (m *MockInfraProxyClient) GetRoleExpandedRunList(ctx context.Context, in *request.ExpandedRunList, opts ...grpc.CallOption) (*response.ExpandedRunList, error) { +// DeleteEnvironment mocks base method +func (m *MockInfraProxyClient) DeleteEnvironment(ctx context.Context, in *request.Environment, opts ...grpc.CallOption) (*response.Environment, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetRoleExpandedRunList", varargs...) - ret0, _ := ret[0].(*response.ExpandedRunList) + ret := m.ctrl.Call(m, "DeleteEnvironment", varargs...) + ret0, _ := ret[0].(*response.Environment) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetRoleExpandedRunList indicates an expected call of GetRoleExpandedRunList. -func (mr *MockInfraProxyClientMockRecorder) GetRoleExpandedRunList(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// DeleteEnvironment indicates an expected call of DeleteEnvironment +func (mr *MockInfraProxyClientMockRecorder) DeleteEnvironment(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRoleExpandedRunList", reflect.TypeOf((*MockInfraProxyClient)(nil).GetRoleExpandedRunList), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteEnvironment", reflect.TypeOf((*MockInfraProxyClient)(nil).DeleteEnvironment), varargs...) } -// GetRoles mocks base method. -func (m *MockInfraProxyClient) GetRoles(ctx context.Context, in *request.Roles, opts ...grpc.CallOption) (*response.Roles, error) { +// UpdateEnvironment mocks base method +func (m *MockInfraProxyClient) UpdateEnvironment(ctx context.Context, in *request.UpdateEnvironment, opts ...grpc.CallOption) (*response.Environment, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetRoles", varargs...) - ret0, _ := ret[0].(*response.Roles) + ret := m.ctrl.Call(m, "UpdateEnvironment", varargs...) + ret0, _ := ret[0].(*response.Environment) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetRoles indicates an expected call of GetRoles. -func (mr *MockInfraProxyClientMockRecorder) GetRoles(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// UpdateEnvironment indicates an expected call of UpdateEnvironment +func (mr *MockInfraProxyClientMockRecorder) UpdateEnvironment(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRoles", reflect.TypeOf((*MockInfraProxyClient)(nil).GetRoles), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateEnvironment", reflect.TypeOf((*MockInfraProxyClient)(nil).UpdateEnvironment), varargs...) } -// GetServer mocks base method. -func (m *MockInfraProxyClient) GetServer(ctx context.Context, in *request.GetServer, opts ...grpc.CallOption) (*response.GetServer, error) { +// GetEnvironmentRecipes mocks base method +func (m *MockInfraProxyClient) GetEnvironmentRecipes(ctx context.Context, in *request.Environment, opts ...grpc.CallOption) (*response.EnvironmentRecipesList, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetServer", varargs...) - ret0, _ := ret[0].(*response.GetServer) + ret := m.ctrl.Call(m, "GetEnvironmentRecipes", varargs...) + ret0, _ := ret[0].(*response.EnvironmentRecipesList) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetServer indicates an expected call of GetServer. -func (mr *MockInfraProxyClientMockRecorder) GetServer(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetEnvironmentRecipes indicates an expected call of GetEnvironmentRecipes +func (mr *MockInfraProxyClientMockRecorder) GetEnvironmentRecipes(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetServer", reflect.TypeOf((*MockInfraProxyClient)(nil).GetServer), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetEnvironmentRecipes", reflect.TypeOf((*MockInfraProxyClient)(nil).GetEnvironmentRecipes), varargs...) } -// GetServerStatus mocks base method. -func (m *MockInfraProxyClient) GetServerStatus(ctx context.Context, in *request.GetServerStatus, opts ...grpc.CallOption) (*response.GetServerStatus, error) { +// GetNodes mocks base method +func (m *MockInfraProxyClient) GetNodes(ctx context.Context, in *request.Nodes, opts ...grpc.CallOption) (*response.Nodes, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetServerStatus", varargs...) - ret0, _ := ret[0].(*response.GetServerStatus) + ret := m.ctrl.Call(m, "GetNodes", varargs...) + ret0, _ := ret[0].(*response.Nodes) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetServerStatus indicates an expected call of GetServerStatus. -func (mr *MockInfraProxyClientMockRecorder) GetServerStatus(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetNodes indicates an expected call of GetNodes +func (mr *MockInfraProxyClientMockRecorder) GetNodes(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetServerStatus", reflect.TypeOf((*MockInfraProxyClient)(nil).GetServerStatus), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodes", reflect.TypeOf((*MockInfraProxyClient)(nil).GetNodes), varargs...) } -// GetServerUsersList mocks base method. -func (m *MockInfraProxyClient) GetServerUsersList(ctx context.Context, in *request.ServerUsers, opts ...grpc.CallOption) (*response.ServerUsers, error) { +// DeleteNode mocks base method +func (m *MockInfraProxyClient) DeleteNode(ctx context.Context, in *request.Node, opts ...grpc.CallOption) (*response.DeleteNode, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetServerUsersList", varargs...) - ret0, _ := ret[0].(*response.ServerUsers) + ret := m.ctrl.Call(m, "DeleteNode", varargs...) + ret0, _ := ret[0].(*response.DeleteNode) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetServerUsersList indicates an expected call of GetServerUsersList. -func (mr *MockInfraProxyClientMockRecorder) GetServerUsersList(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// DeleteNode indicates an expected call of DeleteNode +func (mr *MockInfraProxyClientMockRecorder) DeleteNode(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetServerUsersList", reflect.TypeOf((*MockInfraProxyClient)(nil).GetServerUsersList), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteNode", reflect.TypeOf((*MockInfraProxyClient)(nil).DeleteNode), varargs...) } -// GetServers mocks base method. -func (m *MockInfraProxyClient) GetServers(ctx context.Context, in *request.GetServers, opts ...grpc.CallOption) (*response.GetServers, error) { +// UpdateNode mocks base method +func (m *MockInfraProxyClient) UpdateNode(ctx context.Context, in *request.NodeDetails, opts ...grpc.CallOption) (*response.Node, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetServers", varargs...) - ret0, _ := ret[0].(*response.GetServers) + ret := m.ctrl.Call(m, "UpdateNode", varargs...) + ret0, _ := ret[0].(*response.Node) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetServers indicates an expected call of GetServers. -func (mr *MockInfraProxyClientMockRecorder) GetServers(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// UpdateNode indicates an expected call of UpdateNode +func (mr *MockInfraProxyClientMockRecorder) UpdateNode(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetServers", reflect.TypeOf((*MockInfraProxyClient)(nil).GetServers), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNode", reflect.TypeOf((*MockInfraProxyClient)(nil).UpdateNode), varargs...) } -// ResetClientKey mocks base method. -func (m *MockInfraProxyClient) ResetClientKey(ctx context.Context, in *request.ClientKey, opts ...grpc.CallOption) (*response.ResetClient, error) { +// UpdateNodeAttributes mocks base method +func (m *MockInfraProxyClient) UpdateNodeAttributes(ctx context.Context, in *request.UpdateNodeAttributes, opts ...grpc.CallOption) (*response.UpdateNodeAttributes, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "ResetClientKey", varargs...) - ret0, _ := ret[0].(*response.ResetClient) + ret := m.ctrl.Call(m, "UpdateNodeAttributes", varargs...) + ret0, _ := ret[0].(*response.UpdateNodeAttributes) ret1, _ := ret[1].(error) return ret0, ret1 } -// ResetClientKey indicates an expected call of ResetClientKey. -func (mr *MockInfraProxyClientMockRecorder) ResetClientKey(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// UpdateNodeAttributes indicates an expected call of UpdateNodeAttributes +func (mr *MockInfraProxyClientMockRecorder) UpdateNodeAttributes(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ResetClientKey", reflect.TypeOf((*MockInfraProxyClient)(nil).ResetClientKey), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNodeAttributes", reflect.TypeOf((*MockInfraProxyClient)(nil).UpdateNodeAttributes), varargs...) } -// ResetOrgAdminKey mocks base method. -func (m *MockInfraProxyClient) ResetOrgAdminKey(ctx context.Context, in *request.ResetOrgAdminKey, opts ...grpc.CallOption) (*response.ResetOrgAdminKey, error) { +// GetNode mocks base method +func (m *MockInfraProxyClient) GetNode(ctx context.Context, in *request.Node, opts ...grpc.CallOption) (*response.Node, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "ResetOrgAdminKey", varargs...) - ret0, _ := ret[0].(*response.ResetOrgAdminKey) + ret := m.ctrl.Call(m, "GetNode", varargs...) + ret0, _ := ret[0].(*response.Node) ret1, _ := ret[1].(error) return ret0, ret1 } -// ResetOrgAdminKey indicates an expected call of ResetOrgAdminKey. -func (mr *MockInfraProxyClientMockRecorder) ResetOrgAdminKey(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetNode indicates an expected call of GetNode +func (mr *MockInfraProxyClientMockRecorder) GetNode(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ResetOrgAdminKey", reflect.TypeOf((*MockInfraProxyClient)(nil).ResetOrgAdminKey), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNode", reflect.TypeOf((*MockInfraProxyClient)(nil).GetNode), varargs...) } -// UpdateDataBagItem mocks base method. -func (m *MockInfraProxyClient) UpdateDataBagItem(ctx context.Context, in *request.UpdateDataBagItem, opts ...grpc.CallOption) (*response.UpdateDataBagItem, error) { +// GetNodeExpandedRunList mocks base method +func (m *MockInfraProxyClient) GetNodeExpandedRunList(ctx context.Context, in *request.NodeExpandedRunList, opts ...grpc.CallOption) (*response.NodeExpandedRunList, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "UpdateDataBagItem", varargs...) - ret0, _ := ret[0].(*response.UpdateDataBagItem) + ret := m.ctrl.Call(m, "GetNodeExpandedRunList", varargs...) + ret0, _ := ret[0].(*response.NodeExpandedRunList) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateDataBagItem indicates an expected call of UpdateDataBagItem. -func (mr *MockInfraProxyClientMockRecorder) UpdateDataBagItem(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetNodeExpandedRunList indicates an expected call of GetNodeExpandedRunList +func (mr *MockInfraProxyClientMockRecorder) GetNodeExpandedRunList(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateDataBagItem", reflect.TypeOf((*MockInfraProxyClient)(nil).UpdateDataBagItem), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodeExpandedRunList", reflect.TypeOf((*MockInfraProxyClient)(nil).GetNodeExpandedRunList), varargs...) } -// UpdateEnvironment mocks base method. -func (m *MockInfraProxyClient) UpdateEnvironment(ctx context.Context, in *request.UpdateEnvironment, opts ...grpc.CallOption) (*response.Environment, error) { +// UpdateNodeTags mocks base method +func (m *MockInfraProxyClient) UpdateNodeTags(ctx context.Context, in *request.UpdateNodeTags, opts ...grpc.CallOption) (*response.UpdateNodeTags, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "UpdateEnvironment", varargs...) - ret0, _ := ret[0].(*response.Environment) + ret := m.ctrl.Call(m, "UpdateNodeTags", varargs...) + ret0, _ := ret[0].(*response.UpdateNodeTags) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateEnvironment indicates an expected call of UpdateEnvironment. -func (mr *MockInfraProxyClientMockRecorder) UpdateEnvironment(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// UpdateNodeTags indicates an expected call of UpdateNodeTags +func (mr *MockInfraProxyClientMockRecorder) UpdateNodeTags(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateEnvironment", reflect.TypeOf((*MockInfraProxyClient)(nil).UpdateEnvironment), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNodeTags", reflect.TypeOf((*MockInfraProxyClient)(nil).UpdateNodeTags), varargs...) } -// UpdateNode mocks base method. -func (m *MockInfraProxyClient) UpdateNode(ctx context.Context, in *request.NodeDetails, opts ...grpc.CallOption) (*response.Node, error) { +// UpdateNodeEnvironment mocks base method +func (m *MockInfraProxyClient) UpdateNodeEnvironment(ctx context.Context, in *request.UpdateNodeEnvironment, opts ...grpc.CallOption) (*response.UpdateNodeEnvironment, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "UpdateNode", varargs...) - ret0, _ := ret[0].(*response.Node) + ret := m.ctrl.Call(m, "UpdateNodeEnvironment", varargs...) + ret0, _ := ret[0].(*response.UpdateNodeEnvironment) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateNode indicates an expected call of UpdateNode. -func (mr *MockInfraProxyClientMockRecorder) UpdateNode(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// UpdateNodeEnvironment indicates an expected call of UpdateNodeEnvironment +func (mr *MockInfraProxyClientMockRecorder) UpdateNodeEnvironment(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNode", reflect.TypeOf((*MockInfraProxyClient)(nil).UpdateNode), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNodeEnvironment", reflect.TypeOf((*MockInfraProxyClient)(nil).UpdateNodeEnvironment), varargs...) } -// UpdateNodeAttributes mocks base method. -func (m *MockInfraProxyClient) UpdateNodeAttributes(ctx context.Context, in *request.UpdateNodeAttributes, opts ...grpc.CallOption) (*response.UpdateNodeAttributes, error) { +// GetPolicyfiles mocks base method +func (m *MockInfraProxyClient) GetPolicyfiles(ctx context.Context, in *request.Policyfiles, opts ...grpc.CallOption) (*response.Policyfiles, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "UpdateNodeAttributes", varargs...) - ret0, _ := ret[0].(*response.UpdateNodeAttributes) + ret := m.ctrl.Call(m, "GetPolicyfiles", varargs...) + ret0, _ := ret[0].(*response.Policyfiles) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateNodeAttributes indicates an expected call of UpdateNodeAttributes. -func (mr *MockInfraProxyClientMockRecorder) UpdateNodeAttributes(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetPolicyfiles indicates an expected call of GetPolicyfiles +func (mr *MockInfraProxyClientMockRecorder) GetPolicyfiles(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNodeAttributes", reflect.TypeOf((*MockInfraProxyClient)(nil).UpdateNodeAttributes), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPolicyfiles", reflect.TypeOf((*MockInfraProxyClient)(nil).GetPolicyfiles), varargs...) } -// UpdateNodeEnvironment mocks base method. -func (m *MockInfraProxyClient) UpdateNodeEnvironment(ctx context.Context, in *request.UpdateNodeEnvironment, opts ...grpc.CallOption) (*response.UpdateNodeEnvironment, error) { +// GetPolicyfile mocks base method +func (m *MockInfraProxyClient) GetPolicyfile(ctx context.Context, in *request.Policyfile, opts ...grpc.CallOption) (*response.Policyfile, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "UpdateNodeEnvironment", varargs...) - ret0, _ := ret[0].(*response.UpdateNodeEnvironment) + ret := m.ctrl.Call(m, "GetPolicyfile", varargs...) + ret0, _ := ret[0].(*response.Policyfile) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateNodeEnvironment indicates an expected call of UpdateNodeEnvironment. -func (mr *MockInfraProxyClientMockRecorder) UpdateNodeEnvironment(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetPolicyfile indicates an expected call of GetPolicyfile +func (mr *MockInfraProxyClientMockRecorder) GetPolicyfile(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNodeEnvironment", reflect.TypeOf((*MockInfraProxyClient)(nil).UpdateNodeEnvironment), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPolicyfile", reflect.TypeOf((*MockInfraProxyClient)(nil).GetPolicyfile), varargs...) } -// UpdateNodeTags mocks base method. -func (m *MockInfraProxyClient) UpdateNodeTags(ctx context.Context, in *request.UpdateNodeTags, opts ...grpc.CallOption) (*response.UpdateNodeTags, error) { +// DeletePolicyfile mocks base method +func (m *MockInfraProxyClient) DeletePolicyfile(ctx context.Context, in *request.DeletePolicyfile, opts ...grpc.CallOption) (*response.DeletePolicyfile, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "UpdateNodeTags", varargs...) - ret0, _ := ret[0].(*response.UpdateNodeTags) + ret := m.ctrl.Call(m, "DeletePolicyfile", varargs...) + ret0, _ := ret[0].(*response.DeletePolicyfile) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateNodeTags indicates an expected call of UpdateNodeTags. -func (mr *MockInfraProxyClientMockRecorder) UpdateNodeTags(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// DeletePolicyfile indicates an expected call of DeletePolicyfile +func (mr *MockInfraProxyClientMockRecorder) DeletePolicyfile(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNodeTags", reflect.TypeOf((*MockInfraProxyClient)(nil).UpdateNodeTags), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeletePolicyfile", reflect.TypeOf((*MockInfraProxyClient)(nil).DeletePolicyfile), varargs...) } -// UpdateOrg mocks base method. -func (m *MockInfraProxyClient) UpdateOrg(ctx context.Context, in *request.UpdateOrg, opts ...grpc.CallOption) (*response.UpdateOrg, error) { +// GetPolicyfileRevisions mocks base method +func (m *MockInfraProxyClient) GetPolicyfileRevisions(ctx context.Context, in *request.PolicyfileRevisions, opts ...grpc.CallOption) (*response.PolicyfileRevisions, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "UpdateOrg", varargs...) - ret0, _ := ret[0].(*response.UpdateOrg) + ret := m.ctrl.Call(m, "GetPolicyfileRevisions", varargs...) + ret0, _ := ret[0].(*response.PolicyfileRevisions) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateOrg indicates an expected call of UpdateOrg. -func (mr *MockInfraProxyClientMockRecorder) UpdateOrg(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetPolicyfileRevisions indicates an expected call of GetPolicyfileRevisions +func (mr *MockInfraProxyClientMockRecorder) GetPolicyfileRevisions(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateOrg", reflect.TypeOf((*MockInfraProxyClient)(nil).UpdateOrg), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPolicyfileRevisions", reflect.TypeOf((*MockInfraProxyClient)(nil).GetPolicyfileRevisions), varargs...) } -// UpdateRole mocks base method. -func (m *MockInfraProxyClient) UpdateRole(ctx context.Context, in *request.UpdateRole, opts ...grpc.CallOption) (*response.Role, error) { +// GetPolicygroup mocks base method +func (m *MockInfraProxyClient) GetPolicygroup(ctx context.Context, in *request.Policygroup, opts ...grpc.CallOption) (*response.Policygroup, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "UpdateRole", varargs...) - ret0, _ := ret[0].(*response.Role) + ret := m.ctrl.Call(m, "GetPolicygroup", varargs...) + ret0, _ := ret[0].(*response.Policygroup) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateRole indicates an expected call of UpdateRole. -func (mr *MockInfraProxyClientMockRecorder) UpdateRole(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetPolicygroup indicates an expected call of GetPolicygroup +func (mr *MockInfraProxyClientMockRecorder) GetPolicygroup(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateRole", reflect.TypeOf((*MockInfraProxyClient)(nil).UpdateRole), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPolicygroup", reflect.TypeOf((*MockInfraProxyClient)(nil).GetPolicygroup), varargs...) } -// UpdateServer mocks base method. -func (m *MockInfraProxyClient) UpdateServer(ctx context.Context, in *request.UpdateServer, opts ...grpc.CallOption) (*response.UpdateServer, error) { +// GetOrgUsersList mocks base method +func (m *MockInfraProxyClient) GetOrgUsersList(ctx context.Context, in *request.OrgUsers, opts ...grpc.CallOption) (*response.OrgUsers, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "UpdateServer", varargs...) - ret0, _ := ret[0].(*response.UpdateServer) + ret := m.ctrl.Call(m, "GetOrgUsersList", varargs...) + ret0, _ := ret[0].(*response.OrgUsers) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateServer indicates an expected call of UpdateServer. -func (mr *MockInfraProxyClientMockRecorder) UpdateServer(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetOrgUsersList indicates an expected call of GetOrgUsersList +func (mr *MockInfraProxyClientMockRecorder) GetOrgUsersList(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateServer", reflect.TypeOf((*MockInfraProxyClient)(nil).UpdateServer), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrgUsersList", reflect.TypeOf((*MockInfraProxyClient)(nil).GetOrgUsersList), varargs...) } -// UpdateWebuiKey mocks base method. -func (m *MockInfraProxyClient) UpdateWebuiKey(ctx context.Context, in *request.UpdateWebuiKey, opts ...grpc.CallOption) (*response.UpdateWebuiKey, error) { +// GetServerUsersList mocks base method +func (m *MockInfraProxyClient) GetServerUsersList(ctx context.Context, in *request.ServerUsers, opts ...grpc.CallOption) (*response.ServerUsers, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "UpdateWebuiKey", varargs...) - ret0, _ := ret[0].(*response.UpdateWebuiKey) + ret := m.ctrl.Call(m, "GetServerUsersList", varargs...) + ret0, _ := ret[0].(*response.ServerUsers) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateWebuiKey indicates an expected call of UpdateWebuiKey. -func (mr *MockInfraProxyClientMockRecorder) UpdateWebuiKey(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetServerUsersList indicates an expected call of GetServerUsersList +func (mr *MockInfraProxyClientMockRecorder) GetServerUsersList(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWebuiKey", reflect.TypeOf((*MockInfraProxyClient)(nil).UpdateWebuiKey), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetServerUsersList", reflect.TypeOf((*MockInfraProxyClient)(nil).GetServerUsersList), varargs...) } -// ValidateWebuiKey mocks base method. -func (m *MockInfraProxyClient) ValidateWebuiKey(ctx context.Context, in *request.ValidateWebuiKey, opts ...grpc.CallOption) (*response.ValidateWebuiKey, error) { +// GetAutomateInfraServerUsersList mocks base method +func (m *MockInfraProxyClient) GetAutomateInfraServerUsersList(ctx context.Context, in *request.AutomateInfraServerUsers, opts ...grpc.CallOption) (*response.AutomateInfraServerUsers, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "ValidateWebuiKey", varargs...) - ret0, _ := ret[0].(*response.ValidateWebuiKey) + ret := m.ctrl.Call(m, "GetAutomateInfraServerUsersList", varargs...) + ret0, _ := ret[0].(*response.AutomateInfraServerUsers) ret1, _ := ret[1].(error) return ret0, ret1 } -// ValidateWebuiKey indicates an expected call of ValidateWebuiKey. -func (mr *MockInfraProxyClientMockRecorder) ValidateWebuiKey(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetAutomateInfraServerUsersList indicates an expected call of GetAutomateInfraServerUsersList +func (mr *MockInfraProxyClientMockRecorder) GetAutomateInfraServerUsersList(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ValidateWebuiKey", reflect.TypeOf((*MockInfraProxyClient)(nil).ValidateWebuiKey), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAutomateInfraServerUsersList", reflect.TypeOf((*MockInfraProxyClient)(nil).GetAutomateInfraServerUsersList), varargs...) } -// MockInfraProxyServer is a mock of InfraProxyServer interface. +// MockInfraProxyServer is a mock of InfraProxyServer interface type MockInfraProxyServer struct { ctrl *gomock.Controller recorder *MockInfraProxyServerMockRecorder } -// MockInfraProxyServerMockRecorder is the mock recorder for MockInfraProxyServer. +// MockInfraProxyServerMockRecorder is the mock recorder for MockInfraProxyServer type MockInfraProxyServerMockRecorder struct { mock *MockInfraProxyServer } -// NewMockInfraProxyServer creates a new mock instance. +// NewMockInfraProxyServer creates a new mock instance func NewMockInfraProxyServer(ctrl *gomock.Controller) *MockInfraProxyServer { mock := &MockInfraProxyServer{ctrl: ctrl} mock.recorder = &MockInfraProxyServerMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockInfraProxyServer) EXPECT() *MockInfraProxyServerMockRecorder { return m.recorder } -// CreateClient mocks base method. -func (m *MockInfraProxyServer) CreateClient(arg0 context.Context, arg1 *request.CreateClient) (*response.CreateClient, error) { +// GetServers mocks base method +func (m *MockInfraProxyServer) GetServers(arg0 context.Context, arg1 *request.GetServers) (*response.GetServers, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CreateClient", arg0, arg1) - ret0, _ := ret[0].(*response.CreateClient) + ret := m.ctrl.Call(m, "GetServers", arg0, arg1) + ret0, _ := ret[0].(*response.GetServers) ret1, _ := ret[1].(error) return ret0, ret1 } -// CreateClient indicates an expected call of CreateClient. -func (mr *MockInfraProxyServerMockRecorder) CreateClient(arg0, arg1 interface{}) *gomock.Call { +// GetServers indicates an expected call of GetServers +func (mr *MockInfraProxyServerMockRecorder) GetServers(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateClient", reflect.TypeOf((*MockInfraProxyServer)(nil).CreateClient), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetServers", reflect.TypeOf((*MockInfraProxyServer)(nil).GetServers), arg0, arg1) } -// CreateDataBag mocks base method. -func (m *MockInfraProxyServer) CreateDataBag(arg0 context.Context, arg1 *request.CreateDataBag) (*response.CreateDataBag, error) { +// GetServerStatus mocks base method +func (m *MockInfraProxyServer) GetServerStatus(arg0 context.Context, arg1 *request.GetServerStatus) (*response.GetServerStatus, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CreateDataBag", arg0, arg1) - ret0, _ := ret[0].(*response.CreateDataBag) + ret := m.ctrl.Call(m, "GetServerStatus", arg0, arg1) + ret0, _ := ret[0].(*response.GetServerStatus) ret1, _ := ret[1].(error) return ret0, ret1 } -// CreateDataBag indicates an expected call of CreateDataBag. -func (mr *MockInfraProxyServerMockRecorder) CreateDataBag(arg0, arg1 interface{}) *gomock.Call { +// GetServerStatus indicates an expected call of GetServerStatus +func (mr *MockInfraProxyServerMockRecorder) GetServerStatus(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateDataBag", reflect.TypeOf((*MockInfraProxyServer)(nil).CreateDataBag), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetServerStatus", reflect.TypeOf((*MockInfraProxyServer)(nil).GetServerStatus), arg0, arg1) } -// CreateDataBagItem mocks base method. -func (m *MockInfraProxyServer) CreateDataBagItem(arg0 context.Context, arg1 *request.CreateDataBagItem) (*response.CreateDataBagItem, error) { +// ValidateWebuiKey mocks base method +func (m *MockInfraProxyServer) ValidateWebuiKey(arg0 context.Context, arg1 *request.ValidateWebuiKey) (*response.ValidateWebuiKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CreateDataBagItem", arg0, arg1) - ret0, _ := ret[0].(*response.CreateDataBagItem) + ret := m.ctrl.Call(m, "ValidateWebuiKey", arg0, arg1) + ret0, _ := ret[0].(*response.ValidateWebuiKey) ret1, _ := ret[1].(error) return ret0, ret1 } -// CreateDataBagItem indicates an expected call of CreateDataBagItem. -func (mr *MockInfraProxyServerMockRecorder) CreateDataBagItem(arg0, arg1 interface{}) *gomock.Call { +// ValidateWebuiKey indicates an expected call of ValidateWebuiKey +func (mr *MockInfraProxyServerMockRecorder) ValidateWebuiKey(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateDataBagItem", reflect.TypeOf((*MockInfraProxyServer)(nil).CreateDataBagItem), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ValidateWebuiKey", reflect.TypeOf((*MockInfraProxyServer)(nil).ValidateWebuiKey), arg0, arg1) } -// CreateEnvironment mocks base method. -func (m *MockInfraProxyServer) CreateEnvironment(arg0 context.Context, arg1 *request.CreateEnvironment) (*response.Environment, error) { +// UpdateWebuiKey mocks base method +func (m *MockInfraProxyServer) UpdateWebuiKey(arg0 context.Context, arg1 *request.UpdateWebuiKey) (*response.UpdateWebuiKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CreateEnvironment", arg0, arg1) - ret0, _ := ret[0].(*response.Environment) + ret := m.ctrl.Call(m, "UpdateWebuiKey", arg0, arg1) + ret0, _ := ret[0].(*response.UpdateWebuiKey) ret1, _ := ret[1].(error) return ret0, ret1 } -// CreateEnvironment indicates an expected call of CreateEnvironment. -func (mr *MockInfraProxyServerMockRecorder) CreateEnvironment(arg0, arg1 interface{}) *gomock.Call { +// UpdateWebuiKey indicates an expected call of UpdateWebuiKey +func (mr *MockInfraProxyServerMockRecorder) UpdateWebuiKey(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateEnvironment", reflect.TypeOf((*MockInfraProxyServer)(nil).CreateEnvironment), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWebuiKey", reflect.TypeOf((*MockInfraProxyServer)(nil).UpdateWebuiKey), arg0, arg1) } -// CreateOrg mocks base method. -func (m *MockInfraProxyServer) CreateOrg(arg0 context.Context, arg1 *request.CreateOrg) (*response.CreateOrg, error) { +// GetServer mocks base method +func (m *MockInfraProxyServer) GetServer(arg0 context.Context, arg1 *request.GetServer) (*response.GetServer, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CreateOrg", arg0, arg1) - ret0, _ := ret[0].(*response.CreateOrg) + ret := m.ctrl.Call(m, "GetServer", arg0, arg1) + ret0, _ := ret[0].(*response.GetServer) ret1, _ := ret[1].(error) return ret0, ret1 } -// CreateOrg indicates an expected call of CreateOrg. -func (mr *MockInfraProxyServerMockRecorder) CreateOrg(arg0, arg1 interface{}) *gomock.Call { +// GetServer indicates an expected call of GetServer +func (mr *MockInfraProxyServerMockRecorder) GetServer(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateOrg", reflect.TypeOf((*MockInfraProxyServer)(nil).CreateOrg), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetServer", reflect.TypeOf((*MockInfraProxyServer)(nil).GetServer), arg0, arg1) } -// CreateRole mocks base method. -func (m *MockInfraProxyServer) CreateRole(arg0 context.Context, arg1 *request.CreateRole) (*response.Role, error) { +// CreateServer mocks base method +func (m *MockInfraProxyServer) CreateServer(arg0 context.Context, arg1 *request.CreateServer) (*response.CreateServer, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CreateRole", arg0, arg1) - ret0, _ := ret[0].(*response.Role) + ret := m.ctrl.Call(m, "CreateServer", arg0, arg1) + ret0, _ := ret[0].(*response.CreateServer) ret1, _ := ret[1].(error) return ret0, ret1 } -// CreateRole indicates an expected call of CreateRole. -func (mr *MockInfraProxyServerMockRecorder) CreateRole(arg0, arg1 interface{}) *gomock.Call { +// CreateServer indicates an expected call of CreateServer +func (mr *MockInfraProxyServerMockRecorder) CreateServer(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateRole", reflect.TypeOf((*MockInfraProxyServer)(nil).CreateRole), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateServer", reflect.TypeOf((*MockInfraProxyServer)(nil).CreateServer), arg0, arg1) } -// CreateServer mocks base method. -func (m *MockInfraProxyServer) CreateServer(arg0 context.Context, arg1 *request.CreateServer) (*response.CreateServer, error) { +// UpdateServer mocks base method +func (m *MockInfraProxyServer) UpdateServer(arg0 context.Context, arg1 *request.UpdateServer) (*response.UpdateServer, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CreateServer", arg0, arg1) - ret0, _ := ret[0].(*response.CreateServer) + ret := m.ctrl.Call(m, "UpdateServer", arg0, arg1) + ret0, _ := ret[0].(*response.UpdateServer) ret1, _ := ret[1].(error) return ret0, ret1 } -// CreateServer indicates an expected call of CreateServer. -func (mr *MockInfraProxyServerMockRecorder) CreateServer(arg0, arg1 interface{}) *gomock.Call { +// UpdateServer indicates an expected call of UpdateServer +func (mr *MockInfraProxyServerMockRecorder) UpdateServer(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateServer", reflect.TypeOf((*MockInfraProxyServer)(nil).CreateServer), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateServer", reflect.TypeOf((*MockInfraProxyServer)(nil).UpdateServer), arg0, arg1) } -// DeleteClient mocks base method. -func (m *MockInfraProxyServer) DeleteClient(arg0 context.Context, arg1 *request.Client) (*response.Client, error) { +// DeleteServer mocks base method +func (m *MockInfraProxyServer) DeleteServer(arg0 context.Context, arg1 *request.DeleteServer) (*response.DeleteServer, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteClient", arg0, arg1) - ret0, _ := ret[0].(*response.Client) + ret := m.ctrl.Call(m, "DeleteServer", arg0, arg1) + ret0, _ := ret[0].(*response.DeleteServer) ret1, _ := ret[1].(error) return ret0, ret1 } -// DeleteClient indicates an expected call of DeleteClient. -func (mr *MockInfraProxyServerMockRecorder) DeleteClient(arg0, arg1 interface{}) *gomock.Call { +// DeleteServer indicates an expected call of DeleteServer +func (mr *MockInfraProxyServerMockRecorder) DeleteServer(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteClient", reflect.TypeOf((*MockInfraProxyServer)(nil).DeleteClient), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteServer", reflect.TypeOf((*MockInfraProxyServer)(nil).DeleteServer), arg0, arg1) } -// DeleteDataBag mocks base method. -func (m *MockInfraProxyServer) DeleteDataBag(arg0 context.Context, arg1 *request.DataBag) (*response.DataBag, error) { +// GetOrgs mocks base method +func (m *MockInfraProxyServer) GetOrgs(arg0 context.Context, arg1 *request.GetOrgs) (*response.GetOrgs, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteDataBag", arg0, arg1) - ret0, _ := ret[0].(*response.DataBag) + ret := m.ctrl.Call(m, "GetOrgs", arg0, arg1) + ret0, _ := ret[0].(*response.GetOrgs) ret1, _ := ret[1].(error) return ret0, ret1 } -// DeleteDataBag indicates an expected call of DeleteDataBag. -func (mr *MockInfraProxyServerMockRecorder) DeleteDataBag(arg0, arg1 interface{}) *gomock.Call { +// GetOrgs indicates an expected call of GetOrgs +func (mr *MockInfraProxyServerMockRecorder) GetOrgs(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteDataBag", reflect.TypeOf((*MockInfraProxyServer)(nil).DeleteDataBag), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrgs", reflect.TypeOf((*MockInfraProxyServer)(nil).GetOrgs), arg0, arg1) } -// DeleteDataBagItem mocks base method. -func (m *MockInfraProxyServer) DeleteDataBagItem(arg0 context.Context, arg1 *request.DataBagItem) (*response.DataBagItem, error) { +// GetOrg mocks base method +func (m *MockInfraProxyServer) GetOrg(arg0 context.Context, arg1 *request.GetOrg) (*response.GetOrg, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteDataBagItem", arg0, arg1) - ret0, _ := ret[0].(*response.DataBagItem) + ret := m.ctrl.Call(m, "GetOrg", arg0, arg1) + ret0, _ := ret[0].(*response.GetOrg) ret1, _ := ret[1].(error) return ret0, ret1 } -// DeleteDataBagItem indicates an expected call of DeleteDataBagItem. -func (mr *MockInfraProxyServerMockRecorder) DeleteDataBagItem(arg0, arg1 interface{}) *gomock.Call { +// GetOrg indicates an expected call of GetOrg +func (mr *MockInfraProxyServerMockRecorder) GetOrg(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteDataBagItem", reflect.TypeOf((*MockInfraProxyServer)(nil).DeleteDataBagItem), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrg", reflect.TypeOf((*MockInfraProxyServer)(nil).GetOrg), arg0, arg1) } -// DeleteEnvironment mocks base method. -func (m *MockInfraProxyServer) DeleteEnvironment(arg0 context.Context, arg1 *request.Environment) (*response.Environment, error) { +// CreateOrg mocks base method +func (m *MockInfraProxyServer) CreateOrg(arg0 context.Context, arg1 *request.CreateOrg) (*response.CreateOrg, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteEnvironment", arg0, arg1) - ret0, _ := ret[0].(*response.Environment) + ret := m.ctrl.Call(m, "CreateOrg", arg0, arg1) + ret0, _ := ret[0].(*response.CreateOrg) ret1, _ := ret[1].(error) return ret0, ret1 } -// DeleteEnvironment indicates an expected call of DeleteEnvironment. -func (mr *MockInfraProxyServerMockRecorder) DeleteEnvironment(arg0, arg1 interface{}) *gomock.Call { +// CreateOrg indicates an expected call of CreateOrg +func (mr *MockInfraProxyServerMockRecorder) CreateOrg(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteEnvironment", reflect.TypeOf((*MockInfraProxyServer)(nil).DeleteEnvironment), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateOrg", reflect.TypeOf((*MockInfraProxyServer)(nil).CreateOrg), arg0, arg1) } -// DeleteNode mocks base method. -func (m *MockInfraProxyServer) DeleteNode(arg0 context.Context, arg1 *request.Node) (*response.DeleteNode, error) { +// UpdateOrg mocks base method +func (m *MockInfraProxyServer) UpdateOrg(arg0 context.Context, arg1 *request.UpdateOrg) (*response.UpdateOrg, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteNode", arg0, arg1) - ret0, _ := ret[0].(*response.DeleteNode) + ret := m.ctrl.Call(m, "UpdateOrg", arg0, arg1) + ret0, _ := ret[0].(*response.UpdateOrg) ret1, _ := ret[1].(error) return ret0, ret1 } -// DeleteNode indicates an expected call of DeleteNode. -func (mr *MockInfraProxyServerMockRecorder) DeleteNode(arg0, arg1 interface{}) *gomock.Call { +// UpdateOrg indicates an expected call of UpdateOrg +func (mr *MockInfraProxyServerMockRecorder) UpdateOrg(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteNode", reflect.TypeOf((*MockInfraProxyServer)(nil).DeleteNode), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateOrg", reflect.TypeOf((*MockInfraProxyServer)(nil).UpdateOrg), arg0, arg1) } -// DeleteOrg mocks base method. +// DeleteOrg mocks base method func (m *MockInfraProxyServer) DeleteOrg(arg0 context.Context, arg1 *request.DeleteOrg) (*response.DeleteOrg, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "DeleteOrg", arg0, arg1) @@ -1469,728 +1468,728 @@ func (m *MockInfraProxyServer) DeleteOrg(arg0 context.Context, arg1 *request.Del return ret0, ret1 } -// DeleteOrg indicates an expected call of DeleteOrg. +// DeleteOrg indicates an expected call of DeleteOrg func (mr *MockInfraProxyServerMockRecorder) DeleteOrg(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOrg", reflect.TypeOf((*MockInfraProxyServer)(nil).DeleteOrg), arg0, arg1) } -// DeletePolicyfile mocks base method. -func (m *MockInfraProxyServer) DeletePolicyfile(arg0 context.Context, arg1 *request.DeletePolicyfile) (*response.DeletePolicyfile, error) { +// ResetOrgAdminKey mocks base method +func (m *MockInfraProxyServer) ResetOrgAdminKey(arg0 context.Context, arg1 *request.ResetOrgAdminKey) (*response.ResetOrgAdminKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeletePolicyfile", arg0, arg1) - ret0, _ := ret[0].(*response.DeletePolicyfile) + ret := m.ctrl.Call(m, "ResetOrgAdminKey", arg0, arg1) + ret0, _ := ret[0].(*response.ResetOrgAdminKey) ret1, _ := ret[1].(error) return ret0, ret1 } -// DeletePolicyfile indicates an expected call of DeletePolicyfile. -func (mr *MockInfraProxyServerMockRecorder) DeletePolicyfile(arg0, arg1 interface{}) *gomock.Call { +// ResetOrgAdminKey indicates an expected call of ResetOrgAdminKey +func (mr *MockInfraProxyServerMockRecorder) ResetOrgAdminKey(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeletePolicyfile", reflect.TypeOf((*MockInfraProxyServer)(nil).DeletePolicyfile), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ResetOrgAdminKey", reflect.TypeOf((*MockInfraProxyServer)(nil).ResetOrgAdminKey), arg0, arg1) } -// DeleteRole mocks base method. -func (m *MockInfraProxyServer) DeleteRole(arg0 context.Context, arg1 *request.Role) (*response.Role, error) { +// GetInfraServerOrgs mocks base method +func (m *MockInfraProxyServer) GetInfraServerOrgs(arg0 context.Context, arg1 *request.GetInfraServerOrgs) (*response.GetInfraServerOrgs, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteRole", arg0, arg1) - ret0, _ := ret[0].(*response.Role) + ret := m.ctrl.Call(m, "GetInfraServerOrgs", arg0, arg1) + ret0, _ := ret[0].(*response.GetInfraServerOrgs) ret1, _ := ret[1].(error) return ret0, ret1 } -// DeleteRole indicates an expected call of DeleteRole. -func (mr *MockInfraProxyServerMockRecorder) DeleteRole(arg0, arg1 interface{}) *gomock.Call { +// GetInfraServerOrgs indicates an expected call of GetInfraServerOrgs +func (mr *MockInfraProxyServerMockRecorder) GetInfraServerOrgs(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteRole", reflect.TypeOf((*MockInfraProxyServer)(nil).DeleteRole), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetInfraServerOrgs", reflect.TypeOf((*MockInfraProxyServer)(nil).GetInfraServerOrgs), arg0, arg1) } -// DeleteServer mocks base method. -func (m *MockInfraProxyServer) DeleteServer(arg0 context.Context, arg1 *request.DeleteServer) (*response.DeleteServer, error) { +// GetCookbooks mocks base method +func (m *MockInfraProxyServer) GetCookbooks(arg0 context.Context, arg1 *request.Cookbooks) (*response.Cookbooks, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteServer", arg0, arg1) - ret0, _ := ret[0].(*response.DeleteServer) + ret := m.ctrl.Call(m, "GetCookbooks", arg0, arg1) + ret0, _ := ret[0].(*response.Cookbooks) ret1, _ := ret[1].(error) return ret0, ret1 } -// DeleteServer indicates an expected call of DeleteServer. -func (mr *MockInfraProxyServerMockRecorder) DeleteServer(arg0, arg1 interface{}) *gomock.Call { +// GetCookbooks indicates an expected call of GetCookbooks +func (mr *MockInfraProxyServerMockRecorder) GetCookbooks(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteServer", reflect.TypeOf((*MockInfraProxyServer)(nil).DeleteServer), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCookbooks", reflect.TypeOf((*MockInfraProxyServer)(nil).GetCookbooks), arg0, arg1) } -// GetAutomateInfraServerUsersList mocks base method. -func (m *MockInfraProxyServer) GetAutomateInfraServerUsersList(arg0 context.Context, arg1 *request.AutomateInfraServerUsers) (*response.AutomateInfraServerUsers, error) { +// GetCookbookVersions mocks base method +func (m *MockInfraProxyServer) GetCookbookVersions(arg0 context.Context, arg1 *request.CookbookVersions) (*response.CookbookVersions, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAutomateInfraServerUsersList", arg0, arg1) - ret0, _ := ret[0].(*response.AutomateInfraServerUsers) + ret := m.ctrl.Call(m, "GetCookbookVersions", arg0, arg1) + ret0, _ := ret[0].(*response.CookbookVersions) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetAutomateInfraServerUsersList indicates an expected call of GetAutomateInfraServerUsersList. -func (mr *MockInfraProxyServerMockRecorder) GetAutomateInfraServerUsersList(arg0, arg1 interface{}) *gomock.Call { +// GetCookbookVersions indicates an expected call of GetCookbookVersions +func (mr *MockInfraProxyServerMockRecorder) GetCookbookVersions(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAutomateInfraServerUsersList", reflect.TypeOf((*MockInfraProxyServer)(nil).GetAutomateInfraServerUsersList), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCookbookVersions", reflect.TypeOf((*MockInfraProxyServer)(nil).GetCookbookVersions), arg0, arg1) } -// GetClient mocks base method. -func (m *MockInfraProxyServer) GetClient(arg0 context.Context, arg1 *request.Client) (*response.Client, error) { +// GetCookbook mocks base method +func (m *MockInfraProxyServer) GetCookbook(arg0 context.Context, arg1 *request.Cookbook) (*response.Cookbook, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetClient", arg0, arg1) - ret0, _ := ret[0].(*response.Client) - ret1, _ := ret[1].(error) + ret := m.ctrl.Call(m, "GetCookbook", arg0, arg1) + ret0, _ := ret[0].(*response.Cookbook) + ret1, _ := ret[1].(error) return ret0, ret1 } -// GetClient indicates an expected call of GetClient. -func (mr *MockInfraProxyServerMockRecorder) GetClient(arg0, arg1 interface{}) *gomock.Call { +// GetCookbook indicates an expected call of GetCookbook +func (mr *MockInfraProxyServerMockRecorder) GetCookbook(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetClient", reflect.TypeOf((*MockInfraProxyServer)(nil).GetClient), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCookbook", reflect.TypeOf((*MockInfraProxyServer)(nil).GetCookbook), arg0, arg1) } -// GetClients mocks base method. -func (m *MockInfraProxyServer) GetClients(arg0 context.Context, arg1 *request.Clients) (*response.Clients, error) { +// GetCookbookFileContent mocks base method +func (m *MockInfraProxyServer) GetCookbookFileContent(arg0 context.Context, arg1 *request.CookbookFileContent) (*response.CookbookFileContent, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetClients", arg0, arg1) - ret0, _ := ret[0].(*response.Clients) + ret := m.ctrl.Call(m, "GetCookbookFileContent", arg0, arg1) + ret0, _ := ret[0].(*response.CookbookFileContent) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetClients indicates an expected call of GetClients. -func (mr *MockInfraProxyServerMockRecorder) GetClients(arg0, arg1 interface{}) *gomock.Call { +// GetCookbookFileContent indicates an expected call of GetCookbookFileContent +func (mr *MockInfraProxyServerMockRecorder) GetCookbookFileContent(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetClients", reflect.TypeOf((*MockInfraProxyServer)(nil).GetClients), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCookbookFileContent", reflect.TypeOf((*MockInfraProxyServer)(nil).GetCookbookFileContent), arg0, arg1) } -// GetCookbook mocks base method. -func (m *MockInfraProxyServer) GetCookbook(arg0 context.Context, arg1 *request.Cookbook) (*response.Cookbook, error) { +// GetRoles mocks base method +func (m *MockInfraProxyServer) GetRoles(arg0 context.Context, arg1 *request.Roles) (*response.Roles, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetCookbook", arg0, arg1) - ret0, _ := ret[0].(*response.Cookbook) + ret := m.ctrl.Call(m, "GetRoles", arg0, arg1) + ret0, _ := ret[0].(*response.Roles) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetCookbook indicates an expected call of GetCookbook. -func (mr *MockInfraProxyServerMockRecorder) GetCookbook(arg0, arg1 interface{}) *gomock.Call { +// GetRoles indicates an expected call of GetRoles +func (mr *MockInfraProxyServerMockRecorder) GetRoles(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCookbook", reflect.TypeOf((*MockInfraProxyServer)(nil).GetCookbook), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRoles", reflect.TypeOf((*MockInfraProxyServer)(nil).GetRoles), arg0, arg1) } -// GetCookbookFileContent mocks base method. -func (m *MockInfraProxyServer) GetCookbookFileContent(arg0 context.Context, arg1 *request.CookbookFileContent) (*response.CookbookFileContent, error) { +// GetRole mocks base method +func (m *MockInfraProxyServer) GetRole(arg0 context.Context, arg1 *request.Role) (*response.Role, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetCookbookFileContent", arg0, arg1) - ret0, _ := ret[0].(*response.CookbookFileContent) + ret := m.ctrl.Call(m, "GetRole", arg0, arg1) + ret0, _ := ret[0].(*response.Role) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetCookbookFileContent indicates an expected call of GetCookbookFileContent. -func (mr *MockInfraProxyServerMockRecorder) GetCookbookFileContent(arg0, arg1 interface{}) *gomock.Call { +// GetRole indicates an expected call of GetRole +func (mr *MockInfraProxyServerMockRecorder) GetRole(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCookbookFileContent", reflect.TypeOf((*MockInfraProxyServer)(nil).GetCookbookFileContent), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRole", reflect.TypeOf((*MockInfraProxyServer)(nil).GetRole), arg0, arg1) } -// GetCookbookVersions mocks base method. -func (m *MockInfraProxyServer) GetCookbookVersions(arg0 context.Context, arg1 *request.CookbookVersions) (*response.CookbookVersions, error) { +// GetRoleExpandedRunList mocks base method +func (m *MockInfraProxyServer) GetRoleExpandedRunList(arg0 context.Context, arg1 *request.ExpandedRunList) (*response.ExpandedRunList, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetCookbookVersions", arg0, arg1) - ret0, _ := ret[0].(*response.CookbookVersions) + ret := m.ctrl.Call(m, "GetRoleExpandedRunList", arg0, arg1) + ret0, _ := ret[0].(*response.ExpandedRunList) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetCookbookVersions indicates an expected call of GetCookbookVersions. -func (mr *MockInfraProxyServerMockRecorder) GetCookbookVersions(arg0, arg1 interface{}) *gomock.Call { +// GetRoleExpandedRunList indicates an expected call of GetRoleExpandedRunList +func (mr *MockInfraProxyServerMockRecorder) GetRoleExpandedRunList(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCookbookVersions", reflect.TypeOf((*MockInfraProxyServer)(nil).GetCookbookVersions), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRoleExpandedRunList", reflect.TypeOf((*MockInfraProxyServer)(nil).GetRoleExpandedRunList), arg0, arg1) } -// GetCookbooks mocks base method. -func (m *MockInfraProxyServer) GetCookbooks(arg0 context.Context, arg1 *request.Cookbooks) (*response.Cookbooks, error) { +// GetRoleEnvironments mocks base method +func (m *MockInfraProxyServer) GetRoleEnvironments(arg0 context.Context, arg1 *request.Role) (*response.RoleEnvironments, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetCookbooks", arg0, arg1) - ret0, _ := ret[0].(*response.Cookbooks) + ret := m.ctrl.Call(m, "GetRoleEnvironments", arg0, arg1) + ret0, _ := ret[0].(*response.RoleEnvironments) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetCookbooks indicates an expected call of GetCookbooks. -func (mr *MockInfraProxyServerMockRecorder) GetCookbooks(arg0, arg1 interface{}) *gomock.Call { +// GetRoleEnvironments indicates an expected call of GetRoleEnvironments +func (mr *MockInfraProxyServerMockRecorder) GetRoleEnvironments(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCookbooks", reflect.TypeOf((*MockInfraProxyServer)(nil).GetCookbooks), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRoleEnvironments", reflect.TypeOf((*MockInfraProxyServer)(nil).GetRoleEnvironments), arg0, arg1) } -// GetDataBagItem mocks base method. -func (m *MockInfraProxyServer) GetDataBagItem(arg0 context.Context, arg1 *request.DataBagItem) (*response.DataBagItem, error) { +// CreateRole mocks base method +func (m *MockInfraProxyServer) CreateRole(arg0 context.Context, arg1 *request.CreateRole) (*response.Role, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetDataBagItem", arg0, arg1) - ret0, _ := ret[0].(*response.DataBagItem) + ret := m.ctrl.Call(m, "CreateRole", arg0, arg1) + ret0, _ := ret[0].(*response.Role) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetDataBagItem indicates an expected call of GetDataBagItem. -func (mr *MockInfraProxyServerMockRecorder) GetDataBagItem(arg0, arg1 interface{}) *gomock.Call { +// CreateRole indicates an expected call of CreateRole +func (mr *MockInfraProxyServerMockRecorder) CreateRole(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDataBagItem", reflect.TypeOf((*MockInfraProxyServer)(nil).GetDataBagItem), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateRole", reflect.TypeOf((*MockInfraProxyServer)(nil).CreateRole), arg0, arg1) } -// GetDataBagItems mocks base method. -func (m *MockInfraProxyServer) GetDataBagItems(arg0 context.Context, arg1 *request.DataBagItems) (*response.DataBagItems, error) { +// DeleteRole mocks base method +func (m *MockInfraProxyServer) DeleteRole(arg0 context.Context, arg1 *request.Role) (*response.Role, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetDataBagItems", arg0, arg1) - ret0, _ := ret[0].(*response.DataBagItems) + ret := m.ctrl.Call(m, "DeleteRole", arg0, arg1) + ret0, _ := ret[0].(*response.Role) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetDataBagItems indicates an expected call of GetDataBagItems. -func (mr *MockInfraProxyServerMockRecorder) GetDataBagItems(arg0, arg1 interface{}) *gomock.Call { +// DeleteRole indicates an expected call of DeleteRole +func (mr *MockInfraProxyServerMockRecorder) DeleteRole(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDataBagItems", reflect.TypeOf((*MockInfraProxyServer)(nil).GetDataBagItems), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteRole", reflect.TypeOf((*MockInfraProxyServer)(nil).DeleteRole), arg0, arg1) } -// GetDataBags mocks base method. -func (m *MockInfraProxyServer) GetDataBags(arg0 context.Context, arg1 *request.DataBags) (*response.DataBags, error) { +// UpdateRole mocks base method +func (m *MockInfraProxyServer) UpdateRole(arg0 context.Context, arg1 *request.UpdateRole) (*response.Role, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetDataBags", arg0, arg1) - ret0, _ := ret[0].(*response.DataBags) + ret := m.ctrl.Call(m, "UpdateRole", arg0, arg1) + ret0, _ := ret[0].(*response.Role) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetDataBags indicates an expected call of GetDataBags. -func (mr *MockInfraProxyServerMockRecorder) GetDataBags(arg0, arg1 interface{}) *gomock.Call { +// UpdateRole indicates an expected call of UpdateRole +func (mr *MockInfraProxyServerMockRecorder) UpdateRole(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDataBags", reflect.TypeOf((*MockInfraProxyServer)(nil).GetDataBags), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateRole", reflect.TypeOf((*MockInfraProxyServer)(nil).UpdateRole), arg0, arg1) } -// GetEnvironment mocks base method. -func (m *MockInfraProxyServer) GetEnvironment(arg0 context.Context, arg1 *request.Environment) (*response.Environment, error) { +// GetClients mocks base method +func (m *MockInfraProxyServer) GetClients(arg0 context.Context, arg1 *request.Clients) (*response.Clients, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetEnvironment", arg0, arg1) - ret0, _ := ret[0].(*response.Environment) + ret := m.ctrl.Call(m, "GetClients", arg0, arg1) + ret0, _ := ret[0].(*response.Clients) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetEnvironment indicates an expected call of GetEnvironment. -func (mr *MockInfraProxyServerMockRecorder) GetEnvironment(arg0, arg1 interface{}) *gomock.Call { +// GetClients indicates an expected call of GetClients +func (mr *MockInfraProxyServerMockRecorder) GetClients(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetEnvironment", reflect.TypeOf((*MockInfraProxyServer)(nil).GetEnvironment), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetClients", reflect.TypeOf((*MockInfraProxyServer)(nil).GetClients), arg0, arg1) } -// GetEnvironmentRecipes mocks base method. -func (m *MockInfraProxyServer) GetEnvironmentRecipes(arg0 context.Context, arg1 *request.Environment) (*response.EnvironmentRecipesList, error) { +// GetClient mocks base method +func (m *MockInfraProxyServer) GetClient(arg0 context.Context, arg1 *request.Client) (*response.Client, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetEnvironmentRecipes", arg0, arg1) - ret0, _ := ret[0].(*response.EnvironmentRecipesList) + ret := m.ctrl.Call(m, "GetClient", arg0, arg1) + ret0, _ := ret[0].(*response.Client) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetEnvironmentRecipes indicates an expected call of GetEnvironmentRecipes. -func (mr *MockInfraProxyServerMockRecorder) GetEnvironmentRecipes(arg0, arg1 interface{}) *gomock.Call { +// GetClient indicates an expected call of GetClient +func (mr *MockInfraProxyServerMockRecorder) GetClient(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetEnvironmentRecipes", reflect.TypeOf((*MockInfraProxyServer)(nil).GetEnvironmentRecipes), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetClient", reflect.TypeOf((*MockInfraProxyServer)(nil).GetClient), arg0, arg1) } -// GetEnvironments mocks base method. -func (m *MockInfraProxyServer) GetEnvironments(arg0 context.Context, arg1 *request.Environments) (*response.Environments, error) { +// CreateClient mocks base method +func (m *MockInfraProxyServer) CreateClient(arg0 context.Context, arg1 *request.CreateClient) (*response.CreateClient, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetEnvironments", arg0, arg1) - ret0, _ := ret[0].(*response.Environments) + ret := m.ctrl.Call(m, "CreateClient", arg0, arg1) + ret0, _ := ret[0].(*response.CreateClient) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetEnvironments indicates an expected call of GetEnvironments. -func (mr *MockInfraProxyServerMockRecorder) GetEnvironments(arg0, arg1 interface{}) *gomock.Call { +// CreateClient indicates an expected call of CreateClient +func (mr *MockInfraProxyServerMockRecorder) CreateClient(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetEnvironments", reflect.TypeOf((*MockInfraProxyServer)(nil).GetEnvironments), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateClient", reflect.TypeOf((*MockInfraProxyServer)(nil).CreateClient), arg0, arg1) } -// GetInfraServerOrgs mocks base method. -func (m *MockInfraProxyServer) GetInfraServerOrgs(arg0 context.Context, arg1 *request.GetInfraServerOrgs) (*response.GetInfraServerOrgs, error) { +// DeleteClient mocks base method +func (m *MockInfraProxyServer) DeleteClient(arg0 context.Context, arg1 *request.Client) (*response.Client, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetInfraServerOrgs", arg0, arg1) - ret0, _ := ret[0].(*response.GetInfraServerOrgs) + ret := m.ctrl.Call(m, "DeleteClient", arg0, arg1) + ret0, _ := ret[0].(*response.Client) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetInfraServerOrgs indicates an expected call of GetInfraServerOrgs. -func (mr *MockInfraProxyServerMockRecorder) GetInfraServerOrgs(arg0, arg1 interface{}) *gomock.Call { +// DeleteClient indicates an expected call of DeleteClient +func (mr *MockInfraProxyServerMockRecorder) DeleteClient(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetInfraServerOrgs", reflect.TypeOf((*MockInfraProxyServer)(nil).GetInfraServerOrgs), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteClient", reflect.TypeOf((*MockInfraProxyServer)(nil).DeleteClient), arg0, arg1) } -// GetNode mocks base method. -func (m *MockInfraProxyServer) GetNode(arg0 context.Context, arg1 *request.Node) (*response.Node, error) { +// ResetClientKey mocks base method +func (m *MockInfraProxyServer) ResetClientKey(arg0 context.Context, arg1 *request.ClientKey) (*response.ResetClient, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetNode", arg0, arg1) - ret0, _ := ret[0].(*response.Node) + ret := m.ctrl.Call(m, "ResetClientKey", arg0, arg1) + ret0, _ := ret[0].(*response.ResetClient) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetNode indicates an expected call of GetNode. -func (mr *MockInfraProxyServerMockRecorder) GetNode(arg0, arg1 interface{}) *gomock.Call { +// ResetClientKey indicates an expected call of ResetClientKey +func (mr *MockInfraProxyServerMockRecorder) ResetClientKey(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNode", reflect.TypeOf((*MockInfraProxyServer)(nil).GetNode), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ResetClientKey", reflect.TypeOf((*MockInfraProxyServer)(nil).ResetClientKey), arg0, arg1) } -// GetNodeExpandedRunList mocks base method. -func (m *MockInfraProxyServer) GetNodeExpandedRunList(arg0 context.Context, arg1 *request.NodeExpandedRunList) (*response.NodeExpandedRunList, error) { +// GetDataBags mocks base method +func (m *MockInfraProxyServer) GetDataBags(arg0 context.Context, arg1 *request.DataBags) (*response.DataBags, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetNodeExpandedRunList", arg0, arg1) - ret0, _ := ret[0].(*response.NodeExpandedRunList) + ret := m.ctrl.Call(m, "GetDataBags", arg0, arg1) + ret0, _ := ret[0].(*response.DataBags) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetNodeExpandedRunList indicates an expected call of GetNodeExpandedRunList. -func (mr *MockInfraProxyServerMockRecorder) GetNodeExpandedRunList(arg0, arg1 interface{}) *gomock.Call { +// GetDataBags indicates an expected call of GetDataBags +func (mr *MockInfraProxyServerMockRecorder) GetDataBags(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodeExpandedRunList", reflect.TypeOf((*MockInfraProxyServer)(nil).GetNodeExpandedRunList), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDataBags", reflect.TypeOf((*MockInfraProxyServer)(nil).GetDataBags), arg0, arg1) } -// GetNodes mocks base method. -func (m *MockInfraProxyServer) GetNodes(arg0 context.Context, arg1 *request.Nodes) (*response.Nodes, error) { +// GetDataBagItems mocks base method +func (m *MockInfraProxyServer) GetDataBagItems(arg0 context.Context, arg1 *request.DataBagItems) (*response.DataBagItems, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetNodes", arg0, arg1) - ret0, _ := ret[0].(*response.Nodes) + ret := m.ctrl.Call(m, "GetDataBagItems", arg0, arg1) + ret0, _ := ret[0].(*response.DataBagItems) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetNodes indicates an expected call of GetNodes. -func (mr *MockInfraProxyServerMockRecorder) GetNodes(arg0, arg1 interface{}) *gomock.Call { +// GetDataBagItems indicates an expected call of GetDataBagItems +func (mr *MockInfraProxyServerMockRecorder) GetDataBagItems(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodes", reflect.TypeOf((*MockInfraProxyServer)(nil).GetNodes), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDataBagItems", reflect.TypeOf((*MockInfraProxyServer)(nil).GetDataBagItems), arg0, arg1) } -// GetOrg mocks base method. -func (m *MockInfraProxyServer) GetOrg(arg0 context.Context, arg1 *request.GetOrg) (*response.GetOrg, error) { +// GetDataBagItem mocks base method +func (m *MockInfraProxyServer) GetDataBagItem(arg0 context.Context, arg1 *request.DataBagItem) (*response.DataBagItem, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetOrg", arg0, arg1) - ret0, _ := ret[0].(*response.GetOrg) + ret := m.ctrl.Call(m, "GetDataBagItem", arg0, arg1) + ret0, _ := ret[0].(*response.DataBagItem) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetOrg indicates an expected call of GetOrg. -func (mr *MockInfraProxyServerMockRecorder) GetOrg(arg0, arg1 interface{}) *gomock.Call { +// GetDataBagItem indicates an expected call of GetDataBagItem +func (mr *MockInfraProxyServerMockRecorder) GetDataBagItem(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrg", reflect.TypeOf((*MockInfraProxyServer)(nil).GetOrg), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDataBagItem", reflect.TypeOf((*MockInfraProxyServer)(nil).GetDataBagItem), arg0, arg1) } -// GetOrgUsersList mocks base method. -func (m *MockInfraProxyServer) GetOrgUsersList(arg0 context.Context, arg1 *request.OrgUsers) (*response.OrgUsers, error) { +// CreateDataBag mocks base method +func (m *MockInfraProxyServer) CreateDataBag(arg0 context.Context, arg1 *request.CreateDataBag) (*response.CreateDataBag, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetOrgUsersList", arg0, arg1) - ret0, _ := ret[0].(*response.OrgUsers) + ret := m.ctrl.Call(m, "CreateDataBag", arg0, arg1) + ret0, _ := ret[0].(*response.CreateDataBag) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetOrgUsersList indicates an expected call of GetOrgUsersList. -func (mr *MockInfraProxyServerMockRecorder) GetOrgUsersList(arg0, arg1 interface{}) *gomock.Call { +// CreateDataBag indicates an expected call of CreateDataBag +func (mr *MockInfraProxyServerMockRecorder) CreateDataBag(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrgUsersList", reflect.TypeOf((*MockInfraProxyServer)(nil).GetOrgUsersList), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateDataBag", reflect.TypeOf((*MockInfraProxyServer)(nil).CreateDataBag), arg0, arg1) } -// GetOrgs mocks base method. -func (m *MockInfraProxyServer) GetOrgs(arg0 context.Context, arg1 *request.GetOrgs) (*response.GetOrgs, error) { +// CreateDataBagItem mocks base method +func (m *MockInfraProxyServer) CreateDataBagItem(arg0 context.Context, arg1 *request.CreateDataBagItem) (*response.CreateDataBagItem, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetOrgs", arg0, arg1) - ret0, _ := ret[0].(*response.GetOrgs) + ret := m.ctrl.Call(m, "CreateDataBagItem", arg0, arg1) + ret0, _ := ret[0].(*response.CreateDataBagItem) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetOrgs indicates an expected call of GetOrgs. -func (mr *MockInfraProxyServerMockRecorder) GetOrgs(arg0, arg1 interface{}) *gomock.Call { +// CreateDataBagItem indicates an expected call of CreateDataBagItem +func (mr *MockInfraProxyServerMockRecorder) CreateDataBagItem(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrgs", reflect.TypeOf((*MockInfraProxyServer)(nil).GetOrgs), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateDataBagItem", reflect.TypeOf((*MockInfraProxyServer)(nil).CreateDataBagItem), arg0, arg1) } -// GetPolicyfile mocks base method. -func (m *MockInfraProxyServer) GetPolicyfile(arg0 context.Context, arg1 *request.Policyfile) (*response.Policyfile, error) { +// DeleteDataBag mocks base method +func (m *MockInfraProxyServer) DeleteDataBag(arg0 context.Context, arg1 *request.DataBag) (*response.DataBag, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetPolicyfile", arg0, arg1) - ret0, _ := ret[0].(*response.Policyfile) + ret := m.ctrl.Call(m, "DeleteDataBag", arg0, arg1) + ret0, _ := ret[0].(*response.DataBag) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetPolicyfile indicates an expected call of GetPolicyfile. -func (mr *MockInfraProxyServerMockRecorder) GetPolicyfile(arg0, arg1 interface{}) *gomock.Call { +// DeleteDataBag indicates an expected call of DeleteDataBag +func (mr *MockInfraProxyServerMockRecorder) DeleteDataBag(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPolicyfile", reflect.TypeOf((*MockInfraProxyServer)(nil).GetPolicyfile), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteDataBag", reflect.TypeOf((*MockInfraProxyServer)(nil).DeleteDataBag), arg0, arg1) } -// GetPolicyfileRevisions mocks base method. -func (m *MockInfraProxyServer) GetPolicyfileRevisions(arg0 context.Context, arg1 *request.PolicyfileRevisions) (*response.PolicyfileRevisions, error) { +// DeleteDataBagItem mocks base method +func (m *MockInfraProxyServer) DeleteDataBagItem(arg0 context.Context, arg1 *request.DataBagItem) (*response.DataBagItem, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetPolicyfileRevisions", arg0, arg1) - ret0, _ := ret[0].(*response.PolicyfileRevisions) + ret := m.ctrl.Call(m, "DeleteDataBagItem", arg0, arg1) + ret0, _ := ret[0].(*response.DataBagItem) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetPolicyfileRevisions indicates an expected call of GetPolicyfileRevisions. -func (mr *MockInfraProxyServerMockRecorder) GetPolicyfileRevisions(arg0, arg1 interface{}) *gomock.Call { +// DeleteDataBagItem indicates an expected call of DeleteDataBagItem +func (mr *MockInfraProxyServerMockRecorder) DeleteDataBagItem(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPolicyfileRevisions", reflect.TypeOf((*MockInfraProxyServer)(nil).GetPolicyfileRevisions), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteDataBagItem", reflect.TypeOf((*MockInfraProxyServer)(nil).DeleteDataBagItem), arg0, arg1) } -// GetPolicyfiles mocks base method. -func (m *MockInfraProxyServer) GetPolicyfiles(arg0 context.Context, arg1 *request.Policyfiles) (*response.Policyfiles, error) { +// UpdateDataBagItem mocks base method +func (m *MockInfraProxyServer) UpdateDataBagItem(arg0 context.Context, arg1 *request.UpdateDataBagItem) (*response.UpdateDataBagItem, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetPolicyfiles", arg0, arg1) - ret0, _ := ret[0].(*response.Policyfiles) + ret := m.ctrl.Call(m, "UpdateDataBagItem", arg0, arg1) + ret0, _ := ret[0].(*response.UpdateDataBagItem) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetPolicyfiles indicates an expected call of GetPolicyfiles. -func (mr *MockInfraProxyServerMockRecorder) GetPolicyfiles(arg0, arg1 interface{}) *gomock.Call { +// UpdateDataBagItem indicates an expected call of UpdateDataBagItem +func (mr *MockInfraProxyServerMockRecorder) UpdateDataBagItem(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPolicyfiles", reflect.TypeOf((*MockInfraProxyServer)(nil).GetPolicyfiles), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateDataBagItem", reflect.TypeOf((*MockInfraProxyServer)(nil).UpdateDataBagItem), arg0, arg1) } -// GetPolicygroup mocks base method. -func (m *MockInfraProxyServer) GetPolicygroup(arg0 context.Context, arg1 *request.Policygroup) (*response.Policygroup, error) { +// GetEnvironments mocks base method +func (m *MockInfraProxyServer) GetEnvironments(arg0 context.Context, arg1 *request.Environments) (*response.Environments, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetPolicygroup", arg0, arg1) - ret0, _ := ret[0].(*response.Policygroup) + ret := m.ctrl.Call(m, "GetEnvironments", arg0, arg1) + ret0, _ := ret[0].(*response.Environments) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetPolicygroup indicates an expected call of GetPolicygroup. -func (mr *MockInfraProxyServerMockRecorder) GetPolicygroup(arg0, arg1 interface{}) *gomock.Call { +// GetEnvironments indicates an expected call of GetEnvironments +func (mr *MockInfraProxyServerMockRecorder) GetEnvironments(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPolicygroup", reflect.TypeOf((*MockInfraProxyServer)(nil).GetPolicygroup), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetEnvironments", reflect.TypeOf((*MockInfraProxyServer)(nil).GetEnvironments), arg0, arg1) } -// GetRole mocks base method. -func (m *MockInfraProxyServer) GetRole(arg0 context.Context, arg1 *request.Role) (*response.Role, error) { +// GetEnvironment mocks base method +func (m *MockInfraProxyServer) GetEnvironment(arg0 context.Context, arg1 *request.Environment) (*response.Environment, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetRole", arg0, arg1) - ret0, _ := ret[0].(*response.Role) + ret := m.ctrl.Call(m, "GetEnvironment", arg0, arg1) + ret0, _ := ret[0].(*response.Environment) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetRole indicates an expected call of GetRole. -func (mr *MockInfraProxyServerMockRecorder) GetRole(arg0, arg1 interface{}) *gomock.Call { +// GetEnvironment indicates an expected call of GetEnvironment +func (mr *MockInfraProxyServerMockRecorder) GetEnvironment(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRole", reflect.TypeOf((*MockInfraProxyServer)(nil).GetRole), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetEnvironment", reflect.TypeOf((*MockInfraProxyServer)(nil).GetEnvironment), arg0, arg1) } -// GetRoleEnvironments mocks base method. -func (m *MockInfraProxyServer) GetRoleEnvironments(arg0 context.Context, arg1 *request.Role) (*response.RoleEnvironments, error) { +// CreateEnvironment mocks base method +func (m *MockInfraProxyServer) CreateEnvironment(arg0 context.Context, arg1 *request.CreateEnvironment) (*response.Environment, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetRoleEnvironments", arg0, arg1) - ret0, _ := ret[0].(*response.RoleEnvironments) + ret := m.ctrl.Call(m, "CreateEnvironment", arg0, arg1) + ret0, _ := ret[0].(*response.Environment) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetRoleEnvironments indicates an expected call of GetRoleEnvironments. -func (mr *MockInfraProxyServerMockRecorder) GetRoleEnvironments(arg0, arg1 interface{}) *gomock.Call { +// CreateEnvironment indicates an expected call of CreateEnvironment +func (mr *MockInfraProxyServerMockRecorder) CreateEnvironment(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRoleEnvironments", reflect.TypeOf((*MockInfraProxyServer)(nil).GetRoleEnvironments), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateEnvironment", reflect.TypeOf((*MockInfraProxyServer)(nil).CreateEnvironment), arg0, arg1) } -// GetRoleExpandedRunList mocks base method. -func (m *MockInfraProxyServer) GetRoleExpandedRunList(arg0 context.Context, arg1 *request.ExpandedRunList) (*response.ExpandedRunList, error) { +// DeleteEnvironment mocks base method +func (m *MockInfraProxyServer) DeleteEnvironment(arg0 context.Context, arg1 *request.Environment) (*response.Environment, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetRoleExpandedRunList", arg0, arg1) - ret0, _ := ret[0].(*response.ExpandedRunList) + ret := m.ctrl.Call(m, "DeleteEnvironment", arg0, arg1) + ret0, _ := ret[0].(*response.Environment) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetRoleExpandedRunList indicates an expected call of GetRoleExpandedRunList. -func (mr *MockInfraProxyServerMockRecorder) GetRoleExpandedRunList(arg0, arg1 interface{}) *gomock.Call { +// DeleteEnvironment indicates an expected call of DeleteEnvironment +func (mr *MockInfraProxyServerMockRecorder) DeleteEnvironment(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRoleExpandedRunList", reflect.TypeOf((*MockInfraProxyServer)(nil).GetRoleExpandedRunList), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteEnvironment", reflect.TypeOf((*MockInfraProxyServer)(nil).DeleteEnvironment), arg0, arg1) } -// GetRoles mocks base method. -func (m *MockInfraProxyServer) GetRoles(arg0 context.Context, arg1 *request.Roles) (*response.Roles, error) { +// UpdateEnvironment mocks base method +func (m *MockInfraProxyServer) UpdateEnvironment(arg0 context.Context, arg1 *request.UpdateEnvironment) (*response.Environment, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetRoles", arg0, arg1) - ret0, _ := ret[0].(*response.Roles) + ret := m.ctrl.Call(m, "UpdateEnvironment", arg0, arg1) + ret0, _ := ret[0].(*response.Environment) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetRoles indicates an expected call of GetRoles. -func (mr *MockInfraProxyServerMockRecorder) GetRoles(arg0, arg1 interface{}) *gomock.Call { +// UpdateEnvironment indicates an expected call of UpdateEnvironment +func (mr *MockInfraProxyServerMockRecorder) UpdateEnvironment(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRoles", reflect.TypeOf((*MockInfraProxyServer)(nil).GetRoles), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateEnvironment", reflect.TypeOf((*MockInfraProxyServer)(nil).UpdateEnvironment), arg0, arg1) } -// GetServer mocks base method. -func (m *MockInfraProxyServer) GetServer(arg0 context.Context, arg1 *request.GetServer) (*response.GetServer, error) { +// GetEnvironmentRecipes mocks base method +func (m *MockInfraProxyServer) GetEnvironmentRecipes(arg0 context.Context, arg1 *request.Environment) (*response.EnvironmentRecipesList, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetServer", arg0, arg1) - ret0, _ := ret[0].(*response.GetServer) + ret := m.ctrl.Call(m, "GetEnvironmentRecipes", arg0, arg1) + ret0, _ := ret[0].(*response.EnvironmentRecipesList) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetServer indicates an expected call of GetServer. -func (mr *MockInfraProxyServerMockRecorder) GetServer(arg0, arg1 interface{}) *gomock.Call { +// GetEnvironmentRecipes indicates an expected call of GetEnvironmentRecipes +func (mr *MockInfraProxyServerMockRecorder) GetEnvironmentRecipes(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetServer", reflect.TypeOf((*MockInfraProxyServer)(nil).GetServer), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetEnvironmentRecipes", reflect.TypeOf((*MockInfraProxyServer)(nil).GetEnvironmentRecipes), arg0, arg1) } -// GetServerStatus mocks base method. -func (m *MockInfraProxyServer) GetServerStatus(arg0 context.Context, arg1 *request.GetServerStatus) (*response.GetServerStatus, error) { +// GetNodes mocks base method +func (m *MockInfraProxyServer) GetNodes(arg0 context.Context, arg1 *request.Nodes) (*response.Nodes, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetServerStatus", arg0, arg1) - ret0, _ := ret[0].(*response.GetServerStatus) + ret := m.ctrl.Call(m, "GetNodes", arg0, arg1) + ret0, _ := ret[0].(*response.Nodes) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetServerStatus indicates an expected call of GetServerStatus. -func (mr *MockInfraProxyServerMockRecorder) GetServerStatus(arg0, arg1 interface{}) *gomock.Call { +// GetNodes indicates an expected call of GetNodes +func (mr *MockInfraProxyServerMockRecorder) GetNodes(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetServerStatus", reflect.TypeOf((*MockInfraProxyServer)(nil).GetServerStatus), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodes", reflect.TypeOf((*MockInfraProxyServer)(nil).GetNodes), arg0, arg1) } -// GetServerUsersList mocks base method. -func (m *MockInfraProxyServer) GetServerUsersList(arg0 context.Context, arg1 *request.ServerUsers) (*response.ServerUsers, error) { +// DeleteNode mocks base method +func (m *MockInfraProxyServer) DeleteNode(arg0 context.Context, arg1 *request.Node) (*response.DeleteNode, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetServerUsersList", arg0, arg1) - ret0, _ := ret[0].(*response.ServerUsers) + ret := m.ctrl.Call(m, "DeleteNode", arg0, arg1) + ret0, _ := ret[0].(*response.DeleteNode) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetServerUsersList indicates an expected call of GetServerUsersList. -func (mr *MockInfraProxyServerMockRecorder) GetServerUsersList(arg0, arg1 interface{}) *gomock.Call { +// DeleteNode indicates an expected call of DeleteNode +func (mr *MockInfraProxyServerMockRecorder) DeleteNode(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetServerUsersList", reflect.TypeOf((*MockInfraProxyServer)(nil).GetServerUsersList), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteNode", reflect.TypeOf((*MockInfraProxyServer)(nil).DeleteNode), arg0, arg1) } -// GetServers mocks base method. -func (m *MockInfraProxyServer) GetServers(arg0 context.Context, arg1 *request.GetServers) (*response.GetServers, error) { +// UpdateNode mocks base method +func (m *MockInfraProxyServer) UpdateNode(arg0 context.Context, arg1 *request.NodeDetails) (*response.Node, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetServers", arg0, arg1) - ret0, _ := ret[0].(*response.GetServers) + ret := m.ctrl.Call(m, "UpdateNode", arg0, arg1) + ret0, _ := ret[0].(*response.Node) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetServers indicates an expected call of GetServers. -func (mr *MockInfraProxyServerMockRecorder) GetServers(arg0, arg1 interface{}) *gomock.Call { +// UpdateNode indicates an expected call of UpdateNode +func (mr *MockInfraProxyServerMockRecorder) UpdateNode(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetServers", reflect.TypeOf((*MockInfraProxyServer)(nil).GetServers), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNode", reflect.TypeOf((*MockInfraProxyServer)(nil).UpdateNode), arg0, arg1) } -// ResetClientKey mocks base method. -func (m *MockInfraProxyServer) ResetClientKey(arg0 context.Context, arg1 *request.ClientKey) (*response.ResetClient, error) { +// UpdateNodeAttributes mocks base method +func (m *MockInfraProxyServer) UpdateNodeAttributes(arg0 context.Context, arg1 *request.UpdateNodeAttributes) (*response.UpdateNodeAttributes, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ResetClientKey", arg0, arg1) - ret0, _ := ret[0].(*response.ResetClient) + ret := m.ctrl.Call(m, "UpdateNodeAttributes", arg0, arg1) + ret0, _ := ret[0].(*response.UpdateNodeAttributes) ret1, _ := ret[1].(error) return ret0, ret1 } -// ResetClientKey indicates an expected call of ResetClientKey. -func (mr *MockInfraProxyServerMockRecorder) ResetClientKey(arg0, arg1 interface{}) *gomock.Call { +// UpdateNodeAttributes indicates an expected call of UpdateNodeAttributes +func (mr *MockInfraProxyServerMockRecorder) UpdateNodeAttributes(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ResetClientKey", reflect.TypeOf((*MockInfraProxyServer)(nil).ResetClientKey), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNodeAttributes", reflect.TypeOf((*MockInfraProxyServer)(nil).UpdateNodeAttributes), arg0, arg1) } -// ResetOrgAdminKey mocks base method. -func (m *MockInfraProxyServer) ResetOrgAdminKey(arg0 context.Context, arg1 *request.ResetOrgAdminKey) (*response.ResetOrgAdminKey, error) { +// GetNode mocks base method +func (m *MockInfraProxyServer) GetNode(arg0 context.Context, arg1 *request.Node) (*response.Node, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ResetOrgAdminKey", arg0, arg1) - ret0, _ := ret[0].(*response.ResetOrgAdminKey) + ret := m.ctrl.Call(m, "GetNode", arg0, arg1) + ret0, _ := ret[0].(*response.Node) ret1, _ := ret[1].(error) return ret0, ret1 } -// ResetOrgAdminKey indicates an expected call of ResetOrgAdminKey. -func (mr *MockInfraProxyServerMockRecorder) ResetOrgAdminKey(arg0, arg1 interface{}) *gomock.Call { +// GetNode indicates an expected call of GetNode +func (mr *MockInfraProxyServerMockRecorder) GetNode(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ResetOrgAdminKey", reflect.TypeOf((*MockInfraProxyServer)(nil).ResetOrgAdminKey), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNode", reflect.TypeOf((*MockInfraProxyServer)(nil).GetNode), arg0, arg1) } -// UpdateDataBagItem mocks base method. -func (m *MockInfraProxyServer) UpdateDataBagItem(arg0 context.Context, arg1 *request.UpdateDataBagItem) (*response.UpdateDataBagItem, error) { +// GetNodeExpandedRunList mocks base method +func (m *MockInfraProxyServer) GetNodeExpandedRunList(arg0 context.Context, arg1 *request.NodeExpandedRunList) (*response.NodeExpandedRunList, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateDataBagItem", arg0, arg1) - ret0, _ := ret[0].(*response.UpdateDataBagItem) + ret := m.ctrl.Call(m, "GetNodeExpandedRunList", arg0, arg1) + ret0, _ := ret[0].(*response.NodeExpandedRunList) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateDataBagItem indicates an expected call of UpdateDataBagItem. -func (mr *MockInfraProxyServerMockRecorder) UpdateDataBagItem(arg0, arg1 interface{}) *gomock.Call { +// GetNodeExpandedRunList indicates an expected call of GetNodeExpandedRunList +func (mr *MockInfraProxyServerMockRecorder) GetNodeExpandedRunList(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateDataBagItem", reflect.TypeOf((*MockInfraProxyServer)(nil).UpdateDataBagItem), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodeExpandedRunList", reflect.TypeOf((*MockInfraProxyServer)(nil).GetNodeExpandedRunList), arg0, arg1) } -// UpdateEnvironment mocks base method. -func (m *MockInfraProxyServer) UpdateEnvironment(arg0 context.Context, arg1 *request.UpdateEnvironment) (*response.Environment, error) { +// UpdateNodeTags mocks base method +func (m *MockInfraProxyServer) UpdateNodeTags(arg0 context.Context, arg1 *request.UpdateNodeTags) (*response.UpdateNodeTags, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateEnvironment", arg0, arg1) - ret0, _ := ret[0].(*response.Environment) + ret := m.ctrl.Call(m, "UpdateNodeTags", arg0, arg1) + ret0, _ := ret[0].(*response.UpdateNodeTags) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateEnvironment indicates an expected call of UpdateEnvironment. -func (mr *MockInfraProxyServerMockRecorder) UpdateEnvironment(arg0, arg1 interface{}) *gomock.Call { +// UpdateNodeTags indicates an expected call of UpdateNodeTags +func (mr *MockInfraProxyServerMockRecorder) UpdateNodeTags(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateEnvironment", reflect.TypeOf((*MockInfraProxyServer)(nil).UpdateEnvironment), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNodeTags", reflect.TypeOf((*MockInfraProxyServer)(nil).UpdateNodeTags), arg0, arg1) } -// UpdateNode mocks base method. -func (m *MockInfraProxyServer) UpdateNode(arg0 context.Context, arg1 *request.NodeDetails) (*response.Node, error) { +// UpdateNodeEnvironment mocks base method +func (m *MockInfraProxyServer) UpdateNodeEnvironment(arg0 context.Context, arg1 *request.UpdateNodeEnvironment) (*response.UpdateNodeEnvironment, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateNode", arg0, arg1) - ret0, _ := ret[0].(*response.Node) + ret := m.ctrl.Call(m, "UpdateNodeEnvironment", arg0, arg1) + ret0, _ := ret[0].(*response.UpdateNodeEnvironment) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateNode indicates an expected call of UpdateNode. -func (mr *MockInfraProxyServerMockRecorder) UpdateNode(arg0, arg1 interface{}) *gomock.Call { +// UpdateNodeEnvironment indicates an expected call of UpdateNodeEnvironment +func (mr *MockInfraProxyServerMockRecorder) UpdateNodeEnvironment(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNode", reflect.TypeOf((*MockInfraProxyServer)(nil).UpdateNode), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNodeEnvironment", reflect.TypeOf((*MockInfraProxyServer)(nil).UpdateNodeEnvironment), arg0, arg1) } -// UpdateNodeAttributes mocks base method. -func (m *MockInfraProxyServer) UpdateNodeAttributes(arg0 context.Context, arg1 *request.UpdateNodeAttributes) (*response.UpdateNodeAttributes, error) { +// GetPolicyfiles mocks base method +func (m *MockInfraProxyServer) GetPolicyfiles(arg0 context.Context, arg1 *request.Policyfiles) (*response.Policyfiles, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateNodeAttributes", arg0, arg1) - ret0, _ := ret[0].(*response.UpdateNodeAttributes) + ret := m.ctrl.Call(m, "GetPolicyfiles", arg0, arg1) + ret0, _ := ret[0].(*response.Policyfiles) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateNodeAttributes indicates an expected call of UpdateNodeAttributes. -func (mr *MockInfraProxyServerMockRecorder) UpdateNodeAttributes(arg0, arg1 interface{}) *gomock.Call { +// GetPolicyfiles indicates an expected call of GetPolicyfiles +func (mr *MockInfraProxyServerMockRecorder) GetPolicyfiles(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNodeAttributes", reflect.TypeOf((*MockInfraProxyServer)(nil).UpdateNodeAttributes), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPolicyfiles", reflect.TypeOf((*MockInfraProxyServer)(nil).GetPolicyfiles), arg0, arg1) } -// UpdateNodeEnvironment mocks base method. -func (m *MockInfraProxyServer) UpdateNodeEnvironment(arg0 context.Context, arg1 *request.UpdateNodeEnvironment) (*response.UpdateNodeEnvironment, error) { +// GetPolicyfile mocks base method +func (m *MockInfraProxyServer) GetPolicyfile(arg0 context.Context, arg1 *request.Policyfile) (*response.Policyfile, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateNodeEnvironment", arg0, arg1) - ret0, _ := ret[0].(*response.UpdateNodeEnvironment) + ret := m.ctrl.Call(m, "GetPolicyfile", arg0, arg1) + ret0, _ := ret[0].(*response.Policyfile) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateNodeEnvironment indicates an expected call of UpdateNodeEnvironment. -func (mr *MockInfraProxyServerMockRecorder) UpdateNodeEnvironment(arg0, arg1 interface{}) *gomock.Call { +// GetPolicyfile indicates an expected call of GetPolicyfile +func (mr *MockInfraProxyServerMockRecorder) GetPolicyfile(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNodeEnvironment", reflect.TypeOf((*MockInfraProxyServer)(nil).UpdateNodeEnvironment), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPolicyfile", reflect.TypeOf((*MockInfraProxyServer)(nil).GetPolicyfile), arg0, arg1) } -// UpdateNodeTags mocks base method. -func (m *MockInfraProxyServer) UpdateNodeTags(arg0 context.Context, arg1 *request.UpdateNodeTags) (*response.UpdateNodeTags, error) { +// DeletePolicyfile mocks base method +func (m *MockInfraProxyServer) DeletePolicyfile(arg0 context.Context, arg1 *request.DeletePolicyfile) (*response.DeletePolicyfile, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateNodeTags", arg0, arg1) - ret0, _ := ret[0].(*response.UpdateNodeTags) + ret := m.ctrl.Call(m, "DeletePolicyfile", arg0, arg1) + ret0, _ := ret[0].(*response.DeletePolicyfile) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateNodeTags indicates an expected call of UpdateNodeTags. -func (mr *MockInfraProxyServerMockRecorder) UpdateNodeTags(arg0, arg1 interface{}) *gomock.Call { +// DeletePolicyfile indicates an expected call of DeletePolicyfile +func (mr *MockInfraProxyServerMockRecorder) DeletePolicyfile(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNodeTags", reflect.TypeOf((*MockInfraProxyServer)(nil).UpdateNodeTags), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeletePolicyfile", reflect.TypeOf((*MockInfraProxyServer)(nil).DeletePolicyfile), arg0, arg1) } -// UpdateOrg mocks base method. -func (m *MockInfraProxyServer) UpdateOrg(arg0 context.Context, arg1 *request.UpdateOrg) (*response.UpdateOrg, error) { +// GetPolicyfileRevisions mocks base method +func (m *MockInfraProxyServer) GetPolicyfileRevisions(arg0 context.Context, arg1 *request.PolicyfileRevisions) (*response.PolicyfileRevisions, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateOrg", arg0, arg1) - ret0, _ := ret[0].(*response.UpdateOrg) + ret := m.ctrl.Call(m, "GetPolicyfileRevisions", arg0, arg1) + ret0, _ := ret[0].(*response.PolicyfileRevisions) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateOrg indicates an expected call of UpdateOrg. -func (mr *MockInfraProxyServerMockRecorder) UpdateOrg(arg0, arg1 interface{}) *gomock.Call { +// GetPolicyfileRevisions indicates an expected call of GetPolicyfileRevisions +func (mr *MockInfraProxyServerMockRecorder) GetPolicyfileRevisions(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateOrg", reflect.TypeOf((*MockInfraProxyServer)(nil).UpdateOrg), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPolicyfileRevisions", reflect.TypeOf((*MockInfraProxyServer)(nil).GetPolicyfileRevisions), arg0, arg1) } -// UpdateRole mocks base method. -func (m *MockInfraProxyServer) UpdateRole(arg0 context.Context, arg1 *request.UpdateRole) (*response.Role, error) { +// GetPolicygroup mocks base method +func (m *MockInfraProxyServer) GetPolicygroup(arg0 context.Context, arg1 *request.Policygroup) (*response.Policygroup, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateRole", arg0, arg1) - ret0, _ := ret[0].(*response.Role) + ret := m.ctrl.Call(m, "GetPolicygroup", arg0, arg1) + ret0, _ := ret[0].(*response.Policygroup) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateRole indicates an expected call of UpdateRole. -func (mr *MockInfraProxyServerMockRecorder) UpdateRole(arg0, arg1 interface{}) *gomock.Call { +// GetPolicygroup indicates an expected call of GetPolicygroup +func (mr *MockInfraProxyServerMockRecorder) GetPolicygroup(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateRole", reflect.TypeOf((*MockInfraProxyServer)(nil).UpdateRole), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPolicygroup", reflect.TypeOf((*MockInfraProxyServer)(nil).GetPolicygroup), arg0, arg1) } -// UpdateServer mocks base method. -func (m *MockInfraProxyServer) UpdateServer(arg0 context.Context, arg1 *request.UpdateServer) (*response.UpdateServer, error) { +// GetOrgUsersList mocks base method +func (m *MockInfraProxyServer) GetOrgUsersList(arg0 context.Context, arg1 *request.OrgUsers) (*response.OrgUsers, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateServer", arg0, arg1) - ret0, _ := ret[0].(*response.UpdateServer) + ret := m.ctrl.Call(m, "GetOrgUsersList", arg0, arg1) + ret0, _ := ret[0].(*response.OrgUsers) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateServer indicates an expected call of UpdateServer. -func (mr *MockInfraProxyServerMockRecorder) UpdateServer(arg0, arg1 interface{}) *gomock.Call { +// GetOrgUsersList indicates an expected call of GetOrgUsersList +func (mr *MockInfraProxyServerMockRecorder) GetOrgUsersList(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateServer", reflect.TypeOf((*MockInfraProxyServer)(nil).UpdateServer), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrgUsersList", reflect.TypeOf((*MockInfraProxyServer)(nil).GetOrgUsersList), arg0, arg1) } -// UpdateWebuiKey mocks base method. -func (m *MockInfraProxyServer) UpdateWebuiKey(arg0 context.Context, arg1 *request.UpdateWebuiKey) (*response.UpdateWebuiKey, error) { +// GetServerUsersList mocks base method +func (m *MockInfraProxyServer) GetServerUsersList(arg0 context.Context, arg1 *request.ServerUsers) (*response.ServerUsers, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateWebuiKey", arg0, arg1) - ret0, _ := ret[0].(*response.UpdateWebuiKey) + ret := m.ctrl.Call(m, "GetServerUsersList", arg0, arg1) + ret0, _ := ret[0].(*response.ServerUsers) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateWebuiKey indicates an expected call of UpdateWebuiKey. -func (mr *MockInfraProxyServerMockRecorder) UpdateWebuiKey(arg0, arg1 interface{}) *gomock.Call { +// GetServerUsersList indicates an expected call of GetServerUsersList +func (mr *MockInfraProxyServerMockRecorder) GetServerUsersList(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWebuiKey", reflect.TypeOf((*MockInfraProxyServer)(nil).UpdateWebuiKey), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetServerUsersList", reflect.TypeOf((*MockInfraProxyServer)(nil).GetServerUsersList), arg0, arg1) } -// ValidateWebuiKey mocks base method. -func (m *MockInfraProxyServer) ValidateWebuiKey(arg0 context.Context, arg1 *request.ValidateWebuiKey) (*response.ValidateWebuiKey, error) { +// GetAutomateInfraServerUsersList mocks base method +func (m *MockInfraProxyServer) GetAutomateInfraServerUsersList(arg0 context.Context, arg1 *request.AutomateInfraServerUsers) (*response.AutomateInfraServerUsers, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ValidateWebuiKey", arg0, arg1) - ret0, _ := ret[0].(*response.ValidateWebuiKey) + ret := m.ctrl.Call(m, "GetAutomateInfraServerUsersList", arg0, arg1) + ret0, _ := ret[0].(*response.AutomateInfraServerUsers) ret1, _ := ret[1].(error) return ret0, ret1 } -// ValidateWebuiKey indicates an expected call of ValidateWebuiKey. -func (mr *MockInfraProxyServerMockRecorder) ValidateWebuiKey(arg0, arg1 interface{}) *gomock.Call { +// GetAutomateInfraServerUsersList indicates an expected call of GetAutomateInfraServerUsersList +func (mr *MockInfraProxyServerMockRecorder) GetAutomateInfraServerUsersList(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ValidateWebuiKey", reflect.TypeOf((*MockInfraProxyServer)(nil).ValidateWebuiKey), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAutomateInfraServerUsersList", reflect.TypeOf((*MockInfraProxyServer)(nil).GetAutomateInfraServerUsersList), arg0, arg1) } diff --git a/api/external/infra_proxy/migrations/request/migrations.pb.go b/api/external/infra_proxy/migrations/request/migrations.pb.go index e0b18487e5a..2ada9f7a503 100644 --- a/api/external/infra_proxy/migrations/request/migrations.pb.go +++ b/api/external/infra_proxy/migrations/request/migrations.pb.go @@ -11,6 +11,7 @@ import ( protoreflect "google.golang.org/protobuf/reflect/protoreflect" protoimpl "google.golang.org/protobuf/runtime/protoimpl" reflect "reflect" + sync "sync" ) const ( @@ -24,6 +25,187 @@ const ( // of the legacy proto package is being used. const _ = proto.ProtoPackageIsVersion4 +type UploadZipFileRequest struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields + + //Chef infra server id for the file + ServerId string `protobuf:"bytes,1,opt,name=serverId,proto3" json:"serverId,omitempty"` + //Chunk of the data in bytes + Chunk *Chunk `protobuf:"bytes,2,opt,name=chunk,proto3" json:"chunk,omitempty"` + //Metadata for the request + Meta *Metadata `protobuf:"bytes,3,opt,name=meta,proto3" json:"meta,omitempty"` +} + +func (x *UploadZipFileRequest) Reset() { + *x = UploadZipFileRequest{} + if protoimpl.UnsafeEnabled { + mi := &file_external_infra_proxy_migrations_request_migrations_proto_msgTypes[0] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) + } +} + +func (x *UploadZipFileRequest) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*UploadZipFileRequest) ProtoMessage() {} + +func (x *UploadZipFileRequest) ProtoReflect() protoreflect.Message { + mi := &file_external_infra_proxy_migrations_request_migrations_proto_msgTypes[0] + if protoimpl.UnsafeEnabled && x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use UploadZipFileRequest.ProtoReflect.Descriptor instead. +func (*UploadZipFileRequest) Descriptor() ([]byte, []int) { + return file_external_infra_proxy_migrations_request_migrations_proto_rawDescGZIP(), []int{0} +} + +func (x *UploadZipFileRequest) GetServerId() string { + if x != nil { + return x.ServerId + } + return "" +} + +func (x *UploadZipFileRequest) GetChunk() *Chunk { + if x != nil { + return x.Chunk + } + return nil +} + +func (x *UploadZipFileRequest) GetMeta() *Metadata { + if x != nil { + return x.Meta + } + return nil +} + +type Metadata struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields + + //Migration File name + Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` + //Content type of the request + ContentType string `protobuf:"bytes,2,opt,name=contentType,proto3" json:"contentType,omitempty"` +} + +func (x *Metadata) Reset() { + *x = Metadata{} + if protoimpl.UnsafeEnabled { + mi := &file_external_infra_proxy_migrations_request_migrations_proto_msgTypes[1] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) + } +} + +func (x *Metadata) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*Metadata) ProtoMessage() {} + +func (x *Metadata) ProtoReflect() protoreflect.Message { + mi := &file_external_infra_proxy_migrations_request_migrations_proto_msgTypes[1] + if protoimpl.UnsafeEnabled && x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use Metadata.ProtoReflect.Descriptor instead. +func (*Metadata) Descriptor() ([]byte, []int) { + return file_external_infra_proxy_migrations_request_migrations_proto_rawDescGZIP(), []int{1} +} + +func (x *Metadata) GetName() string { + if x != nil { + return x.Name + } + return "" +} + +func (x *Metadata) GetContentType() string { + if x != nil { + return x.ContentType + } + return "" +} + +// Zip content in byte form. +type Chunk struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields + + // Byte data + Data []byte `protobuf:"bytes,1,opt,name=data,proto3" json:"data,omitempty"` + // Stream pointer + Position int64 `protobuf:"varint,2,opt,name=position,proto3" json:"position,omitempty"` +} + +func (x *Chunk) Reset() { + *x = Chunk{} + if protoimpl.UnsafeEnabled { + mi := &file_external_infra_proxy_migrations_request_migrations_proto_msgTypes[2] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) + } +} + +func (x *Chunk) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*Chunk) ProtoMessage() {} + +func (x *Chunk) ProtoReflect() protoreflect.Message { + mi := &file_external_infra_proxy_migrations_request_migrations_proto_msgTypes[2] + if protoimpl.UnsafeEnabled && x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use Chunk.ProtoReflect.Descriptor instead. +func (*Chunk) Descriptor() ([]byte, []int) { + return file_external_infra_proxy_migrations_request_migrations_proto_rawDescGZIP(), []int{2} +} + +func (x *Chunk) GetData() []byte { + if x != nil { + return x.Data + } + return nil +} + +func (x *Chunk) GetPosition() int64 { + if x != nil { + return x.Position + } + return 0 +} + var File_external_infra_proxy_migrations_request_migrations_proto protoreflect.FileDescriptor var file_external_infra_proxy_migrations_request_migrations_proto_rawDesc = []byte{ @@ -33,21 +215,61 @@ var file_external_infra_proxy_migrations_request_migrations_proto_rawDesc = []by 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x30, 0x63, 0x68, 0x65, 0x66, 0x2e, 0x61, 0x75, 0x74, 0x6f, 0x6d, 0x61, 0x74, 0x65, 0x2e, 0x61, 0x70, 0x69, 0x2e, 0x69, 0x6e, 0x66, 0x72, 0x61, 0x5f, 0x70, 0x72, 0x6f, 0x78, 0x79, 0x2e, 0x6d, 0x69, 0x67, 0x72, 0x61, 0x74, - 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x72, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x42, 0x46, 0x5a, 0x44, - 0x67, 0x69, 0x74, 0x68, 0x75, 0x62, 0x2e, 0x63, 0x6f, 0x6d, 0x2f, 0x63, 0x68, 0x65, 0x66, 0x2f, - 0x61, 0x75, 0x74, 0x6f, 0x6d, 0x61, 0x74, 0x65, 0x2f, 0x61, 0x70, 0x69, 0x2f, 0x65, 0x78, 0x74, - 0x65, 0x72, 0x6e, 0x61, 0x6c, 0x2f, 0x69, 0x6e, 0x66, 0x72, 0x61, 0x5f, 0x70, 0x72, 0x6f, 0x78, - 0x79, 0x2f, 0x6d, 0x69, 0x67, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2f, 0x72, 0x65, 0x71, - 0x75, 0x65, 0x73, 0x74, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33, + 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x72, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x22, 0xd1, 0x01, 0x0a, + 0x14, 0x55, 0x70, 0x6c, 0x6f, 0x61, 0x64, 0x5a, 0x69, 0x70, 0x46, 0x69, 0x6c, 0x65, 0x52, 0x65, + 0x71, 0x75, 0x65, 0x73, 0x74, 0x12, 0x1a, 0x0a, 0x08, 0x73, 0x65, 0x72, 0x76, 0x65, 0x72, 0x49, + 0x64, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x08, 0x73, 0x65, 0x72, 0x76, 0x65, 0x72, 0x49, + 0x64, 0x12, 0x4d, 0x0a, 0x05, 0x63, 0x68, 0x75, 0x6e, 0x6b, 0x18, 0x02, 0x20, 0x01, 0x28, 0x0b, + 0x32, 0x37, 0x2e, 0x63, 0x68, 0x65, 0x66, 0x2e, 0x61, 0x75, 0x74, 0x6f, 0x6d, 0x61, 0x74, 0x65, + 0x2e, 0x61, 0x70, 0x69, 0x2e, 0x69, 0x6e, 0x66, 0x72, 0x61, 0x5f, 0x70, 0x72, 0x6f, 0x78, 0x79, + 0x2e, 0x6d, 0x69, 0x67, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x72, 0x65, 0x71, 0x75, + 0x65, 0x73, 0x74, 0x2e, 0x43, 0x68, 0x75, 0x6e, 0x6b, 0x52, 0x05, 0x63, 0x68, 0x75, 0x6e, 0x6b, + 0x12, 0x4e, 0x0a, 0x04, 0x6d, 0x65, 0x74, 0x61, 0x18, 0x03, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x3a, + 0x2e, 0x63, 0x68, 0x65, 0x66, 0x2e, 0x61, 0x75, 0x74, 0x6f, 0x6d, 0x61, 0x74, 0x65, 0x2e, 0x61, + 0x70, 0x69, 0x2e, 0x69, 0x6e, 0x66, 0x72, 0x61, 0x5f, 0x70, 0x72, 0x6f, 0x78, 0x79, 0x2e, 0x6d, + 0x69, 0x67, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x72, 0x65, 0x71, 0x75, 0x65, 0x73, + 0x74, 0x2e, 0x4d, 0x65, 0x74, 0x61, 0x64, 0x61, 0x74, 0x61, 0x52, 0x04, 0x6d, 0x65, 0x74, 0x61, + 0x22, 0x40, 0x0a, 0x08, 0x4d, 0x65, 0x74, 0x61, 0x64, 0x61, 0x74, 0x61, 0x12, 0x12, 0x0a, 0x04, + 0x6e, 0x61, 0x6d, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x04, 0x6e, 0x61, 0x6d, 0x65, + 0x12, 0x20, 0x0a, 0x0b, 0x63, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, 0x54, 0x79, 0x70, 0x65, 0x18, + 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0b, 0x63, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, 0x54, 0x79, + 0x70, 0x65, 0x22, 0x37, 0x0a, 0x05, 0x43, 0x68, 0x75, 0x6e, 0x6b, 0x12, 0x12, 0x0a, 0x04, 0x64, + 0x61, 0x74, 0x61, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x04, 0x64, 0x61, 0x74, 0x61, 0x12, + 0x1a, 0x0a, 0x08, 0x70, 0x6f, 0x73, 0x69, 0x74, 0x69, 0x6f, 0x6e, 0x18, 0x02, 0x20, 0x01, 0x28, + 0x03, 0x52, 0x08, 0x70, 0x6f, 0x73, 0x69, 0x74, 0x69, 0x6f, 0x6e, 0x42, 0x46, 0x5a, 0x44, 0x67, + 0x69, 0x74, 0x68, 0x75, 0x62, 0x2e, 0x63, 0x6f, 0x6d, 0x2f, 0x63, 0x68, 0x65, 0x66, 0x2f, 0x61, + 0x75, 0x74, 0x6f, 0x6d, 0x61, 0x74, 0x65, 0x2f, 0x61, 0x70, 0x69, 0x2f, 0x65, 0x78, 0x74, 0x65, + 0x72, 0x6e, 0x61, 0x6c, 0x2f, 0x69, 0x6e, 0x66, 0x72, 0x61, 0x5f, 0x70, 0x72, 0x6f, 0x78, 0x79, + 0x2f, 0x6d, 0x69, 0x67, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2f, 0x72, 0x65, 0x71, 0x75, + 0x65, 0x73, 0x74, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33, +} + +var ( + file_external_infra_proxy_migrations_request_migrations_proto_rawDescOnce sync.Once + file_external_infra_proxy_migrations_request_migrations_proto_rawDescData = file_external_infra_proxy_migrations_request_migrations_proto_rawDesc +) + +func file_external_infra_proxy_migrations_request_migrations_proto_rawDescGZIP() []byte { + file_external_infra_proxy_migrations_request_migrations_proto_rawDescOnce.Do(func() { + file_external_infra_proxy_migrations_request_migrations_proto_rawDescData = protoimpl.X.CompressGZIP(file_external_infra_proxy_migrations_request_migrations_proto_rawDescData) + }) + return file_external_infra_proxy_migrations_request_migrations_proto_rawDescData } -var file_external_infra_proxy_migrations_request_migrations_proto_goTypes = []interface{}{} +var file_external_infra_proxy_migrations_request_migrations_proto_msgTypes = make([]protoimpl.MessageInfo, 3) +var file_external_infra_proxy_migrations_request_migrations_proto_goTypes = []interface{}{ + (*UploadZipFileRequest)(nil), // 0: chef.automate.api.infra_proxy.migrations.request.UploadZipFileRequest + (*Metadata)(nil), // 1: chef.automate.api.infra_proxy.migrations.request.Metadata + (*Chunk)(nil), // 2: chef.automate.api.infra_proxy.migrations.request.Chunk +} var file_external_infra_proxy_migrations_request_migrations_proto_depIdxs = []int32{ - 0, // [0:0] is the sub-list for method output_type - 0, // [0:0] is the sub-list for method input_type - 0, // [0:0] is the sub-list for extension type_name - 0, // [0:0] is the sub-list for extension extendee - 0, // [0:0] is the sub-list for field type_name + 2, // 0: chef.automate.api.infra_proxy.migrations.request.UploadZipFileRequest.chunk:type_name -> chef.automate.api.infra_proxy.migrations.request.Chunk + 1, // 1: chef.automate.api.infra_proxy.migrations.request.UploadZipFileRequest.meta:type_name -> chef.automate.api.infra_proxy.migrations.request.Metadata + 2, // [2:2] is the sub-list for method output_type + 2, // [2:2] is the sub-list for method input_type + 2, // [2:2] is the sub-list for extension type_name + 2, // [2:2] is the sub-list for extension extendee + 0, // [0:2] is the sub-list for field type_name } func init() { file_external_infra_proxy_migrations_request_migrations_proto_init() } @@ -55,18 +277,57 @@ func file_external_infra_proxy_migrations_request_migrations_proto_init() { if File_external_infra_proxy_migrations_request_migrations_proto != nil { return } + if !protoimpl.UnsafeEnabled { + file_external_infra_proxy_migrations_request_migrations_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} { + switch v := v.(*UploadZipFileRequest); i { + case 0: + return &v.state + case 1: + return &v.sizeCache + case 2: + return &v.unknownFields + default: + return nil + } + } + file_external_infra_proxy_migrations_request_migrations_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} { + switch v := v.(*Metadata); i { + case 0: + return &v.state + case 1: + return &v.sizeCache + case 2: + return &v.unknownFields + default: + return nil + } + } + file_external_infra_proxy_migrations_request_migrations_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} { + switch v := v.(*Chunk); i { + case 0: + return &v.state + case 1: + return &v.sizeCache + case 2: + return &v.unknownFields + default: + return nil + } + } + } type x struct{} out := protoimpl.TypeBuilder{ File: protoimpl.DescBuilder{ GoPackagePath: reflect.TypeOf(x{}).PkgPath(), RawDescriptor: file_external_infra_proxy_migrations_request_migrations_proto_rawDesc, NumEnums: 0, - NumMessages: 0, + NumMessages: 3, NumExtensions: 0, NumServices: 0, }, GoTypes: file_external_infra_proxy_migrations_request_migrations_proto_goTypes, DependencyIndexes: file_external_infra_proxy_migrations_request_migrations_proto_depIdxs, + MessageInfos: file_external_infra_proxy_migrations_request_migrations_proto_msgTypes, }.Build() File_external_infra_proxy_migrations_request_migrations_proto = out.File file_external_infra_proxy_migrations_request_migrations_proto_rawDesc = nil diff --git a/api/external/infra_proxy/migrations/request/migrations.proto b/api/external/infra_proxy/migrations/request/migrations.proto index bdfba85debf..bc47a0e82e5 100644 --- a/api/external/infra_proxy/migrations/request/migrations.proto +++ b/api/external/infra_proxy/migrations/request/migrations.proto @@ -2,3 +2,27 @@ syntax = "proto3"; package chef.automate.api.infra_proxy.migrations.request; option go_package = "github.com/chef/automate/api/external/infra_proxy/migrations/request"; + +message UploadZipFileRequest { + //Chef infra server id for the file + string serverId =1; + //Chunk of the data in bytes + Chunk chunk = 2; + //Metadata for the request + Metadata meta = 3; +} + +message Metadata { + //Migration File name + string name = 1; + //Content type of the request + string contentType =2; +} + +// Zip content in byte form. +message Chunk { + // Byte data + bytes data = 1; + // Stream pointer + int64 position = 2; +} diff --git a/api/external/infra_proxy/migrations/response/migrations.pb.go b/api/external/infra_proxy/migrations/response/migrations.pb.go index 69f80ea4408..ccf8298643f 100644 --- a/api/external/infra_proxy/migrations/response/migrations.pb.go +++ b/api/external/infra_proxy/migrations/response/migrations.pb.go @@ -11,6 +11,7 @@ import ( protoreflect "google.golang.org/protobuf/reflect/protoreflect" protoimpl "google.golang.org/protobuf/runtime/protoimpl" reflect "reflect" + sync "sync" ) const ( @@ -24,6 +25,72 @@ const ( // of the legacy proto package is being used. const _ = proto.ProtoPackageIsVersion4 +type UploadZipFileResponse struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields + + // Success responce for zip upload. + Success bool `protobuf:"varint,1,opt,name=success,proto3" json:"success,omitempty"` + // Migration ID + MigrationId string `protobuf:"bytes,2,opt,name=migrationId,proto3" json:"migrationId,omitempty"` + // Error data in case of failure. + Errors []string `protobuf:"bytes,3,rep,name=errors,proto3" json:"errors,omitempty"` +} + +func (x *UploadZipFileResponse) Reset() { + *x = UploadZipFileResponse{} + if protoimpl.UnsafeEnabled { + mi := &file_external_infra_proxy_migrations_response_migrations_proto_msgTypes[0] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) + } +} + +func (x *UploadZipFileResponse) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*UploadZipFileResponse) ProtoMessage() {} + +func (x *UploadZipFileResponse) ProtoReflect() protoreflect.Message { + mi := &file_external_infra_proxy_migrations_response_migrations_proto_msgTypes[0] + if protoimpl.UnsafeEnabled && x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use UploadZipFileResponse.ProtoReflect.Descriptor instead. +func (*UploadZipFileResponse) Descriptor() ([]byte, []int) { + return file_external_infra_proxy_migrations_response_migrations_proto_rawDescGZIP(), []int{0} +} + +func (x *UploadZipFileResponse) GetSuccess() bool { + if x != nil { + return x.Success + } + return false +} + +func (x *UploadZipFileResponse) GetMigrationId() string { + if x != nil { + return x.MigrationId + } + return "" +} + +func (x *UploadZipFileResponse) GetErrors() []string { + if x != nil { + return x.Errors + } + return nil +} + var File_external_infra_proxy_migrations_response_migrations_proto protoreflect.FileDescriptor var file_external_infra_proxy_migrations_response_migrations_proto_rawDesc = []byte{ @@ -33,15 +100,37 @@ var file_external_infra_proxy_migrations_response_migrations_proto_rawDesc = []b 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x31, 0x63, 0x68, 0x65, 0x66, 0x2e, 0x61, 0x75, 0x74, 0x6f, 0x6d, 0x61, 0x74, 0x65, 0x2e, 0x61, 0x70, 0x69, 0x2e, 0x69, 0x6e, 0x66, 0x72, 0x61, 0x5f, 0x70, 0x72, 0x6f, 0x78, 0x79, 0x2e, 0x6d, 0x69, 0x67, 0x72, 0x61, - 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x72, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x42, 0x47, - 0x5a, 0x45, 0x67, 0x69, 0x74, 0x68, 0x75, 0x62, 0x2e, 0x63, 0x6f, 0x6d, 0x2f, 0x63, 0x68, 0x65, - 0x66, 0x2f, 0x61, 0x75, 0x74, 0x6f, 0x6d, 0x61, 0x74, 0x65, 0x2f, 0x61, 0x70, 0x69, 0x2f, 0x65, - 0x78, 0x74, 0x65, 0x72, 0x6e, 0x61, 0x6c, 0x2f, 0x69, 0x6e, 0x66, 0x72, 0x61, 0x5f, 0x70, 0x72, - 0x6f, 0x78, 0x79, 0x2f, 0x6d, 0x69, 0x67, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2f, 0x72, - 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33, + 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x72, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x22, 0x6b, + 0x0a, 0x15, 0x55, 0x70, 0x6c, 0x6f, 0x61, 0x64, 0x5a, 0x69, 0x70, 0x46, 0x69, 0x6c, 0x65, 0x52, + 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x12, 0x18, 0x0a, 0x07, 0x73, 0x75, 0x63, 0x63, 0x65, + 0x73, 0x73, 0x18, 0x01, 0x20, 0x01, 0x28, 0x08, 0x52, 0x07, 0x73, 0x75, 0x63, 0x63, 0x65, 0x73, + 0x73, 0x12, 0x20, 0x0a, 0x0b, 0x6d, 0x69, 0x67, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x49, 0x64, + 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0b, 0x6d, 0x69, 0x67, 0x72, 0x61, 0x74, 0x69, 0x6f, + 0x6e, 0x49, 0x64, 0x12, 0x16, 0x0a, 0x06, 0x65, 0x72, 0x72, 0x6f, 0x72, 0x73, 0x18, 0x03, 0x20, + 0x03, 0x28, 0x09, 0x52, 0x06, 0x65, 0x72, 0x72, 0x6f, 0x72, 0x73, 0x42, 0x47, 0x5a, 0x45, 0x67, + 0x69, 0x74, 0x68, 0x75, 0x62, 0x2e, 0x63, 0x6f, 0x6d, 0x2f, 0x63, 0x68, 0x65, 0x66, 0x2f, 0x61, + 0x75, 0x74, 0x6f, 0x6d, 0x61, 0x74, 0x65, 0x2f, 0x61, 0x70, 0x69, 0x2f, 0x65, 0x78, 0x74, 0x65, + 0x72, 0x6e, 0x61, 0x6c, 0x2f, 0x69, 0x6e, 0x66, 0x72, 0x61, 0x5f, 0x70, 0x72, 0x6f, 0x78, 0x79, + 0x2f, 0x6d, 0x69, 0x67, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2f, 0x72, 0x65, 0x73, 0x70, + 0x6f, 0x6e, 0x73, 0x65, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33, } -var file_external_infra_proxy_migrations_response_migrations_proto_goTypes = []interface{}{} +var ( + file_external_infra_proxy_migrations_response_migrations_proto_rawDescOnce sync.Once + file_external_infra_proxy_migrations_response_migrations_proto_rawDescData = file_external_infra_proxy_migrations_response_migrations_proto_rawDesc +) + +func file_external_infra_proxy_migrations_response_migrations_proto_rawDescGZIP() []byte { + file_external_infra_proxy_migrations_response_migrations_proto_rawDescOnce.Do(func() { + file_external_infra_proxy_migrations_response_migrations_proto_rawDescData = protoimpl.X.CompressGZIP(file_external_infra_proxy_migrations_response_migrations_proto_rawDescData) + }) + return file_external_infra_proxy_migrations_response_migrations_proto_rawDescData +} + +var file_external_infra_proxy_migrations_response_migrations_proto_msgTypes = make([]protoimpl.MessageInfo, 1) +var file_external_infra_proxy_migrations_response_migrations_proto_goTypes = []interface{}{ + (*UploadZipFileResponse)(nil), // 0: chef.automate.api.infra_proxy.migrations.response.UploadZipFileResponse +} var file_external_infra_proxy_migrations_response_migrations_proto_depIdxs = []int32{ 0, // [0:0] is the sub-list for method output_type 0, // [0:0] is the sub-list for method input_type @@ -55,18 +144,33 @@ func file_external_infra_proxy_migrations_response_migrations_proto_init() { if File_external_infra_proxy_migrations_response_migrations_proto != nil { return } + if !protoimpl.UnsafeEnabled { + file_external_infra_proxy_migrations_response_migrations_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} { + switch v := v.(*UploadZipFileResponse); i { + case 0: + return &v.state + case 1: + return &v.sizeCache + case 2: + return &v.unknownFields + default: + return nil + } + } + } type x struct{} out := protoimpl.TypeBuilder{ File: protoimpl.DescBuilder{ GoPackagePath: reflect.TypeOf(x{}).PkgPath(), RawDescriptor: file_external_infra_proxy_migrations_response_migrations_proto_rawDesc, NumEnums: 0, - NumMessages: 0, + NumMessages: 1, NumExtensions: 0, NumServices: 0, }, GoTypes: file_external_infra_proxy_migrations_response_migrations_proto_goTypes, DependencyIndexes: file_external_infra_proxy_migrations_response_migrations_proto_depIdxs, + MessageInfos: file_external_infra_proxy_migrations_response_migrations_proto_msgTypes, }.Build() File_external_infra_proxy_migrations_response_migrations_proto = out.File file_external_infra_proxy_migrations_response_migrations_proto_rawDesc = nil diff --git a/api/external/infra_proxy/migrations/response/migrations.proto b/api/external/infra_proxy/migrations/response/migrations.proto index e8dd9f7904e..6badc66c175 100644 --- a/api/external/infra_proxy/migrations/response/migrations.proto +++ b/api/external/infra_proxy/migrations/response/migrations.proto @@ -2,3 +2,13 @@ syntax = "proto3"; package chef.automate.api.infra_proxy.migrations.response; option go_package = "github.com/chef/automate/api/external/infra_proxy/migrations/response"; + + +message UploadZipFileResponse { + // Success responce for zip upload. + bool success = 1; + // Migration ID + string migrationId = 2; + // Error data in case of failure. + repeated string errors = 3; +} diff --git a/api/external/secrets/secrets.pb.client_mock.go b/api/external/secrets/secrets.pb.client_mock.go index fa05e0cfb1c..3ed33712312 100644 --- a/api/external/secrets/secrets.pb.client_mock.go +++ b/api/external/secrets/secrets.pb.client_mock.go @@ -6,36 +6,35 @@ package secrets import ( context "context" - reflect "reflect" - gomock "github.com/golang/mock/gomock" grpc "google.golang.org/grpc" + reflect "reflect" ) -// MockSecretsServiceClient is a mock of SecretsServiceClient interface. +// MockSecretsServiceClient is a mock of SecretsServiceClient interface type MockSecretsServiceClient struct { ctrl *gomock.Controller recorder *MockSecretsServiceClientMockRecorder } -// MockSecretsServiceClientMockRecorder is the mock recorder for MockSecretsServiceClient. +// MockSecretsServiceClientMockRecorder is the mock recorder for MockSecretsServiceClient type MockSecretsServiceClientMockRecorder struct { mock *MockSecretsServiceClient } -// NewMockSecretsServiceClient creates a new mock instance. +// NewMockSecretsServiceClient creates a new mock instance func NewMockSecretsServiceClient(ctrl *gomock.Controller) *MockSecretsServiceClient { mock := &MockSecretsServiceClient{ctrl: ctrl} mock.recorder = &MockSecretsServiceClientMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockSecretsServiceClient) EXPECT() *MockSecretsServiceClientMockRecorder { return m.recorder } -// Create mocks base method. +// Create mocks base method func (m *MockSecretsServiceClient) Create(ctx context.Context, in *Secret, opts ...grpc.CallOption) (*Id, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} @@ -48,117 +47,117 @@ func (m *MockSecretsServiceClient) Create(ctx context.Context, in *Secret, opts return ret0, ret1 } -// Create indicates an expected call of Create. +// Create indicates an expected call of Create func (mr *MockSecretsServiceClientMockRecorder) Create(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockSecretsServiceClient)(nil).Create), varargs...) } -// Delete mocks base method. -func (m *MockSecretsServiceClient) Delete(ctx context.Context, in *Id, opts ...grpc.CallOption) (*DeleteResponse, error) { +// Read mocks base method +func (m *MockSecretsServiceClient) Read(ctx context.Context, in *Id, opts ...grpc.CallOption) (*Secret, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "Delete", varargs...) - ret0, _ := ret[0].(*DeleteResponse) + ret := m.ctrl.Call(m, "Read", varargs...) + ret0, _ := ret[0].(*Secret) ret1, _ := ret[1].(error) return ret0, ret1 } -// Delete indicates an expected call of Delete. -func (mr *MockSecretsServiceClientMockRecorder) Delete(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// Read indicates an expected call of Read +func (mr *MockSecretsServiceClientMockRecorder) Read(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockSecretsServiceClient)(nil).Delete), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Read", reflect.TypeOf((*MockSecretsServiceClient)(nil).Read), varargs...) } -// List mocks base method. -func (m *MockSecretsServiceClient) List(ctx context.Context, in *Query, opts ...grpc.CallOption) (*Secrets, error) { +// Update mocks base method +func (m *MockSecretsServiceClient) Update(ctx context.Context, in *Secret, opts ...grpc.CallOption) (*UpdateResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "List", varargs...) - ret0, _ := ret[0].(*Secrets) + ret := m.ctrl.Call(m, "Update", varargs...) + ret0, _ := ret[0].(*UpdateResponse) ret1, _ := ret[1].(error) return ret0, ret1 } -// List indicates an expected call of List. -func (mr *MockSecretsServiceClientMockRecorder) List(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// Update indicates an expected call of Update +func (mr *MockSecretsServiceClientMockRecorder) Update(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockSecretsServiceClient)(nil).List), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockSecretsServiceClient)(nil).Update), varargs...) } -// Read mocks base method. -func (m *MockSecretsServiceClient) Read(ctx context.Context, in *Id, opts ...grpc.CallOption) (*Secret, error) { +// Delete mocks base method +func (m *MockSecretsServiceClient) Delete(ctx context.Context, in *Id, opts ...grpc.CallOption) (*DeleteResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "Read", varargs...) - ret0, _ := ret[0].(*Secret) + ret := m.ctrl.Call(m, "Delete", varargs...) + ret0, _ := ret[0].(*DeleteResponse) ret1, _ := ret[1].(error) return ret0, ret1 } -// Read indicates an expected call of Read. -func (mr *MockSecretsServiceClientMockRecorder) Read(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// Delete indicates an expected call of Delete +func (mr *MockSecretsServiceClientMockRecorder) Delete(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Read", reflect.TypeOf((*MockSecretsServiceClient)(nil).Read), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockSecretsServiceClient)(nil).Delete), varargs...) } -// Update mocks base method. -func (m *MockSecretsServiceClient) Update(ctx context.Context, in *Secret, opts ...grpc.CallOption) (*UpdateResponse, error) { +// List mocks base method +func (m *MockSecretsServiceClient) List(ctx context.Context, in *Query, opts ...grpc.CallOption) (*Secrets, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "Update", varargs...) - ret0, _ := ret[0].(*UpdateResponse) + ret := m.ctrl.Call(m, "List", varargs...) + ret0, _ := ret[0].(*Secrets) ret1, _ := ret[1].(error) return ret0, ret1 } -// Update indicates an expected call of Update. -func (mr *MockSecretsServiceClientMockRecorder) Update(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// List indicates an expected call of List +func (mr *MockSecretsServiceClientMockRecorder) List(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockSecretsServiceClient)(nil).Update), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockSecretsServiceClient)(nil).List), varargs...) } -// MockSecretsServiceServer is a mock of SecretsServiceServer interface. +// MockSecretsServiceServer is a mock of SecretsServiceServer interface type MockSecretsServiceServer struct { ctrl *gomock.Controller recorder *MockSecretsServiceServerMockRecorder } -// MockSecretsServiceServerMockRecorder is the mock recorder for MockSecretsServiceServer. +// MockSecretsServiceServerMockRecorder is the mock recorder for MockSecretsServiceServer type MockSecretsServiceServerMockRecorder struct { mock *MockSecretsServiceServer } -// NewMockSecretsServiceServer creates a new mock instance. +// NewMockSecretsServiceServer creates a new mock instance func NewMockSecretsServiceServer(ctrl *gomock.Controller) *MockSecretsServiceServer { mock := &MockSecretsServiceServer{ctrl: ctrl} mock.recorder = &MockSecretsServiceServerMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockSecretsServiceServer) EXPECT() *MockSecretsServiceServerMockRecorder { return m.recorder } -// Create mocks base method. +// Create mocks base method func (m *MockSecretsServiceServer) Create(arg0 context.Context, arg1 *Secret) (*Id, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Create", arg0, arg1) @@ -167,68 +166,68 @@ func (m *MockSecretsServiceServer) Create(arg0 context.Context, arg1 *Secret) (* return ret0, ret1 } -// Create indicates an expected call of Create. +// Create indicates an expected call of Create func (mr *MockSecretsServiceServerMockRecorder) Create(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockSecretsServiceServer)(nil).Create), arg0, arg1) } -// Delete mocks base method. -func (m *MockSecretsServiceServer) Delete(arg0 context.Context, arg1 *Id) (*DeleteResponse, error) { +// Read mocks base method +func (m *MockSecretsServiceServer) Read(arg0 context.Context, arg1 *Id) (*Secret, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Delete", arg0, arg1) - ret0, _ := ret[0].(*DeleteResponse) + ret := m.ctrl.Call(m, "Read", arg0, arg1) + ret0, _ := ret[0].(*Secret) ret1, _ := ret[1].(error) return ret0, ret1 } -// Delete indicates an expected call of Delete. -func (mr *MockSecretsServiceServerMockRecorder) Delete(arg0, arg1 interface{}) *gomock.Call { +// Read indicates an expected call of Read +func (mr *MockSecretsServiceServerMockRecorder) Read(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockSecretsServiceServer)(nil).Delete), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Read", reflect.TypeOf((*MockSecretsServiceServer)(nil).Read), arg0, arg1) } -// List mocks base method. -func (m *MockSecretsServiceServer) List(arg0 context.Context, arg1 *Query) (*Secrets, error) { +// Update mocks base method +func (m *MockSecretsServiceServer) Update(arg0 context.Context, arg1 *Secret) (*UpdateResponse, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "List", arg0, arg1) - ret0, _ := ret[0].(*Secrets) + ret := m.ctrl.Call(m, "Update", arg0, arg1) + ret0, _ := ret[0].(*UpdateResponse) ret1, _ := ret[1].(error) return ret0, ret1 } -// List indicates an expected call of List. -func (mr *MockSecretsServiceServerMockRecorder) List(arg0, arg1 interface{}) *gomock.Call { +// Update indicates an expected call of Update +func (mr *MockSecretsServiceServerMockRecorder) Update(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockSecretsServiceServer)(nil).List), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockSecretsServiceServer)(nil).Update), arg0, arg1) } -// Read mocks base method. -func (m *MockSecretsServiceServer) Read(arg0 context.Context, arg1 *Id) (*Secret, error) { +// Delete mocks base method +func (m *MockSecretsServiceServer) Delete(arg0 context.Context, arg1 *Id) (*DeleteResponse, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Read", arg0, arg1) - ret0, _ := ret[0].(*Secret) + ret := m.ctrl.Call(m, "Delete", arg0, arg1) + ret0, _ := ret[0].(*DeleteResponse) ret1, _ := ret[1].(error) return ret0, ret1 } -// Read indicates an expected call of Read. -func (mr *MockSecretsServiceServerMockRecorder) Read(arg0, arg1 interface{}) *gomock.Call { +// Delete indicates an expected call of Delete +func (mr *MockSecretsServiceServerMockRecorder) Delete(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Read", reflect.TypeOf((*MockSecretsServiceServer)(nil).Read), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockSecretsServiceServer)(nil).Delete), arg0, arg1) } -// Update mocks base method. -func (m *MockSecretsServiceServer) Update(arg0 context.Context, arg1 *Secret) (*UpdateResponse, error) { +// List mocks base method +func (m *MockSecretsServiceServer) List(arg0 context.Context, arg1 *Query) (*Secrets, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Update", arg0, arg1) - ret0, _ := ret[0].(*UpdateResponse) + ret := m.ctrl.Call(m, "List", arg0, arg1) + ret0, _ := ret[0].(*Secrets) ret1, _ := ret[1].(error) return ret0, ret1 } -// Update indicates an expected call of Update. -func (mr *MockSecretsServiceServerMockRecorder) Update(arg0, arg1 interface{}) *gomock.Call { +// List indicates an expected call of List +func (mr *MockSecretsServiceServerMockRecorder) List(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockSecretsServiceServer)(nil).Update), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockSecretsServiceServer)(nil).List), arg0, arg1) } diff --git a/api/interservice/authn/authenticate.pb.client_mock.go b/api/interservice/authn/authenticate.pb.client_mock.go index bf1ed348801..f3f257bf0ec 100644 --- a/api/interservice/authn/authenticate.pb.client_mock.go +++ b/api/interservice/authn/authenticate.pb.client_mock.go @@ -6,36 +6,35 @@ package authn import ( context "context" - reflect "reflect" - gomock "github.com/golang/mock/gomock" grpc "google.golang.org/grpc" + reflect "reflect" ) -// MockAuthenticationServiceClient is a mock of AuthenticationServiceClient interface. +// MockAuthenticationServiceClient is a mock of AuthenticationServiceClient interface type MockAuthenticationServiceClient struct { ctrl *gomock.Controller recorder *MockAuthenticationServiceClientMockRecorder } -// MockAuthenticationServiceClientMockRecorder is the mock recorder for MockAuthenticationServiceClient. +// MockAuthenticationServiceClientMockRecorder is the mock recorder for MockAuthenticationServiceClient type MockAuthenticationServiceClientMockRecorder struct { mock *MockAuthenticationServiceClient } -// NewMockAuthenticationServiceClient creates a new mock instance. +// NewMockAuthenticationServiceClient creates a new mock instance func NewMockAuthenticationServiceClient(ctrl *gomock.Controller) *MockAuthenticationServiceClient { mock := &MockAuthenticationServiceClient{ctrl: ctrl} mock.recorder = &MockAuthenticationServiceClientMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockAuthenticationServiceClient) EXPECT() *MockAuthenticationServiceClientMockRecorder { return m.recorder } -// Authenticate mocks base method. +// Authenticate mocks base method func (m *MockAuthenticationServiceClient) Authenticate(ctx context.Context, in *AuthenticateRequest, opts ...grpc.CallOption) (*AuthenticateResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} @@ -48,37 +47,37 @@ func (m *MockAuthenticationServiceClient) Authenticate(ctx context.Context, in * return ret0, ret1 } -// Authenticate indicates an expected call of Authenticate. +// Authenticate indicates an expected call of Authenticate func (mr *MockAuthenticationServiceClientMockRecorder) Authenticate(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Authenticate", reflect.TypeOf((*MockAuthenticationServiceClient)(nil).Authenticate), varargs...) } -// MockAuthenticationServiceServer is a mock of AuthenticationServiceServer interface. +// MockAuthenticationServiceServer is a mock of AuthenticationServiceServer interface type MockAuthenticationServiceServer struct { ctrl *gomock.Controller recorder *MockAuthenticationServiceServerMockRecorder } -// MockAuthenticationServiceServerMockRecorder is the mock recorder for MockAuthenticationServiceServer. +// MockAuthenticationServiceServerMockRecorder is the mock recorder for MockAuthenticationServiceServer type MockAuthenticationServiceServerMockRecorder struct { mock *MockAuthenticationServiceServer } -// NewMockAuthenticationServiceServer creates a new mock instance. +// NewMockAuthenticationServiceServer creates a new mock instance func NewMockAuthenticationServiceServer(ctrl *gomock.Controller) *MockAuthenticationServiceServer { mock := &MockAuthenticationServiceServer{ctrl: ctrl} mock.recorder = &MockAuthenticationServiceServerMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockAuthenticationServiceServer) EXPECT() *MockAuthenticationServiceServerMockRecorder { return m.recorder } -// Authenticate mocks base method. +// Authenticate mocks base method func (m *MockAuthenticationServiceServer) Authenticate(arg0 context.Context, arg1 *AuthenticateRequest) (*AuthenticateResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Authenticate", arg0, arg1) @@ -87,7 +86,7 @@ func (m *MockAuthenticationServiceServer) Authenticate(arg0 context.Context, arg return ret0, ret1 } -// Authenticate indicates an expected call of Authenticate. +// Authenticate indicates an expected call of Authenticate func (mr *MockAuthenticationServiceServerMockRecorder) Authenticate(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Authenticate", reflect.TypeOf((*MockAuthenticationServiceServer)(nil).Authenticate), arg0, arg1) diff --git a/api/interservice/authz/authz.pb.client_mock.go b/api/interservice/authz/authz.pb.client_mock.go index e5a5897e871..a9721ce5107 100644 --- a/api/interservice/authz/authz.pb.client_mock.go +++ b/api/interservice/authz/authz.pb.client_mock.go @@ -6,97 +6,96 @@ package authz import ( context "context" - reflect "reflect" - version "github.com/chef/automate/api/external/common/version" gomock "github.com/golang/mock/gomock" grpc "google.golang.org/grpc" + reflect "reflect" ) -// MockAuthorizationServiceClient is a mock of AuthorizationServiceClient interface. +// MockAuthorizationServiceClient is a mock of AuthorizationServiceClient interface type MockAuthorizationServiceClient struct { ctrl *gomock.Controller recorder *MockAuthorizationServiceClientMockRecorder } -// MockAuthorizationServiceClientMockRecorder is the mock recorder for MockAuthorizationServiceClient. +// MockAuthorizationServiceClientMockRecorder is the mock recorder for MockAuthorizationServiceClient type MockAuthorizationServiceClientMockRecorder struct { mock *MockAuthorizationServiceClient } -// NewMockAuthorizationServiceClient creates a new mock instance. +// NewMockAuthorizationServiceClient creates a new mock instance func NewMockAuthorizationServiceClient(ctrl *gomock.Controller) *MockAuthorizationServiceClient { mock := &MockAuthorizationServiceClient{ctrl: ctrl} mock.recorder = &MockAuthorizationServiceClientMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockAuthorizationServiceClient) EXPECT() *MockAuthorizationServiceClientMockRecorder { return m.recorder } -// FilterAuthorizedPairs mocks base method. -func (m *MockAuthorizationServiceClient) FilterAuthorizedPairs(ctx context.Context, in *FilterAuthorizedPairsReq, opts ...grpc.CallOption) (*FilterAuthorizedPairsResp, error) { +// GetVersion mocks base method +func (m *MockAuthorizationServiceClient) GetVersion(ctx context.Context, in *version.VersionInfoRequest, opts ...grpc.CallOption) (*version.VersionInfo, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "FilterAuthorizedPairs", varargs...) - ret0, _ := ret[0].(*FilterAuthorizedPairsResp) + ret := m.ctrl.Call(m, "GetVersion", varargs...) + ret0, _ := ret[0].(*version.VersionInfo) ret1, _ := ret[1].(error) return ret0, ret1 } -// FilterAuthorizedPairs indicates an expected call of FilterAuthorizedPairs. -func (mr *MockAuthorizationServiceClientMockRecorder) FilterAuthorizedPairs(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetVersion indicates an expected call of GetVersion +func (mr *MockAuthorizationServiceClientMockRecorder) GetVersion(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FilterAuthorizedPairs", reflect.TypeOf((*MockAuthorizationServiceClient)(nil).FilterAuthorizedPairs), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetVersion", reflect.TypeOf((*MockAuthorizationServiceClient)(nil).GetVersion), varargs...) } -// FilterAuthorizedProjects mocks base method. -func (m *MockAuthorizationServiceClient) FilterAuthorizedProjects(ctx context.Context, in *FilterAuthorizedProjectsReq, opts ...grpc.CallOption) (*FilterAuthorizedProjectsResp, error) { +// FilterAuthorizedPairs mocks base method +func (m *MockAuthorizationServiceClient) FilterAuthorizedPairs(ctx context.Context, in *FilterAuthorizedPairsReq, opts ...grpc.CallOption) (*FilterAuthorizedPairsResp, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "FilterAuthorizedProjects", varargs...) - ret0, _ := ret[0].(*FilterAuthorizedProjectsResp) + ret := m.ctrl.Call(m, "FilterAuthorizedPairs", varargs...) + ret0, _ := ret[0].(*FilterAuthorizedPairsResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// FilterAuthorizedProjects indicates an expected call of FilterAuthorizedProjects. -func (mr *MockAuthorizationServiceClientMockRecorder) FilterAuthorizedProjects(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// FilterAuthorizedPairs indicates an expected call of FilterAuthorizedPairs +func (mr *MockAuthorizationServiceClientMockRecorder) FilterAuthorizedPairs(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FilterAuthorizedProjects", reflect.TypeOf((*MockAuthorizationServiceClient)(nil).FilterAuthorizedProjects), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FilterAuthorizedPairs", reflect.TypeOf((*MockAuthorizationServiceClient)(nil).FilterAuthorizedPairs), varargs...) } -// GetVersion mocks base method. -func (m *MockAuthorizationServiceClient) GetVersion(ctx context.Context, in *version.VersionInfoRequest, opts ...grpc.CallOption) (*version.VersionInfo, error) { +// FilterAuthorizedProjects mocks base method +func (m *MockAuthorizationServiceClient) FilterAuthorizedProjects(ctx context.Context, in *FilterAuthorizedProjectsReq, opts ...grpc.CallOption) (*FilterAuthorizedProjectsResp, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetVersion", varargs...) - ret0, _ := ret[0].(*version.VersionInfo) + ret := m.ctrl.Call(m, "FilterAuthorizedProjects", varargs...) + ret0, _ := ret[0].(*FilterAuthorizedProjectsResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetVersion indicates an expected call of GetVersion. -func (mr *MockAuthorizationServiceClientMockRecorder) GetVersion(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// FilterAuthorizedProjects indicates an expected call of FilterAuthorizedProjects +func (mr *MockAuthorizationServiceClientMockRecorder) FilterAuthorizedProjects(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetVersion", reflect.TypeOf((*MockAuthorizationServiceClient)(nil).GetVersion), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FilterAuthorizedProjects", reflect.TypeOf((*MockAuthorizationServiceClient)(nil).FilterAuthorizedProjects), varargs...) } -// ProjectsAuthorized mocks base method. +// ProjectsAuthorized mocks base method func (m *MockAuthorizationServiceClient) ProjectsAuthorized(ctx context.Context, in *ProjectsAuthorizedReq, opts ...grpc.CallOption) (*ProjectsAuthorizedResp, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} @@ -109,14 +108,14 @@ func (m *MockAuthorizationServiceClient) ProjectsAuthorized(ctx context.Context, return ret0, ret1 } -// ProjectsAuthorized indicates an expected call of ProjectsAuthorized. +// ProjectsAuthorized indicates an expected call of ProjectsAuthorized func (mr *MockAuthorizationServiceClientMockRecorder) ProjectsAuthorized(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProjectsAuthorized", reflect.TypeOf((*MockAuthorizationServiceClient)(nil).ProjectsAuthorized), varargs...) } -// ValidateProjectAssignment mocks base method. +// ValidateProjectAssignment mocks base method func (m *MockAuthorizationServiceClient) ValidateProjectAssignment(ctx context.Context, in *ValidateProjectAssignmentReq, opts ...grpc.CallOption) (*ValidateProjectAssignmentResp, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} @@ -129,37 +128,52 @@ func (m *MockAuthorizationServiceClient) ValidateProjectAssignment(ctx context.C return ret0, ret1 } -// ValidateProjectAssignment indicates an expected call of ValidateProjectAssignment. +// ValidateProjectAssignment indicates an expected call of ValidateProjectAssignment func (mr *MockAuthorizationServiceClientMockRecorder) ValidateProjectAssignment(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ValidateProjectAssignment", reflect.TypeOf((*MockAuthorizationServiceClient)(nil).ValidateProjectAssignment), varargs...) } -// MockAuthorizationServiceServer is a mock of AuthorizationServiceServer interface. +// MockAuthorizationServiceServer is a mock of AuthorizationServiceServer interface type MockAuthorizationServiceServer struct { ctrl *gomock.Controller recorder *MockAuthorizationServiceServerMockRecorder } -// MockAuthorizationServiceServerMockRecorder is the mock recorder for MockAuthorizationServiceServer. +// MockAuthorizationServiceServerMockRecorder is the mock recorder for MockAuthorizationServiceServer type MockAuthorizationServiceServerMockRecorder struct { mock *MockAuthorizationServiceServer } -// NewMockAuthorizationServiceServer creates a new mock instance. +// NewMockAuthorizationServiceServer creates a new mock instance func NewMockAuthorizationServiceServer(ctrl *gomock.Controller) *MockAuthorizationServiceServer { mock := &MockAuthorizationServiceServer{ctrl: ctrl} mock.recorder = &MockAuthorizationServiceServerMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockAuthorizationServiceServer) EXPECT() *MockAuthorizationServiceServerMockRecorder { return m.recorder } -// FilterAuthorizedPairs mocks base method. +// GetVersion mocks base method +func (m *MockAuthorizationServiceServer) GetVersion(arg0 context.Context, arg1 *version.VersionInfoRequest) (*version.VersionInfo, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetVersion", arg0, arg1) + ret0, _ := ret[0].(*version.VersionInfo) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetVersion indicates an expected call of GetVersion +func (mr *MockAuthorizationServiceServerMockRecorder) GetVersion(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetVersion", reflect.TypeOf((*MockAuthorizationServiceServer)(nil).GetVersion), arg0, arg1) +} + +// FilterAuthorizedPairs mocks base method func (m *MockAuthorizationServiceServer) FilterAuthorizedPairs(arg0 context.Context, arg1 *FilterAuthorizedPairsReq) (*FilterAuthorizedPairsResp, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "FilterAuthorizedPairs", arg0, arg1) @@ -168,13 +182,13 @@ func (m *MockAuthorizationServiceServer) FilterAuthorizedPairs(arg0 context.Cont return ret0, ret1 } -// FilterAuthorizedPairs indicates an expected call of FilterAuthorizedPairs. +// FilterAuthorizedPairs indicates an expected call of FilterAuthorizedPairs func (mr *MockAuthorizationServiceServerMockRecorder) FilterAuthorizedPairs(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FilterAuthorizedPairs", reflect.TypeOf((*MockAuthorizationServiceServer)(nil).FilterAuthorizedPairs), arg0, arg1) } -// FilterAuthorizedProjects mocks base method. +// FilterAuthorizedProjects mocks base method func (m *MockAuthorizationServiceServer) FilterAuthorizedProjects(arg0 context.Context, arg1 *FilterAuthorizedProjectsReq) (*FilterAuthorizedProjectsResp, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "FilterAuthorizedProjects", arg0, arg1) @@ -183,28 +197,13 @@ func (m *MockAuthorizationServiceServer) FilterAuthorizedProjects(arg0 context.C return ret0, ret1 } -// FilterAuthorizedProjects indicates an expected call of FilterAuthorizedProjects. +// FilterAuthorizedProjects indicates an expected call of FilterAuthorizedProjects func (mr *MockAuthorizationServiceServerMockRecorder) FilterAuthorizedProjects(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FilterAuthorizedProjects", reflect.TypeOf((*MockAuthorizationServiceServer)(nil).FilterAuthorizedProjects), arg0, arg1) } -// GetVersion mocks base method. -func (m *MockAuthorizationServiceServer) GetVersion(arg0 context.Context, arg1 *version.VersionInfoRequest) (*version.VersionInfo, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetVersion", arg0, arg1) - ret0, _ := ret[0].(*version.VersionInfo) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// GetVersion indicates an expected call of GetVersion. -func (mr *MockAuthorizationServiceServerMockRecorder) GetVersion(arg0, arg1 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetVersion", reflect.TypeOf((*MockAuthorizationServiceServer)(nil).GetVersion), arg0, arg1) -} - -// ProjectsAuthorized mocks base method. +// ProjectsAuthorized mocks base method func (m *MockAuthorizationServiceServer) ProjectsAuthorized(arg0 context.Context, arg1 *ProjectsAuthorizedReq) (*ProjectsAuthorizedResp, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ProjectsAuthorized", arg0, arg1) @@ -213,13 +212,13 @@ func (m *MockAuthorizationServiceServer) ProjectsAuthorized(arg0 context.Context return ret0, ret1 } -// ProjectsAuthorized indicates an expected call of ProjectsAuthorized. +// ProjectsAuthorized indicates an expected call of ProjectsAuthorized func (mr *MockAuthorizationServiceServerMockRecorder) ProjectsAuthorized(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProjectsAuthorized", reflect.TypeOf((*MockAuthorizationServiceServer)(nil).ProjectsAuthorized), arg0, arg1) } -// ValidateProjectAssignment mocks base method. +// ValidateProjectAssignment mocks base method func (m *MockAuthorizationServiceServer) ValidateProjectAssignment(arg0 context.Context, arg1 *ValidateProjectAssignmentReq) (*ValidateProjectAssignmentResp, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ValidateProjectAssignment", arg0, arg1) @@ -228,7 +227,7 @@ func (m *MockAuthorizationServiceServer) ValidateProjectAssignment(arg0 context. return ret0, ret1 } -// ValidateProjectAssignment indicates an expected call of ValidateProjectAssignment. +// ValidateProjectAssignment indicates an expected call of ValidateProjectAssignment func (mr *MockAuthorizationServiceServerMockRecorder) ValidateProjectAssignment(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ValidateProjectAssignment", reflect.TypeOf((*MockAuthorizationServiceServer)(nil).ValidateProjectAssignment), arg0, arg1) diff --git a/api/interservice/authz/project.pb.client_mock.go b/api/interservice/authz/project.pb.client_mock.go index c05a64e73e9..2a23ea6fe9d 100644 --- a/api/interservice/authz/project.pb.client_mock.go +++ b/api/interservice/authz/project.pb.client_mock.go @@ -6,296 +6,295 @@ package authz import ( context "context" - reflect "reflect" - gomock "github.com/golang/mock/gomock" grpc "google.golang.org/grpc" + reflect "reflect" ) -// MockProjectsServiceClient is a mock of ProjectsServiceClient interface. +// MockProjectsServiceClient is a mock of ProjectsServiceClient interface type MockProjectsServiceClient struct { ctrl *gomock.Controller recorder *MockProjectsServiceClientMockRecorder } -// MockProjectsServiceClientMockRecorder is the mock recorder for MockProjectsServiceClient. +// MockProjectsServiceClientMockRecorder is the mock recorder for MockProjectsServiceClient type MockProjectsServiceClientMockRecorder struct { mock *MockProjectsServiceClient } -// NewMockProjectsServiceClient creates a new mock instance. +// NewMockProjectsServiceClient creates a new mock instance func NewMockProjectsServiceClient(ctrl *gomock.Controller) *MockProjectsServiceClient { mock := &MockProjectsServiceClient{ctrl: ctrl} mock.recorder = &MockProjectsServiceClientMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockProjectsServiceClient) EXPECT() *MockProjectsServiceClientMockRecorder { return m.recorder } -// ApplyRulesCancel mocks base method. -func (m *MockProjectsServiceClient) ApplyRulesCancel(ctx context.Context, in *ApplyRulesCancelReq, opts ...grpc.CallOption) (*ApplyRulesCancelResp, error) { +// UpdateProject mocks base method +func (m *MockProjectsServiceClient) UpdateProject(ctx context.Context, in *UpdateProjectReq, opts ...grpc.CallOption) (*UpdateProjectResp, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "ApplyRulesCancel", varargs...) - ret0, _ := ret[0].(*ApplyRulesCancelResp) + ret := m.ctrl.Call(m, "UpdateProject", varargs...) + ret0, _ := ret[0].(*UpdateProjectResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// ApplyRulesCancel indicates an expected call of ApplyRulesCancel. -func (mr *MockProjectsServiceClientMockRecorder) ApplyRulesCancel(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// UpdateProject indicates an expected call of UpdateProject +func (mr *MockProjectsServiceClientMockRecorder) UpdateProject(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ApplyRulesCancel", reflect.TypeOf((*MockProjectsServiceClient)(nil).ApplyRulesCancel), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateProject", reflect.TypeOf((*MockProjectsServiceClient)(nil).UpdateProject), varargs...) } -// ApplyRulesStart mocks base method. -func (m *MockProjectsServiceClient) ApplyRulesStart(ctx context.Context, in *ApplyRulesStartReq, opts ...grpc.CallOption) (*ApplyRulesStartResp, error) { +// CreateProject mocks base method +func (m *MockProjectsServiceClient) CreateProject(ctx context.Context, in *CreateProjectReq, opts ...grpc.CallOption) (*CreateProjectResp, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "ApplyRulesStart", varargs...) - ret0, _ := ret[0].(*ApplyRulesStartResp) + ret := m.ctrl.Call(m, "CreateProject", varargs...) + ret0, _ := ret[0].(*CreateProjectResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// ApplyRulesStart indicates an expected call of ApplyRulesStart. -func (mr *MockProjectsServiceClientMockRecorder) ApplyRulesStart(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// CreateProject indicates an expected call of CreateProject +func (mr *MockProjectsServiceClientMockRecorder) CreateProject(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ApplyRulesStart", reflect.TypeOf((*MockProjectsServiceClient)(nil).ApplyRulesStart), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateProject", reflect.TypeOf((*MockProjectsServiceClient)(nil).CreateProject), varargs...) } -// ApplyRulesStatus mocks base method. -func (m *MockProjectsServiceClient) ApplyRulesStatus(ctx context.Context, in *ApplyRulesStatusReq, opts ...grpc.CallOption) (*ApplyRulesStatusResp, error) { +// GetProject mocks base method +func (m *MockProjectsServiceClient) GetProject(ctx context.Context, in *GetProjectReq, opts ...grpc.CallOption) (*GetProjectResp, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "ApplyRulesStatus", varargs...) - ret0, _ := ret[0].(*ApplyRulesStatusResp) + ret := m.ctrl.Call(m, "GetProject", varargs...) + ret0, _ := ret[0].(*GetProjectResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// ApplyRulesStatus indicates an expected call of ApplyRulesStatus. -func (mr *MockProjectsServiceClientMockRecorder) ApplyRulesStatus(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetProject indicates an expected call of GetProject +func (mr *MockProjectsServiceClientMockRecorder) GetProject(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ApplyRulesStatus", reflect.TypeOf((*MockProjectsServiceClient)(nil).ApplyRulesStatus), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProject", reflect.TypeOf((*MockProjectsServiceClient)(nil).GetProject), varargs...) } -// CreateProject mocks base method. -func (m *MockProjectsServiceClient) CreateProject(ctx context.Context, in *CreateProjectReq, opts ...grpc.CallOption) (*CreateProjectResp, error) { +// DeleteProject mocks base method +func (m *MockProjectsServiceClient) DeleteProject(ctx context.Context, in *DeleteProjectReq, opts ...grpc.CallOption) (*DeleteProjectResp, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "CreateProject", varargs...) - ret0, _ := ret[0].(*CreateProjectResp) + ret := m.ctrl.Call(m, "DeleteProject", varargs...) + ret0, _ := ret[0].(*DeleteProjectResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// CreateProject indicates an expected call of CreateProject. -func (mr *MockProjectsServiceClientMockRecorder) CreateProject(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// DeleteProject indicates an expected call of DeleteProject +func (mr *MockProjectsServiceClientMockRecorder) DeleteProject(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateProject", reflect.TypeOf((*MockProjectsServiceClient)(nil).CreateProject), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteProject", reflect.TypeOf((*MockProjectsServiceClient)(nil).DeleteProject), varargs...) } -// CreateRule mocks base method. -func (m *MockProjectsServiceClient) CreateRule(ctx context.Context, in *CreateRuleReq, opts ...grpc.CallOption) (*CreateRuleResp, error) { +// ListProjects mocks base method +func (m *MockProjectsServiceClient) ListProjects(ctx context.Context, in *ListProjectsReq, opts ...grpc.CallOption) (*ListProjectsResp, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "CreateRule", varargs...) - ret0, _ := ret[0].(*CreateRuleResp) + ret := m.ctrl.Call(m, "ListProjects", varargs...) + ret0, _ := ret[0].(*ListProjectsResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// CreateRule indicates an expected call of CreateRule. -func (mr *MockProjectsServiceClientMockRecorder) CreateRule(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// ListProjects indicates an expected call of ListProjects +func (mr *MockProjectsServiceClientMockRecorder) ListProjects(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateRule", reflect.TypeOf((*MockProjectsServiceClient)(nil).CreateRule), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListProjects", reflect.TypeOf((*MockProjectsServiceClient)(nil).ListProjects), varargs...) } -// DeleteProject mocks base method. -func (m *MockProjectsServiceClient) DeleteProject(ctx context.Context, in *DeleteProjectReq, opts ...grpc.CallOption) (*DeleteProjectResp, error) { +// ListProjectsForIntrospection mocks base method +func (m *MockProjectsServiceClient) ListProjectsForIntrospection(ctx context.Context, in *ListProjectsReq, opts ...grpc.CallOption) (*ListProjectsResp, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "DeleteProject", varargs...) - ret0, _ := ret[0].(*DeleteProjectResp) + ret := m.ctrl.Call(m, "ListProjectsForIntrospection", varargs...) + ret0, _ := ret[0].(*ListProjectsResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// DeleteProject indicates an expected call of DeleteProject. -func (mr *MockProjectsServiceClientMockRecorder) DeleteProject(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// ListProjectsForIntrospection indicates an expected call of ListProjectsForIntrospection +func (mr *MockProjectsServiceClientMockRecorder) ListProjectsForIntrospection(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteProject", reflect.TypeOf((*MockProjectsServiceClient)(nil).DeleteProject), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListProjectsForIntrospection", reflect.TypeOf((*MockProjectsServiceClient)(nil).ListProjectsForIntrospection), varargs...) } -// DeleteRule mocks base method. -func (m *MockProjectsServiceClient) DeleteRule(ctx context.Context, in *DeleteRuleReq, opts ...grpc.CallOption) (*DeleteRuleResp, error) { +// ApplyRulesStart mocks base method +func (m *MockProjectsServiceClient) ApplyRulesStart(ctx context.Context, in *ApplyRulesStartReq, opts ...grpc.CallOption) (*ApplyRulesStartResp, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "DeleteRule", varargs...) - ret0, _ := ret[0].(*DeleteRuleResp) + ret := m.ctrl.Call(m, "ApplyRulesStart", varargs...) + ret0, _ := ret[0].(*ApplyRulesStartResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// DeleteRule indicates an expected call of DeleteRule. -func (mr *MockProjectsServiceClientMockRecorder) DeleteRule(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// ApplyRulesStart indicates an expected call of ApplyRulesStart +func (mr *MockProjectsServiceClientMockRecorder) ApplyRulesStart(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteRule", reflect.TypeOf((*MockProjectsServiceClient)(nil).DeleteRule), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ApplyRulesStart", reflect.TypeOf((*MockProjectsServiceClient)(nil).ApplyRulesStart), varargs...) } -// GetProject mocks base method. -func (m *MockProjectsServiceClient) GetProject(ctx context.Context, in *GetProjectReq, opts ...grpc.CallOption) (*GetProjectResp, error) { +// ApplyRulesCancel mocks base method +func (m *MockProjectsServiceClient) ApplyRulesCancel(ctx context.Context, in *ApplyRulesCancelReq, opts ...grpc.CallOption) (*ApplyRulesCancelResp, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetProject", varargs...) - ret0, _ := ret[0].(*GetProjectResp) + ret := m.ctrl.Call(m, "ApplyRulesCancel", varargs...) + ret0, _ := ret[0].(*ApplyRulesCancelResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetProject indicates an expected call of GetProject. -func (mr *MockProjectsServiceClientMockRecorder) GetProject(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// ApplyRulesCancel indicates an expected call of ApplyRulesCancel +func (mr *MockProjectsServiceClientMockRecorder) ApplyRulesCancel(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProject", reflect.TypeOf((*MockProjectsServiceClient)(nil).GetProject), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ApplyRulesCancel", reflect.TypeOf((*MockProjectsServiceClient)(nil).ApplyRulesCancel), varargs...) } -// GetRule mocks base method. -func (m *MockProjectsServiceClient) GetRule(ctx context.Context, in *GetRuleReq, opts ...grpc.CallOption) (*GetRuleResp, error) { +// ApplyRulesStatus mocks base method +func (m *MockProjectsServiceClient) ApplyRulesStatus(ctx context.Context, in *ApplyRulesStatusReq, opts ...grpc.CallOption) (*ApplyRulesStatusResp, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetRule", varargs...) - ret0, _ := ret[0].(*GetRuleResp) + ret := m.ctrl.Call(m, "ApplyRulesStatus", varargs...) + ret0, _ := ret[0].(*ApplyRulesStatusResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetRule indicates an expected call of GetRule. -func (mr *MockProjectsServiceClientMockRecorder) GetRule(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// ApplyRulesStatus indicates an expected call of ApplyRulesStatus +func (mr *MockProjectsServiceClientMockRecorder) ApplyRulesStatus(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRule", reflect.TypeOf((*MockProjectsServiceClient)(nil).GetRule), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ApplyRulesStatus", reflect.TypeOf((*MockProjectsServiceClient)(nil).ApplyRulesStatus), varargs...) } -// ListProjects mocks base method. -func (m *MockProjectsServiceClient) ListProjects(ctx context.Context, in *ListProjectsReq, opts ...grpc.CallOption) (*ListProjectsResp, error) { +// CreateRule mocks base method +func (m *MockProjectsServiceClient) CreateRule(ctx context.Context, in *CreateRuleReq, opts ...grpc.CallOption) (*CreateRuleResp, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "ListProjects", varargs...) - ret0, _ := ret[0].(*ListProjectsResp) + ret := m.ctrl.Call(m, "CreateRule", varargs...) + ret0, _ := ret[0].(*CreateRuleResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// ListProjects indicates an expected call of ListProjects. -func (mr *MockProjectsServiceClientMockRecorder) ListProjects(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// CreateRule indicates an expected call of CreateRule +func (mr *MockProjectsServiceClientMockRecorder) CreateRule(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListProjects", reflect.TypeOf((*MockProjectsServiceClient)(nil).ListProjects), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateRule", reflect.TypeOf((*MockProjectsServiceClient)(nil).CreateRule), varargs...) } -// ListProjectsForIntrospection mocks base method. -func (m *MockProjectsServiceClient) ListProjectsForIntrospection(ctx context.Context, in *ListProjectsReq, opts ...grpc.CallOption) (*ListProjectsResp, error) { +// UpdateRule mocks base method +func (m *MockProjectsServiceClient) UpdateRule(ctx context.Context, in *UpdateRuleReq, opts ...grpc.CallOption) (*UpdateRuleResp, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "ListProjectsForIntrospection", varargs...) - ret0, _ := ret[0].(*ListProjectsResp) + ret := m.ctrl.Call(m, "UpdateRule", varargs...) + ret0, _ := ret[0].(*UpdateRuleResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// ListProjectsForIntrospection indicates an expected call of ListProjectsForIntrospection. -func (mr *MockProjectsServiceClientMockRecorder) ListProjectsForIntrospection(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// UpdateRule indicates an expected call of UpdateRule +func (mr *MockProjectsServiceClientMockRecorder) UpdateRule(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListProjectsForIntrospection", reflect.TypeOf((*MockProjectsServiceClient)(nil).ListProjectsForIntrospection), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateRule", reflect.TypeOf((*MockProjectsServiceClient)(nil).UpdateRule), varargs...) } -// ListRules mocks base method. -func (m *MockProjectsServiceClient) ListRules(ctx context.Context, in *ListRulesReq, opts ...grpc.CallOption) (*ListRulesResp, error) { +// GetRule mocks base method +func (m *MockProjectsServiceClient) GetRule(ctx context.Context, in *GetRuleReq, opts ...grpc.CallOption) (*GetRuleResp, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "ListRules", varargs...) - ret0, _ := ret[0].(*ListRulesResp) + ret := m.ctrl.Call(m, "GetRule", varargs...) + ret0, _ := ret[0].(*GetRuleResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// ListRules indicates an expected call of ListRules. -func (mr *MockProjectsServiceClientMockRecorder) ListRules(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetRule indicates an expected call of GetRule +func (mr *MockProjectsServiceClientMockRecorder) GetRule(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListRules", reflect.TypeOf((*MockProjectsServiceClient)(nil).ListRules), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRule", reflect.TypeOf((*MockProjectsServiceClient)(nil).GetRule), varargs...) } -// ListRulesForAllProjects mocks base method. -func (m *MockProjectsServiceClient) ListRulesForAllProjects(ctx context.Context, in *ListRulesForAllProjectsReq, opts ...grpc.CallOption) (*ListRulesForAllProjectsResp, error) { +// ListRules mocks base method +func (m *MockProjectsServiceClient) ListRules(ctx context.Context, in *ListRulesReq, opts ...grpc.CallOption) (*ListRulesResp, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "ListRulesForAllProjects", varargs...) - ret0, _ := ret[0].(*ListRulesForAllProjectsResp) + ret := m.ctrl.Call(m, "ListRules", varargs...) + ret0, _ := ret[0].(*ListRulesResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// ListRulesForAllProjects indicates an expected call of ListRulesForAllProjects. -func (mr *MockProjectsServiceClientMockRecorder) ListRulesForAllProjects(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// ListRules indicates an expected call of ListRules +func (mr *MockProjectsServiceClientMockRecorder) ListRules(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListRulesForAllProjects", reflect.TypeOf((*MockProjectsServiceClient)(nil).ListRulesForAllProjects), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListRules", reflect.TypeOf((*MockProjectsServiceClient)(nil).ListRules), varargs...) } -// ListRulesForProject mocks base method. +// ListRulesForProject mocks base method func (m *MockProjectsServiceClient) ListRulesForProject(ctx context.Context, in *ListRulesForProjectReq, opts ...grpc.CallOption) (*ListRulesForProjectResp, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} @@ -308,272 +307,272 @@ func (m *MockProjectsServiceClient) ListRulesForProject(ctx context.Context, in return ret0, ret1 } -// ListRulesForProject indicates an expected call of ListRulesForProject. +// ListRulesForProject indicates an expected call of ListRulesForProject func (mr *MockProjectsServiceClientMockRecorder) ListRulesForProject(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListRulesForProject", reflect.TypeOf((*MockProjectsServiceClient)(nil).ListRulesForProject), varargs...) } -// UpdateProject mocks base method. -func (m *MockProjectsServiceClient) UpdateProject(ctx context.Context, in *UpdateProjectReq, opts ...grpc.CallOption) (*UpdateProjectResp, error) { +// DeleteRule mocks base method +func (m *MockProjectsServiceClient) DeleteRule(ctx context.Context, in *DeleteRuleReq, opts ...grpc.CallOption) (*DeleteRuleResp, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "UpdateProject", varargs...) - ret0, _ := ret[0].(*UpdateProjectResp) + ret := m.ctrl.Call(m, "DeleteRule", varargs...) + ret0, _ := ret[0].(*DeleteRuleResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateProject indicates an expected call of UpdateProject. -func (mr *MockProjectsServiceClientMockRecorder) UpdateProject(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// DeleteRule indicates an expected call of DeleteRule +func (mr *MockProjectsServiceClientMockRecorder) DeleteRule(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateProject", reflect.TypeOf((*MockProjectsServiceClient)(nil).UpdateProject), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteRule", reflect.TypeOf((*MockProjectsServiceClient)(nil).DeleteRule), varargs...) } -// UpdateRule mocks base method. -func (m *MockProjectsServiceClient) UpdateRule(ctx context.Context, in *UpdateRuleReq, opts ...grpc.CallOption) (*UpdateRuleResp, error) { +// ListRulesForAllProjects mocks base method +func (m *MockProjectsServiceClient) ListRulesForAllProjects(ctx context.Context, in *ListRulesForAllProjectsReq, opts ...grpc.CallOption) (*ListRulesForAllProjectsResp, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "UpdateRule", varargs...) - ret0, _ := ret[0].(*UpdateRuleResp) + ret := m.ctrl.Call(m, "ListRulesForAllProjects", varargs...) + ret0, _ := ret[0].(*ListRulesForAllProjectsResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateRule indicates an expected call of UpdateRule. -func (mr *MockProjectsServiceClientMockRecorder) UpdateRule(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// ListRulesForAllProjects indicates an expected call of ListRulesForAllProjects +func (mr *MockProjectsServiceClientMockRecorder) ListRulesForAllProjects(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateRule", reflect.TypeOf((*MockProjectsServiceClient)(nil).UpdateRule), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListRulesForAllProjects", reflect.TypeOf((*MockProjectsServiceClient)(nil).ListRulesForAllProjects), varargs...) } -// MockProjectsServiceServer is a mock of ProjectsServiceServer interface. +// MockProjectsServiceServer is a mock of ProjectsServiceServer interface type MockProjectsServiceServer struct { ctrl *gomock.Controller recorder *MockProjectsServiceServerMockRecorder } -// MockProjectsServiceServerMockRecorder is the mock recorder for MockProjectsServiceServer. +// MockProjectsServiceServerMockRecorder is the mock recorder for MockProjectsServiceServer type MockProjectsServiceServerMockRecorder struct { mock *MockProjectsServiceServer } -// NewMockProjectsServiceServer creates a new mock instance. +// NewMockProjectsServiceServer creates a new mock instance func NewMockProjectsServiceServer(ctrl *gomock.Controller) *MockProjectsServiceServer { mock := &MockProjectsServiceServer{ctrl: ctrl} mock.recorder = &MockProjectsServiceServerMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockProjectsServiceServer) EXPECT() *MockProjectsServiceServerMockRecorder { return m.recorder } -// ApplyRulesCancel mocks base method. -func (m *MockProjectsServiceServer) ApplyRulesCancel(arg0 context.Context, arg1 *ApplyRulesCancelReq) (*ApplyRulesCancelResp, error) { +// UpdateProject mocks base method +func (m *MockProjectsServiceServer) UpdateProject(arg0 context.Context, arg1 *UpdateProjectReq) (*UpdateProjectResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ApplyRulesCancel", arg0, arg1) - ret0, _ := ret[0].(*ApplyRulesCancelResp) + ret := m.ctrl.Call(m, "UpdateProject", arg0, arg1) + ret0, _ := ret[0].(*UpdateProjectResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// ApplyRulesCancel indicates an expected call of ApplyRulesCancel. -func (mr *MockProjectsServiceServerMockRecorder) ApplyRulesCancel(arg0, arg1 interface{}) *gomock.Call { +// UpdateProject indicates an expected call of UpdateProject +func (mr *MockProjectsServiceServerMockRecorder) UpdateProject(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ApplyRulesCancel", reflect.TypeOf((*MockProjectsServiceServer)(nil).ApplyRulesCancel), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateProject", reflect.TypeOf((*MockProjectsServiceServer)(nil).UpdateProject), arg0, arg1) } -// ApplyRulesStart mocks base method. -func (m *MockProjectsServiceServer) ApplyRulesStart(arg0 context.Context, arg1 *ApplyRulesStartReq) (*ApplyRulesStartResp, error) { +// CreateProject mocks base method +func (m *MockProjectsServiceServer) CreateProject(arg0 context.Context, arg1 *CreateProjectReq) (*CreateProjectResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ApplyRulesStart", arg0, arg1) - ret0, _ := ret[0].(*ApplyRulesStartResp) + ret := m.ctrl.Call(m, "CreateProject", arg0, arg1) + ret0, _ := ret[0].(*CreateProjectResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// ApplyRulesStart indicates an expected call of ApplyRulesStart. -func (mr *MockProjectsServiceServerMockRecorder) ApplyRulesStart(arg0, arg1 interface{}) *gomock.Call { +// CreateProject indicates an expected call of CreateProject +func (mr *MockProjectsServiceServerMockRecorder) CreateProject(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ApplyRulesStart", reflect.TypeOf((*MockProjectsServiceServer)(nil).ApplyRulesStart), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateProject", reflect.TypeOf((*MockProjectsServiceServer)(nil).CreateProject), arg0, arg1) } -// ApplyRulesStatus mocks base method. -func (m *MockProjectsServiceServer) ApplyRulesStatus(arg0 context.Context, arg1 *ApplyRulesStatusReq) (*ApplyRulesStatusResp, error) { +// GetProject mocks base method +func (m *MockProjectsServiceServer) GetProject(arg0 context.Context, arg1 *GetProjectReq) (*GetProjectResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ApplyRulesStatus", arg0, arg1) - ret0, _ := ret[0].(*ApplyRulesStatusResp) + ret := m.ctrl.Call(m, "GetProject", arg0, arg1) + ret0, _ := ret[0].(*GetProjectResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// ApplyRulesStatus indicates an expected call of ApplyRulesStatus. -func (mr *MockProjectsServiceServerMockRecorder) ApplyRulesStatus(arg0, arg1 interface{}) *gomock.Call { +// GetProject indicates an expected call of GetProject +func (mr *MockProjectsServiceServerMockRecorder) GetProject(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ApplyRulesStatus", reflect.TypeOf((*MockProjectsServiceServer)(nil).ApplyRulesStatus), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProject", reflect.TypeOf((*MockProjectsServiceServer)(nil).GetProject), arg0, arg1) } -// CreateProject mocks base method. -func (m *MockProjectsServiceServer) CreateProject(arg0 context.Context, arg1 *CreateProjectReq) (*CreateProjectResp, error) { +// DeleteProject mocks base method +func (m *MockProjectsServiceServer) DeleteProject(arg0 context.Context, arg1 *DeleteProjectReq) (*DeleteProjectResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CreateProject", arg0, arg1) - ret0, _ := ret[0].(*CreateProjectResp) + ret := m.ctrl.Call(m, "DeleteProject", arg0, arg1) + ret0, _ := ret[0].(*DeleteProjectResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// CreateProject indicates an expected call of CreateProject. -func (mr *MockProjectsServiceServerMockRecorder) CreateProject(arg0, arg1 interface{}) *gomock.Call { +// DeleteProject indicates an expected call of DeleteProject +func (mr *MockProjectsServiceServerMockRecorder) DeleteProject(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateProject", reflect.TypeOf((*MockProjectsServiceServer)(nil).CreateProject), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteProject", reflect.TypeOf((*MockProjectsServiceServer)(nil).DeleteProject), arg0, arg1) } -// CreateRule mocks base method. -func (m *MockProjectsServiceServer) CreateRule(arg0 context.Context, arg1 *CreateRuleReq) (*CreateRuleResp, error) { +// ListProjects mocks base method +func (m *MockProjectsServiceServer) ListProjects(arg0 context.Context, arg1 *ListProjectsReq) (*ListProjectsResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CreateRule", arg0, arg1) - ret0, _ := ret[0].(*CreateRuleResp) + ret := m.ctrl.Call(m, "ListProjects", arg0, arg1) + ret0, _ := ret[0].(*ListProjectsResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// CreateRule indicates an expected call of CreateRule. -func (mr *MockProjectsServiceServerMockRecorder) CreateRule(arg0, arg1 interface{}) *gomock.Call { +// ListProjects indicates an expected call of ListProjects +func (mr *MockProjectsServiceServerMockRecorder) ListProjects(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateRule", reflect.TypeOf((*MockProjectsServiceServer)(nil).CreateRule), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListProjects", reflect.TypeOf((*MockProjectsServiceServer)(nil).ListProjects), arg0, arg1) } -// DeleteProject mocks base method. -func (m *MockProjectsServiceServer) DeleteProject(arg0 context.Context, arg1 *DeleteProjectReq) (*DeleteProjectResp, error) { +// ListProjectsForIntrospection mocks base method +func (m *MockProjectsServiceServer) ListProjectsForIntrospection(arg0 context.Context, arg1 *ListProjectsReq) (*ListProjectsResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteProject", arg0, arg1) - ret0, _ := ret[0].(*DeleteProjectResp) + ret := m.ctrl.Call(m, "ListProjectsForIntrospection", arg0, arg1) + ret0, _ := ret[0].(*ListProjectsResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// DeleteProject indicates an expected call of DeleteProject. -func (mr *MockProjectsServiceServerMockRecorder) DeleteProject(arg0, arg1 interface{}) *gomock.Call { +// ListProjectsForIntrospection indicates an expected call of ListProjectsForIntrospection +func (mr *MockProjectsServiceServerMockRecorder) ListProjectsForIntrospection(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteProject", reflect.TypeOf((*MockProjectsServiceServer)(nil).DeleteProject), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListProjectsForIntrospection", reflect.TypeOf((*MockProjectsServiceServer)(nil).ListProjectsForIntrospection), arg0, arg1) } -// DeleteRule mocks base method. -func (m *MockProjectsServiceServer) DeleteRule(arg0 context.Context, arg1 *DeleteRuleReq) (*DeleteRuleResp, error) { +// ApplyRulesStart mocks base method +func (m *MockProjectsServiceServer) ApplyRulesStart(arg0 context.Context, arg1 *ApplyRulesStartReq) (*ApplyRulesStartResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteRule", arg0, arg1) - ret0, _ := ret[0].(*DeleteRuleResp) + ret := m.ctrl.Call(m, "ApplyRulesStart", arg0, arg1) + ret0, _ := ret[0].(*ApplyRulesStartResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// DeleteRule indicates an expected call of DeleteRule. -func (mr *MockProjectsServiceServerMockRecorder) DeleteRule(arg0, arg1 interface{}) *gomock.Call { +// ApplyRulesStart indicates an expected call of ApplyRulesStart +func (mr *MockProjectsServiceServerMockRecorder) ApplyRulesStart(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteRule", reflect.TypeOf((*MockProjectsServiceServer)(nil).DeleteRule), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ApplyRulesStart", reflect.TypeOf((*MockProjectsServiceServer)(nil).ApplyRulesStart), arg0, arg1) } -// GetProject mocks base method. -func (m *MockProjectsServiceServer) GetProject(arg0 context.Context, arg1 *GetProjectReq) (*GetProjectResp, error) { +// ApplyRulesCancel mocks base method +func (m *MockProjectsServiceServer) ApplyRulesCancel(arg0 context.Context, arg1 *ApplyRulesCancelReq) (*ApplyRulesCancelResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetProject", arg0, arg1) - ret0, _ := ret[0].(*GetProjectResp) + ret := m.ctrl.Call(m, "ApplyRulesCancel", arg0, arg1) + ret0, _ := ret[0].(*ApplyRulesCancelResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetProject indicates an expected call of GetProject. -func (mr *MockProjectsServiceServerMockRecorder) GetProject(arg0, arg1 interface{}) *gomock.Call { +// ApplyRulesCancel indicates an expected call of ApplyRulesCancel +func (mr *MockProjectsServiceServerMockRecorder) ApplyRulesCancel(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProject", reflect.TypeOf((*MockProjectsServiceServer)(nil).GetProject), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ApplyRulesCancel", reflect.TypeOf((*MockProjectsServiceServer)(nil).ApplyRulesCancel), arg0, arg1) } -// GetRule mocks base method. -func (m *MockProjectsServiceServer) GetRule(arg0 context.Context, arg1 *GetRuleReq) (*GetRuleResp, error) { +// ApplyRulesStatus mocks base method +func (m *MockProjectsServiceServer) ApplyRulesStatus(arg0 context.Context, arg1 *ApplyRulesStatusReq) (*ApplyRulesStatusResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetRule", arg0, arg1) - ret0, _ := ret[0].(*GetRuleResp) + ret := m.ctrl.Call(m, "ApplyRulesStatus", arg0, arg1) + ret0, _ := ret[0].(*ApplyRulesStatusResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetRule indicates an expected call of GetRule. -func (mr *MockProjectsServiceServerMockRecorder) GetRule(arg0, arg1 interface{}) *gomock.Call { +// ApplyRulesStatus indicates an expected call of ApplyRulesStatus +func (mr *MockProjectsServiceServerMockRecorder) ApplyRulesStatus(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRule", reflect.TypeOf((*MockProjectsServiceServer)(nil).GetRule), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ApplyRulesStatus", reflect.TypeOf((*MockProjectsServiceServer)(nil).ApplyRulesStatus), arg0, arg1) } -// ListProjects mocks base method. -func (m *MockProjectsServiceServer) ListProjects(arg0 context.Context, arg1 *ListProjectsReq) (*ListProjectsResp, error) { +// CreateRule mocks base method +func (m *MockProjectsServiceServer) CreateRule(arg0 context.Context, arg1 *CreateRuleReq) (*CreateRuleResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ListProjects", arg0, arg1) - ret0, _ := ret[0].(*ListProjectsResp) + ret := m.ctrl.Call(m, "CreateRule", arg0, arg1) + ret0, _ := ret[0].(*CreateRuleResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// ListProjects indicates an expected call of ListProjects. -func (mr *MockProjectsServiceServerMockRecorder) ListProjects(arg0, arg1 interface{}) *gomock.Call { +// CreateRule indicates an expected call of CreateRule +func (mr *MockProjectsServiceServerMockRecorder) CreateRule(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListProjects", reflect.TypeOf((*MockProjectsServiceServer)(nil).ListProjects), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateRule", reflect.TypeOf((*MockProjectsServiceServer)(nil).CreateRule), arg0, arg1) } -// ListProjectsForIntrospection mocks base method. -func (m *MockProjectsServiceServer) ListProjectsForIntrospection(arg0 context.Context, arg1 *ListProjectsReq) (*ListProjectsResp, error) { +// UpdateRule mocks base method +func (m *MockProjectsServiceServer) UpdateRule(arg0 context.Context, arg1 *UpdateRuleReq) (*UpdateRuleResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ListProjectsForIntrospection", arg0, arg1) - ret0, _ := ret[0].(*ListProjectsResp) + ret := m.ctrl.Call(m, "UpdateRule", arg0, arg1) + ret0, _ := ret[0].(*UpdateRuleResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// ListProjectsForIntrospection indicates an expected call of ListProjectsForIntrospection. -func (mr *MockProjectsServiceServerMockRecorder) ListProjectsForIntrospection(arg0, arg1 interface{}) *gomock.Call { +// UpdateRule indicates an expected call of UpdateRule +func (mr *MockProjectsServiceServerMockRecorder) UpdateRule(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListProjectsForIntrospection", reflect.TypeOf((*MockProjectsServiceServer)(nil).ListProjectsForIntrospection), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateRule", reflect.TypeOf((*MockProjectsServiceServer)(nil).UpdateRule), arg0, arg1) } -// ListRules mocks base method. -func (m *MockProjectsServiceServer) ListRules(arg0 context.Context, arg1 *ListRulesReq) (*ListRulesResp, error) { +// GetRule mocks base method +func (m *MockProjectsServiceServer) GetRule(arg0 context.Context, arg1 *GetRuleReq) (*GetRuleResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ListRules", arg0, arg1) - ret0, _ := ret[0].(*ListRulesResp) + ret := m.ctrl.Call(m, "GetRule", arg0, arg1) + ret0, _ := ret[0].(*GetRuleResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// ListRules indicates an expected call of ListRules. -func (mr *MockProjectsServiceServerMockRecorder) ListRules(arg0, arg1 interface{}) *gomock.Call { +// GetRule indicates an expected call of GetRule +func (mr *MockProjectsServiceServerMockRecorder) GetRule(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListRules", reflect.TypeOf((*MockProjectsServiceServer)(nil).ListRules), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRule", reflect.TypeOf((*MockProjectsServiceServer)(nil).GetRule), arg0, arg1) } -// ListRulesForAllProjects mocks base method. -func (m *MockProjectsServiceServer) ListRulesForAllProjects(arg0 context.Context, arg1 *ListRulesForAllProjectsReq) (*ListRulesForAllProjectsResp, error) { +// ListRules mocks base method +func (m *MockProjectsServiceServer) ListRules(arg0 context.Context, arg1 *ListRulesReq) (*ListRulesResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ListRulesForAllProjects", arg0, arg1) - ret0, _ := ret[0].(*ListRulesForAllProjectsResp) + ret := m.ctrl.Call(m, "ListRules", arg0, arg1) + ret0, _ := ret[0].(*ListRulesResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// ListRulesForAllProjects indicates an expected call of ListRulesForAllProjects. -func (mr *MockProjectsServiceServerMockRecorder) ListRulesForAllProjects(arg0, arg1 interface{}) *gomock.Call { +// ListRules indicates an expected call of ListRules +func (mr *MockProjectsServiceServerMockRecorder) ListRules(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListRulesForAllProjects", reflect.TypeOf((*MockProjectsServiceServer)(nil).ListRulesForAllProjects), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListRules", reflect.TypeOf((*MockProjectsServiceServer)(nil).ListRules), arg0, arg1) } -// ListRulesForProject mocks base method. +// ListRulesForProject mocks base method func (m *MockProjectsServiceServer) ListRulesForProject(arg0 context.Context, arg1 *ListRulesForProjectReq) (*ListRulesForProjectResp, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ListRulesForProject", arg0, arg1) @@ -582,38 +581,38 @@ func (m *MockProjectsServiceServer) ListRulesForProject(arg0 context.Context, ar return ret0, ret1 } -// ListRulesForProject indicates an expected call of ListRulesForProject. +// ListRulesForProject indicates an expected call of ListRulesForProject func (mr *MockProjectsServiceServerMockRecorder) ListRulesForProject(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListRulesForProject", reflect.TypeOf((*MockProjectsServiceServer)(nil).ListRulesForProject), arg0, arg1) } -// UpdateProject mocks base method. -func (m *MockProjectsServiceServer) UpdateProject(arg0 context.Context, arg1 *UpdateProjectReq) (*UpdateProjectResp, error) { +// DeleteRule mocks base method +func (m *MockProjectsServiceServer) DeleteRule(arg0 context.Context, arg1 *DeleteRuleReq) (*DeleteRuleResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateProject", arg0, arg1) - ret0, _ := ret[0].(*UpdateProjectResp) + ret := m.ctrl.Call(m, "DeleteRule", arg0, arg1) + ret0, _ := ret[0].(*DeleteRuleResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateProject indicates an expected call of UpdateProject. -func (mr *MockProjectsServiceServerMockRecorder) UpdateProject(arg0, arg1 interface{}) *gomock.Call { +// DeleteRule indicates an expected call of DeleteRule +func (mr *MockProjectsServiceServerMockRecorder) DeleteRule(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateProject", reflect.TypeOf((*MockProjectsServiceServer)(nil).UpdateProject), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteRule", reflect.TypeOf((*MockProjectsServiceServer)(nil).DeleteRule), arg0, arg1) } -// UpdateRule mocks base method. -func (m *MockProjectsServiceServer) UpdateRule(arg0 context.Context, arg1 *UpdateRuleReq) (*UpdateRuleResp, error) { +// ListRulesForAllProjects mocks base method +func (m *MockProjectsServiceServer) ListRulesForAllProjects(arg0 context.Context, arg1 *ListRulesForAllProjectsReq) (*ListRulesForAllProjectsResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateRule", arg0, arg1) - ret0, _ := ret[0].(*UpdateRuleResp) + ret := m.ctrl.Call(m, "ListRulesForAllProjects", arg0, arg1) + ret0, _ := ret[0].(*ListRulesForAllProjectsResp) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateRule indicates an expected call of UpdateRule. -func (mr *MockProjectsServiceServerMockRecorder) UpdateRule(arg0, arg1 interface{}) *gomock.Call { +// ListRulesForAllProjects indicates an expected call of ListRulesForAllProjects +func (mr *MockProjectsServiceServerMockRecorder) ListRulesForAllProjects(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateRule", reflect.TypeOf((*MockProjectsServiceServer)(nil).UpdateRule), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListRulesForAllProjects", reflect.TypeOf((*MockProjectsServiceServer)(nil).ListRulesForAllProjects), arg0, arg1) } diff --git a/api/interservice/cfgmgmt/service/cfgmgmt.pb.client_mock.go b/api/interservice/cfgmgmt/service/cfgmgmt.pb.client_mock.go index 1d158540d13..b21f0e210c3 100644 --- a/api/interservice/cfgmgmt/service/cfgmgmt.pb.client_mock.go +++ b/api/interservice/cfgmgmt/service/cfgmgmt.pb.client_mock.go @@ -6,8 +6,6 @@ package service import ( context "context" - reflect "reflect" - request "github.com/chef/automate/api/external/cfgmgmt/request" response "github.com/chef/automate/api/external/cfgmgmt/response" request0 "github.com/chef/automate/api/interservice/cfgmgmt/request" @@ -16,32 +14,33 @@ import ( grpc "google.golang.org/grpc" metadata "google.golang.org/grpc/metadata" structpb "google.golang.org/protobuf/types/known/structpb" + reflect "reflect" ) -// MockCfgMgmtServiceClient is a mock of CfgMgmtServiceClient interface. +// MockCfgMgmtServiceClient is a mock of CfgMgmtServiceClient interface type MockCfgMgmtServiceClient struct { ctrl *gomock.Controller recorder *MockCfgMgmtServiceClientMockRecorder } -// MockCfgMgmtServiceClientMockRecorder is the mock recorder for MockCfgMgmtServiceClient. +// MockCfgMgmtServiceClientMockRecorder is the mock recorder for MockCfgMgmtServiceClient type MockCfgMgmtServiceClientMockRecorder struct { mock *MockCfgMgmtServiceClient } -// NewMockCfgMgmtServiceClient creates a new mock instance. +// NewMockCfgMgmtServiceClient creates a new mock instance func NewMockCfgMgmtServiceClient(ctrl *gomock.Controller) *MockCfgMgmtServiceClient { mock := &MockCfgMgmtServiceClient{ctrl: ctrl} mock.recorder = &MockCfgMgmtServiceClientMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockCfgMgmtServiceClient) EXPECT() *MockCfgMgmtServiceClientMockRecorder { return m.recorder } -// CreateRollout mocks base method. +// CreateRollout mocks base method func (m *MockCfgMgmtServiceClient) CreateRollout(arg0 context.Context, arg1 *request.CreateRollout, arg2 ...grpc.CallOption) (*response.Rollout, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -54,14 +53,14 @@ func (m *MockCfgMgmtServiceClient) CreateRollout(arg0 context.Context, arg1 *req return ret0, ret1 } -// CreateRollout indicates an expected call of CreateRollout. +// CreateRollout indicates an expected call of CreateRollout func (mr *MockCfgMgmtServiceClientMockRecorder) CreateRollout(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateRollout", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).CreateRollout), varargs...) } -// CreateRolloutTest mocks base method. +// CreateRolloutTest mocks base method func (m *MockCfgMgmtServiceClient) CreateRolloutTest(arg0 context.Context, arg1 *request.CreateRolloutTest, arg2 ...grpc.CallOption) (*response.CreateRolloutTest, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -74,14 +73,14 @@ func (m *MockCfgMgmtServiceClient) CreateRolloutTest(arg0 context.Context, arg1 return ret0, ret1 } -// CreateRolloutTest indicates an expected call of CreateRolloutTest. +// CreateRolloutTest indicates an expected call of CreateRolloutTest func (mr *MockCfgMgmtServiceClientMockRecorder) CreateRolloutTest(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateRolloutTest", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).CreateRolloutTest), varargs...) } -// GetAttributes mocks base method. +// GetAttributes mocks base method func (m *MockCfgMgmtServiceClient) GetAttributes(arg0 context.Context, arg1 *request0.Node, arg2 ...grpc.CallOption) (*response0.NodeAttribute, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -94,14 +93,14 @@ func (m *MockCfgMgmtServiceClient) GetAttributes(arg0 context.Context, arg1 *req return ret0, ret1 } -// GetAttributes indicates an expected call of GetAttributes. +// GetAttributes indicates an expected call of GetAttributes func (mr *MockCfgMgmtServiceClientMockRecorder) GetAttributes(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAttributes", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).GetAttributes), varargs...) } -// GetCheckInCountsTimeSeries mocks base method. +// GetCheckInCountsTimeSeries mocks base method func (m *MockCfgMgmtServiceClient) GetCheckInCountsTimeSeries(arg0 context.Context, arg1 *request0.CheckInCountsTimeSeries, arg2 ...grpc.CallOption) (*response0.CheckInCountsTimeSeries, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -114,14 +113,14 @@ func (m *MockCfgMgmtServiceClient) GetCheckInCountsTimeSeries(arg0 context.Conte return ret0, ret1 } -// GetCheckInCountsTimeSeries indicates an expected call of GetCheckInCountsTimeSeries. +// GetCheckInCountsTimeSeries indicates an expected call of GetCheckInCountsTimeSeries func (mr *MockCfgMgmtServiceClientMockRecorder) GetCheckInCountsTimeSeries(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCheckInCountsTimeSeries", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).GetCheckInCountsTimeSeries), varargs...) } -// GetErrors mocks base method. +// GetErrors mocks base method func (m *MockCfgMgmtServiceClient) GetErrors(arg0 context.Context, arg1 *request.Errors, arg2 ...grpc.CallOption) (*response.Errors, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -134,14 +133,14 @@ func (m *MockCfgMgmtServiceClient) GetErrors(arg0 context.Context, arg1 *request return ret0, ret1 } -// GetErrors indicates an expected call of GetErrors. +// GetErrors indicates an expected call of GetErrors func (mr *MockCfgMgmtServiceClientMockRecorder) GetErrors(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetErrors", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).GetErrors), varargs...) } -// GetHealth mocks base method. +// GetHealth mocks base method func (m *MockCfgMgmtServiceClient) GetHealth(arg0 context.Context, arg1 *request0.Health, arg2 ...grpc.CallOption) (*response0.Health, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -154,14 +153,14 @@ func (m *MockCfgMgmtServiceClient) GetHealth(arg0 context.Context, arg1 *request return ret0, ret1 } -// GetHealth indicates an expected call of GetHealth. +// GetHealth indicates an expected call of GetHealth func (mr *MockCfgMgmtServiceClientMockRecorder) GetHealth(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetHealth", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).GetHealth), varargs...) } -// GetInventoryNodes mocks base method. +// GetInventoryNodes mocks base method func (m *MockCfgMgmtServiceClient) GetInventoryNodes(arg0 context.Context, arg1 *request0.InventoryNodes, arg2 ...grpc.CallOption) (*response0.InventoryNodes, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -174,14 +173,14 @@ func (m *MockCfgMgmtServiceClient) GetInventoryNodes(arg0 context.Context, arg1 return ret0, ret1 } -// GetInventoryNodes indicates an expected call of GetInventoryNodes. +// GetInventoryNodes indicates an expected call of GetInventoryNodes func (mr *MockCfgMgmtServiceClientMockRecorder) GetInventoryNodes(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetInventoryNodes", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).GetInventoryNodes), varargs...) } -// GetMissingNodeDurationCounts mocks base method. +// GetMissingNodeDurationCounts mocks base method func (m *MockCfgMgmtServiceClient) GetMissingNodeDurationCounts(arg0 context.Context, arg1 *request0.MissingNodeDurationCounts, arg2 ...grpc.CallOption) (*response0.MissingNodeDurationCounts, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -194,14 +193,14 @@ func (m *MockCfgMgmtServiceClient) GetMissingNodeDurationCounts(arg0 context.Con return ret0, ret1 } -// GetMissingNodeDurationCounts indicates an expected call of GetMissingNodeDurationCounts. +// GetMissingNodeDurationCounts indicates an expected call of GetMissingNodeDurationCounts func (mr *MockCfgMgmtServiceClientMockRecorder) GetMissingNodeDurationCounts(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetMissingNodeDurationCounts", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).GetMissingNodeDurationCounts), varargs...) } -// GetNodeMetadataCounts mocks base method. +// GetNodeMetadataCounts mocks base method func (m *MockCfgMgmtServiceClient) GetNodeMetadataCounts(arg0 context.Context, arg1 *request0.NodeMetadataCounts, arg2 ...grpc.CallOption) (*response0.NodeMetadataCounts, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -214,14 +213,14 @@ func (m *MockCfgMgmtServiceClient) GetNodeMetadataCounts(arg0 context.Context, a return ret0, ret1 } -// GetNodeMetadataCounts indicates an expected call of GetNodeMetadataCounts. +// GetNodeMetadataCounts indicates an expected call of GetNodeMetadataCounts func (mr *MockCfgMgmtServiceClientMockRecorder) GetNodeMetadataCounts(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodeMetadataCounts", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).GetNodeMetadataCounts), varargs...) } -// GetNodeRun mocks base method. +// GetNodeRun mocks base method func (m *MockCfgMgmtServiceClient) GetNodeRun(arg0 context.Context, arg1 *request0.NodeRun, arg2 ...grpc.CallOption) (*response0.Run, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -234,14 +233,14 @@ func (m *MockCfgMgmtServiceClient) GetNodeRun(arg0 context.Context, arg1 *reques return ret0, ret1 } -// GetNodeRun indicates an expected call of GetNodeRun. +// GetNodeRun indicates an expected call of GetNodeRun func (mr *MockCfgMgmtServiceClientMockRecorder) GetNodeRun(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodeRun", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).GetNodeRun), varargs...) } -// GetNodeRunsDailyStatusTimeSeries mocks base method. +// GetNodeRunsDailyStatusTimeSeries mocks base method func (m *MockCfgMgmtServiceClient) GetNodeRunsDailyStatusTimeSeries(arg0 context.Context, arg1 *request0.NodeRunsDailyStatusTimeSeries, arg2 ...grpc.CallOption) (*response0.NodeRunsDailyStatusTimeSeries, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -254,14 +253,14 @@ func (m *MockCfgMgmtServiceClient) GetNodeRunsDailyStatusTimeSeries(arg0 context return ret0, ret1 } -// GetNodeRunsDailyStatusTimeSeries indicates an expected call of GetNodeRunsDailyStatusTimeSeries. +// GetNodeRunsDailyStatusTimeSeries indicates an expected call of GetNodeRunsDailyStatusTimeSeries func (mr *MockCfgMgmtServiceClientMockRecorder) GetNodeRunsDailyStatusTimeSeries(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodeRunsDailyStatusTimeSeries", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).GetNodeRunsDailyStatusTimeSeries), varargs...) } -// GetNodes mocks base method. +// GetNodes mocks base method func (m *MockCfgMgmtServiceClient) GetNodes(arg0 context.Context, arg1 *request0.Nodes, arg2 ...grpc.CallOption) (*structpb.ListValue, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -274,14 +273,14 @@ func (m *MockCfgMgmtServiceClient) GetNodes(arg0 context.Context, arg1 *request0 return ret0, ret1 } -// GetNodes indicates an expected call of GetNodes. +// GetNodes indicates an expected call of GetNodes func (mr *MockCfgMgmtServiceClientMockRecorder) GetNodes(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodes", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).GetNodes), varargs...) } -// GetNodesCounts mocks base method. +// GetNodesCounts mocks base method func (m *MockCfgMgmtServiceClient) GetNodesCounts(arg0 context.Context, arg1 *request0.NodesCounts, arg2 ...grpc.CallOption) (*response0.NodesCounts, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -294,14 +293,14 @@ func (m *MockCfgMgmtServiceClient) GetNodesCounts(arg0 context.Context, arg1 *re return ret0, ret1 } -// GetNodesCounts indicates an expected call of GetNodesCounts. +// GetNodesCounts indicates an expected call of GetNodesCounts func (mr *MockCfgMgmtServiceClientMockRecorder) GetNodesCounts(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodesCounts", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).GetNodesCounts), varargs...) } -// GetNodesUsageCount mocks base method. +// GetNodesUsageCount mocks base method func (m *MockCfgMgmtServiceClient) GetNodesUsageCount(arg0 context.Context, arg1 *request0.GetNodesUsageCountRequest, arg2 ...grpc.CallOption) (*response0.GetNodesUsageCountResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -314,14 +313,14 @@ func (m *MockCfgMgmtServiceClient) GetNodesUsageCount(arg0 context.Context, arg1 return ret0, ret1 } -// GetNodesUsageCount indicates an expected call of GetNodesUsageCount. +// GetNodesUsageCount indicates an expected call of GetNodesUsageCount func (mr *MockCfgMgmtServiceClientMockRecorder) GetNodesUsageCount(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodesUsageCount", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).GetNodesUsageCount), varargs...) } -// GetOrganizations mocks base method. +// GetOrganizations mocks base method func (m *MockCfgMgmtServiceClient) GetOrganizations(arg0 context.Context, arg1 *request0.Organizations, arg2 ...grpc.CallOption) (*structpb.ListValue, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -334,14 +333,14 @@ func (m *MockCfgMgmtServiceClient) GetOrganizations(arg0 context.Context, arg1 * return ret0, ret1 } -// GetOrganizations indicates an expected call of GetOrganizations. +// GetOrganizations indicates an expected call of GetOrganizations func (mr *MockCfgMgmtServiceClientMockRecorder) GetOrganizations(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrganizations", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).GetOrganizations), varargs...) } -// GetPolicyCookbooks mocks base method. +// GetPolicyCookbooks mocks base method func (m *MockCfgMgmtServiceClient) GetPolicyCookbooks(arg0 context.Context, arg1 *request0.PolicyRevision, arg2 ...grpc.CallOption) (*response0.PolicyCookbooks, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -354,14 +353,14 @@ func (m *MockCfgMgmtServiceClient) GetPolicyCookbooks(arg0 context.Context, arg1 return ret0, ret1 } -// GetPolicyCookbooks indicates an expected call of GetPolicyCookbooks. +// GetPolicyCookbooks indicates an expected call of GetPolicyCookbooks func (mr *MockCfgMgmtServiceClientMockRecorder) GetPolicyCookbooks(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPolicyCookbooks", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).GetPolicyCookbooks), varargs...) } -// GetRolloutById mocks base method. +// GetRolloutById mocks base method func (m *MockCfgMgmtServiceClient) GetRolloutById(arg0 context.Context, arg1 *request.RolloutById, arg2 ...grpc.CallOption) (*response.Rollout, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -374,14 +373,14 @@ func (m *MockCfgMgmtServiceClient) GetRolloutById(arg0 context.Context, arg1 *re return ret0, ret1 } -// GetRolloutById indicates an expected call of GetRolloutById. +// GetRolloutById indicates an expected call of GetRolloutById func (mr *MockCfgMgmtServiceClientMockRecorder) GetRolloutById(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRolloutById", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).GetRolloutById), varargs...) } -// GetRollouts mocks base method. +// GetRollouts mocks base method func (m *MockCfgMgmtServiceClient) GetRollouts(arg0 context.Context, arg1 *request.Rollouts, arg2 ...grpc.CallOption) (*response.Rollouts, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -394,14 +393,14 @@ func (m *MockCfgMgmtServiceClient) GetRollouts(arg0 context.Context, arg1 *reque return ret0, ret1 } -// GetRollouts indicates an expected call of GetRollouts. +// GetRollouts indicates an expected call of GetRollouts func (mr *MockCfgMgmtServiceClientMockRecorder) GetRollouts(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRollouts", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).GetRollouts), varargs...) } -// GetRuns mocks base method. +// GetRuns mocks base method func (m *MockCfgMgmtServiceClient) GetRuns(arg0 context.Context, arg1 *request0.Runs, arg2 ...grpc.CallOption) (*structpb.ListValue, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -414,14 +413,14 @@ func (m *MockCfgMgmtServiceClient) GetRuns(arg0 context.Context, arg1 *request0. return ret0, ret1 } -// GetRuns indicates an expected call of GetRuns. +// GetRuns indicates an expected call of GetRuns func (mr *MockCfgMgmtServiceClientMockRecorder) GetRuns(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRuns", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).GetRuns), varargs...) } -// GetRunsCounts mocks base method. +// GetRunsCounts mocks base method func (m *MockCfgMgmtServiceClient) GetRunsCounts(arg0 context.Context, arg1 *request0.RunsCounts, arg2 ...grpc.CallOption) (*response0.RunsCounts, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -434,14 +433,14 @@ func (m *MockCfgMgmtServiceClient) GetRunsCounts(arg0 context.Context, arg1 *req return ret0, ret1 } -// GetRunsCounts indicates an expected call of GetRunsCounts. +// GetRunsCounts indicates an expected call of GetRunsCounts func (mr *MockCfgMgmtServiceClientMockRecorder) GetRunsCounts(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRunsCounts", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).GetRunsCounts), varargs...) } -// GetSourceFqdns mocks base method. +// GetSourceFqdns mocks base method func (m *MockCfgMgmtServiceClient) GetSourceFqdns(arg0 context.Context, arg1 *request0.SourceFQDNS, arg2 ...grpc.CallOption) (*structpb.ListValue, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -454,14 +453,14 @@ func (m *MockCfgMgmtServiceClient) GetSourceFqdns(arg0 context.Context, arg1 *re return ret0, ret1 } -// GetSourceFqdns indicates an expected call of GetSourceFqdns. +// GetSourceFqdns indicates an expected call of GetSourceFqdns func (mr *MockCfgMgmtServiceClientMockRecorder) GetSourceFqdns(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSourceFqdns", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).GetSourceFqdns), varargs...) } -// GetSuggestions mocks base method. +// GetSuggestions mocks base method func (m *MockCfgMgmtServiceClient) GetSuggestions(arg0 context.Context, arg1 *request0.Suggestion, arg2 ...grpc.CallOption) (*structpb.ListValue, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -474,14 +473,14 @@ func (m *MockCfgMgmtServiceClient) GetSuggestions(arg0 context.Context, arg1 *re return ret0, ret1 } -// GetSuggestions indicates an expected call of GetSuggestions. +// GetSuggestions indicates an expected call of GetSuggestions func (mr *MockCfgMgmtServiceClientMockRecorder) GetSuggestions(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSuggestions", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).GetSuggestions), varargs...) } -// GetVersion mocks base method. +// GetVersion mocks base method func (m *MockCfgMgmtServiceClient) GetVersion(arg0 context.Context, arg1 *request0.VersionInfo, arg2 ...grpc.CallOption) (*response0.VersionInfo, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -494,14 +493,14 @@ func (m *MockCfgMgmtServiceClient) GetVersion(arg0 context.Context, arg1 *reques return ret0, ret1 } -// GetVersion indicates an expected call of GetVersion. +// GetVersion indicates an expected call of GetVersion func (mr *MockCfgMgmtServiceClientMockRecorder) GetVersion(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetVersion", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).GetVersion), varargs...) } -// HandlePolicyUpdateAction mocks base method. +// HandlePolicyUpdateAction mocks base method func (m *MockCfgMgmtServiceClient) HandlePolicyUpdateAction(arg0 context.Context, arg1 *request0.PolicyUpdateAction, arg2 ...grpc.CallOption) (*response0.PolicyUpdateAction, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -514,14 +513,14 @@ func (m *MockCfgMgmtServiceClient) HandlePolicyUpdateAction(arg0 context.Context return ret0, ret1 } -// HandlePolicyUpdateAction indicates an expected call of HandlePolicyUpdateAction. +// HandlePolicyUpdateAction indicates an expected call of HandlePolicyUpdateAction func (mr *MockCfgMgmtServiceClientMockRecorder) HandlePolicyUpdateAction(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HandlePolicyUpdateAction", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).HandlePolicyUpdateAction), varargs...) } -// ListNodeSegmentsWithRolloutProgress mocks base method. +// ListNodeSegmentsWithRolloutProgress mocks base method func (m *MockCfgMgmtServiceClient) ListNodeSegmentsWithRolloutProgress(arg0 context.Context, arg1 *request.ListNodeSegmentsWithRolloutProgress, arg2 ...grpc.CallOption) (*response.NodeSegmentsWithRolloutProgress, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -534,14 +533,14 @@ func (m *MockCfgMgmtServiceClient) ListNodeSegmentsWithRolloutProgress(arg0 cont return ret0, ret1 } -// ListNodeSegmentsWithRolloutProgress indicates an expected call of ListNodeSegmentsWithRolloutProgress. +// ListNodeSegmentsWithRolloutProgress indicates an expected call of ListNodeSegmentsWithRolloutProgress func (mr *MockCfgMgmtServiceClientMockRecorder) ListNodeSegmentsWithRolloutProgress(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListNodeSegmentsWithRolloutProgress", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).ListNodeSegmentsWithRolloutProgress), varargs...) } -// NodeExport mocks base method. +// NodeExport mocks base method func (m *MockCfgMgmtServiceClient) NodeExport(arg0 context.Context, arg1 *request0.NodeExport, arg2 ...grpc.CallOption) (CfgMgmtService_NodeExportClient, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -554,14 +553,14 @@ func (m *MockCfgMgmtServiceClient) NodeExport(arg0 context.Context, arg1 *reques return ret0, ret1 } -// NodeExport indicates an expected call of NodeExport. +// NodeExport indicates an expected call of NodeExport func (mr *MockCfgMgmtServiceClientMockRecorder) NodeExport(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NodeExport", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).NodeExport), varargs...) } -// ReportExport mocks base method. +// ReportExport mocks base method func (m *MockCfgMgmtServiceClient) ReportExport(arg0 context.Context, arg1 *request0.ReportExport, arg2 ...grpc.CallOption) (CfgMgmtService_ReportExportClient, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -574,14 +573,14 @@ func (m *MockCfgMgmtServiceClient) ReportExport(arg0 context.Context, arg1 *requ return ret0, ret1 } -// ReportExport indicates an expected call of ReportExport. +// ReportExport indicates an expected call of ReportExport func (mr *MockCfgMgmtServiceClientMockRecorder) ReportExport(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReportExport", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).ReportExport), varargs...) } -// UpdateTelemetryReported mocks base method. +// UpdateTelemetryReported mocks base method func (m *MockCfgMgmtServiceClient) UpdateTelemetryReported(arg0 context.Context, arg1 *request0.UpdateTelemetryReportedRequest, arg2 ...grpc.CallOption) (*response0.UpdateTelemetryReportedResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -594,37 +593,37 @@ func (m *MockCfgMgmtServiceClient) UpdateTelemetryReported(arg0 context.Context, return ret0, ret1 } -// UpdateTelemetryReported indicates an expected call of UpdateTelemetryReported. +// UpdateTelemetryReported indicates an expected call of UpdateTelemetryReported func (mr *MockCfgMgmtServiceClientMockRecorder) UpdateTelemetryReported(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTelemetryReported", reflect.TypeOf((*MockCfgMgmtServiceClient)(nil).UpdateTelemetryReported), varargs...) } -// MockCfgMgmtService_NodeExportClient is a mock of CfgMgmtService_NodeExportClient interface. +// MockCfgMgmtService_NodeExportClient is a mock of CfgMgmtService_NodeExportClient interface type MockCfgMgmtService_NodeExportClient struct { ctrl *gomock.Controller recorder *MockCfgMgmtService_NodeExportClientMockRecorder } -// MockCfgMgmtService_NodeExportClientMockRecorder is the mock recorder for MockCfgMgmtService_NodeExportClient. +// MockCfgMgmtService_NodeExportClientMockRecorder is the mock recorder for MockCfgMgmtService_NodeExportClient type MockCfgMgmtService_NodeExportClientMockRecorder struct { mock *MockCfgMgmtService_NodeExportClient } -// NewMockCfgMgmtService_NodeExportClient creates a new mock instance. +// NewMockCfgMgmtService_NodeExportClient creates a new mock instance func NewMockCfgMgmtService_NodeExportClient(ctrl *gomock.Controller) *MockCfgMgmtService_NodeExportClient { mock := &MockCfgMgmtService_NodeExportClient{ctrl: ctrl} mock.recorder = &MockCfgMgmtService_NodeExportClientMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockCfgMgmtService_NodeExportClient) EXPECT() *MockCfgMgmtService_NodeExportClientMockRecorder { return m.recorder } -// CloseSend mocks base method. +// CloseSend mocks base method func (m *MockCfgMgmtService_NodeExportClient) CloseSend() error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "CloseSend") @@ -632,13 +631,13 @@ func (m *MockCfgMgmtService_NodeExportClient) CloseSend() error { return ret0 } -// CloseSend indicates an expected call of CloseSend. +// CloseSend indicates an expected call of CloseSend func (mr *MockCfgMgmtService_NodeExportClientMockRecorder) CloseSend() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseSend", reflect.TypeOf((*MockCfgMgmtService_NodeExportClient)(nil).CloseSend)) } -// Context mocks base method. +// Context mocks base method func (m *MockCfgMgmtService_NodeExportClient) Context() context.Context { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Context") @@ -646,13 +645,13 @@ func (m *MockCfgMgmtService_NodeExportClient) Context() context.Context { return ret0 } -// Context indicates an expected call of Context. +// Context indicates an expected call of Context func (mr *MockCfgMgmtService_NodeExportClientMockRecorder) Context() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockCfgMgmtService_NodeExportClient)(nil).Context)) } -// Header mocks base method. +// Header mocks base method func (m *MockCfgMgmtService_NodeExportClient) Header() (metadata.MD, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Header") @@ -661,13 +660,13 @@ func (m *MockCfgMgmtService_NodeExportClient) Header() (metadata.MD, error) { return ret0, ret1 } -// Header indicates an expected call of Header. +// Header indicates an expected call of Header func (mr *MockCfgMgmtService_NodeExportClientMockRecorder) Header() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Header", reflect.TypeOf((*MockCfgMgmtService_NodeExportClient)(nil).Header)) } -// Recv mocks base method. +// Recv mocks base method func (m *MockCfgMgmtService_NodeExportClient) Recv() (*response0.ExportData, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Recv") @@ -676,13 +675,13 @@ func (m *MockCfgMgmtService_NodeExportClient) Recv() (*response0.ExportData, err return ret0, ret1 } -// Recv indicates an expected call of Recv. +// Recv indicates an expected call of Recv func (mr *MockCfgMgmtService_NodeExportClientMockRecorder) Recv() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Recv", reflect.TypeOf((*MockCfgMgmtService_NodeExportClient)(nil).Recv)) } -// RecvMsg mocks base method. +// RecvMsg mocks base method func (m *MockCfgMgmtService_NodeExportClient) RecvMsg(arg0 interface{}) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "RecvMsg", arg0) @@ -690,13 +689,13 @@ func (m *MockCfgMgmtService_NodeExportClient) RecvMsg(arg0 interface{}) error { return ret0 } -// RecvMsg indicates an expected call of RecvMsg. +// RecvMsg indicates an expected call of RecvMsg func (mr *MockCfgMgmtService_NodeExportClientMockRecorder) RecvMsg(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockCfgMgmtService_NodeExportClient)(nil).RecvMsg), arg0) } -// SendMsg mocks base method. +// SendMsg mocks base method func (m *MockCfgMgmtService_NodeExportClient) SendMsg(arg0 interface{}) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "SendMsg", arg0) @@ -704,13 +703,13 @@ func (m *MockCfgMgmtService_NodeExportClient) SendMsg(arg0 interface{}) error { return ret0 } -// SendMsg indicates an expected call of SendMsg. +// SendMsg indicates an expected call of SendMsg func (mr *MockCfgMgmtService_NodeExportClientMockRecorder) SendMsg(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockCfgMgmtService_NodeExportClient)(nil).SendMsg), arg0) } -// Trailer mocks base method. +// Trailer mocks base method func (m *MockCfgMgmtService_NodeExportClient) Trailer() metadata.MD { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Trailer") @@ -718,36 +717,36 @@ func (m *MockCfgMgmtService_NodeExportClient) Trailer() metadata.MD { return ret0 } -// Trailer indicates an expected call of Trailer. +// Trailer indicates an expected call of Trailer func (mr *MockCfgMgmtService_NodeExportClientMockRecorder) Trailer() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Trailer", reflect.TypeOf((*MockCfgMgmtService_NodeExportClient)(nil).Trailer)) } -// MockCfgMgmtService_ReportExportClient is a mock of CfgMgmtService_ReportExportClient interface. +// MockCfgMgmtService_ReportExportClient is a mock of CfgMgmtService_ReportExportClient interface type MockCfgMgmtService_ReportExportClient struct { ctrl *gomock.Controller recorder *MockCfgMgmtService_ReportExportClientMockRecorder } -// MockCfgMgmtService_ReportExportClientMockRecorder is the mock recorder for MockCfgMgmtService_ReportExportClient. +// MockCfgMgmtService_ReportExportClientMockRecorder is the mock recorder for MockCfgMgmtService_ReportExportClient type MockCfgMgmtService_ReportExportClientMockRecorder struct { mock *MockCfgMgmtService_ReportExportClient } -// NewMockCfgMgmtService_ReportExportClient creates a new mock instance. +// NewMockCfgMgmtService_ReportExportClient creates a new mock instance func NewMockCfgMgmtService_ReportExportClient(ctrl *gomock.Controller) *MockCfgMgmtService_ReportExportClient { mock := &MockCfgMgmtService_ReportExportClient{ctrl: ctrl} mock.recorder = &MockCfgMgmtService_ReportExportClientMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockCfgMgmtService_ReportExportClient) EXPECT() *MockCfgMgmtService_ReportExportClientMockRecorder { return m.recorder } -// CloseSend mocks base method. +// CloseSend mocks base method func (m *MockCfgMgmtService_ReportExportClient) CloseSend() error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "CloseSend") @@ -755,13 +754,13 @@ func (m *MockCfgMgmtService_ReportExportClient) CloseSend() error { return ret0 } -// CloseSend indicates an expected call of CloseSend. +// CloseSend indicates an expected call of CloseSend func (mr *MockCfgMgmtService_ReportExportClientMockRecorder) CloseSend() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseSend", reflect.TypeOf((*MockCfgMgmtService_ReportExportClient)(nil).CloseSend)) } -// Context mocks base method. +// Context mocks base method func (m *MockCfgMgmtService_ReportExportClient) Context() context.Context { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Context") @@ -769,13 +768,13 @@ func (m *MockCfgMgmtService_ReportExportClient) Context() context.Context { return ret0 } -// Context indicates an expected call of Context. +// Context indicates an expected call of Context func (mr *MockCfgMgmtService_ReportExportClientMockRecorder) Context() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockCfgMgmtService_ReportExportClient)(nil).Context)) } -// Header mocks base method. +// Header mocks base method func (m *MockCfgMgmtService_ReportExportClient) Header() (metadata.MD, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Header") @@ -784,13 +783,13 @@ func (m *MockCfgMgmtService_ReportExportClient) Header() (metadata.MD, error) { return ret0, ret1 } -// Header indicates an expected call of Header. +// Header indicates an expected call of Header func (mr *MockCfgMgmtService_ReportExportClientMockRecorder) Header() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Header", reflect.TypeOf((*MockCfgMgmtService_ReportExportClient)(nil).Header)) } -// Recv mocks base method. +// Recv mocks base method func (m *MockCfgMgmtService_ReportExportClient) Recv() (*response0.ReportExportData, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Recv") @@ -799,13 +798,13 @@ func (m *MockCfgMgmtService_ReportExportClient) Recv() (*response0.ReportExportD return ret0, ret1 } -// Recv indicates an expected call of Recv. +// Recv indicates an expected call of Recv func (mr *MockCfgMgmtService_ReportExportClientMockRecorder) Recv() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Recv", reflect.TypeOf((*MockCfgMgmtService_ReportExportClient)(nil).Recv)) } -// RecvMsg mocks base method. +// RecvMsg mocks base method func (m *MockCfgMgmtService_ReportExportClient) RecvMsg(arg0 interface{}) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "RecvMsg", arg0) @@ -813,13 +812,13 @@ func (m *MockCfgMgmtService_ReportExportClient) RecvMsg(arg0 interface{}) error return ret0 } -// RecvMsg indicates an expected call of RecvMsg. +// RecvMsg indicates an expected call of RecvMsg func (mr *MockCfgMgmtService_ReportExportClientMockRecorder) RecvMsg(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockCfgMgmtService_ReportExportClient)(nil).RecvMsg), arg0) } -// SendMsg mocks base method. +// SendMsg mocks base method func (m *MockCfgMgmtService_ReportExportClient) SendMsg(arg0 interface{}) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "SendMsg", arg0) @@ -827,13 +826,13 @@ func (m *MockCfgMgmtService_ReportExportClient) SendMsg(arg0 interface{}) error return ret0 } -// SendMsg indicates an expected call of SendMsg. +// SendMsg indicates an expected call of SendMsg func (mr *MockCfgMgmtService_ReportExportClientMockRecorder) SendMsg(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockCfgMgmtService_ReportExportClient)(nil).SendMsg), arg0) } -// Trailer mocks base method. +// Trailer mocks base method func (m *MockCfgMgmtService_ReportExportClient) Trailer() metadata.MD { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Trailer") @@ -841,36 +840,36 @@ func (m *MockCfgMgmtService_ReportExportClient) Trailer() metadata.MD { return ret0 } -// Trailer indicates an expected call of Trailer. +// Trailer indicates an expected call of Trailer func (mr *MockCfgMgmtService_ReportExportClientMockRecorder) Trailer() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Trailer", reflect.TypeOf((*MockCfgMgmtService_ReportExportClient)(nil).Trailer)) } -// MockCfgMgmtServiceServer is a mock of CfgMgmtServiceServer interface. +// MockCfgMgmtServiceServer is a mock of CfgMgmtServiceServer interface type MockCfgMgmtServiceServer struct { ctrl *gomock.Controller recorder *MockCfgMgmtServiceServerMockRecorder } -// MockCfgMgmtServiceServerMockRecorder is the mock recorder for MockCfgMgmtServiceServer. +// MockCfgMgmtServiceServerMockRecorder is the mock recorder for MockCfgMgmtServiceServer type MockCfgMgmtServiceServerMockRecorder struct { mock *MockCfgMgmtServiceServer } -// NewMockCfgMgmtServiceServer creates a new mock instance. +// NewMockCfgMgmtServiceServer creates a new mock instance func NewMockCfgMgmtServiceServer(ctrl *gomock.Controller) *MockCfgMgmtServiceServer { mock := &MockCfgMgmtServiceServer{ctrl: ctrl} mock.recorder = &MockCfgMgmtServiceServerMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockCfgMgmtServiceServer) EXPECT() *MockCfgMgmtServiceServerMockRecorder { return m.recorder } -// CreateRollout mocks base method. +// CreateRollout mocks base method func (m *MockCfgMgmtServiceServer) CreateRollout(arg0 context.Context, arg1 *request.CreateRollout) (*response.Rollout, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "CreateRollout", arg0, arg1) @@ -879,13 +878,13 @@ func (m *MockCfgMgmtServiceServer) CreateRollout(arg0 context.Context, arg1 *req return ret0, ret1 } -// CreateRollout indicates an expected call of CreateRollout. +// CreateRollout indicates an expected call of CreateRollout func (mr *MockCfgMgmtServiceServerMockRecorder) CreateRollout(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateRollout", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).CreateRollout), arg0, arg1) } -// CreateRolloutTest mocks base method. +// CreateRolloutTest mocks base method func (m *MockCfgMgmtServiceServer) CreateRolloutTest(arg0 context.Context, arg1 *request.CreateRolloutTest) (*response.CreateRolloutTest, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "CreateRolloutTest", arg0, arg1) @@ -894,13 +893,13 @@ func (m *MockCfgMgmtServiceServer) CreateRolloutTest(arg0 context.Context, arg1 return ret0, ret1 } -// CreateRolloutTest indicates an expected call of CreateRolloutTest. +// CreateRolloutTest indicates an expected call of CreateRolloutTest func (mr *MockCfgMgmtServiceServerMockRecorder) CreateRolloutTest(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateRolloutTest", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).CreateRolloutTest), arg0, arg1) } -// GetAttributes mocks base method. +// GetAttributes mocks base method func (m *MockCfgMgmtServiceServer) GetAttributes(arg0 context.Context, arg1 *request0.Node) (*response0.NodeAttribute, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetAttributes", arg0, arg1) @@ -909,13 +908,13 @@ func (m *MockCfgMgmtServiceServer) GetAttributes(arg0 context.Context, arg1 *req return ret0, ret1 } -// GetAttributes indicates an expected call of GetAttributes. +// GetAttributes indicates an expected call of GetAttributes func (mr *MockCfgMgmtServiceServerMockRecorder) GetAttributes(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAttributes", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).GetAttributes), arg0, arg1) } -// GetCheckInCountsTimeSeries mocks base method. +// GetCheckInCountsTimeSeries mocks base method func (m *MockCfgMgmtServiceServer) GetCheckInCountsTimeSeries(arg0 context.Context, arg1 *request0.CheckInCountsTimeSeries) (*response0.CheckInCountsTimeSeries, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetCheckInCountsTimeSeries", arg0, arg1) @@ -924,13 +923,13 @@ func (m *MockCfgMgmtServiceServer) GetCheckInCountsTimeSeries(arg0 context.Conte return ret0, ret1 } -// GetCheckInCountsTimeSeries indicates an expected call of GetCheckInCountsTimeSeries. +// GetCheckInCountsTimeSeries indicates an expected call of GetCheckInCountsTimeSeries func (mr *MockCfgMgmtServiceServerMockRecorder) GetCheckInCountsTimeSeries(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCheckInCountsTimeSeries", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).GetCheckInCountsTimeSeries), arg0, arg1) } -// GetErrors mocks base method. +// GetErrors mocks base method func (m *MockCfgMgmtServiceServer) GetErrors(arg0 context.Context, arg1 *request.Errors) (*response.Errors, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetErrors", arg0, arg1) @@ -939,13 +938,13 @@ func (m *MockCfgMgmtServiceServer) GetErrors(arg0 context.Context, arg1 *request return ret0, ret1 } -// GetErrors indicates an expected call of GetErrors. +// GetErrors indicates an expected call of GetErrors func (mr *MockCfgMgmtServiceServerMockRecorder) GetErrors(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetErrors", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).GetErrors), arg0, arg1) } -// GetHealth mocks base method. +// GetHealth mocks base method func (m *MockCfgMgmtServiceServer) GetHealth(arg0 context.Context, arg1 *request0.Health) (*response0.Health, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetHealth", arg0, arg1) @@ -954,13 +953,13 @@ func (m *MockCfgMgmtServiceServer) GetHealth(arg0 context.Context, arg1 *request return ret0, ret1 } -// GetHealth indicates an expected call of GetHealth. +// GetHealth indicates an expected call of GetHealth func (mr *MockCfgMgmtServiceServerMockRecorder) GetHealth(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetHealth", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).GetHealth), arg0, arg1) } -// GetInventoryNodes mocks base method. +// GetInventoryNodes mocks base method func (m *MockCfgMgmtServiceServer) GetInventoryNodes(arg0 context.Context, arg1 *request0.InventoryNodes) (*response0.InventoryNodes, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetInventoryNodes", arg0, arg1) @@ -969,13 +968,13 @@ func (m *MockCfgMgmtServiceServer) GetInventoryNodes(arg0 context.Context, arg1 return ret0, ret1 } -// GetInventoryNodes indicates an expected call of GetInventoryNodes. +// GetInventoryNodes indicates an expected call of GetInventoryNodes func (mr *MockCfgMgmtServiceServerMockRecorder) GetInventoryNodes(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetInventoryNodes", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).GetInventoryNodes), arg0, arg1) } -// GetMissingNodeDurationCounts mocks base method. +// GetMissingNodeDurationCounts mocks base method func (m *MockCfgMgmtServiceServer) GetMissingNodeDurationCounts(arg0 context.Context, arg1 *request0.MissingNodeDurationCounts) (*response0.MissingNodeDurationCounts, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetMissingNodeDurationCounts", arg0, arg1) @@ -984,13 +983,13 @@ func (m *MockCfgMgmtServiceServer) GetMissingNodeDurationCounts(arg0 context.Con return ret0, ret1 } -// GetMissingNodeDurationCounts indicates an expected call of GetMissingNodeDurationCounts. +// GetMissingNodeDurationCounts indicates an expected call of GetMissingNodeDurationCounts func (mr *MockCfgMgmtServiceServerMockRecorder) GetMissingNodeDurationCounts(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetMissingNodeDurationCounts", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).GetMissingNodeDurationCounts), arg0, arg1) } -// GetNodeMetadataCounts mocks base method. +// GetNodeMetadataCounts mocks base method func (m *MockCfgMgmtServiceServer) GetNodeMetadataCounts(arg0 context.Context, arg1 *request0.NodeMetadataCounts) (*response0.NodeMetadataCounts, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetNodeMetadataCounts", arg0, arg1) @@ -999,13 +998,13 @@ func (m *MockCfgMgmtServiceServer) GetNodeMetadataCounts(arg0 context.Context, a return ret0, ret1 } -// GetNodeMetadataCounts indicates an expected call of GetNodeMetadataCounts. +// GetNodeMetadataCounts indicates an expected call of GetNodeMetadataCounts func (mr *MockCfgMgmtServiceServerMockRecorder) GetNodeMetadataCounts(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodeMetadataCounts", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).GetNodeMetadataCounts), arg0, arg1) } -// GetNodeRun mocks base method. +// GetNodeRun mocks base method func (m *MockCfgMgmtServiceServer) GetNodeRun(arg0 context.Context, arg1 *request0.NodeRun) (*response0.Run, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetNodeRun", arg0, arg1) @@ -1014,13 +1013,13 @@ func (m *MockCfgMgmtServiceServer) GetNodeRun(arg0 context.Context, arg1 *reques return ret0, ret1 } -// GetNodeRun indicates an expected call of GetNodeRun. +// GetNodeRun indicates an expected call of GetNodeRun func (mr *MockCfgMgmtServiceServerMockRecorder) GetNodeRun(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodeRun", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).GetNodeRun), arg0, arg1) } -// GetNodeRunsDailyStatusTimeSeries mocks base method. +// GetNodeRunsDailyStatusTimeSeries mocks base method func (m *MockCfgMgmtServiceServer) GetNodeRunsDailyStatusTimeSeries(arg0 context.Context, arg1 *request0.NodeRunsDailyStatusTimeSeries) (*response0.NodeRunsDailyStatusTimeSeries, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetNodeRunsDailyStatusTimeSeries", arg0, arg1) @@ -1029,13 +1028,13 @@ func (m *MockCfgMgmtServiceServer) GetNodeRunsDailyStatusTimeSeries(arg0 context return ret0, ret1 } -// GetNodeRunsDailyStatusTimeSeries indicates an expected call of GetNodeRunsDailyStatusTimeSeries. +// GetNodeRunsDailyStatusTimeSeries indicates an expected call of GetNodeRunsDailyStatusTimeSeries func (mr *MockCfgMgmtServiceServerMockRecorder) GetNodeRunsDailyStatusTimeSeries(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodeRunsDailyStatusTimeSeries", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).GetNodeRunsDailyStatusTimeSeries), arg0, arg1) } -// GetNodes mocks base method. +// GetNodes mocks base method func (m *MockCfgMgmtServiceServer) GetNodes(arg0 context.Context, arg1 *request0.Nodes) (*structpb.ListValue, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetNodes", arg0, arg1) @@ -1044,13 +1043,13 @@ func (m *MockCfgMgmtServiceServer) GetNodes(arg0 context.Context, arg1 *request0 return ret0, ret1 } -// GetNodes indicates an expected call of GetNodes. +// GetNodes indicates an expected call of GetNodes func (mr *MockCfgMgmtServiceServerMockRecorder) GetNodes(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodes", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).GetNodes), arg0, arg1) } -// GetNodesCounts mocks base method. +// GetNodesCounts mocks base method func (m *MockCfgMgmtServiceServer) GetNodesCounts(arg0 context.Context, arg1 *request0.NodesCounts) (*response0.NodesCounts, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetNodesCounts", arg0, arg1) @@ -1059,13 +1058,13 @@ func (m *MockCfgMgmtServiceServer) GetNodesCounts(arg0 context.Context, arg1 *re return ret0, ret1 } -// GetNodesCounts indicates an expected call of GetNodesCounts. +// GetNodesCounts indicates an expected call of GetNodesCounts func (mr *MockCfgMgmtServiceServerMockRecorder) GetNodesCounts(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodesCounts", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).GetNodesCounts), arg0, arg1) } -// GetNodesUsageCount mocks base method. +// GetNodesUsageCount mocks base method func (m *MockCfgMgmtServiceServer) GetNodesUsageCount(arg0 context.Context, arg1 *request0.GetNodesUsageCountRequest) (*response0.GetNodesUsageCountResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetNodesUsageCount", arg0, arg1) @@ -1074,13 +1073,13 @@ func (m *MockCfgMgmtServiceServer) GetNodesUsageCount(arg0 context.Context, arg1 return ret0, ret1 } -// GetNodesUsageCount indicates an expected call of GetNodesUsageCount. +// GetNodesUsageCount indicates an expected call of GetNodesUsageCount func (mr *MockCfgMgmtServiceServerMockRecorder) GetNodesUsageCount(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodesUsageCount", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).GetNodesUsageCount), arg0, arg1) } -// GetOrganizations mocks base method. +// GetOrganizations mocks base method func (m *MockCfgMgmtServiceServer) GetOrganizations(arg0 context.Context, arg1 *request0.Organizations) (*structpb.ListValue, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetOrganizations", arg0, arg1) @@ -1089,13 +1088,13 @@ func (m *MockCfgMgmtServiceServer) GetOrganizations(arg0 context.Context, arg1 * return ret0, ret1 } -// GetOrganizations indicates an expected call of GetOrganizations. +// GetOrganizations indicates an expected call of GetOrganizations func (mr *MockCfgMgmtServiceServerMockRecorder) GetOrganizations(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrganizations", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).GetOrganizations), arg0, arg1) } -// GetPolicyCookbooks mocks base method. +// GetPolicyCookbooks mocks base method func (m *MockCfgMgmtServiceServer) GetPolicyCookbooks(arg0 context.Context, arg1 *request0.PolicyRevision) (*response0.PolicyCookbooks, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetPolicyCookbooks", arg0, arg1) @@ -1104,13 +1103,13 @@ func (m *MockCfgMgmtServiceServer) GetPolicyCookbooks(arg0 context.Context, arg1 return ret0, ret1 } -// GetPolicyCookbooks indicates an expected call of GetPolicyCookbooks. +// GetPolicyCookbooks indicates an expected call of GetPolicyCookbooks func (mr *MockCfgMgmtServiceServerMockRecorder) GetPolicyCookbooks(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPolicyCookbooks", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).GetPolicyCookbooks), arg0, arg1) } -// GetRolloutById mocks base method. +// GetRolloutById mocks base method func (m *MockCfgMgmtServiceServer) GetRolloutById(arg0 context.Context, arg1 *request.RolloutById) (*response.Rollout, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetRolloutById", arg0, arg1) @@ -1119,13 +1118,13 @@ func (m *MockCfgMgmtServiceServer) GetRolloutById(arg0 context.Context, arg1 *re return ret0, ret1 } -// GetRolloutById indicates an expected call of GetRolloutById. +// GetRolloutById indicates an expected call of GetRolloutById func (mr *MockCfgMgmtServiceServerMockRecorder) GetRolloutById(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRolloutById", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).GetRolloutById), arg0, arg1) } -// GetRollouts mocks base method. +// GetRollouts mocks base method func (m *MockCfgMgmtServiceServer) GetRollouts(arg0 context.Context, arg1 *request.Rollouts) (*response.Rollouts, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetRollouts", arg0, arg1) @@ -1134,13 +1133,13 @@ func (m *MockCfgMgmtServiceServer) GetRollouts(arg0 context.Context, arg1 *reque return ret0, ret1 } -// GetRollouts indicates an expected call of GetRollouts. +// GetRollouts indicates an expected call of GetRollouts func (mr *MockCfgMgmtServiceServerMockRecorder) GetRollouts(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRollouts", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).GetRollouts), arg0, arg1) } -// GetRuns mocks base method. +// GetRuns mocks base method func (m *MockCfgMgmtServiceServer) GetRuns(arg0 context.Context, arg1 *request0.Runs) (*structpb.ListValue, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetRuns", arg0, arg1) @@ -1149,13 +1148,13 @@ func (m *MockCfgMgmtServiceServer) GetRuns(arg0 context.Context, arg1 *request0. return ret0, ret1 } -// GetRuns indicates an expected call of GetRuns. +// GetRuns indicates an expected call of GetRuns func (mr *MockCfgMgmtServiceServerMockRecorder) GetRuns(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRuns", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).GetRuns), arg0, arg1) } -// GetRunsCounts mocks base method. +// GetRunsCounts mocks base method func (m *MockCfgMgmtServiceServer) GetRunsCounts(arg0 context.Context, arg1 *request0.RunsCounts) (*response0.RunsCounts, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetRunsCounts", arg0, arg1) @@ -1164,13 +1163,13 @@ func (m *MockCfgMgmtServiceServer) GetRunsCounts(arg0 context.Context, arg1 *req return ret0, ret1 } -// GetRunsCounts indicates an expected call of GetRunsCounts. +// GetRunsCounts indicates an expected call of GetRunsCounts func (mr *MockCfgMgmtServiceServerMockRecorder) GetRunsCounts(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRunsCounts", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).GetRunsCounts), arg0, arg1) } -// GetSourceFqdns mocks base method. +// GetSourceFqdns mocks base method func (m *MockCfgMgmtServiceServer) GetSourceFqdns(arg0 context.Context, arg1 *request0.SourceFQDNS) (*structpb.ListValue, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetSourceFqdns", arg0, arg1) @@ -1179,13 +1178,13 @@ func (m *MockCfgMgmtServiceServer) GetSourceFqdns(arg0 context.Context, arg1 *re return ret0, ret1 } -// GetSourceFqdns indicates an expected call of GetSourceFqdns. +// GetSourceFqdns indicates an expected call of GetSourceFqdns func (mr *MockCfgMgmtServiceServerMockRecorder) GetSourceFqdns(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSourceFqdns", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).GetSourceFqdns), arg0, arg1) } -// GetSuggestions mocks base method. +// GetSuggestions mocks base method func (m *MockCfgMgmtServiceServer) GetSuggestions(arg0 context.Context, arg1 *request0.Suggestion) (*structpb.ListValue, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetSuggestions", arg0, arg1) @@ -1194,13 +1193,13 @@ func (m *MockCfgMgmtServiceServer) GetSuggestions(arg0 context.Context, arg1 *re return ret0, ret1 } -// GetSuggestions indicates an expected call of GetSuggestions. +// GetSuggestions indicates an expected call of GetSuggestions func (mr *MockCfgMgmtServiceServerMockRecorder) GetSuggestions(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSuggestions", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).GetSuggestions), arg0, arg1) } -// GetVersion mocks base method. +// GetVersion mocks base method func (m *MockCfgMgmtServiceServer) GetVersion(arg0 context.Context, arg1 *request0.VersionInfo) (*response0.VersionInfo, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetVersion", arg0, arg1) @@ -1209,13 +1208,13 @@ func (m *MockCfgMgmtServiceServer) GetVersion(arg0 context.Context, arg1 *reques return ret0, ret1 } -// GetVersion indicates an expected call of GetVersion. +// GetVersion indicates an expected call of GetVersion func (mr *MockCfgMgmtServiceServerMockRecorder) GetVersion(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetVersion", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).GetVersion), arg0, arg1) } -// HandlePolicyUpdateAction mocks base method. +// HandlePolicyUpdateAction mocks base method func (m *MockCfgMgmtServiceServer) HandlePolicyUpdateAction(arg0 context.Context, arg1 *request0.PolicyUpdateAction) (*response0.PolicyUpdateAction, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "HandlePolicyUpdateAction", arg0, arg1) @@ -1224,13 +1223,13 @@ func (m *MockCfgMgmtServiceServer) HandlePolicyUpdateAction(arg0 context.Context return ret0, ret1 } -// HandlePolicyUpdateAction indicates an expected call of HandlePolicyUpdateAction. +// HandlePolicyUpdateAction indicates an expected call of HandlePolicyUpdateAction func (mr *MockCfgMgmtServiceServerMockRecorder) HandlePolicyUpdateAction(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HandlePolicyUpdateAction", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).HandlePolicyUpdateAction), arg0, arg1) } -// ListNodeSegmentsWithRolloutProgress mocks base method. +// ListNodeSegmentsWithRolloutProgress mocks base method func (m *MockCfgMgmtServiceServer) ListNodeSegmentsWithRolloutProgress(arg0 context.Context, arg1 *request.ListNodeSegmentsWithRolloutProgress) (*response.NodeSegmentsWithRolloutProgress, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ListNodeSegmentsWithRolloutProgress", arg0, arg1) @@ -1239,13 +1238,13 @@ func (m *MockCfgMgmtServiceServer) ListNodeSegmentsWithRolloutProgress(arg0 cont return ret0, ret1 } -// ListNodeSegmentsWithRolloutProgress indicates an expected call of ListNodeSegmentsWithRolloutProgress. +// ListNodeSegmentsWithRolloutProgress indicates an expected call of ListNodeSegmentsWithRolloutProgress func (mr *MockCfgMgmtServiceServerMockRecorder) ListNodeSegmentsWithRolloutProgress(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListNodeSegmentsWithRolloutProgress", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).ListNodeSegmentsWithRolloutProgress), arg0, arg1) } -// NodeExport mocks base method. +// NodeExport mocks base method func (m *MockCfgMgmtServiceServer) NodeExport(arg0 *request0.NodeExport, arg1 CfgMgmtService_NodeExportServer) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "NodeExport", arg0, arg1) @@ -1253,13 +1252,13 @@ func (m *MockCfgMgmtServiceServer) NodeExport(arg0 *request0.NodeExport, arg1 Cf return ret0 } -// NodeExport indicates an expected call of NodeExport. +// NodeExport indicates an expected call of NodeExport func (mr *MockCfgMgmtServiceServerMockRecorder) NodeExport(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NodeExport", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).NodeExport), arg0, arg1) } -// ReportExport mocks base method. +// ReportExport mocks base method func (m *MockCfgMgmtServiceServer) ReportExport(arg0 *request0.ReportExport, arg1 CfgMgmtService_ReportExportServer) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ReportExport", arg0, arg1) @@ -1267,13 +1266,13 @@ func (m *MockCfgMgmtServiceServer) ReportExport(arg0 *request0.ReportExport, arg return ret0 } -// ReportExport indicates an expected call of ReportExport. +// ReportExport indicates an expected call of ReportExport func (mr *MockCfgMgmtServiceServerMockRecorder) ReportExport(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReportExport", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).ReportExport), arg0, arg1) } -// UpdateTelemetryReported mocks base method. +// UpdateTelemetryReported mocks base method func (m *MockCfgMgmtServiceServer) UpdateTelemetryReported(arg0 context.Context, arg1 *request0.UpdateTelemetryReportedRequest) (*response0.UpdateTelemetryReportedResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "UpdateTelemetryReported", arg0, arg1) @@ -1282,36 +1281,36 @@ func (m *MockCfgMgmtServiceServer) UpdateTelemetryReported(arg0 context.Context, return ret0, ret1 } -// UpdateTelemetryReported indicates an expected call of UpdateTelemetryReported. +// UpdateTelemetryReported indicates an expected call of UpdateTelemetryReported func (mr *MockCfgMgmtServiceServerMockRecorder) UpdateTelemetryReported(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTelemetryReported", reflect.TypeOf((*MockCfgMgmtServiceServer)(nil).UpdateTelemetryReported), arg0, arg1) } -// MockCfgMgmtService_NodeExportServer is a mock of CfgMgmtService_NodeExportServer interface. +// MockCfgMgmtService_NodeExportServer is a mock of CfgMgmtService_NodeExportServer interface type MockCfgMgmtService_NodeExportServer struct { ctrl *gomock.Controller recorder *MockCfgMgmtService_NodeExportServerMockRecorder } -// MockCfgMgmtService_NodeExportServerMockRecorder is the mock recorder for MockCfgMgmtService_NodeExportServer. +// MockCfgMgmtService_NodeExportServerMockRecorder is the mock recorder for MockCfgMgmtService_NodeExportServer type MockCfgMgmtService_NodeExportServerMockRecorder struct { mock *MockCfgMgmtService_NodeExportServer } -// NewMockCfgMgmtService_NodeExportServer creates a new mock instance. +// NewMockCfgMgmtService_NodeExportServer creates a new mock instance func NewMockCfgMgmtService_NodeExportServer(ctrl *gomock.Controller) *MockCfgMgmtService_NodeExportServer { mock := &MockCfgMgmtService_NodeExportServer{ctrl: ctrl} mock.recorder = &MockCfgMgmtService_NodeExportServerMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockCfgMgmtService_NodeExportServer) EXPECT() *MockCfgMgmtService_NodeExportServerMockRecorder { return m.recorder } -// Context mocks base method. +// Context mocks base method func (m *MockCfgMgmtService_NodeExportServer) Context() context.Context { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Context") @@ -1319,13 +1318,13 @@ func (m *MockCfgMgmtService_NodeExportServer) Context() context.Context { return ret0 } -// Context indicates an expected call of Context. +// Context indicates an expected call of Context func (mr *MockCfgMgmtService_NodeExportServerMockRecorder) Context() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockCfgMgmtService_NodeExportServer)(nil).Context)) } -// RecvMsg mocks base method. +// RecvMsg mocks base method func (m *MockCfgMgmtService_NodeExportServer) RecvMsg(arg0 interface{}) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "RecvMsg", arg0) @@ -1333,13 +1332,13 @@ func (m *MockCfgMgmtService_NodeExportServer) RecvMsg(arg0 interface{}) error { return ret0 } -// RecvMsg indicates an expected call of RecvMsg. +// RecvMsg indicates an expected call of RecvMsg func (mr *MockCfgMgmtService_NodeExportServerMockRecorder) RecvMsg(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockCfgMgmtService_NodeExportServer)(nil).RecvMsg), arg0) } -// Send mocks base method. +// Send mocks base method func (m *MockCfgMgmtService_NodeExportServer) Send(arg0 *response0.ExportData) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Send", arg0) @@ -1347,13 +1346,13 @@ func (m *MockCfgMgmtService_NodeExportServer) Send(arg0 *response0.ExportData) e return ret0 } -// Send indicates an expected call of Send. +// Send indicates an expected call of Send func (mr *MockCfgMgmtService_NodeExportServerMockRecorder) Send(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockCfgMgmtService_NodeExportServer)(nil).Send), arg0) } -// SendHeader mocks base method. +// SendHeader mocks base method func (m *MockCfgMgmtService_NodeExportServer) SendHeader(arg0 metadata.MD) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "SendHeader", arg0) @@ -1361,13 +1360,13 @@ func (m *MockCfgMgmtService_NodeExportServer) SendHeader(arg0 metadata.MD) error return ret0 } -// SendHeader indicates an expected call of SendHeader. +// SendHeader indicates an expected call of SendHeader func (mr *MockCfgMgmtService_NodeExportServerMockRecorder) SendHeader(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendHeader", reflect.TypeOf((*MockCfgMgmtService_NodeExportServer)(nil).SendHeader), arg0) } -// SendMsg mocks base method. +// SendMsg mocks base method func (m *MockCfgMgmtService_NodeExportServer) SendMsg(arg0 interface{}) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "SendMsg", arg0) @@ -1375,13 +1374,13 @@ func (m *MockCfgMgmtService_NodeExportServer) SendMsg(arg0 interface{}) error { return ret0 } -// SendMsg indicates an expected call of SendMsg. +// SendMsg indicates an expected call of SendMsg func (mr *MockCfgMgmtService_NodeExportServerMockRecorder) SendMsg(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockCfgMgmtService_NodeExportServer)(nil).SendMsg), arg0) } -// SetHeader mocks base method. +// SetHeader mocks base method func (m *MockCfgMgmtService_NodeExportServer) SetHeader(arg0 metadata.MD) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "SetHeader", arg0) @@ -1389,48 +1388,48 @@ func (m *MockCfgMgmtService_NodeExportServer) SetHeader(arg0 metadata.MD) error return ret0 } -// SetHeader indicates an expected call of SetHeader. +// SetHeader indicates an expected call of SetHeader func (mr *MockCfgMgmtService_NodeExportServerMockRecorder) SetHeader(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetHeader", reflect.TypeOf((*MockCfgMgmtService_NodeExportServer)(nil).SetHeader), arg0) } -// SetTrailer mocks base method. +// SetTrailer mocks base method func (m *MockCfgMgmtService_NodeExportServer) SetTrailer(arg0 metadata.MD) { m.ctrl.T.Helper() m.ctrl.Call(m, "SetTrailer", arg0) } -// SetTrailer indicates an expected call of SetTrailer. +// SetTrailer indicates an expected call of SetTrailer func (mr *MockCfgMgmtService_NodeExportServerMockRecorder) SetTrailer(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTrailer", reflect.TypeOf((*MockCfgMgmtService_NodeExportServer)(nil).SetTrailer), arg0) } -// MockCfgMgmtService_ReportExportServer is a mock of CfgMgmtService_ReportExportServer interface. +// MockCfgMgmtService_ReportExportServer is a mock of CfgMgmtService_ReportExportServer interface type MockCfgMgmtService_ReportExportServer struct { ctrl *gomock.Controller recorder *MockCfgMgmtService_ReportExportServerMockRecorder } -// MockCfgMgmtService_ReportExportServerMockRecorder is the mock recorder for MockCfgMgmtService_ReportExportServer. +// MockCfgMgmtService_ReportExportServerMockRecorder is the mock recorder for MockCfgMgmtService_ReportExportServer type MockCfgMgmtService_ReportExportServerMockRecorder struct { mock *MockCfgMgmtService_ReportExportServer } -// NewMockCfgMgmtService_ReportExportServer creates a new mock instance. +// NewMockCfgMgmtService_ReportExportServer creates a new mock instance func NewMockCfgMgmtService_ReportExportServer(ctrl *gomock.Controller) *MockCfgMgmtService_ReportExportServer { mock := &MockCfgMgmtService_ReportExportServer{ctrl: ctrl} mock.recorder = &MockCfgMgmtService_ReportExportServerMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockCfgMgmtService_ReportExportServer) EXPECT() *MockCfgMgmtService_ReportExportServerMockRecorder { return m.recorder } -// Context mocks base method. +// Context mocks base method func (m *MockCfgMgmtService_ReportExportServer) Context() context.Context { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Context") @@ -1438,13 +1437,13 @@ func (m *MockCfgMgmtService_ReportExportServer) Context() context.Context { return ret0 } -// Context indicates an expected call of Context. +// Context indicates an expected call of Context func (mr *MockCfgMgmtService_ReportExportServerMockRecorder) Context() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockCfgMgmtService_ReportExportServer)(nil).Context)) } -// RecvMsg mocks base method. +// RecvMsg mocks base method func (m *MockCfgMgmtService_ReportExportServer) RecvMsg(arg0 interface{}) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "RecvMsg", arg0) @@ -1452,13 +1451,13 @@ func (m *MockCfgMgmtService_ReportExportServer) RecvMsg(arg0 interface{}) error return ret0 } -// RecvMsg indicates an expected call of RecvMsg. +// RecvMsg indicates an expected call of RecvMsg func (mr *MockCfgMgmtService_ReportExportServerMockRecorder) RecvMsg(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockCfgMgmtService_ReportExportServer)(nil).RecvMsg), arg0) } -// Send mocks base method. +// Send mocks base method func (m *MockCfgMgmtService_ReportExportServer) Send(arg0 *response0.ReportExportData) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Send", arg0) @@ -1466,13 +1465,13 @@ func (m *MockCfgMgmtService_ReportExportServer) Send(arg0 *response0.ReportExpor return ret0 } -// Send indicates an expected call of Send. +// Send indicates an expected call of Send func (mr *MockCfgMgmtService_ReportExportServerMockRecorder) Send(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockCfgMgmtService_ReportExportServer)(nil).Send), arg0) } -// SendHeader mocks base method. +// SendHeader mocks base method func (m *MockCfgMgmtService_ReportExportServer) SendHeader(arg0 metadata.MD) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "SendHeader", arg0) @@ -1480,13 +1479,13 @@ func (m *MockCfgMgmtService_ReportExportServer) SendHeader(arg0 metadata.MD) err return ret0 } -// SendHeader indicates an expected call of SendHeader. +// SendHeader indicates an expected call of SendHeader func (mr *MockCfgMgmtService_ReportExportServerMockRecorder) SendHeader(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendHeader", reflect.TypeOf((*MockCfgMgmtService_ReportExportServer)(nil).SendHeader), arg0) } -// SendMsg mocks base method. +// SendMsg mocks base method func (m *MockCfgMgmtService_ReportExportServer) SendMsg(arg0 interface{}) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "SendMsg", arg0) @@ -1494,13 +1493,13 @@ func (m *MockCfgMgmtService_ReportExportServer) SendMsg(arg0 interface{}) error return ret0 } -// SendMsg indicates an expected call of SendMsg. +// SendMsg indicates an expected call of SendMsg func (mr *MockCfgMgmtService_ReportExportServerMockRecorder) SendMsg(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockCfgMgmtService_ReportExportServer)(nil).SendMsg), arg0) } -// SetHeader mocks base method. +// SetHeader mocks base method func (m *MockCfgMgmtService_ReportExportServer) SetHeader(arg0 metadata.MD) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "SetHeader", arg0) @@ -1508,19 +1507,19 @@ func (m *MockCfgMgmtService_ReportExportServer) SetHeader(arg0 metadata.MD) erro return ret0 } -// SetHeader indicates an expected call of SetHeader. +// SetHeader indicates an expected call of SetHeader func (mr *MockCfgMgmtService_ReportExportServerMockRecorder) SetHeader(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetHeader", reflect.TypeOf((*MockCfgMgmtService_ReportExportServer)(nil).SetHeader), arg0) } -// SetTrailer mocks base method. +// SetTrailer mocks base method func (m *MockCfgMgmtService_ReportExportServer) SetTrailer(arg0 metadata.MD) { m.ctrl.T.Helper() m.ctrl.Call(m, "SetTrailer", arg0) } -// SetTrailer indicates an expected call of SetTrailer. +// SetTrailer indicates an expected call of SetTrailer func (mr *MockCfgMgmtService_ReportExportServerMockRecorder) SetTrailer(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTrailer", reflect.TypeOf((*MockCfgMgmtService_ReportExportServer)(nil).SetTrailer), arg0) diff --git a/api/interservice/compliance/ingest/ingest/compliance.pb.client_mock.go b/api/interservice/compliance/ingest/ingest/compliance.pb.client_mock.go index 8c30af6b834..47c8d50fb6f 100644 --- a/api/interservice/compliance/ingest/ingest/compliance.pb.client_mock.go +++ b/api/interservice/compliance/ingest/ingest/compliance.pb.client_mock.go @@ -6,79 +6,78 @@ package ingest import ( context "context" - reflect "reflect" - compliance "github.com/chef/automate/api/interservice/compliance/ingest/events/compliance" event "github.com/chef/automate/api/interservice/event" gomock "github.com/golang/mock/gomock" grpc "google.golang.org/grpc" emptypb "google.golang.org/protobuf/types/known/emptypb" + reflect "reflect" ) -// MockComplianceIngesterServiceClient is a mock of ComplianceIngesterServiceClient interface. +// MockComplianceIngesterServiceClient is a mock of ComplianceIngesterServiceClient interface type MockComplianceIngesterServiceClient struct { ctrl *gomock.Controller recorder *MockComplianceIngesterServiceClientMockRecorder } -// MockComplianceIngesterServiceClientMockRecorder is the mock recorder for MockComplianceIngesterServiceClient. +// MockComplianceIngesterServiceClientMockRecorder is the mock recorder for MockComplianceIngesterServiceClient type MockComplianceIngesterServiceClientMockRecorder struct { mock *MockComplianceIngesterServiceClient } -// NewMockComplianceIngesterServiceClient creates a new mock instance. +// NewMockComplianceIngesterServiceClient creates a new mock instance func NewMockComplianceIngesterServiceClient(ctrl *gomock.Controller) *MockComplianceIngesterServiceClient { mock := &MockComplianceIngesterServiceClient{ctrl: ctrl} mock.recorder = &MockComplianceIngesterServiceClientMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockComplianceIngesterServiceClient) EXPECT() *MockComplianceIngesterServiceClientMockRecorder { return m.recorder } -// HandleEvent mocks base method. -func (m *MockComplianceIngesterServiceClient) HandleEvent(ctx context.Context, in *event.EventMsg, opts ...grpc.CallOption) (*event.EventResponse, error) { +// ProcessComplianceReport mocks base method +func (m *MockComplianceIngesterServiceClient) ProcessComplianceReport(ctx context.Context, in *compliance.Report, opts ...grpc.CallOption) (*emptypb.Empty, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "HandleEvent", varargs...) - ret0, _ := ret[0].(*event.EventResponse) + ret := m.ctrl.Call(m, "ProcessComplianceReport", varargs...) + ret0, _ := ret[0].(*emptypb.Empty) ret1, _ := ret[1].(error) return ret0, ret1 } -// HandleEvent indicates an expected call of HandleEvent. -func (mr *MockComplianceIngesterServiceClientMockRecorder) HandleEvent(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// ProcessComplianceReport indicates an expected call of ProcessComplianceReport +func (mr *MockComplianceIngesterServiceClientMockRecorder) ProcessComplianceReport(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HandleEvent", reflect.TypeOf((*MockComplianceIngesterServiceClient)(nil).HandleEvent), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProcessComplianceReport", reflect.TypeOf((*MockComplianceIngesterServiceClient)(nil).ProcessComplianceReport), varargs...) } -// ProcessComplianceReport mocks base method. -func (m *MockComplianceIngesterServiceClient) ProcessComplianceReport(ctx context.Context, in *compliance.Report, opts ...grpc.CallOption) (*emptypb.Empty, error) { +// HandleEvent mocks base method +func (m *MockComplianceIngesterServiceClient) HandleEvent(ctx context.Context, in *event.EventMsg, opts ...grpc.CallOption) (*event.EventResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "ProcessComplianceReport", varargs...) - ret0, _ := ret[0].(*emptypb.Empty) + ret := m.ctrl.Call(m, "HandleEvent", varargs...) + ret0, _ := ret[0].(*event.EventResponse) ret1, _ := ret[1].(error) return ret0, ret1 } -// ProcessComplianceReport indicates an expected call of ProcessComplianceReport. -func (mr *MockComplianceIngesterServiceClientMockRecorder) ProcessComplianceReport(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// HandleEvent indicates an expected call of HandleEvent +func (mr *MockComplianceIngesterServiceClientMockRecorder) HandleEvent(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProcessComplianceReport", reflect.TypeOf((*MockComplianceIngesterServiceClient)(nil).ProcessComplianceReport), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HandleEvent", reflect.TypeOf((*MockComplianceIngesterServiceClient)(nil).HandleEvent), varargs...) } -// ProjectUpdateStatus mocks base method. +// ProjectUpdateStatus mocks base method func (m *MockComplianceIngesterServiceClient) ProjectUpdateStatus(ctx context.Context, in *ProjectUpdateStatusReq, opts ...grpc.CallOption) (*ProjectUpdateStatusResp, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} @@ -91,67 +90,67 @@ func (m *MockComplianceIngesterServiceClient) ProjectUpdateStatus(ctx context.Co return ret0, ret1 } -// ProjectUpdateStatus indicates an expected call of ProjectUpdateStatus. +// ProjectUpdateStatus indicates an expected call of ProjectUpdateStatus func (mr *MockComplianceIngesterServiceClientMockRecorder) ProjectUpdateStatus(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProjectUpdateStatus", reflect.TypeOf((*MockComplianceIngesterServiceClient)(nil).ProjectUpdateStatus), varargs...) } -// MockComplianceIngesterServiceServer is a mock of ComplianceIngesterServiceServer interface. +// MockComplianceIngesterServiceServer is a mock of ComplianceIngesterServiceServer interface type MockComplianceIngesterServiceServer struct { ctrl *gomock.Controller recorder *MockComplianceIngesterServiceServerMockRecorder } -// MockComplianceIngesterServiceServerMockRecorder is the mock recorder for MockComplianceIngesterServiceServer. +// MockComplianceIngesterServiceServerMockRecorder is the mock recorder for MockComplianceIngesterServiceServer type MockComplianceIngesterServiceServerMockRecorder struct { mock *MockComplianceIngesterServiceServer } -// NewMockComplianceIngesterServiceServer creates a new mock instance. +// NewMockComplianceIngesterServiceServer creates a new mock instance func NewMockComplianceIngesterServiceServer(ctrl *gomock.Controller) *MockComplianceIngesterServiceServer { mock := &MockComplianceIngesterServiceServer{ctrl: ctrl} mock.recorder = &MockComplianceIngesterServiceServerMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockComplianceIngesterServiceServer) EXPECT() *MockComplianceIngesterServiceServerMockRecorder { return m.recorder } -// HandleEvent mocks base method. -func (m *MockComplianceIngesterServiceServer) HandleEvent(arg0 context.Context, arg1 *event.EventMsg) (*event.EventResponse, error) { +// ProcessComplianceReport mocks base method +func (m *MockComplianceIngesterServiceServer) ProcessComplianceReport(arg0 context.Context, arg1 *compliance.Report) (*emptypb.Empty, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "HandleEvent", arg0, arg1) - ret0, _ := ret[0].(*event.EventResponse) + ret := m.ctrl.Call(m, "ProcessComplianceReport", arg0, arg1) + ret0, _ := ret[0].(*emptypb.Empty) ret1, _ := ret[1].(error) return ret0, ret1 } -// HandleEvent indicates an expected call of HandleEvent. -func (mr *MockComplianceIngesterServiceServerMockRecorder) HandleEvent(arg0, arg1 interface{}) *gomock.Call { +// ProcessComplianceReport indicates an expected call of ProcessComplianceReport +func (mr *MockComplianceIngesterServiceServerMockRecorder) ProcessComplianceReport(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HandleEvent", reflect.TypeOf((*MockComplianceIngesterServiceServer)(nil).HandleEvent), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProcessComplianceReport", reflect.TypeOf((*MockComplianceIngesterServiceServer)(nil).ProcessComplianceReport), arg0, arg1) } -// ProcessComplianceReport mocks base method. -func (m *MockComplianceIngesterServiceServer) ProcessComplianceReport(arg0 context.Context, arg1 *compliance.Report) (*emptypb.Empty, error) { +// HandleEvent mocks base method +func (m *MockComplianceIngesterServiceServer) HandleEvent(arg0 context.Context, arg1 *event.EventMsg) (*event.EventResponse, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ProcessComplianceReport", arg0, arg1) - ret0, _ := ret[0].(*emptypb.Empty) + ret := m.ctrl.Call(m, "HandleEvent", arg0, arg1) + ret0, _ := ret[0].(*event.EventResponse) ret1, _ := ret[1].(error) return ret0, ret1 } -// ProcessComplianceReport indicates an expected call of ProcessComplianceReport. -func (mr *MockComplianceIngesterServiceServerMockRecorder) ProcessComplianceReport(arg0, arg1 interface{}) *gomock.Call { +// HandleEvent indicates an expected call of HandleEvent +func (mr *MockComplianceIngesterServiceServerMockRecorder) HandleEvent(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProcessComplianceReport", reflect.TypeOf((*MockComplianceIngesterServiceServer)(nil).ProcessComplianceReport), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HandleEvent", reflect.TypeOf((*MockComplianceIngesterServiceServer)(nil).HandleEvent), arg0, arg1) } -// ProjectUpdateStatus mocks base method. +// ProjectUpdateStatus mocks base method func (m *MockComplianceIngesterServiceServer) ProjectUpdateStatus(arg0 context.Context, arg1 *ProjectUpdateStatusReq) (*ProjectUpdateStatusResp, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ProjectUpdateStatus", arg0, arg1) @@ -160,7 +159,7 @@ func (m *MockComplianceIngesterServiceServer) ProjectUpdateStatus(arg0 context.C return ret0, ret1 } -// ProjectUpdateStatus indicates an expected call of ProjectUpdateStatus. +// ProjectUpdateStatus indicates an expected call of ProjectUpdateStatus func (mr *MockComplianceIngesterServiceServerMockRecorder) ProjectUpdateStatus(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProjectUpdateStatus", reflect.TypeOf((*MockComplianceIngesterServiceServer)(nil).ProjectUpdateStatus), arg0, arg1) diff --git a/api/interservice/event/event.pb.client_mock.go b/api/interservice/event/event.pb.client_mock.go index 119e4a2157c..0cfd1910616 100644 --- a/api/interservice/event/event.pb.client_mock.go +++ b/api/interservice/event/event.pb.client_mock.go @@ -6,36 +6,35 @@ package event import ( context "context" - reflect "reflect" - gomock "github.com/golang/mock/gomock" grpc "google.golang.org/grpc" + reflect "reflect" ) -// MockEventServiceClient is a mock of EventServiceClient interface. +// MockEventServiceClient is a mock of EventServiceClient interface type MockEventServiceClient struct { ctrl *gomock.Controller recorder *MockEventServiceClientMockRecorder } -// MockEventServiceClientMockRecorder is the mock recorder for MockEventServiceClient. +// MockEventServiceClientMockRecorder is the mock recorder for MockEventServiceClient type MockEventServiceClientMockRecorder struct { mock *MockEventServiceClient } -// NewMockEventServiceClient creates a new mock instance. +// NewMockEventServiceClient creates a new mock instance func NewMockEventServiceClient(ctrl *gomock.Controller) *MockEventServiceClient { mock := &MockEventServiceClient{ctrl: ctrl} mock.recorder = &MockEventServiceClientMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockEventServiceClient) EXPECT() *MockEventServiceClientMockRecorder { return m.recorder } -// Publish mocks base method. +// Publish mocks base method func (m *MockEventServiceClient) Publish(ctx context.Context, in *PublishRequest, opts ...grpc.CallOption) (*PublishResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} @@ -48,97 +47,97 @@ func (m *MockEventServiceClient) Publish(ctx context.Context, in *PublishRequest return ret0, ret1 } -// Publish indicates an expected call of Publish. +// Publish indicates an expected call of Publish func (mr *MockEventServiceClientMockRecorder) Publish(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Publish", reflect.TypeOf((*MockEventServiceClient)(nil).Publish), varargs...) } -// Start mocks base method. -func (m *MockEventServiceClient) Start(ctx context.Context, in *StartRequest, opts ...grpc.CallOption) (*StartResponse, error) { +// Subscribe mocks base method +func (m *MockEventServiceClient) Subscribe(ctx context.Context, in *SubscribeRequest, opts ...grpc.CallOption) (*SubscribeResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "Start", varargs...) - ret0, _ := ret[0].(*StartResponse) + ret := m.ctrl.Call(m, "Subscribe", varargs...) + ret0, _ := ret[0].(*SubscribeResponse) ret1, _ := ret[1].(error) return ret0, ret1 } -// Start indicates an expected call of Start. -func (mr *MockEventServiceClientMockRecorder) Start(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// Subscribe indicates an expected call of Subscribe +func (mr *MockEventServiceClientMockRecorder) Subscribe(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Start", reflect.TypeOf((*MockEventServiceClient)(nil).Start), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Subscribe", reflect.TypeOf((*MockEventServiceClient)(nil).Subscribe), varargs...) } -// Stop mocks base method. -func (m *MockEventServiceClient) Stop(ctx context.Context, in *StopRequest, opts ...grpc.CallOption) (*StopResponse, error) { +// Start mocks base method +func (m *MockEventServiceClient) Start(ctx context.Context, in *StartRequest, opts ...grpc.CallOption) (*StartResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "Stop", varargs...) - ret0, _ := ret[0].(*StopResponse) + ret := m.ctrl.Call(m, "Start", varargs...) + ret0, _ := ret[0].(*StartResponse) ret1, _ := ret[1].(error) return ret0, ret1 } -// Stop indicates an expected call of Stop. -func (mr *MockEventServiceClientMockRecorder) Stop(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// Start indicates an expected call of Start +func (mr *MockEventServiceClientMockRecorder) Start(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stop", reflect.TypeOf((*MockEventServiceClient)(nil).Stop), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Start", reflect.TypeOf((*MockEventServiceClient)(nil).Start), varargs...) } -// Subscribe mocks base method. -func (m *MockEventServiceClient) Subscribe(ctx context.Context, in *SubscribeRequest, opts ...grpc.CallOption) (*SubscribeResponse, error) { +// Stop mocks base method +func (m *MockEventServiceClient) Stop(ctx context.Context, in *StopRequest, opts ...grpc.CallOption) (*StopResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "Subscribe", varargs...) - ret0, _ := ret[0].(*SubscribeResponse) + ret := m.ctrl.Call(m, "Stop", varargs...) + ret0, _ := ret[0].(*StopResponse) ret1, _ := ret[1].(error) return ret0, ret1 } -// Subscribe indicates an expected call of Subscribe. -func (mr *MockEventServiceClientMockRecorder) Subscribe(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// Stop indicates an expected call of Stop +func (mr *MockEventServiceClientMockRecorder) Stop(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Subscribe", reflect.TypeOf((*MockEventServiceClient)(nil).Subscribe), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stop", reflect.TypeOf((*MockEventServiceClient)(nil).Stop), varargs...) } -// MockEventServiceServer is a mock of EventServiceServer interface. +// MockEventServiceServer is a mock of EventServiceServer interface type MockEventServiceServer struct { ctrl *gomock.Controller recorder *MockEventServiceServerMockRecorder } -// MockEventServiceServerMockRecorder is the mock recorder for MockEventServiceServer. +// MockEventServiceServerMockRecorder is the mock recorder for MockEventServiceServer type MockEventServiceServerMockRecorder struct { mock *MockEventServiceServer } -// NewMockEventServiceServer creates a new mock instance. +// NewMockEventServiceServer creates a new mock instance func NewMockEventServiceServer(ctrl *gomock.Controller) *MockEventServiceServer { mock := &MockEventServiceServer{ctrl: ctrl} mock.recorder = &MockEventServiceServerMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockEventServiceServer) EXPECT() *MockEventServiceServerMockRecorder { return m.recorder } -// Publish mocks base method. +// Publish mocks base method func (m *MockEventServiceServer) Publish(arg0 context.Context, arg1 *PublishRequest) (*PublishResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Publish", arg0, arg1) @@ -147,13 +146,28 @@ func (m *MockEventServiceServer) Publish(arg0 context.Context, arg1 *PublishRequ return ret0, ret1 } -// Publish indicates an expected call of Publish. +// Publish indicates an expected call of Publish func (mr *MockEventServiceServerMockRecorder) Publish(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Publish", reflect.TypeOf((*MockEventServiceServer)(nil).Publish), arg0, arg1) } -// Start mocks base method. +// Subscribe mocks base method +func (m *MockEventServiceServer) Subscribe(arg0 context.Context, arg1 *SubscribeRequest) (*SubscribeResponse, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Subscribe", arg0, arg1) + ret0, _ := ret[0].(*SubscribeResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// Subscribe indicates an expected call of Subscribe +func (mr *MockEventServiceServerMockRecorder) Subscribe(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Subscribe", reflect.TypeOf((*MockEventServiceServer)(nil).Subscribe), arg0, arg1) +} + +// Start mocks base method func (m *MockEventServiceServer) Start(arg0 context.Context, arg1 *StartRequest) (*StartResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Start", arg0, arg1) @@ -162,13 +176,13 @@ func (m *MockEventServiceServer) Start(arg0 context.Context, arg1 *StartRequest) return ret0, ret1 } -// Start indicates an expected call of Start. +// Start indicates an expected call of Start func (mr *MockEventServiceServerMockRecorder) Start(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Start", reflect.TypeOf((*MockEventServiceServer)(nil).Start), arg0, arg1) } -// Stop mocks base method. +// Stop mocks base method func (m *MockEventServiceServer) Stop(arg0 context.Context, arg1 *StopRequest) (*StopResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Stop", arg0, arg1) @@ -177,23 +191,8 @@ func (m *MockEventServiceServer) Stop(arg0 context.Context, arg1 *StopRequest) ( return ret0, ret1 } -// Stop indicates an expected call of Stop. +// Stop indicates an expected call of Stop func (mr *MockEventServiceServerMockRecorder) Stop(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stop", reflect.TypeOf((*MockEventServiceServer)(nil).Stop), arg0, arg1) } - -// Subscribe mocks base method. -func (m *MockEventServiceServer) Subscribe(arg0 context.Context, arg1 *SubscribeRequest) (*SubscribeResponse, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Subscribe", arg0, arg1) - ret0, _ := ret[0].(*SubscribeResponse) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// Subscribe indicates an expected call of Subscribe. -func (mr *MockEventServiceServerMockRecorder) Subscribe(arg0, arg1 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Subscribe", reflect.TypeOf((*MockEventServiceServer)(nil).Subscribe), arg0, arg1) -} diff --git a/api/interservice/event_feed/event_feed.pb.client_mock.go b/api/interservice/event_feed/event_feed.pb.client_mock.go index 8484e9f7787..c6d9c7674ea 100644 --- a/api/interservice/event_feed/event_feed.pb.client_mock.go +++ b/api/interservice/event_feed/event_feed.pb.client_mock.go @@ -6,38 +6,37 @@ package event_feed import ( context "context" - reflect "reflect" - event "github.com/chef/automate/api/interservice/event" gomock "github.com/golang/mock/gomock" grpc "google.golang.org/grpc" metadata "google.golang.org/grpc/metadata" + reflect "reflect" ) -// MockEventFeedServiceClient is a mock of EventFeedServiceClient interface. +// MockEventFeedServiceClient is a mock of EventFeedServiceClient interface type MockEventFeedServiceClient struct { ctrl *gomock.Controller recorder *MockEventFeedServiceClientMockRecorder } -// MockEventFeedServiceClientMockRecorder is the mock recorder for MockEventFeedServiceClient. +// MockEventFeedServiceClientMockRecorder is the mock recorder for MockEventFeedServiceClient type MockEventFeedServiceClientMockRecorder struct { mock *MockEventFeedServiceClient } -// NewMockEventFeedServiceClient creates a new mock instance. +// NewMockEventFeedServiceClient creates a new mock instance func NewMockEventFeedServiceClient(ctrl *gomock.Controller) *MockEventFeedServiceClient { mock := &MockEventFeedServiceClient{ctrl: ctrl} mock.recorder = &MockEventFeedServiceClientMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockEventFeedServiceClient) EXPECT() *MockEventFeedServiceClientMockRecorder { return m.recorder } -// EventExport mocks base method. +// EventExport mocks base method func (m *MockEventFeedServiceClient) EventExport(arg0 context.Context, arg1 *EventExportRequest, arg2 ...grpc.CallOption) (EventFeedService_EventExportClient, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -50,14 +49,14 @@ func (m *MockEventFeedServiceClient) EventExport(arg0 context.Context, arg1 *Eve return ret0, ret1 } -// EventExport indicates an expected call of EventExport. +// EventExport indicates an expected call of EventExport func (mr *MockEventFeedServiceClientMockRecorder) EventExport(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "EventExport", reflect.TypeOf((*MockEventFeedServiceClient)(nil).EventExport), varargs...) } -// GetFeed mocks base method. +// GetFeed mocks base method func (m *MockEventFeedServiceClient) GetFeed(arg0 context.Context, arg1 *FeedRequest, arg2 ...grpc.CallOption) (*FeedResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -70,14 +69,14 @@ func (m *MockEventFeedServiceClient) GetFeed(arg0 context.Context, arg1 *FeedReq return ret0, ret1 } -// GetFeed indicates an expected call of GetFeed. +// GetFeed indicates an expected call of GetFeed func (mr *MockEventFeedServiceClientMockRecorder) GetFeed(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFeed", reflect.TypeOf((*MockEventFeedServiceClient)(nil).GetFeed), varargs...) } -// GetFeedSummary mocks base method. +// GetFeedSummary mocks base method func (m *MockEventFeedServiceClient) GetFeedSummary(arg0 context.Context, arg1 *FeedSummaryRequest, arg2 ...grpc.CallOption) (*FeedSummaryResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -90,14 +89,14 @@ func (m *MockEventFeedServiceClient) GetFeedSummary(arg0 context.Context, arg1 * return ret0, ret1 } -// GetFeedSummary indicates an expected call of GetFeedSummary. +// GetFeedSummary indicates an expected call of GetFeedSummary func (mr *MockEventFeedServiceClientMockRecorder) GetFeedSummary(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFeedSummary", reflect.TypeOf((*MockEventFeedServiceClient)(nil).GetFeedSummary), varargs...) } -// GetFeedTimeline mocks base method. +// GetFeedTimeline mocks base method func (m *MockEventFeedServiceClient) GetFeedTimeline(arg0 context.Context, arg1 *FeedTimelineRequest, arg2 ...grpc.CallOption) (*FeedTimelineResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -110,14 +109,14 @@ func (m *MockEventFeedServiceClient) GetFeedTimeline(arg0 context.Context, arg1 return ret0, ret1 } -// GetFeedTimeline indicates an expected call of GetFeedTimeline. +// GetFeedTimeline indicates an expected call of GetFeedTimeline func (mr *MockEventFeedServiceClientMockRecorder) GetFeedTimeline(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFeedTimeline", reflect.TypeOf((*MockEventFeedServiceClient)(nil).GetFeedTimeline), varargs...) } -// HandleEvent mocks base method. +// HandleEvent mocks base method func (m *MockEventFeedServiceClient) HandleEvent(arg0 context.Context, arg1 *event.EventMsg, arg2 ...grpc.CallOption) (*event.EventResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} @@ -130,37 +129,37 @@ func (m *MockEventFeedServiceClient) HandleEvent(arg0 context.Context, arg1 *eve return ret0, ret1 } -// HandleEvent indicates an expected call of HandleEvent. +// HandleEvent indicates an expected call of HandleEvent func (mr *MockEventFeedServiceClientMockRecorder) HandleEvent(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HandleEvent", reflect.TypeOf((*MockEventFeedServiceClient)(nil).HandleEvent), varargs...) } -// MockEventFeedService_EventExportClient is a mock of EventFeedService_EventExportClient interface. +// MockEventFeedService_EventExportClient is a mock of EventFeedService_EventExportClient interface type MockEventFeedService_EventExportClient struct { ctrl *gomock.Controller recorder *MockEventFeedService_EventExportClientMockRecorder } -// MockEventFeedService_EventExportClientMockRecorder is the mock recorder for MockEventFeedService_EventExportClient. +// MockEventFeedService_EventExportClientMockRecorder is the mock recorder for MockEventFeedService_EventExportClient type MockEventFeedService_EventExportClientMockRecorder struct { mock *MockEventFeedService_EventExportClient } -// NewMockEventFeedService_EventExportClient creates a new mock instance. +// NewMockEventFeedService_EventExportClient creates a new mock instance func NewMockEventFeedService_EventExportClient(ctrl *gomock.Controller) *MockEventFeedService_EventExportClient { mock := &MockEventFeedService_EventExportClient{ctrl: ctrl} mock.recorder = &MockEventFeedService_EventExportClientMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockEventFeedService_EventExportClient) EXPECT() *MockEventFeedService_EventExportClientMockRecorder { return m.recorder } -// CloseSend mocks base method. +// CloseSend mocks base method func (m *MockEventFeedService_EventExportClient) CloseSend() error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "CloseSend") @@ -168,13 +167,13 @@ func (m *MockEventFeedService_EventExportClient) CloseSend() error { return ret0 } -// CloseSend indicates an expected call of CloseSend. +// CloseSend indicates an expected call of CloseSend func (mr *MockEventFeedService_EventExportClientMockRecorder) CloseSend() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseSend", reflect.TypeOf((*MockEventFeedService_EventExportClient)(nil).CloseSend)) } -// Context mocks base method. +// Context mocks base method func (m *MockEventFeedService_EventExportClient) Context() context.Context { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Context") @@ -182,13 +181,13 @@ func (m *MockEventFeedService_EventExportClient) Context() context.Context { return ret0 } -// Context indicates an expected call of Context. +// Context indicates an expected call of Context func (mr *MockEventFeedService_EventExportClientMockRecorder) Context() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockEventFeedService_EventExportClient)(nil).Context)) } -// Header mocks base method. +// Header mocks base method func (m *MockEventFeedService_EventExportClient) Header() (metadata.MD, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Header") @@ -197,13 +196,13 @@ func (m *MockEventFeedService_EventExportClient) Header() (metadata.MD, error) { return ret0, ret1 } -// Header indicates an expected call of Header. +// Header indicates an expected call of Header func (mr *MockEventFeedService_EventExportClientMockRecorder) Header() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Header", reflect.TypeOf((*MockEventFeedService_EventExportClient)(nil).Header)) } -// Recv mocks base method. +// Recv mocks base method func (m *MockEventFeedService_EventExportClient) Recv() (*EventExportResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Recv") @@ -212,13 +211,13 @@ func (m *MockEventFeedService_EventExportClient) Recv() (*EventExportResponse, e return ret0, ret1 } -// Recv indicates an expected call of Recv. +// Recv indicates an expected call of Recv func (mr *MockEventFeedService_EventExportClientMockRecorder) Recv() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Recv", reflect.TypeOf((*MockEventFeedService_EventExportClient)(nil).Recv)) } -// RecvMsg mocks base method. +// RecvMsg mocks base method func (m *MockEventFeedService_EventExportClient) RecvMsg(arg0 interface{}) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "RecvMsg", arg0) @@ -226,13 +225,13 @@ func (m *MockEventFeedService_EventExportClient) RecvMsg(arg0 interface{}) error return ret0 } -// RecvMsg indicates an expected call of RecvMsg. +// RecvMsg indicates an expected call of RecvMsg func (mr *MockEventFeedService_EventExportClientMockRecorder) RecvMsg(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockEventFeedService_EventExportClient)(nil).RecvMsg), arg0) } -// SendMsg mocks base method. +// SendMsg mocks base method func (m *MockEventFeedService_EventExportClient) SendMsg(arg0 interface{}) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "SendMsg", arg0) @@ -240,13 +239,13 @@ func (m *MockEventFeedService_EventExportClient) SendMsg(arg0 interface{}) error return ret0 } -// SendMsg indicates an expected call of SendMsg. +// SendMsg indicates an expected call of SendMsg func (mr *MockEventFeedService_EventExportClientMockRecorder) SendMsg(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockEventFeedService_EventExportClient)(nil).SendMsg), arg0) } -// Trailer mocks base method. +// Trailer mocks base method func (m *MockEventFeedService_EventExportClient) Trailer() metadata.MD { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Trailer") @@ -254,36 +253,36 @@ func (m *MockEventFeedService_EventExportClient) Trailer() metadata.MD { return ret0 } -// Trailer indicates an expected call of Trailer. +// Trailer indicates an expected call of Trailer func (mr *MockEventFeedService_EventExportClientMockRecorder) Trailer() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Trailer", reflect.TypeOf((*MockEventFeedService_EventExportClient)(nil).Trailer)) } -// MockEventFeedServiceServer is a mock of EventFeedServiceServer interface. +// MockEventFeedServiceServer is a mock of EventFeedServiceServer interface type MockEventFeedServiceServer struct { ctrl *gomock.Controller recorder *MockEventFeedServiceServerMockRecorder } -// MockEventFeedServiceServerMockRecorder is the mock recorder for MockEventFeedServiceServer. +// MockEventFeedServiceServerMockRecorder is the mock recorder for MockEventFeedServiceServer type MockEventFeedServiceServerMockRecorder struct { mock *MockEventFeedServiceServer } -// NewMockEventFeedServiceServer creates a new mock instance. +// NewMockEventFeedServiceServer creates a new mock instance func NewMockEventFeedServiceServer(ctrl *gomock.Controller) *MockEventFeedServiceServer { mock := &MockEventFeedServiceServer{ctrl: ctrl} mock.recorder = &MockEventFeedServiceServerMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockEventFeedServiceServer) EXPECT() *MockEventFeedServiceServerMockRecorder { return m.recorder } -// EventExport mocks base method. +// EventExport mocks base method func (m *MockEventFeedServiceServer) EventExport(arg0 *EventExportRequest, arg1 EventFeedService_EventExportServer) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "EventExport", arg0, arg1) @@ -291,13 +290,13 @@ func (m *MockEventFeedServiceServer) EventExport(arg0 *EventExportRequest, arg1 return ret0 } -// EventExport indicates an expected call of EventExport. +// EventExport indicates an expected call of EventExport func (mr *MockEventFeedServiceServerMockRecorder) EventExport(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "EventExport", reflect.TypeOf((*MockEventFeedServiceServer)(nil).EventExport), arg0, arg1) } -// GetFeed mocks base method. +// GetFeed mocks base method func (m *MockEventFeedServiceServer) GetFeed(arg0 context.Context, arg1 *FeedRequest) (*FeedResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetFeed", arg0, arg1) @@ -306,13 +305,13 @@ func (m *MockEventFeedServiceServer) GetFeed(arg0 context.Context, arg1 *FeedReq return ret0, ret1 } -// GetFeed indicates an expected call of GetFeed. +// GetFeed indicates an expected call of GetFeed func (mr *MockEventFeedServiceServerMockRecorder) GetFeed(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFeed", reflect.TypeOf((*MockEventFeedServiceServer)(nil).GetFeed), arg0, arg1) } -// GetFeedSummary mocks base method. +// GetFeedSummary mocks base method func (m *MockEventFeedServiceServer) GetFeedSummary(arg0 context.Context, arg1 *FeedSummaryRequest) (*FeedSummaryResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetFeedSummary", arg0, arg1) @@ -321,13 +320,13 @@ func (m *MockEventFeedServiceServer) GetFeedSummary(arg0 context.Context, arg1 * return ret0, ret1 } -// GetFeedSummary indicates an expected call of GetFeedSummary. +// GetFeedSummary indicates an expected call of GetFeedSummary func (mr *MockEventFeedServiceServerMockRecorder) GetFeedSummary(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFeedSummary", reflect.TypeOf((*MockEventFeedServiceServer)(nil).GetFeedSummary), arg0, arg1) } -// GetFeedTimeline mocks base method. +// GetFeedTimeline mocks base method func (m *MockEventFeedServiceServer) GetFeedTimeline(arg0 context.Context, arg1 *FeedTimelineRequest) (*FeedTimelineResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetFeedTimeline", arg0, arg1) @@ -336,13 +335,13 @@ func (m *MockEventFeedServiceServer) GetFeedTimeline(arg0 context.Context, arg1 return ret0, ret1 } -// GetFeedTimeline indicates an expected call of GetFeedTimeline. +// GetFeedTimeline indicates an expected call of GetFeedTimeline func (mr *MockEventFeedServiceServerMockRecorder) GetFeedTimeline(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFeedTimeline", reflect.TypeOf((*MockEventFeedServiceServer)(nil).GetFeedTimeline), arg0, arg1) } -// HandleEvent mocks base method. +// HandleEvent mocks base method func (m *MockEventFeedServiceServer) HandleEvent(arg0 context.Context, arg1 *event.EventMsg) (*event.EventResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "HandleEvent", arg0, arg1) @@ -351,36 +350,36 @@ func (m *MockEventFeedServiceServer) HandleEvent(arg0 context.Context, arg1 *eve return ret0, ret1 } -// HandleEvent indicates an expected call of HandleEvent. +// HandleEvent indicates an expected call of HandleEvent func (mr *MockEventFeedServiceServerMockRecorder) HandleEvent(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HandleEvent", reflect.TypeOf((*MockEventFeedServiceServer)(nil).HandleEvent), arg0, arg1) } -// MockEventFeedService_EventExportServer is a mock of EventFeedService_EventExportServer interface. +// MockEventFeedService_EventExportServer is a mock of EventFeedService_EventExportServer interface type MockEventFeedService_EventExportServer struct { ctrl *gomock.Controller recorder *MockEventFeedService_EventExportServerMockRecorder } -// MockEventFeedService_EventExportServerMockRecorder is the mock recorder for MockEventFeedService_EventExportServer. +// MockEventFeedService_EventExportServerMockRecorder is the mock recorder for MockEventFeedService_EventExportServer type MockEventFeedService_EventExportServerMockRecorder struct { mock *MockEventFeedService_EventExportServer } -// NewMockEventFeedService_EventExportServer creates a new mock instance. +// NewMockEventFeedService_EventExportServer creates a new mock instance func NewMockEventFeedService_EventExportServer(ctrl *gomock.Controller) *MockEventFeedService_EventExportServer { mock := &MockEventFeedService_EventExportServer{ctrl: ctrl} mock.recorder = &MockEventFeedService_EventExportServerMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockEventFeedService_EventExportServer) EXPECT() *MockEventFeedService_EventExportServerMockRecorder { return m.recorder } -// Context mocks base method. +// Context mocks base method func (m *MockEventFeedService_EventExportServer) Context() context.Context { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Context") @@ -388,13 +387,13 @@ func (m *MockEventFeedService_EventExportServer) Context() context.Context { return ret0 } -// Context indicates an expected call of Context. +// Context indicates an expected call of Context func (mr *MockEventFeedService_EventExportServerMockRecorder) Context() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockEventFeedService_EventExportServer)(nil).Context)) } -// RecvMsg mocks base method. +// RecvMsg mocks base method func (m *MockEventFeedService_EventExportServer) RecvMsg(arg0 interface{}) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "RecvMsg", arg0) @@ -402,13 +401,13 @@ func (m *MockEventFeedService_EventExportServer) RecvMsg(arg0 interface{}) error return ret0 } -// RecvMsg indicates an expected call of RecvMsg. +// RecvMsg indicates an expected call of RecvMsg func (mr *MockEventFeedService_EventExportServerMockRecorder) RecvMsg(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockEventFeedService_EventExportServer)(nil).RecvMsg), arg0) } -// Send mocks base method. +// Send mocks base method func (m *MockEventFeedService_EventExportServer) Send(arg0 *EventExportResponse) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Send", arg0) @@ -416,13 +415,13 @@ func (m *MockEventFeedService_EventExportServer) Send(arg0 *EventExportResponse) return ret0 } -// Send indicates an expected call of Send. +// Send indicates an expected call of Send func (mr *MockEventFeedService_EventExportServerMockRecorder) Send(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockEventFeedService_EventExportServer)(nil).Send), arg0) } -// SendHeader mocks base method. +// SendHeader mocks base method func (m *MockEventFeedService_EventExportServer) SendHeader(arg0 metadata.MD) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "SendHeader", arg0) @@ -430,13 +429,13 @@ func (m *MockEventFeedService_EventExportServer) SendHeader(arg0 metadata.MD) er return ret0 } -// SendHeader indicates an expected call of SendHeader. +// SendHeader indicates an expected call of SendHeader func (mr *MockEventFeedService_EventExportServerMockRecorder) SendHeader(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendHeader", reflect.TypeOf((*MockEventFeedService_EventExportServer)(nil).SendHeader), arg0) } -// SendMsg mocks base method. +// SendMsg mocks base method func (m *MockEventFeedService_EventExportServer) SendMsg(arg0 interface{}) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "SendMsg", arg0) @@ -444,13 +443,13 @@ func (m *MockEventFeedService_EventExportServer) SendMsg(arg0 interface{}) error return ret0 } -// SendMsg indicates an expected call of SendMsg. +// SendMsg indicates an expected call of SendMsg func (mr *MockEventFeedService_EventExportServerMockRecorder) SendMsg(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockEventFeedService_EventExportServer)(nil).SendMsg), arg0) } -// SetHeader mocks base method. +// SetHeader mocks base method func (m *MockEventFeedService_EventExportServer) SetHeader(arg0 metadata.MD) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "SetHeader", arg0) @@ -458,19 +457,19 @@ func (m *MockEventFeedService_EventExportServer) SetHeader(arg0 metadata.MD) err return ret0 } -// SetHeader indicates an expected call of SetHeader. +// SetHeader indicates an expected call of SetHeader func (mr *MockEventFeedService_EventExportServerMockRecorder) SetHeader(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetHeader", reflect.TypeOf((*MockEventFeedService_EventExportServer)(nil).SetHeader), arg0) } -// SetTrailer mocks base method. +// SetTrailer mocks base method func (m *MockEventFeedService_EventExportServer) SetTrailer(arg0 metadata.MD) { m.ctrl.T.Helper() m.ctrl.Call(m, "SetTrailer", arg0) } -// SetTrailer indicates an expected call of SetTrailer. +// SetTrailer indicates an expected call of SetTrailer func (mr *MockEventFeedService_EventExportServerMockRecorder) SetTrailer(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTrailer", reflect.TypeOf((*MockEventFeedService_EventExportServer)(nil).SetTrailer), arg0) diff --git a/api/interservice/infra_proxy/migrations/request/migrations.pb.go b/api/interservice/infra_proxy/migrations/request/migrations.pb.go index 213229449f9..125a579a1b4 100644 --- a/api/interservice/infra_proxy/migrations/request/migrations.pb.go +++ b/api/interservice/infra_proxy/migrations/request/migrations.pb.go @@ -11,6 +11,7 @@ import ( protoreflect "google.golang.org/protobuf/reflect/protoreflect" protoimpl "google.golang.org/protobuf/runtime/protoimpl" reflect "reflect" + sync "sync" ) const ( @@ -24,6 +25,187 @@ const ( // of the legacy proto package is being used. const _ = proto.ProtoPackageIsVersion4 +type UploadZipFileRequest struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields + + //Chef infra server id for the file + ServerId string `protobuf:"bytes,1,opt,name=serverId,proto3" json:"serverId,omitempty" toml:"serverId,omitempty" mapstructure:"serverId,omitempty"` + //Chunk of the data in bytes + Chunk *Chunk `protobuf:"bytes,2,opt,name=chunk,proto3" json:"chunk,omitempty" toml:"chunk,omitempty" mapstructure:"chunk,omitempty"` + //Metadata for the request + Meta *Metadata `protobuf:"bytes,3,opt,name=meta,proto3" json:"meta,omitempty" toml:"meta,omitempty" mapstructure:"meta,omitempty"` +} + +func (x *UploadZipFileRequest) Reset() { + *x = UploadZipFileRequest{} + if protoimpl.UnsafeEnabled { + mi := &file_interservice_infra_proxy_migrations_request_migrations_proto_msgTypes[0] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) + } +} + +func (x *UploadZipFileRequest) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*UploadZipFileRequest) ProtoMessage() {} + +func (x *UploadZipFileRequest) ProtoReflect() protoreflect.Message { + mi := &file_interservice_infra_proxy_migrations_request_migrations_proto_msgTypes[0] + if protoimpl.UnsafeEnabled && x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use UploadZipFileRequest.ProtoReflect.Descriptor instead. +func (*UploadZipFileRequest) Descriptor() ([]byte, []int) { + return file_interservice_infra_proxy_migrations_request_migrations_proto_rawDescGZIP(), []int{0} +} + +func (x *UploadZipFileRequest) GetServerId() string { + if x != nil { + return x.ServerId + } + return "" +} + +func (x *UploadZipFileRequest) GetChunk() *Chunk { + if x != nil { + return x.Chunk + } + return nil +} + +func (x *UploadZipFileRequest) GetMeta() *Metadata { + if x != nil { + return x.Meta + } + return nil +} + +type Metadata struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields + + //Migration File name + Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty" toml:"name,omitempty" mapstructure:"name,omitempty"` + //Content type of the request + ContentType string `protobuf:"bytes,2,opt,name=contentType,proto3" json:"contentType,omitempty" toml:"contentType,omitempty" mapstructure:"contentType,omitempty"` +} + +func (x *Metadata) Reset() { + *x = Metadata{} + if protoimpl.UnsafeEnabled { + mi := &file_interservice_infra_proxy_migrations_request_migrations_proto_msgTypes[1] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) + } +} + +func (x *Metadata) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*Metadata) ProtoMessage() {} + +func (x *Metadata) ProtoReflect() protoreflect.Message { + mi := &file_interservice_infra_proxy_migrations_request_migrations_proto_msgTypes[1] + if protoimpl.UnsafeEnabled && x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use Metadata.ProtoReflect.Descriptor instead. +func (*Metadata) Descriptor() ([]byte, []int) { + return file_interservice_infra_proxy_migrations_request_migrations_proto_rawDescGZIP(), []int{1} +} + +func (x *Metadata) GetName() string { + if x != nil { + return x.Name + } + return "" +} + +func (x *Metadata) GetContentType() string { + if x != nil { + return x.ContentType + } + return "" +} + +// Zip content in byte form. +type Chunk struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields + + // Byte data + Data []byte `protobuf:"bytes,1,opt,name=data,proto3" json:"data,omitempty" toml:"data,omitempty" mapstructure:"data,omitempty"` + // Stream pointer + Position int64 `protobuf:"varint,2,opt,name=position,proto3" json:"position,omitempty" toml:"position,omitempty" mapstructure:"position,omitempty"` +} + +func (x *Chunk) Reset() { + *x = Chunk{} + if protoimpl.UnsafeEnabled { + mi := &file_interservice_infra_proxy_migrations_request_migrations_proto_msgTypes[2] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) + } +} + +func (x *Chunk) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*Chunk) ProtoMessage() {} + +func (x *Chunk) ProtoReflect() protoreflect.Message { + mi := &file_interservice_infra_proxy_migrations_request_migrations_proto_msgTypes[2] + if protoimpl.UnsafeEnabled && x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use Chunk.ProtoReflect.Descriptor instead. +func (*Chunk) Descriptor() ([]byte, []int) { + return file_interservice_infra_proxy_migrations_request_migrations_proto_rawDescGZIP(), []int{2} +} + +func (x *Chunk) GetData() []byte { + if x != nil { + return x.Data + } + return nil +} + +func (x *Chunk) GetPosition() int64 { + if x != nil { + return x.Position + } + return 0 +} + var File_interservice_infra_proxy_migrations_request_migrations_proto protoreflect.FileDescriptor var file_interservice_infra_proxy_migrations_request_migrations_proto_rawDesc = []byte{ @@ -34,21 +216,61 @@ var file_interservice_infra_proxy_migrations_request_migrations_proto_rawDesc = 0x63, 0x68, 0x65, 0x66, 0x2e, 0x61, 0x75, 0x74, 0x6f, 0x6d, 0x61, 0x74, 0x65, 0x2e, 0x64, 0x6f, 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x69, 0x6e, 0x66, 0x72, 0x61, 0x5f, 0x70, 0x72, 0x6f, 0x78, 0x79, 0x2e, 0x6d, 0x69, 0x67, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x72, 0x65, 0x71, 0x75, - 0x65, 0x73, 0x74, 0x42, 0x4a, 0x5a, 0x48, 0x67, 0x69, 0x74, 0x68, 0x75, 0x62, 0x2e, 0x63, 0x6f, - 0x6d, 0x2f, 0x63, 0x68, 0x65, 0x66, 0x2f, 0x61, 0x75, 0x74, 0x6f, 0x6d, 0x61, 0x74, 0x65, 0x2f, - 0x61, 0x70, 0x69, 0x2f, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x73, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65, - 0x2f, 0x69, 0x6e, 0x66, 0x72, 0x61, 0x5f, 0x70, 0x72, 0x6f, 0x78, 0x79, 0x2f, 0x6d, 0x69, 0x67, - 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2f, 0x72, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x62, - 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33, + 0x65, 0x73, 0x74, 0x22, 0xd7, 0x01, 0x0a, 0x14, 0x55, 0x70, 0x6c, 0x6f, 0x61, 0x64, 0x5a, 0x69, + 0x70, 0x46, 0x69, 0x6c, 0x65, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x12, 0x1a, 0x0a, 0x08, + 0x73, 0x65, 0x72, 0x76, 0x65, 0x72, 0x49, 0x64, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x08, + 0x73, 0x65, 0x72, 0x76, 0x65, 0x72, 0x49, 0x64, 0x12, 0x50, 0x0a, 0x05, 0x63, 0x68, 0x75, 0x6e, + 0x6b, 0x18, 0x02, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x3a, 0x2e, 0x63, 0x68, 0x65, 0x66, 0x2e, 0x61, + 0x75, 0x74, 0x6f, 0x6d, 0x61, 0x74, 0x65, 0x2e, 0x64, 0x6f, 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x69, + 0x6e, 0x66, 0x72, 0x61, 0x5f, 0x70, 0x72, 0x6f, 0x78, 0x79, 0x2e, 0x6d, 0x69, 0x67, 0x72, 0x61, + 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x72, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x2e, 0x43, 0x68, + 0x75, 0x6e, 0x6b, 0x52, 0x05, 0x63, 0x68, 0x75, 0x6e, 0x6b, 0x12, 0x51, 0x0a, 0x04, 0x6d, 0x65, + 0x74, 0x61, 0x18, 0x03, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x3d, 0x2e, 0x63, 0x68, 0x65, 0x66, 0x2e, + 0x61, 0x75, 0x74, 0x6f, 0x6d, 0x61, 0x74, 0x65, 0x2e, 0x64, 0x6f, 0x6d, 0x61, 0x69, 0x6e, 0x2e, + 0x69, 0x6e, 0x66, 0x72, 0x61, 0x5f, 0x70, 0x72, 0x6f, 0x78, 0x79, 0x2e, 0x6d, 0x69, 0x67, 0x72, + 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x72, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x2e, 0x4d, + 0x65, 0x74, 0x61, 0x64, 0x61, 0x74, 0x61, 0x52, 0x04, 0x6d, 0x65, 0x74, 0x61, 0x22, 0x40, 0x0a, + 0x08, 0x4d, 0x65, 0x74, 0x61, 0x64, 0x61, 0x74, 0x61, 0x12, 0x12, 0x0a, 0x04, 0x6e, 0x61, 0x6d, + 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x04, 0x6e, 0x61, 0x6d, 0x65, 0x12, 0x20, 0x0a, + 0x0b, 0x63, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, 0x54, 0x79, 0x70, 0x65, 0x18, 0x02, 0x20, 0x01, + 0x28, 0x09, 0x52, 0x0b, 0x63, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, 0x54, 0x79, 0x70, 0x65, 0x22, + 0x37, 0x0a, 0x05, 0x43, 0x68, 0x75, 0x6e, 0x6b, 0x12, 0x12, 0x0a, 0x04, 0x64, 0x61, 0x74, 0x61, + 0x18, 0x01, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x04, 0x64, 0x61, 0x74, 0x61, 0x12, 0x1a, 0x0a, 0x08, + 0x70, 0x6f, 0x73, 0x69, 0x74, 0x69, 0x6f, 0x6e, 0x18, 0x02, 0x20, 0x01, 0x28, 0x03, 0x52, 0x08, + 0x70, 0x6f, 0x73, 0x69, 0x74, 0x69, 0x6f, 0x6e, 0x42, 0x4a, 0x5a, 0x48, 0x67, 0x69, 0x74, 0x68, + 0x75, 0x62, 0x2e, 0x63, 0x6f, 0x6d, 0x2f, 0x63, 0x68, 0x65, 0x66, 0x2f, 0x61, 0x75, 0x74, 0x6f, + 0x6d, 0x61, 0x74, 0x65, 0x2f, 0x61, 0x70, 0x69, 0x2f, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x73, 0x65, + 0x72, 0x76, 0x69, 0x63, 0x65, 0x2f, 0x69, 0x6e, 0x66, 0x72, 0x61, 0x5f, 0x70, 0x72, 0x6f, 0x78, + 0x79, 0x2f, 0x6d, 0x69, 0x67, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2f, 0x72, 0x65, 0x71, + 0x75, 0x65, 0x73, 0x74, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33, +} + +var ( + file_interservice_infra_proxy_migrations_request_migrations_proto_rawDescOnce sync.Once + file_interservice_infra_proxy_migrations_request_migrations_proto_rawDescData = file_interservice_infra_proxy_migrations_request_migrations_proto_rawDesc +) + +func file_interservice_infra_proxy_migrations_request_migrations_proto_rawDescGZIP() []byte { + file_interservice_infra_proxy_migrations_request_migrations_proto_rawDescOnce.Do(func() { + file_interservice_infra_proxy_migrations_request_migrations_proto_rawDescData = protoimpl.X.CompressGZIP(file_interservice_infra_proxy_migrations_request_migrations_proto_rawDescData) + }) + return file_interservice_infra_proxy_migrations_request_migrations_proto_rawDescData } -var file_interservice_infra_proxy_migrations_request_migrations_proto_goTypes = []interface{}{} +var file_interservice_infra_proxy_migrations_request_migrations_proto_msgTypes = make([]protoimpl.MessageInfo, 3) +var file_interservice_infra_proxy_migrations_request_migrations_proto_goTypes = []interface{}{ + (*UploadZipFileRequest)(nil), // 0: chef.automate.domain.infra_proxy.migrations.request.UploadZipFileRequest + (*Metadata)(nil), // 1: chef.automate.domain.infra_proxy.migrations.request.Metadata + (*Chunk)(nil), // 2: chef.automate.domain.infra_proxy.migrations.request.Chunk +} var file_interservice_infra_proxy_migrations_request_migrations_proto_depIdxs = []int32{ - 0, // [0:0] is the sub-list for method output_type - 0, // [0:0] is the sub-list for method input_type - 0, // [0:0] is the sub-list for extension type_name - 0, // [0:0] is the sub-list for extension extendee - 0, // [0:0] is the sub-list for field type_name + 2, // 0: chef.automate.domain.infra_proxy.migrations.request.UploadZipFileRequest.chunk:type_name -> chef.automate.domain.infra_proxy.migrations.request.Chunk + 1, // 1: chef.automate.domain.infra_proxy.migrations.request.UploadZipFileRequest.meta:type_name -> chef.automate.domain.infra_proxy.migrations.request.Metadata + 2, // [2:2] is the sub-list for method output_type + 2, // [2:2] is the sub-list for method input_type + 2, // [2:2] is the sub-list for extension type_name + 2, // [2:2] is the sub-list for extension extendee + 0, // [0:2] is the sub-list for field type_name } func init() { file_interservice_infra_proxy_migrations_request_migrations_proto_init() } @@ -56,18 +278,57 @@ func file_interservice_infra_proxy_migrations_request_migrations_proto_init() { if File_interservice_infra_proxy_migrations_request_migrations_proto != nil { return } + if !protoimpl.UnsafeEnabled { + file_interservice_infra_proxy_migrations_request_migrations_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} { + switch v := v.(*UploadZipFileRequest); i { + case 0: + return &v.state + case 1: + return &v.sizeCache + case 2: + return &v.unknownFields + default: + return nil + } + } + file_interservice_infra_proxy_migrations_request_migrations_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} { + switch v := v.(*Metadata); i { + case 0: + return &v.state + case 1: + return &v.sizeCache + case 2: + return &v.unknownFields + default: + return nil + } + } + file_interservice_infra_proxy_migrations_request_migrations_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} { + switch v := v.(*Chunk); i { + case 0: + return &v.state + case 1: + return &v.sizeCache + case 2: + return &v.unknownFields + default: + return nil + } + } + } type x struct{} out := protoimpl.TypeBuilder{ File: protoimpl.DescBuilder{ GoPackagePath: reflect.TypeOf(x{}).PkgPath(), RawDescriptor: file_interservice_infra_proxy_migrations_request_migrations_proto_rawDesc, NumEnums: 0, - NumMessages: 0, + NumMessages: 3, NumExtensions: 0, NumServices: 0, }, GoTypes: file_interservice_infra_proxy_migrations_request_migrations_proto_goTypes, DependencyIndexes: file_interservice_infra_proxy_migrations_request_migrations_proto_depIdxs, + MessageInfos: file_interservice_infra_proxy_migrations_request_migrations_proto_msgTypes, }.Build() File_interservice_infra_proxy_migrations_request_migrations_proto = out.File file_interservice_infra_proxy_migrations_request_migrations_proto_rawDesc = nil diff --git a/api/interservice/infra_proxy/migrations/request/migrations.proto b/api/interservice/infra_proxy/migrations/request/migrations.proto index c13fa805abe..800c571b7ed 100644 --- a/api/interservice/infra_proxy/migrations/request/migrations.proto +++ b/api/interservice/infra_proxy/migrations/request/migrations.proto @@ -2,3 +2,27 @@ syntax = "proto3"; package chef.automate.domain.infra_proxy.migrations.request; option go_package = "github.com/chef/automate/api/interservice/infra_proxy/migrations/request"; + +message UploadZipFileRequest{ + //Chef infra server id for the file + string serverId =1; + //Chunk of the data in bytes + Chunk chunk = 2; + //Metadata for the request + Metadata meta = 3; +} + +message Metadata{ + //Migration File name + string name = 1; + //Content type of the request + string contentType =2; +} + +// Zip content in byte form. +message Chunk { + // Byte data + bytes data = 1; + // Stream pointer + int64 position = 2; +} diff --git a/api/interservice/infra_proxy/migrations/response/migrations.pb.go b/api/interservice/infra_proxy/migrations/response/migrations.pb.go index b0b81311c1e..887d7d0fef7 100644 --- a/api/interservice/infra_proxy/migrations/response/migrations.pb.go +++ b/api/interservice/infra_proxy/migrations/response/migrations.pb.go @@ -11,6 +11,7 @@ import ( protoreflect "google.golang.org/protobuf/reflect/protoreflect" protoimpl "google.golang.org/protobuf/runtime/protoimpl" reflect "reflect" + sync "sync" ) const ( @@ -24,6 +25,72 @@ const ( // of the legacy proto package is being used. const _ = proto.ProtoPackageIsVersion4 +type UploadZipFileResponse struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields + + // Success responce for zip upload. + Success bool `protobuf:"varint,1,opt,name=success,proto3" json:"success,omitempty" toml:"success,omitempty" mapstructure:"success,omitempty"` + // Chef infra server migration ID + MigrationId string `protobuf:"bytes,2,opt,name=migrationId,proto3" json:"migrationId,omitempty" toml:"migrationId,omitempty" mapstructure:"migrationId,omitempty"` + // Error data in case of failure. + Errors []string `protobuf:"bytes,3,rep,name=errors,proto3" json:"errors,omitempty" toml:"errors,omitempty" mapstructure:"errors,omitempty"` +} + +func (x *UploadZipFileResponse) Reset() { + *x = UploadZipFileResponse{} + if protoimpl.UnsafeEnabled { + mi := &file_interservice_infra_proxy_migrations_response_migrations_proto_msgTypes[0] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) + } +} + +func (x *UploadZipFileResponse) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*UploadZipFileResponse) ProtoMessage() {} + +func (x *UploadZipFileResponse) ProtoReflect() protoreflect.Message { + mi := &file_interservice_infra_proxy_migrations_response_migrations_proto_msgTypes[0] + if protoimpl.UnsafeEnabled && x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use UploadZipFileResponse.ProtoReflect.Descriptor instead. +func (*UploadZipFileResponse) Descriptor() ([]byte, []int) { + return file_interservice_infra_proxy_migrations_response_migrations_proto_rawDescGZIP(), []int{0} +} + +func (x *UploadZipFileResponse) GetSuccess() bool { + if x != nil { + return x.Success + } + return false +} + +func (x *UploadZipFileResponse) GetMigrationId() string { + if x != nil { + return x.MigrationId + } + return "" +} + +func (x *UploadZipFileResponse) GetErrors() []string { + if x != nil { + return x.Errors + } + return nil +} + var File_interservice_infra_proxy_migrations_response_migrations_proto protoreflect.FileDescriptor var file_interservice_infra_proxy_migrations_response_migrations_proto_rawDesc = []byte{ @@ -34,15 +101,37 @@ var file_interservice_infra_proxy_migrations_response_migrations_proto_rawDesc = 0x34, 0x63, 0x68, 0x65, 0x66, 0x2e, 0x61, 0x75, 0x74, 0x6f, 0x6d, 0x61, 0x74, 0x65, 0x2e, 0x64, 0x6f, 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x69, 0x6e, 0x66, 0x72, 0x61, 0x5f, 0x70, 0x72, 0x6f, 0x78, 0x79, 0x2e, 0x6d, 0x69, 0x67, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x72, 0x65, 0x73, - 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x42, 0x4b, 0x5a, 0x49, 0x67, 0x69, 0x74, 0x68, 0x75, 0x62, 0x2e, - 0x63, 0x6f, 0x6d, 0x2f, 0x63, 0x68, 0x65, 0x66, 0x2f, 0x61, 0x75, 0x74, 0x6f, 0x6d, 0x61, 0x74, - 0x65, 0x2f, 0x61, 0x70, 0x69, 0x2f, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x73, 0x65, 0x72, 0x76, 0x69, - 0x63, 0x65, 0x2f, 0x69, 0x6e, 0x66, 0x72, 0x61, 0x5f, 0x70, 0x72, 0x6f, 0x78, 0x79, 0x2f, 0x6d, - 0x69, 0x67, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2f, 0x72, 0x65, 0x73, 0x70, 0x6f, 0x6e, - 0x73, 0x65, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33, + 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x22, 0x6b, 0x0a, 0x15, 0x55, 0x70, 0x6c, 0x6f, 0x61, 0x64, 0x5a, + 0x69, 0x70, 0x46, 0x69, 0x6c, 0x65, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x12, 0x18, + 0x0a, 0x07, 0x73, 0x75, 0x63, 0x63, 0x65, 0x73, 0x73, 0x18, 0x01, 0x20, 0x01, 0x28, 0x08, 0x52, + 0x07, 0x73, 0x75, 0x63, 0x63, 0x65, 0x73, 0x73, 0x12, 0x20, 0x0a, 0x0b, 0x6d, 0x69, 0x67, 0x72, + 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x49, 0x64, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0b, 0x6d, + 0x69, 0x67, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x49, 0x64, 0x12, 0x16, 0x0a, 0x06, 0x65, 0x72, + 0x72, 0x6f, 0x72, 0x73, 0x18, 0x03, 0x20, 0x03, 0x28, 0x09, 0x52, 0x06, 0x65, 0x72, 0x72, 0x6f, + 0x72, 0x73, 0x42, 0x4b, 0x5a, 0x49, 0x67, 0x69, 0x74, 0x68, 0x75, 0x62, 0x2e, 0x63, 0x6f, 0x6d, + 0x2f, 0x63, 0x68, 0x65, 0x66, 0x2f, 0x61, 0x75, 0x74, 0x6f, 0x6d, 0x61, 0x74, 0x65, 0x2f, 0x61, + 0x70, 0x69, 0x2f, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x73, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65, 0x2f, + 0x69, 0x6e, 0x66, 0x72, 0x61, 0x5f, 0x70, 0x72, 0x6f, 0x78, 0x79, 0x2f, 0x6d, 0x69, 0x67, 0x72, + 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2f, 0x72, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x62, + 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33, } -var file_interservice_infra_proxy_migrations_response_migrations_proto_goTypes = []interface{}{} +var ( + file_interservice_infra_proxy_migrations_response_migrations_proto_rawDescOnce sync.Once + file_interservice_infra_proxy_migrations_response_migrations_proto_rawDescData = file_interservice_infra_proxy_migrations_response_migrations_proto_rawDesc +) + +func file_interservice_infra_proxy_migrations_response_migrations_proto_rawDescGZIP() []byte { + file_interservice_infra_proxy_migrations_response_migrations_proto_rawDescOnce.Do(func() { + file_interservice_infra_proxy_migrations_response_migrations_proto_rawDescData = protoimpl.X.CompressGZIP(file_interservice_infra_proxy_migrations_response_migrations_proto_rawDescData) + }) + return file_interservice_infra_proxy_migrations_response_migrations_proto_rawDescData +} + +var file_interservice_infra_proxy_migrations_response_migrations_proto_msgTypes = make([]protoimpl.MessageInfo, 1) +var file_interservice_infra_proxy_migrations_response_migrations_proto_goTypes = []interface{}{ + (*UploadZipFileResponse)(nil), // 0: chef.automate.domain.infra_proxy.migrations.response.UploadZipFileResponse +} var file_interservice_infra_proxy_migrations_response_migrations_proto_depIdxs = []int32{ 0, // [0:0] is the sub-list for method output_type 0, // [0:0] is the sub-list for method input_type @@ -56,18 +145,33 @@ func file_interservice_infra_proxy_migrations_response_migrations_proto_init() { if File_interservice_infra_proxy_migrations_response_migrations_proto != nil { return } + if !protoimpl.UnsafeEnabled { + file_interservice_infra_proxy_migrations_response_migrations_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} { + switch v := v.(*UploadZipFileResponse); i { + case 0: + return &v.state + case 1: + return &v.sizeCache + case 2: + return &v.unknownFields + default: + return nil + } + } + } type x struct{} out := protoimpl.TypeBuilder{ File: protoimpl.DescBuilder{ GoPackagePath: reflect.TypeOf(x{}).PkgPath(), RawDescriptor: file_interservice_infra_proxy_migrations_response_migrations_proto_rawDesc, NumEnums: 0, - NumMessages: 0, + NumMessages: 1, NumExtensions: 0, NumServices: 0, }, GoTypes: file_interservice_infra_proxy_migrations_response_migrations_proto_goTypes, DependencyIndexes: file_interservice_infra_proxy_migrations_response_migrations_proto_depIdxs, + MessageInfos: file_interservice_infra_proxy_migrations_response_migrations_proto_msgTypes, }.Build() File_interservice_infra_proxy_migrations_response_migrations_proto = out.File file_interservice_infra_proxy_migrations_response_migrations_proto_rawDesc = nil diff --git a/api/interservice/infra_proxy/migrations/response/migrations.proto b/api/interservice/infra_proxy/migrations/response/migrations.proto index f3b7df260a6..f00ae4003e9 100644 --- a/api/interservice/infra_proxy/migrations/response/migrations.proto +++ b/api/interservice/infra_proxy/migrations/response/migrations.proto @@ -2,3 +2,12 @@ syntax = "proto3"; package chef.automate.domain.infra_proxy.migrations.response; option go_package = "github.com/chef/automate/api/interservice/infra_proxy/migrations/response"; + +message UploadZipFileResponse { + // Success responce for zip upload. + bool success = 1; + // Chef infra server migration ID + string migrationId = 2; + // Error data in case of failure. + repeated string errors = 3; +} diff --git a/api/interservice/infra_proxy/migrations/service/migrations.pb.go b/api/interservice/infra_proxy/migrations/service/migrations.pb.go index c2292dc10b5..f80872b76d1 100644 --- a/api/interservice/infra_proxy/migrations/service/migrations.pb.go +++ b/api/interservice/infra_proxy/migrations/service/migrations.pb.go @@ -7,7 +7,14 @@ package service import ( + context "context" + request "github.com/chef/automate/api/interservice/infra_proxy/migrations/request" + response "github.com/chef/automate/api/interservice/infra_proxy/migrations/response" proto "github.com/golang/protobuf/proto" + _ "google.golang.org/genproto/googleapis/api/annotations" + grpc "google.golang.org/grpc" + codes "google.golang.org/grpc/codes" + status "google.golang.org/grpc/status" protoreflect "google.golang.org/protobuf/reflect/protoreflect" protoimpl "google.golang.org/protobuf/runtime/protoimpl" reflect "reflect" @@ -34,18 +41,45 @@ var file_interservice_infra_proxy_migrations_service_migrations_proto_rawDesc = 0x63, 0x68, 0x65, 0x66, 0x2e, 0x61, 0x75, 0x74, 0x6f, 0x6d, 0x61, 0x74, 0x65, 0x2e, 0x64, 0x6f, 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x69, 0x6e, 0x66, 0x72, 0x61, 0x5f, 0x70, 0x72, 0x6f, 0x78, 0x79, 0x2e, 0x6d, 0x69, 0x67, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x73, 0x65, 0x72, 0x76, - 0x69, 0x63, 0x65, 0x42, 0x4a, 0x5a, 0x48, 0x67, 0x69, 0x74, 0x68, 0x75, 0x62, 0x2e, 0x63, 0x6f, - 0x6d, 0x2f, 0x63, 0x68, 0x65, 0x66, 0x2f, 0x61, 0x75, 0x74, 0x6f, 0x6d, 0x61, 0x74, 0x65, 0x2f, - 0x61, 0x70, 0x69, 0x2f, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x73, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65, + 0x69, 0x63, 0x65, 0x1a, 0x3d, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x73, 0x65, 0x72, 0x76, 0x69, 0x63, + 0x65, 0x2f, 0x69, 0x6e, 0x66, 0x72, 0x61, 0x5f, 0x70, 0x72, 0x6f, 0x78, 0x79, 0x2f, 0x6d, 0x69, + 0x67, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2f, 0x72, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, + 0x65, 0x2f, 0x6d, 0x69, 0x67, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x70, 0x72, 0x6f, + 0x74, 0x6f, 0x1a, 0x3c, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x73, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65, 0x2f, 0x69, 0x6e, 0x66, 0x72, 0x61, 0x5f, 0x70, 0x72, 0x6f, 0x78, 0x79, 0x2f, 0x6d, 0x69, 0x67, - 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2f, 0x73, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65, 0x62, - 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33, + 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2f, 0x72, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x2f, + 0x6d, 0x69, 0x67, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, + 0x1a, 0x1c, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2f, 0x61, 0x70, 0x69, 0x2f, 0x61, 0x6e, 0x6e, + 0x6f, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x32, 0xc1, + 0x01, 0x0a, 0x14, 0x4d, 0x69, 0x67, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x44, 0x61, 0x74, 0x61, + 0x53, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65, 0x12, 0xa8, 0x01, 0x0a, 0x0a, 0x55, 0x70, 0x6c, 0x6f, + 0x61, 0x64, 0x46, 0x69, 0x6c, 0x65, 0x12, 0x49, 0x2e, 0x63, 0x68, 0x65, 0x66, 0x2e, 0x61, 0x75, + 0x74, 0x6f, 0x6d, 0x61, 0x74, 0x65, 0x2e, 0x64, 0x6f, 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x69, 0x6e, + 0x66, 0x72, 0x61, 0x5f, 0x70, 0x72, 0x6f, 0x78, 0x79, 0x2e, 0x6d, 0x69, 0x67, 0x72, 0x61, 0x74, + 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x72, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x2e, 0x55, 0x70, 0x6c, + 0x6f, 0x61, 0x64, 0x5a, 0x69, 0x70, 0x46, 0x69, 0x6c, 0x65, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, + 0x74, 0x1a, 0x4b, 0x2e, 0x63, 0x68, 0x65, 0x66, 0x2e, 0x61, 0x75, 0x74, 0x6f, 0x6d, 0x61, 0x74, + 0x65, 0x2e, 0x64, 0x6f, 0x6d, 0x61, 0x69, 0x6e, 0x2e, 0x69, 0x6e, 0x66, 0x72, 0x61, 0x5f, 0x70, + 0x72, 0x6f, 0x78, 0x79, 0x2e, 0x6d, 0x69, 0x67, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2e, + 0x72, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x2e, 0x55, 0x70, 0x6c, 0x6f, 0x61, 0x64, 0x5a, + 0x69, 0x70, 0x46, 0x69, 0x6c, 0x65, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x22, 0x00, + 0x28, 0x01, 0x42, 0x4a, 0x5a, 0x48, 0x67, 0x69, 0x74, 0x68, 0x75, 0x62, 0x2e, 0x63, 0x6f, 0x6d, + 0x2f, 0x63, 0x68, 0x65, 0x66, 0x2f, 0x61, 0x75, 0x74, 0x6f, 0x6d, 0x61, 0x74, 0x65, 0x2f, 0x61, + 0x70, 0x69, 0x2f, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x73, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65, 0x2f, + 0x69, 0x6e, 0x66, 0x72, 0x61, 0x5f, 0x70, 0x72, 0x6f, 0x78, 0x79, 0x2f, 0x6d, 0x69, 0x67, 0x72, + 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2f, 0x73, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65, 0x62, 0x06, + 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33, } -var file_interservice_infra_proxy_migrations_service_migrations_proto_goTypes = []interface{}{} +var file_interservice_infra_proxy_migrations_service_migrations_proto_goTypes = []interface{}{ + (*request.UploadZipFileRequest)(nil), // 0: chef.automate.domain.infra_proxy.migrations.request.UploadZipFileRequest + (*response.UploadZipFileResponse)(nil), // 1: chef.automate.domain.infra_proxy.migrations.response.UploadZipFileResponse +} var file_interservice_infra_proxy_migrations_service_migrations_proto_depIdxs = []int32{ - 0, // [0:0] is the sub-list for method output_type - 0, // [0:0] is the sub-list for method input_type + 0, // 0: chef.automate.domain.infra_proxy.migrations.service.MigrationDataService.UploadFile:input_type -> chef.automate.domain.infra_proxy.migrations.request.UploadZipFileRequest + 1, // 1: chef.automate.domain.infra_proxy.migrations.service.MigrationDataService.UploadFile:output_type -> chef.automate.domain.infra_proxy.migrations.response.UploadZipFileResponse + 1, // [1:2] is the sub-list for method output_type + 0, // [0:1] is the sub-list for method input_type 0, // [0:0] is the sub-list for extension type_name 0, // [0:0] is the sub-list for extension extendee 0, // [0:0] is the sub-list for field type_name @@ -64,7 +98,7 @@ func file_interservice_infra_proxy_migrations_service_migrations_proto_init() { NumEnums: 0, NumMessages: 0, NumExtensions: 0, - NumServices: 0, + NumServices: 1, }, GoTypes: file_interservice_infra_proxy_migrations_service_migrations_proto_goTypes, DependencyIndexes: file_interservice_infra_proxy_migrations_service_migrations_proto_depIdxs, @@ -74,3 +108,117 @@ func file_interservice_infra_proxy_migrations_service_migrations_proto_init() { file_interservice_infra_proxy_migrations_service_migrations_proto_goTypes = nil file_interservice_infra_proxy_migrations_service_migrations_proto_depIdxs = nil } + +// Reference imports to suppress errors if they are not otherwise used. +var _ context.Context +var _ grpc.ClientConnInterface + +// This is a compile-time assertion to ensure that this generated file +// is compatible with the grpc package it is being compiled against. +const _ = grpc.SupportPackageIsVersion6 + +// MigrationDataServiceClient is the client API for MigrationDataService service. +// +// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://godoc.org/google.golang.org/grpc#ClientConn.NewStream. +type MigrationDataServiceClient interface { + UploadFile(ctx context.Context, opts ...grpc.CallOption) (MigrationDataService_UploadFileClient, error) +} + +type migrationDataServiceClient struct { + cc grpc.ClientConnInterface +} + +func NewMigrationDataServiceClient(cc grpc.ClientConnInterface) MigrationDataServiceClient { + return &migrationDataServiceClient{cc} +} + +func (c *migrationDataServiceClient) UploadFile(ctx context.Context, opts ...grpc.CallOption) (MigrationDataService_UploadFileClient, error) { + stream, err := c.cc.NewStream(ctx, &_MigrationDataService_serviceDesc.Streams[0], "/chef.automate.domain.infra_proxy.migrations.service.MigrationDataService/UploadFile", opts...) + if err != nil { + return nil, err + } + x := &migrationDataServiceUploadFileClient{stream} + return x, nil +} + +type MigrationDataService_UploadFileClient interface { + Send(*request.UploadZipFileRequest) error + CloseAndRecv() (*response.UploadZipFileResponse, error) + grpc.ClientStream +} + +type migrationDataServiceUploadFileClient struct { + grpc.ClientStream +} + +func (x *migrationDataServiceUploadFileClient) Send(m *request.UploadZipFileRequest) error { + return x.ClientStream.SendMsg(m) +} + +func (x *migrationDataServiceUploadFileClient) CloseAndRecv() (*response.UploadZipFileResponse, error) { + if err := x.ClientStream.CloseSend(); err != nil { + return nil, err + } + m := new(response.UploadZipFileResponse) + if err := x.ClientStream.RecvMsg(m); err != nil { + return nil, err + } + return m, nil +} + +// MigrationDataServiceServer is the server API for MigrationDataService service. +type MigrationDataServiceServer interface { + UploadFile(MigrationDataService_UploadFileServer) error +} + +// UnimplementedMigrationDataServiceServer can be embedded to have forward compatible implementations. +type UnimplementedMigrationDataServiceServer struct { +} + +func (*UnimplementedMigrationDataServiceServer) UploadFile(MigrationDataService_UploadFileServer) error { + return status.Errorf(codes.Unimplemented, "method UploadFile not implemented") +} + +func RegisterMigrationDataServiceServer(s *grpc.Server, srv MigrationDataServiceServer) { + s.RegisterService(&_MigrationDataService_serviceDesc, srv) +} + +func _MigrationDataService_UploadFile_Handler(srv interface{}, stream grpc.ServerStream) error { + return srv.(MigrationDataServiceServer).UploadFile(&migrationDataServiceUploadFileServer{stream}) +} + +type MigrationDataService_UploadFileServer interface { + SendAndClose(*response.UploadZipFileResponse) error + Recv() (*request.UploadZipFileRequest, error) + grpc.ServerStream +} + +type migrationDataServiceUploadFileServer struct { + grpc.ServerStream +} + +func (x *migrationDataServiceUploadFileServer) SendAndClose(m *response.UploadZipFileResponse) error { + return x.ServerStream.SendMsg(m) +} + +func (x *migrationDataServiceUploadFileServer) Recv() (*request.UploadZipFileRequest, error) { + m := new(request.UploadZipFileRequest) + if err := x.ServerStream.RecvMsg(m); err != nil { + return nil, err + } + return m, nil +} + +var _MigrationDataService_serviceDesc = grpc.ServiceDesc{ + ServiceName: "chef.automate.domain.infra_proxy.migrations.service.MigrationDataService", + HandlerType: (*MigrationDataServiceServer)(nil), + Methods: []grpc.MethodDesc{}, + Streams: []grpc.StreamDesc{ + { + StreamName: "UploadFile", + Handler: _MigrationDataService_UploadFile_Handler, + ClientStreams: true, + }, + }, + Metadata: "interservice/infra_proxy/migrations/service/migrations.proto", +} diff --git a/api/interservice/infra_proxy/migrations/service/migrations.proto b/api/interservice/infra_proxy/migrations/service/migrations.proto index 5f29921adec..e5a7e28566a 100644 --- a/api/interservice/infra_proxy/migrations/service/migrations.proto +++ b/api/interservice/infra_proxy/migrations/service/migrations.proto @@ -2,3 +2,13 @@ syntax = "proto3"; package chef.automate.domain.infra_proxy.migrations.service; option go_package = "github.com/chef/automate/api/interservice/infra_proxy/migrations/service"; + +import "interservice/infra_proxy/migrations/response/migrations.proto"; +import "interservice/infra_proxy/migrations/request/migrations.proto"; +import "google/api/annotations.proto"; + +service MigrationDataService{ + + rpc UploadFile (stream chef.automate.domain.infra_proxy.migrations.request.UploadZipFileRequest) returns (chef.automate.domain.infra_proxy.migrations.response.UploadZipFileResponse) {}; + +} diff --git a/api/interservice/ingest/chef.pb.client_mock.go b/api/interservice/ingest/chef.pb.client_mock.go index aef5fb237ff..a65e4c19603 100644 --- a/api/interservice/ingest/chef.pb.client_mock.go +++ b/api/interservice/ingest/chef.pb.client_mock.go @@ -6,58 +6,57 @@ package ingest import ( context "context" - reflect "reflect" - request "github.com/chef/automate/api/external/ingest/request" response "github.com/chef/automate/api/external/ingest/response" gomock "github.com/golang/mock/gomock" grpc "google.golang.org/grpc" + reflect "reflect" ) -// MockChefIngesterServiceClient is a mock of ChefIngesterServiceClient interface. +// MockChefIngesterServiceClient is a mock of ChefIngesterServiceClient interface type MockChefIngesterServiceClient struct { ctrl *gomock.Controller recorder *MockChefIngesterServiceClientMockRecorder } -// MockChefIngesterServiceClientMockRecorder is the mock recorder for MockChefIngesterServiceClient. +// MockChefIngesterServiceClientMockRecorder is the mock recorder for MockChefIngesterServiceClient type MockChefIngesterServiceClientMockRecorder struct { mock *MockChefIngesterServiceClient } -// NewMockChefIngesterServiceClient creates a new mock instance. +// NewMockChefIngesterServiceClient creates a new mock instance func NewMockChefIngesterServiceClient(ctrl *gomock.Controller) *MockChefIngesterServiceClient { mock := &MockChefIngesterServiceClient{ctrl: ctrl} mock.recorder = &MockChefIngesterServiceClientMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockChefIngesterServiceClient) EXPECT() *MockChefIngesterServiceClientMockRecorder { return m.recorder } -// GetVersion mocks base method. -func (m *MockChefIngesterServiceClient) GetVersion(ctx context.Context, in *VersionRequest, opts ...grpc.CallOption) (*Version, error) { +// ProcessChefRun mocks base method +func (m *MockChefIngesterServiceClient) ProcessChefRun(ctx context.Context, in *request.Run, opts ...grpc.CallOption) (*response.ProcessChefRunResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetVersion", varargs...) - ret0, _ := ret[0].(*Version) + ret := m.ctrl.Call(m, "ProcessChefRun", varargs...) + ret0, _ := ret[0].(*response.ProcessChefRunResponse) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetVersion indicates an expected call of GetVersion. -func (mr *MockChefIngesterServiceClientMockRecorder) GetVersion(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// ProcessChefRun indicates an expected call of ProcessChefRun +func (mr *MockChefIngesterServiceClientMockRecorder) ProcessChefRun(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetVersion", reflect.TypeOf((*MockChefIngesterServiceClient)(nil).GetVersion), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProcessChefRun", reflect.TypeOf((*MockChefIngesterServiceClient)(nil).ProcessChefRun), varargs...) } -// ProcessChefAction mocks base method. +// ProcessChefAction mocks base method func (m *MockChefIngesterServiceClient) ProcessChefAction(ctx context.Context, in *request.Action, opts ...grpc.CallOption) (*response.ProcessChefActionResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} @@ -70,34 +69,14 @@ func (m *MockChefIngesterServiceClient) ProcessChefAction(ctx context.Context, i return ret0, ret1 } -// ProcessChefAction indicates an expected call of ProcessChefAction. +// ProcessChefAction indicates an expected call of ProcessChefAction func (mr *MockChefIngesterServiceClientMockRecorder) ProcessChefAction(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProcessChefAction", reflect.TypeOf((*MockChefIngesterServiceClient)(nil).ProcessChefAction), varargs...) } -// ProcessChefRun mocks base method. -func (m *MockChefIngesterServiceClient) ProcessChefRun(ctx context.Context, in *request.Run, opts ...grpc.CallOption) (*response.ProcessChefRunResponse, error) { - m.ctrl.T.Helper() - varargs := []interface{}{ctx, in} - for _, a := range opts { - varargs = append(varargs, a) - } - ret := m.ctrl.Call(m, "ProcessChefRun", varargs...) - ret0, _ := ret[0].(*response.ProcessChefRunResponse) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// ProcessChefRun indicates an expected call of ProcessChefRun. -func (mr *MockChefIngesterServiceClientMockRecorder) ProcessChefRun(ctx, in interface{}, opts ...interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProcessChefRun", reflect.TypeOf((*MockChefIngesterServiceClient)(nil).ProcessChefRun), varargs...) -} - -// ProcessLivenessPing mocks base method. +// ProcessLivenessPing mocks base method func (m *MockChefIngesterServiceClient) ProcessLivenessPing(ctx context.Context, in *request.Liveness, opts ...grpc.CallOption) (*response.ProcessLivenessResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} @@ -110,14 +89,14 @@ func (m *MockChefIngesterServiceClient) ProcessLivenessPing(ctx context.Context, return ret0, ret1 } -// ProcessLivenessPing indicates an expected call of ProcessLivenessPing. +// ProcessLivenessPing indicates an expected call of ProcessLivenessPing func (mr *MockChefIngesterServiceClientMockRecorder) ProcessLivenessPing(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProcessLivenessPing", reflect.TypeOf((*MockChefIngesterServiceClient)(nil).ProcessLivenessPing), varargs...) } -// ProcessMultipleNodeDeletes mocks base method. +// ProcessMultipleNodeDeletes mocks base method func (m *MockChefIngesterServiceClient) ProcessMultipleNodeDeletes(ctx context.Context, in *request.MultipleNodeDeleteRequest, opts ...grpc.CallOption) (*response.ProcessMultipleNodeDeleteResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} @@ -130,14 +109,14 @@ func (m *MockChefIngesterServiceClient) ProcessMultipleNodeDeletes(ctx context.C return ret0, ret1 } -// ProcessMultipleNodeDeletes indicates an expected call of ProcessMultipleNodeDeletes. +// ProcessMultipleNodeDeletes indicates an expected call of ProcessMultipleNodeDeletes func (mr *MockChefIngesterServiceClientMockRecorder) ProcessMultipleNodeDeletes(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProcessMultipleNodeDeletes", reflect.TypeOf((*MockChefIngesterServiceClient)(nil).ProcessMultipleNodeDeletes), varargs...) } -// ProcessNodeDelete mocks base method. +// ProcessNodeDelete mocks base method func (m *MockChefIngesterServiceClient) ProcessNodeDelete(ctx context.Context, in *request.Delete, opts ...grpc.CallOption) (*response.ProcessNodeDeleteResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} @@ -150,52 +129,72 @@ func (m *MockChefIngesterServiceClient) ProcessNodeDelete(ctx context.Context, i return ret0, ret1 } -// ProcessNodeDelete indicates an expected call of ProcessNodeDelete. +// ProcessNodeDelete indicates an expected call of ProcessNodeDelete func (mr *MockChefIngesterServiceClientMockRecorder) ProcessNodeDelete(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProcessNodeDelete", reflect.TypeOf((*MockChefIngesterServiceClient)(nil).ProcessNodeDelete), varargs...) } -// MockChefIngesterServiceServer is a mock of ChefIngesterServiceServer interface. +// GetVersion mocks base method +func (m *MockChefIngesterServiceClient) GetVersion(ctx context.Context, in *VersionRequest, opts ...grpc.CallOption) (*Version, error) { + m.ctrl.T.Helper() + varargs := []interface{}{ctx, in} + for _, a := range opts { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "GetVersion", varargs...) + ret0, _ := ret[0].(*Version) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetVersion indicates an expected call of GetVersion +func (mr *MockChefIngesterServiceClientMockRecorder) GetVersion(ctx, in interface{}, opts ...interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]interface{}{ctx, in}, opts...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetVersion", reflect.TypeOf((*MockChefIngesterServiceClient)(nil).GetVersion), varargs...) +} + +// MockChefIngesterServiceServer is a mock of ChefIngesterServiceServer interface type MockChefIngesterServiceServer struct { ctrl *gomock.Controller recorder *MockChefIngesterServiceServerMockRecorder } -// MockChefIngesterServiceServerMockRecorder is the mock recorder for MockChefIngesterServiceServer. +// MockChefIngesterServiceServerMockRecorder is the mock recorder for MockChefIngesterServiceServer type MockChefIngesterServiceServerMockRecorder struct { mock *MockChefIngesterServiceServer } -// NewMockChefIngesterServiceServer creates a new mock instance. +// NewMockChefIngesterServiceServer creates a new mock instance func NewMockChefIngesterServiceServer(ctrl *gomock.Controller) *MockChefIngesterServiceServer { mock := &MockChefIngesterServiceServer{ctrl: ctrl} mock.recorder = &MockChefIngesterServiceServerMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockChefIngesterServiceServer) EXPECT() *MockChefIngesterServiceServerMockRecorder { return m.recorder } -// GetVersion mocks base method. -func (m *MockChefIngesterServiceServer) GetVersion(arg0 context.Context, arg1 *VersionRequest) (*Version, error) { +// ProcessChefRun mocks base method +func (m *MockChefIngesterServiceServer) ProcessChefRun(arg0 context.Context, arg1 *request.Run) (*response.ProcessChefRunResponse, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetVersion", arg0, arg1) - ret0, _ := ret[0].(*Version) + ret := m.ctrl.Call(m, "ProcessChefRun", arg0, arg1) + ret0, _ := ret[0].(*response.ProcessChefRunResponse) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetVersion indicates an expected call of GetVersion. -func (mr *MockChefIngesterServiceServerMockRecorder) GetVersion(arg0, arg1 interface{}) *gomock.Call { +// ProcessChefRun indicates an expected call of ProcessChefRun +func (mr *MockChefIngesterServiceServerMockRecorder) ProcessChefRun(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetVersion", reflect.TypeOf((*MockChefIngesterServiceServer)(nil).GetVersion), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProcessChefRun", reflect.TypeOf((*MockChefIngesterServiceServer)(nil).ProcessChefRun), arg0, arg1) } -// ProcessChefAction mocks base method. +// ProcessChefAction mocks base method func (m *MockChefIngesterServiceServer) ProcessChefAction(arg0 context.Context, arg1 *request.Action) (*response.ProcessChefActionResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ProcessChefAction", arg0, arg1) @@ -204,28 +203,13 @@ func (m *MockChefIngesterServiceServer) ProcessChefAction(arg0 context.Context, return ret0, ret1 } -// ProcessChefAction indicates an expected call of ProcessChefAction. +// ProcessChefAction indicates an expected call of ProcessChefAction func (mr *MockChefIngesterServiceServerMockRecorder) ProcessChefAction(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProcessChefAction", reflect.TypeOf((*MockChefIngesterServiceServer)(nil).ProcessChefAction), arg0, arg1) } -// ProcessChefRun mocks base method. -func (m *MockChefIngesterServiceServer) ProcessChefRun(arg0 context.Context, arg1 *request.Run) (*response.ProcessChefRunResponse, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ProcessChefRun", arg0, arg1) - ret0, _ := ret[0].(*response.ProcessChefRunResponse) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// ProcessChefRun indicates an expected call of ProcessChefRun. -func (mr *MockChefIngesterServiceServerMockRecorder) ProcessChefRun(arg0, arg1 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProcessChefRun", reflect.TypeOf((*MockChefIngesterServiceServer)(nil).ProcessChefRun), arg0, arg1) -} - -// ProcessLivenessPing mocks base method. +// ProcessLivenessPing mocks base method func (m *MockChefIngesterServiceServer) ProcessLivenessPing(arg0 context.Context, arg1 *request.Liveness) (*response.ProcessLivenessResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ProcessLivenessPing", arg0, arg1) @@ -234,13 +218,13 @@ func (m *MockChefIngesterServiceServer) ProcessLivenessPing(arg0 context.Context return ret0, ret1 } -// ProcessLivenessPing indicates an expected call of ProcessLivenessPing. +// ProcessLivenessPing indicates an expected call of ProcessLivenessPing func (mr *MockChefIngesterServiceServerMockRecorder) ProcessLivenessPing(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProcessLivenessPing", reflect.TypeOf((*MockChefIngesterServiceServer)(nil).ProcessLivenessPing), arg0, arg1) } -// ProcessMultipleNodeDeletes mocks base method. +// ProcessMultipleNodeDeletes mocks base method func (m *MockChefIngesterServiceServer) ProcessMultipleNodeDeletes(arg0 context.Context, arg1 *request.MultipleNodeDeleteRequest) (*response.ProcessMultipleNodeDeleteResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ProcessMultipleNodeDeletes", arg0, arg1) @@ -249,13 +233,13 @@ func (m *MockChefIngesterServiceServer) ProcessMultipleNodeDeletes(arg0 context. return ret0, ret1 } -// ProcessMultipleNodeDeletes indicates an expected call of ProcessMultipleNodeDeletes. +// ProcessMultipleNodeDeletes indicates an expected call of ProcessMultipleNodeDeletes func (mr *MockChefIngesterServiceServerMockRecorder) ProcessMultipleNodeDeletes(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProcessMultipleNodeDeletes", reflect.TypeOf((*MockChefIngesterServiceServer)(nil).ProcessMultipleNodeDeletes), arg0, arg1) } -// ProcessNodeDelete mocks base method. +// ProcessNodeDelete mocks base method func (m *MockChefIngesterServiceServer) ProcessNodeDelete(arg0 context.Context, arg1 *request.Delete) (*response.ProcessNodeDeleteResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ProcessNodeDelete", arg0, arg1) @@ -264,8 +248,23 @@ func (m *MockChefIngesterServiceServer) ProcessNodeDelete(arg0 context.Context, return ret0, ret1 } -// ProcessNodeDelete indicates an expected call of ProcessNodeDelete. +// ProcessNodeDelete indicates an expected call of ProcessNodeDelete func (mr *MockChefIngesterServiceServerMockRecorder) ProcessNodeDelete(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProcessNodeDelete", reflect.TypeOf((*MockChefIngesterServiceServer)(nil).ProcessNodeDelete), arg0, arg1) } + +// GetVersion mocks base method +func (m *MockChefIngesterServiceServer) GetVersion(arg0 context.Context, arg1 *VersionRequest) (*Version, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetVersion", arg0, arg1) + ret0, _ := ret[0].(*Version) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetVersion indicates an expected call of GetVersion +func (mr *MockChefIngesterServiceServerMockRecorder) GetVersion(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetVersion", reflect.TypeOf((*MockChefIngesterServiceServer)(nil).GetVersion), arg0, arg1) +} diff --git a/api/interservice/nodemanager/manager/manager.pb.client_mock.go b/api/interservice/nodemanager/manager/manager.pb.client_mock.go index 49a564da534..cbbbe150647 100644 --- a/api/interservice/nodemanager/manager/manager.pb.client_mock.go +++ b/api/interservice/nodemanager/manager/manager.pb.client_mock.go @@ -6,138 +6,137 @@ package manager import ( context "context" - reflect "reflect" - nodes "github.com/chef/automate/api/interservice/nodemanager/nodes" gomock "github.com/golang/mock/gomock" grpc "google.golang.org/grpc" emptypb "google.golang.org/protobuf/types/known/emptypb" + reflect "reflect" ) -// MockNodeManagerServiceClient is a mock of NodeManagerServiceClient interface. +// MockNodeManagerServiceClient is a mock of NodeManagerServiceClient interface type MockNodeManagerServiceClient struct { ctrl *gomock.Controller recorder *MockNodeManagerServiceClientMockRecorder } -// MockNodeManagerServiceClientMockRecorder is the mock recorder for MockNodeManagerServiceClient. +// MockNodeManagerServiceClientMockRecorder is the mock recorder for MockNodeManagerServiceClient type MockNodeManagerServiceClientMockRecorder struct { mock *MockNodeManagerServiceClient } -// NewMockNodeManagerServiceClient creates a new mock instance. +// NewMockNodeManagerServiceClient creates a new mock instance func NewMockNodeManagerServiceClient(ctrl *gomock.Controller) *MockNodeManagerServiceClient { mock := &MockNodeManagerServiceClient{ctrl: ctrl} mock.recorder = &MockNodeManagerServiceClientMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockNodeManagerServiceClient) EXPECT() *MockNodeManagerServiceClientMockRecorder { return m.recorder } -// ChangeNodeState mocks base method. -func (m *MockNodeManagerServiceClient) ChangeNodeState(ctx context.Context, in *NodeState, opts ...grpc.CallOption) (*ChangeNodeStateResponse, error) { +// Create mocks base method +func (m *MockNodeManagerServiceClient) Create(ctx context.Context, in *NodeManager, opts ...grpc.CallOption) (*Ids, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "ChangeNodeState", varargs...) - ret0, _ := ret[0].(*ChangeNodeStateResponse) + ret := m.ctrl.Call(m, "Create", varargs...) + ret0, _ := ret[0].(*Ids) ret1, _ := ret[1].(error) return ret0, ret1 } -// ChangeNodeState indicates an expected call of ChangeNodeState. -func (mr *MockNodeManagerServiceClientMockRecorder) ChangeNodeState(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// Create indicates an expected call of Create +func (mr *MockNodeManagerServiceClientMockRecorder) Create(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ChangeNodeState", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).ChangeNodeState), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).Create), varargs...) } -// Connect mocks base method. -func (m *MockNodeManagerServiceClient) Connect(ctx context.Context, in *NodeManager, opts ...grpc.CallOption) (*emptypb.Empty, error) { +// Read mocks base method +func (m *MockNodeManagerServiceClient) Read(ctx context.Context, in *Id, opts ...grpc.CallOption) (*NodeManager, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "Connect", varargs...) - ret0, _ := ret[0].(*emptypb.Empty) + ret := m.ctrl.Call(m, "Read", varargs...) + ret0, _ := ret[0].(*NodeManager) ret1, _ := ret[1].(error) return ret0, ret1 } -// Connect indicates an expected call of Connect. -func (mr *MockNodeManagerServiceClientMockRecorder) Connect(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// Read indicates an expected call of Read +func (mr *MockNodeManagerServiceClientMockRecorder) Read(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Connect", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).Connect), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Read", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).Read), varargs...) } -// ConnectManager mocks base method. -func (m *MockNodeManagerServiceClient) ConnectManager(ctx context.Context, in *Id, opts ...grpc.CallOption) (*emptypb.Empty, error) { +// Update mocks base method +func (m *MockNodeManagerServiceClient) Update(ctx context.Context, in *NodeManager, opts ...grpc.CallOption) (*emptypb.Empty, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "ConnectManager", varargs...) + ret := m.ctrl.Call(m, "Update", varargs...) ret0, _ := ret[0].(*emptypb.Empty) ret1, _ := ret[1].(error) return ret0, ret1 } -// ConnectManager indicates an expected call of ConnectManager. -func (mr *MockNodeManagerServiceClientMockRecorder) ConnectManager(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// Update indicates an expected call of Update +func (mr *MockNodeManagerServiceClientMockRecorder) Update(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ConnectManager", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).ConnectManager), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).Update), varargs...) } -// Create mocks base method. -func (m *MockNodeManagerServiceClient) Create(ctx context.Context, in *NodeManager, opts ...grpc.CallOption) (*Ids, error) { +// Delete mocks base method +func (m *MockNodeManagerServiceClient) Delete(ctx context.Context, in *Id, opts ...grpc.CallOption) (*emptypb.Empty, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "Create", varargs...) - ret0, _ := ret[0].(*Ids) + ret := m.ctrl.Call(m, "Delete", varargs...) + ret0, _ := ret[0].(*emptypb.Empty) ret1, _ := ret[1].(error) return ret0, ret1 } -// Create indicates an expected call of Create. -func (mr *MockNodeManagerServiceClientMockRecorder) Create(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// Delete indicates an expected call of Delete +func (mr *MockNodeManagerServiceClientMockRecorder) Delete(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).Create), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).Delete), varargs...) } -// Delete mocks base method. -func (m *MockNodeManagerServiceClient) Delete(ctx context.Context, in *Id, opts ...grpc.CallOption) (*emptypb.Empty, error) { +// DeleteWithNodes mocks base method +func (m *MockNodeManagerServiceClient) DeleteWithNodes(ctx context.Context, in *Id, opts ...grpc.CallOption) (*Ids, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "Delete", varargs...) - ret0, _ := ret[0].(*emptypb.Empty) + ret := m.ctrl.Call(m, "DeleteWithNodes", varargs...) + ret0, _ := ret[0].(*Ids) ret1, _ := ret[1].(error) return ret0, ret1 } -// Delete indicates an expected call of Delete. -func (mr *MockNodeManagerServiceClientMockRecorder) Delete(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// DeleteWithNodes indicates an expected call of DeleteWithNodes +func (mr *MockNodeManagerServiceClientMockRecorder) DeleteWithNodes(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).Delete), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteWithNodes", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).DeleteWithNodes), varargs...) } -// DeleteWithNodeStateStopped mocks base method. +// DeleteWithNodeStateStopped mocks base method func (m *MockNodeManagerServiceClient) DeleteWithNodeStateStopped(ctx context.Context, in *Id, opts ...grpc.CallOption) (*emptypb.Empty, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} @@ -150,14 +149,14 @@ func (m *MockNodeManagerServiceClient) DeleteWithNodeStateStopped(ctx context.Co return ret0, ret1 } -// DeleteWithNodeStateStopped indicates an expected call of DeleteWithNodeStateStopped. +// DeleteWithNodeStateStopped indicates an expected call of DeleteWithNodeStateStopped func (mr *MockNodeManagerServiceClientMockRecorder) DeleteWithNodeStateStopped(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteWithNodeStateStopped", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).DeleteWithNodeStateStopped), varargs...) } -// DeleteWithNodeStateTerminated mocks base method. +// DeleteWithNodeStateTerminated mocks base method func (m *MockNodeManagerServiceClient) DeleteWithNodeStateTerminated(ctx context.Context, in *Id, opts ...grpc.CallOption) (*emptypb.Empty, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} @@ -170,292 +169,292 @@ func (m *MockNodeManagerServiceClient) DeleteWithNodeStateTerminated(ctx context return ret0, ret1 } -// DeleteWithNodeStateTerminated indicates an expected call of DeleteWithNodeStateTerminated. +// DeleteWithNodeStateTerminated indicates an expected call of DeleteWithNodeStateTerminated func (mr *MockNodeManagerServiceClientMockRecorder) DeleteWithNodeStateTerminated(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteWithNodeStateTerminated", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).DeleteWithNodeStateTerminated), varargs...) } -// DeleteWithNodes mocks base method. -func (m *MockNodeManagerServiceClient) DeleteWithNodes(ctx context.Context, in *Id, opts ...grpc.CallOption) (*Ids, error) { +// List mocks base method +func (m *MockNodeManagerServiceClient) List(ctx context.Context, in *Query, opts ...grpc.CallOption) (*NodeManagers, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "DeleteWithNodes", varargs...) - ret0, _ := ret[0].(*Ids) + ret := m.ctrl.Call(m, "List", varargs...) + ret0, _ := ret[0].(*NodeManagers) ret1, _ := ret[1].(error) return ret0, ret1 } -// DeleteWithNodes indicates an expected call of DeleteWithNodes. -func (mr *MockNodeManagerServiceClientMockRecorder) DeleteWithNodes(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// List indicates an expected call of List +func (mr *MockNodeManagerServiceClientMockRecorder) List(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteWithNodes", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).DeleteWithNodes), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).List), varargs...) } -// GetNodeWithSecrets mocks base method. -func (m *MockNodeManagerServiceClient) GetNodeWithSecrets(ctx context.Context, in *Id, opts ...grpc.CallOption) (*nodes.Node, error) { +// Connect mocks base method +func (m *MockNodeManagerServiceClient) Connect(ctx context.Context, in *NodeManager, opts ...grpc.CallOption) (*emptypb.Empty, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "GetNodeWithSecrets", varargs...) - ret0, _ := ret[0].(*nodes.Node) + ret := m.ctrl.Call(m, "Connect", varargs...) + ret0, _ := ret[0].(*emptypb.Empty) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetNodeWithSecrets indicates an expected call of GetNodeWithSecrets. -func (mr *MockNodeManagerServiceClientMockRecorder) GetNodeWithSecrets(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// Connect indicates an expected call of Connect +func (mr *MockNodeManagerServiceClientMockRecorder) Connect(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodeWithSecrets", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).GetNodeWithSecrets), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Connect", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).Connect), varargs...) } -// List mocks base method. -func (m *MockNodeManagerServiceClient) List(ctx context.Context, in *Query, opts ...grpc.CallOption) (*NodeManagers, error) { +// ConnectManager mocks base method +func (m *MockNodeManagerServiceClient) ConnectManager(ctx context.Context, in *Id, opts ...grpc.CallOption) (*emptypb.Empty, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "List", varargs...) - ret0, _ := ret[0].(*NodeManagers) + ret := m.ctrl.Call(m, "ConnectManager", varargs...) + ret0, _ := ret[0].(*emptypb.Empty) ret1, _ := ret[1].(error) return ret0, ret1 } -// List indicates an expected call of List. -func (mr *MockNodeManagerServiceClientMockRecorder) List(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// ConnectManager indicates an expected call of ConnectManager +func (mr *MockNodeManagerServiceClientMockRecorder) ConnectManager(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).List), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ConnectManager", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).ConnectManager), varargs...) } -// ProcessNode mocks base method. -func (m *MockNodeManagerServiceClient) ProcessNode(ctx context.Context, in *NodeMetadata, opts ...grpc.CallOption) (*ProcessNodeResponse, error) { +// SearchNodeFields mocks base method +func (m *MockNodeManagerServiceClient) SearchNodeFields(ctx context.Context, in *FieldQuery, opts ...grpc.CallOption) (*Fields, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "ProcessNode", varargs...) - ret0, _ := ret[0].(*ProcessNodeResponse) + ret := m.ctrl.Call(m, "SearchNodeFields", varargs...) + ret0, _ := ret[0].(*Fields) ret1, _ := ret[1].(error) return ret0, ret1 } -// ProcessNode indicates an expected call of ProcessNode. -func (mr *MockNodeManagerServiceClientMockRecorder) ProcessNode(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// SearchNodeFields indicates an expected call of SearchNodeFields +func (mr *MockNodeManagerServiceClientMockRecorder) SearchNodeFields(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProcessNode", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).ProcessNode), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SearchNodeFields", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).SearchNodeFields), varargs...) } -// Read mocks base method. -func (m *MockNodeManagerServiceClient) Read(ctx context.Context, in *Id, opts ...grpc.CallOption) (*NodeManager, error) { +// SearchNodes mocks base method +func (m *MockNodeManagerServiceClient) SearchNodes(ctx context.Context, in *NodeQuery, opts ...grpc.CallOption) (*Nodes, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "Read", varargs...) - ret0, _ := ret[0].(*NodeManager) + ret := m.ctrl.Call(m, "SearchNodes", varargs...) + ret0, _ := ret[0].(*Nodes) ret1, _ := ret[1].(error) return ret0, ret1 } -// Read indicates an expected call of Read. -func (mr *MockNodeManagerServiceClientMockRecorder) Read(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// SearchNodes indicates an expected call of SearchNodes +func (mr *MockNodeManagerServiceClientMockRecorder) SearchNodes(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Read", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).Read), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SearchNodes", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).SearchNodes), varargs...) } -// SearchManagerNodes mocks base method. -func (m *MockNodeManagerServiceClient) SearchManagerNodes(ctx context.Context, in *NodeQuery, opts ...grpc.CallOption) (*ManagerNodes, error) { +// ProcessNode mocks base method +func (m *MockNodeManagerServiceClient) ProcessNode(ctx context.Context, in *NodeMetadata, opts ...grpc.CallOption) (*ProcessNodeResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "SearchManagerNodes", varargs...) - ret0, _ := ret[0].(*ManagerNodes) + ret := m.ctrl.Call(m, "ProcessNode", varargs...) + ret0, _ := ret[0].(*ProcessNodeResponse) ret1, _ := ret[1].(error) return ret0, ret1 } -// SearchManagerNodes indicates an expected call of SearchManagerNodes. -func (mr *MockNodeManagerServiceClientMockRecorder) SearchManagerNodes(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// ProcessNode indicates an expected call of ProcessNode +func (mr *MockNodeManagerServiceClientMockRecorder) ProcessNode(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SearchManagerNodes", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).SearchManagerNodes), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProcessNode", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).ProcessNode), varargs...) } -// SearchNodeFields mocks base method. -func (m *MockNodeManagerServiceClient) SearchNodeFields(ctx context.Context, in *FieldQuery, opts ...grpc.CallOption) (*Fields, error) { +// ChangeNodeState mocks base method +func (m *MockNodeManagerServiceClient) ChangeNodeState(ctx context.Context, in *NodeState, opts ...grpc.CallOption) (*ChangeNodeStateResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "SearchNodeFields", varargs...) - ret0, _ := ret[0].(*Fields) + ret := m.ctrl.Call(m, "ChangeNodeState", varargs...) + ret0, _ := ret[0].(*ChangeNodeStateResponse) ret1, _ := ret[1].(error) return ret0, ret1 } -// SearchNodeFields indicates an expected call of SearchNodeFields. -func (mr *MockNodeManagerServiceClientMockRecorder) SearchNodeFields(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// ChangeNodeState indicates an expected call of ChangeNodeState +func (mr *MockNodeManagerServiceClientMockRecorder) ChangeNodeState(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SearchNodeFields", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).SearchNodeFields), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ChangeNodeState", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).ChangeNodeState), varargs...) } -// SearchNodes mocks base method. -func (m *MockNodeManagerServiceClient) SearchNodes(ctx context.Context, in *NodeQuery, opts ...grpc.CallOption) (*Nodes, error) { +// GetNodeWithSecrets mocks base method +func (m *MockNodeManagerServiceClient) GetNodeWithSecrets(ctx context.Context, in *Id, opts ...grpc.CallOption) (*nodes.Node, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "SearchNodes", varargs...) - ret0, _ := ret[0].(*Nodes) + ret := m.ctrl.Call(m, "GetNodeWithSecrets", varargs...) + ret0, _ := ret[0].(*nodes.Node) ret1, _ := ret[1].(error) return ret0, ret1 } -// SearchNodes indicates an expected call of SearchNodes. -func (mr *MockNodeManagerServiceClientMockRecorder) SearchNodes(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// GetNodeWithSecrets indicates an expected call of GetNodeWithSecrets +func (mr *MockNodeManagerServiceClientMockRecorder) GetNodeWithSecrets(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SearchNodes", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).SearchNodes), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodeWithSecrets", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).GetNodeWithSecrets), varargs...) } -// Update mocks base method. -func (m *MockNodeManagerServiceClient) Update(ctx context.Context, in *NodeManager, opts ...grpc.CallOption) (*emptypb.Empty, error) { +// SearchManagerNodes mocks base method +func (m *MockNodeManagerServiceClient) SearchManagerNodes(ctx context.Context, in *NodeQuery, opts ...grpc.CallOption) (*ManagerNodes, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "Update", varargs...) - ret0, _ := ret[0].(*emptypb.Empty) + ret := m.ctrl.Call(m, "SearchManagerNodes", varargs...) + ret0, _ := ret[0].(*ManagerNodes) ret1, _ := ret[1].(error) return ret0, ret1 } -// Update indicates an expected call of Update. -func (mr *MockNodeManagerServiceClientMockRecorder) Update(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// SearchManagerNodes indicates an expected call of SearchManagerNodes +func (mr *MockNodeManagerServiceClientMockRecorder) SearchManagerNodes(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).Update), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SearchManagerNodes", reflect.TypeOf((*MockNodeManagerServiceClient)(nil).SearchManagerNodes), varargs...) } -// MockNodeManagerServiceServer is a mock of NodeManagerServiceServer interface. +// MockNodeManagerServiceServer is a mock of NodeManagerServiceServer interface type MockNodeManagerServiceServer struct { ctrl *gomock.Controller recorder *MockNodeManagerServiceServerMockRecorder } -// MockNodeManagerServiceServerMockRecorder is the mock recorder for MockNodeManagerServiceServer. +// MockNodeManagerServiceServerMockRecorder is the mock recorder for MockNodeManagerServiceServer type MockNodeManagerServiceServerMockRecorder struct { mock *MockNodeManagerServiceServer } -// NewMockNodeManagerServiceServer creates a new mock instance. +// NewMockNodeManagerServiceServer creates a new mock instance func NewMockNodeManagerServiceServer(ctrl *gomock.Controller) *MockNodeManagerServiceServer { mock := &MockNodeManagerServiceServer{ctrl: ctrl} mock.recorder = &MockNodeManagerServiceServerMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockNodeManagerServiceServer) EXPECT() *MockNodeManagerServiceServerMockRecorder { return m.recorder } -// ChangeNodeState mocks base method. -func (m *MockNodeManagerServiceServer) ChangeNodeState(arg0 context.Context, arg1 *NodeState) (*ChangeNodeStateResponse, error) { +// Create mocks base method +func (m *MockNodeManagerServiceServer) Create(arg0 context.Context, arg1 *NodeManager) (*Ids, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ChangeNodeState", arg0, arg1) - ret0, _ := ret[0].(*ChangeNodeStateResponse) + ret := m.ctrl.Call(m, "Create", arg0, arg1) + ret0, _ := ret[0].(*Ids) ret1, _ := ret[1].(error) return ret0, ret1 } -// ChangeNodeState indicates an expected call of ChangeNodeState. -func (mr *MockNodeManagerServiceServerMockRecorder) ChangeNodeState(arg0, arg1 interface{}) *gomock.Call { +// Create indicates an expected call of Create +func (mr *MockNodeManagerServiceServerMockRecorder) Create(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ChangeNodeState", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).ChangeNodeState), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).Create), arg0, arg1) } -// Connect mocks base method. -func (m *MockNodeManagerServiceServer) Connect(arg0 context.Context, arg1 *NodeManager) (*emptypb.Empty, error) { +// Read mocks base method +func (m *MockNodeManagerServiceServer) Read(arg0 context.Context, arg1 *Id) (*NodeManager, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Connect", arg0, arg1) - ret0, _ := ret[0].(*emptypb.Empty) + ret := m.ctrl.Call(m, "Read", arg0, arg1) + ret0, _ := ret[0].(*NodeManager) ret1, _ := ret[1].(error) return ret0, ret1 } -// Connect indicates an expected call of Connect. -func (mr *MockNodeManagerServiceServerMockRecorder) Connect(arg0, arg1 interface{}) *gomock.Call { +// Read indicates an expected call of Read +func (mr *MockNodeManagerServiceServerMockRecorder) Read(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Connect", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).Connect), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Read", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).Read), arg0, arg1) } -// ConnectManager mocks base method. -func (m *MockNodeManagerServiceServer) ConnectManager(arg0 context.Context, arg1 *Id) (*emptypb.Empty, error) { +// Update mocks base method +func (m *MockNodeManagerServiceServer) Update(arg0 context.Context, arg1 *NodeManager) (*emptypb.Empty, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ConnectManager", arg0, arg1) + ret := m.ctrl.Call(m, "Update", arg0, arg1) ret0, _ := ret[0].(*emptypb.Empty) ret1, _ := ret[1].(error) return ret0, ret1 } -// ConnectManager indicates an expected call of ConnectManager. -func (mr *MockNodeManagerServiceServerMockRecorder) ConnectManager(arg0, arg1 interface{}) *gomock.Call { +// Update indicates an expected call of Update +func (mr *MockNodeManagerServiceServerMockRecorder) Update(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ConnectManager", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).ConnectManager), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).Update), arg0, arg1) } -// Create mocks base method. -func (m *MockNodeManagerServiceServer) Create(arg0 context.Context, arg1 *NodeManager) (*Ids, error) { +// Delete mocks base method +func (m *MockNodeManagerServiceServer) Delete(arg0 context.Context, arg1 *Id) (*emptypb.Empty, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Create", arg0, arg1) - ret0, _ := ret[0].(*Ids) + ret := m.ctrl.Call(m, "Delete", arg0, arg1) + ret0, _ := ret[0].(*emptypb.Empty) ret1, _ := ret[1].(error) return ret0, ret1 } -// Create indicates an expected call of Create. -func (mr *MockNodeManagerServiceServerMockRecorder) Create(arg0, arg1 interface{}) *gomock.Call { +// Delete indicates an expected call of Delete +func (mr *MockNodeManagerServiceServerMockRecorder) Delete(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).Create), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).Delete), arg0, arg1) } -// Delete mocks base method. -func (m *MockNodeManagerServiceServer) Delete(arg0 context.Context, arg1 *Id) (*emptypb.Empty, error) { +// DeleteWithNodes mocks base method +func (m *MockNodeManagerServiceServer) DeleteWithNodes(arg0 context.Context, arg1 *Id) (*Ids, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Delete", arg0, arg1) - ret0, _ := ret[0].(*emptypb.Empty) + ret := m.ctrl.Call(m, "DeleteWithNodes", arg0, arg1) + ret0, _ := ret[0].(*Ids) ret1, _ := ret[1].(error) return ret0, ret1 } -// Delete indicates an expected call of Delete. -func (mr *MockNodeManagerServiceServerMockRecorder) Delete(arg0, arg1 interface{}) *gomock.Call { +// DeleteWithNodes indicates an expected call of DeleteWithNodes +func (mr *MockNodeManagerServiceServerMockRecorder) DeleteWithNodes(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).Delete), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteWithNodes", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).DeleteWithNodes), arg0, arg1) } -// DeleteWithNodeStateStopped mocks base method. +// DeleteWithNodeStateStopped mocks base method func (m *MockNodeManagerServiceServer) DeleteWithNodeStateStopped(arg0 context.Context, arg1 *Id) (*emptypb.Empty, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "DeleteWithNodeStateStopped", arg0, arg1) @@ -464,13 +463,13 @@ func (m *MockNodeManagerServiceServer) DeleteWithNodeStateStopped(arg0 context.C return ret0, ret1 } -// DeleteWithNodeStateStopped indicates an expected call of DeleteWithNodeStateStopped. +// DeleteWithNodeStateStopped indicates an expected call of DeleteWithNodeStateStopped func (mr *MockNodeManagerServiceServerMockRecorder) DeleteWithNodeStateStopped(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteWithNodeStateStopped", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).DeleteWithNodeStateStopped), arg0, arg1) } -// DeleteWithNodeStateTerminated mocks base method. +// DeleteWithNodeStateTerminated mocks base method func (m *MockNodeManagerServiceServer) DeleteWithNodeStateTerminated(arg0 context.Context, arg1 *Id) (*emptypb.Empty, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "DeleteWithNodeStateTerminated", arg0, arg1) @@ -479,143 +478,143 @@ func (m *MockNodeManagerServiceServer) DeleteWithNodeStateTerminated(arg0 contex return ret0, ret1 } -// DeleteWithNodeStateTerminated indicates an expected call of DeleteWithNodeStateTerminated. +// DeleteWithNodeStateTerminated indicates an expected call of DeleteWithNodeStateTerminated func (mr *MockNodeManagerServiceServerMockRecorder) DeleteWithNodeStateTerminated(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteWithNodeStateTerminated", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).DeleteWithNodeStateTerminated), arg0, arg1) } -// DeleteWithNodes mocks base method. -func (m *MockNodeManagerServiceServer) DeleteWithNodes(arg0 context.Context, arg1 *Id) (*Ids, error) { +// List mocks base method +func (m *MockNodeManagerServiceServer) List(arg0 context.Context, arg1 *Query) (*NodeManagers, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteWithNodes", arg0, arg1) - ret0, _ := ret[0].(*Ids) + ret := m.ctrl.Call(m, "List", arg0, arg1) + ret0, _ := ret[0].(*NodeManagers) ret1, _ := ret[1].(error) return ret0, ret1 } -// DeleteWithNodes indicates an expected call of DeleteWithNodes. -func (mr *MockNodeManagerServiceServerMockRecorder) DeleteWithNodes(arg0, arg1 interface{}) *gomock.Call { +// List indicates an expected call of List +func (mr *MockNodeManagerServiceServerMockRecorder) List(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteWithNodes", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).DeleteWithNodes), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).List), arg0, arg1) } -// GetNodeWithSecrets mocks base method. -func (m *MockNodeManagerServiceServer) GetNodeWithSecrets(arg0 context.Context, arg1 *Id) (*nodes.Node, error) { +// Connect mocks base method +func (m *MockNodeManagerServiceServer) Connect(arg0 context.Context, arg1 *NodeManager) (*emptypb.Empty, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetNodeWithSecrets", arg0, arg1) - ret0, _ := ret[0].(*nodes.Node) + ret := m.ctrl.Call(m, "Connect", arg0, arg1) + ret0, _ := ret[0].(*emptypb.Empty) ret1, _ := ret[1].(error) return ret0, ret1 } -// GetNodeWithSecrets indicates an expected call of GetNodeWithSecrets. -func (mr *MockNodeManagerServiceServerMockRecorder) GetNodeWithSecrets(arg0, arg1 interface{}) *gomock.Call { +// Connect indicates an expected call of Connect +func (mr *MockNodeManagerServiceServerMockRecorder) Connect(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodeWithSecrets", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).GetNodeWithSecrets), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Connect", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).Connect), arg0, arg1) } -// List mocks base method. -func (m *MockNodeManagerServiceServer) List(arg0 context.Context, arg1 *Query) (*NodeManagers, error) { +// ConnectManager mocks base method +func (m *MockNodeManagerServiceServer) ConnectManager(arg0 context.Context, arg1 *Id) (*emptypb.Empty, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "List", arg0, arg1) - ret0, _ := ret[0].(*NodeManagers) + ret := m.ctrl.Call(m, "ConnectManager", arg0, arg1) + ret0, _ := ret[0].(*emptypb.Empty) ret1, _ := ret[1].(error) return ret0, ret1 } -// List indicates an expected call of List. -func (mr *MockNodeManagerServiceServerMockRecorder) List(arg0, arg1 interface{}) *gomock.Call { +// ConnectManager indicates an expected call of ConnectManager +func (mr *MockNodeManagerServiceServerMockRecorder) ConnectManager(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).List), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ConnectManager", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).ConnectManager), arg0, arg1) } -// ProcessNode mocks base method. -func (m *MockNodeManagerServiceServer) ProcessNode(arg0 context.Context, arg1 *NodeMetadata) (*ProcessNodeResponse, error) { +// SearchNodeFields mocks base method +func (m *MockNodeManagerServiceServer) SearchNodeFields(arg0 context.Context, arg1 *FieldQuery) (*Fields, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ProcessNode", arg0, arg1) - ret0, _ := ret[0].(*ProcessNodeResponse) + ret := m.ctrl.Call(m, "SearchNodeFields", arg0, arg1) + ret0, _ := ret[0].(*Fields) ret1, _ := ret[1].(error) return ret0, ret1 } -// ProcessNode indicates an expected call of ProcessNode. -func (mr *MockNodeManagerServiceServerMockRecorder) ProcessNode(arg0, arg1 interface{}) *gomock.Call { +// SearchNodeFields indicates an expected call of SearchNodeFields +func (mr *MockNodeManagerServiceServerMockRecorder) SearchNodeFields(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProcessNode", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).ProcessNode), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SearchNodeFields", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).SearchNodeFields), arg0, arg1) } -// Read mocks base method. -func (m *MockNodeManagerServiceServer) Read(arg0 context.Context, arg1 *Id) (*NodeManager, error) { +// SearchNodes mocks base method +func (m *MockNodeManagerServiceServer) SearchNodes(arg0 context.Context, arg1 *NodeQuery) (*Nodes, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Read", arg0, arg1) - ret0, _ := ret[0].(*NodeManager) + ret := m.ctrl.Call(m, "SearchNodes", arg0, arg1) + ret0, _ := ret[0].(*Nodes) ret1, _ := ret[1].(error) return ret0, ret1 } -// Read indicates an expected call of Read. -func (mr *MockNodeManagerServiceServerMockRecorder) Read(arg0, arg1 interface{}) *gomock.Call { +// SearchNodes indicates an expected call of SearchNodes +func (mr *MockNodeManagerServiceServerMockRecorder) SearchNodes(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Read", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).Read), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SearchNodes", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).SearchNodes), arg0, arg1) } -// SearchManagerNodes mocks base method. -func (m *MockNodeManagerServiceServer) SearchManagerNodes(arg0 context.Context, arg1 *NodeQuery) (*ManagerNodes, error) { +// ProcessNode mocks base method +func (m *MockNodeManagerServiceServer) ProcessNode(arg0 context.Context, arg1 *NodeMetadata) (*ProcessNodeResponse, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "SearchManagerNodes", arg0, arg1) - ret0, _ := ret[0].(*ManagerNodes) + ret := m.ctrl.Call(m, "ProcessNode", arg0, arg1) + ret0, _ := ret[0].(*ProcessNodeResponse) ret1, _ := ret[1].(error) return ret0, ret1 } -// SearchManagerNodes indicates an expected call of SearchManagerNodes. -func (mr *MockNodeManagerServiceServerMockRecorder) SearchManagerNodes(arg0, arg1 interface{}) *gomock.Call { +// ProcessNode indicates an expected call of ProcessNode +func (mr *MockNodeManagerServiceServerMockRecorder) ProcessNode(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SearchManagerNodes", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).SearchManagerNodes), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProcessNode", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).ProcessNode), arg0, arg1) } -// SearchNodeFields mocks base method. -func (m *MockNodeManagerServiceServer) SearchNodeFields(arg0 context.Context, arg1 *FieldQuery) (*Fields, error) { +// ChangeNodeState mocks base method +func (m *MockNodeManagerServiceServer) ChangeNodeState(arg0 context.Context, arg1 *NodeState) (*ChangeNodeStateResponse, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "SearchNodeFields", arg0, arg1) - ret0, _ := ret[0].(*Fields) + ret := m.ctrl.Call(m, "ChangeNodeState", arg0, arg1) + ret0, _ := ret[0].(*ChangeNodeStateResponse) ret1, _ := ret[1].(error) return ret0, ret1 } -// SearchNodeFields indicates an expected call of SearchNodeFields. -func (mr *MockNodeManagerServiceServerMockRecorder) SearchNodeFields(arg0, arg1 interface{}) *gomock.Call { +// ChangeNodeState indicates an expected call of ChangeNodeState +func (mr *MockNodeManagerServiceServerMockRecorder) ChangeNodeState(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SearchNodeFields", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).SearchNodeFields), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ChangeNodeState", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).ChangeNodeState), arg0, arg1) } -// SearchNodes mocks base method. -func (m *MockNodeManagerServiceServer) SearchNodes(arg0 context.Context, arg1 *NodeQuery) (*Nodes, error) { +// GetNodeWithSecrets mocks base method +func (m *MockNodeManagerServiceServer) GetNodeWithSecrets(arg0 context.Context, arg1 *Id) (*nodes.Node, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "SearchNodes", arg0, arg1) - ret0, _ := ret[0].(*Nodes) + ret := m.ctrl.Call(m, "GetNodeWithSecrets", arg0, arg1) + ret0, _ := ret[0].(*nodes.Node) ret1, _ := ret[1].(error) return ret0, ret1 } -// SearchNodes indicates an expected call of SearchNodes. -func (mr *MockNodeManagerServiceServerMockRecorder) SearchNodes(arg0, arg1 interface{}) *gomock.Call { +// GetNodeWithSecrets indicates an expected call of GetNodeWithSecrets +func (mr *MockNodeManagerServiceServerMockRecorder) GetNodeWithSecrets(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SearchNodes", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).SearchNodes), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodeWithSecrets", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).GetNodeWithSecrets), arg0, arg1) } -// Update mocks base method. -func (m *MockNodeManagerServiceServer) Update(arg0 context.Context, arg1 *NodeManager) (*emptypb.Empty, error) { +// SearchManagerNodes mocks base method +func (m *MockNodeManagerServiceServer) SearchManagerNodes(arg0 context.Context, arg1 *NodeQuery) (*ManagerNodes, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Update", arg0, arg1) - ret0, _ := ret[0].(*emptypb.Empty) + ret := m.ctrl.Call(m, "SearchManagerNodes", arg0, arg1) + ret0, _ := ret[0].(*ManagerNodes) ret1, _ := ret[1].(error) return ret0, ret1 } -// Update indicates an expected call of Update. -func (mr *MockNodeManagerServiceServerMockRecorder) Update(arg0, arg1 interface{}) *gomock.Call { +// SearchManagerNodes indicates an expected call of SearchManagerNodes +func (mr *MockNodeManagerServiceServerMockRecorder) SearchManagerNodes(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).Update), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SearchManagerNodes", reflect.TypeOf((*MockNodeManagerServiceServer)(nil).SearchManagerNodes), arg0, arg1) } diff --git a/api/interservice/nodemanager/nodes/nodes.pb.client_mock.go b/api/interservice/nodemanager/nodes/nodes.pb.client_mock.go index 301eb199e2d..f208e8a45d1 100644 --- a/api/interservice/nodemanager/nodes/nodes.pb.client_mock.go +++ b/api/interservice/nodemanager/nodes/nodes.pb.client_mock.go @@ -6,197 +6,196 @@ package nodes import ( context "context" - reflect "reflect" - gomock "github.com/golang/mock/gomock" grpc "google.golang.org/grpc" emptypb "google.golang.org/protobuf/types/known/emptypb" + reflect "reflect" ) -// MockNodesServiceClient is a mock of NodesServiceClient interface. +// MockNodesServiceClient is a mock of NodesServiceClient interface type MockNodesServiceClient struct { ctrl *gomock.Controller recorder *MockNodesServiceClientMockRecorder } -// MockNodesServiceClientMockRecorder is the mock recorder for MockNodesServiceClient. +// MockNodesServiceClientMockRecorder is the mock recorder for MockNodesServiceClient type MockNodesServiceClientMockRecorder struct { mock *MockNodesServiceClient } -// NewMockNodesServiceClient creates a new mock instance. +// NewMockNodesServiceClient creates a new mock instance func NewMockNodesServiceClient(ctrl *gomock.Controller) *MockNodesServiceClient { mock := &MockNodesServiceClient{ctrl: ctrl} mock.recorder = &MockNodesServiceClientMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockNodesServiceClient) EXPECT() *MockNodesServiceClientMockRecorder { return m.recorder } -// BulkCreate mocks base method. -func (m *MockNodesServiceClient) BulkCreate(ctx context.Context, in *Nodes, opts ...grpc.CallOption) (*Ids, error) { +// Create mocks base method +func (m *MockNodesServiceClient) Create(ctx context.Context, in *Node, opts ...grpc.CallOption) (*Id, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "BulkCreate", varargs...) - ret0, _ := ret[0].(*Ids) + ret := m.ctrl.Call(m, "Create", varargs...) + ret0, _ := ret[0].(*Id) ret1, _ := ret[1].(error) return ret0, ret1 } -// BulkCreate indicates an expected call of BulkCreate. -func (mr *MockNodesServiceClientMockRecorder) BulkCreate(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// Create indicates an expected call of Create +func (mr *MockNodesServiceClientMockRecorder) Create(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BulkCreate", reflect.TypeOf((*MockNodesServiceClient)(nil).BulkCreate), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockNodesServiceClient)(nil).Create), varargs...) } -// BulkDelete mocks base method. -func (m *MockNodesServiceClient) BulkDelete(ctx context.Context, in *Query, opts ...grpc.CallOption) (*BulkDeleteResponse, error) { +// Read mocks base method +func (m *MockNodesServiceClient) Read(ctx context.Context, in *Id, opts ...grpc.CallOption) (*Node, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "BulkDelete", varargs...) - ret0, _ := ret[0].(*BulkDeleteResponse) + ret := m.ctrl.Call(m, "Read", varargs...) + ret0, _ := ret[0].(*Node) ret1, _ := ret[1].(error) return ret0, ret1 } -// BulkDelete indicates an expected call of BulkDelete. -func (mr *MockNodesServiceClientMockRecorder) BulkDelete(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// Read indicates an expected call of Read +func (mr *MockNodesServiceClientMockRecorder) Read(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BulkDelete", reflect.TypeOf((*MockNodesServiceClient)(nil).BulkDelete), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Read", reflect.TypeOf((*MockNodesServiceClient)(nil).Read), varargs...) } -// BulkDeleteById mocks base method. -func (m *MockNodesServiceClient) BulkDeleteById(ctx context.Context, in *Ids, opts ...grpc.CallOption) (*BulkDeleteResponse, error) { +// Update mocks base method +func (m *MockNodesServiceClient) Update(ctx context.Context, in *Node, opts ...grpc.CallOption) (*emptypb.Empty, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "BulkDeleteById", varargs...) - ret0, _ := ret[0].(*BulkDeleteResponse) + ret := m.ctrl.Call(m, "Update", varargs...) + ret0, _ := ret[0].(*emptypb.Empty) ret1, _ := ret[1].(error) return ret0, ret1 } -// BulkDeleteById indicates an expected call of BulkDeleteById. -func (mr *MockNodesServiceClientMockRecorder) BulkDeleteById(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// Update indicates an expected call of Update +func (mr *MockNodesServiceClientMockRecorder) Update(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BulkDeleteById", reflect.TypeOf((*MockNodesServiceClient)(nil).BulkDeleteById), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockNodesServiceClient)(nil).Update), varargs...) } -// Create mocks base method. -func (m *MockNodesServiceClient) Create(ctx context.Context, in *Node, opts ...grpc.CallOption) (*Id, error) { +// Delete mocks base method +func (m *MockNodesServiceClient) Delete(ctx context.Context, in *Id, opts ...grpc.CallOption) (*emptypb.Empty, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "Create", varargs...) - ret0, _ := ret[0].(*Id) + ret := m.ctrl.Call(m, "Delete", varargs...) + ret0, _ := ret[0].(*emptypb.Empty) ret1, _ := ret[1].(error) return ret0, ret1 } -// Create indicates an expected call of Create. -func (mr *MockNodesServiceClientMockRecorder) Create(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// Delete indicates an expected call of Delete +func (mr *MockNodesServiceClientMockRecorder) Delete(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockNodesServiceClient)(nil).Create), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockNodesServiceClient)(nil).Delete), varargs...) } -// Delete mocks base method. -func (m *MockNodesServiceClient) Delete(ctx context.Context, in *Id, opts ...grpc.CallOption) (*emptypb.Empty, error) { +// List mocks base method +func (m *MockNodesServiceClient) List(ctx context.Context, in *Query, opts ...grpc.CallOption) (*Nodes, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "Delete", varargs...) - ret0, _ := ret[0].(*emptypb.Empty) + ret := m.ctrl.Call(m, "List", varargs...) + ret0, _ := ret[0].(*Nodes) ret1, _ := ret[1].(error) return ret0, ret1 } -// Delete indicates an expected call of Delete. -func (mr *MockNodesServiceClientMockRecorder) Delete(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// List indicates an expected call of List +func (mr *MockNodesServiceClientMockRecorder) List(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockNodesServiceClient)(nil).Delete), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockNodesServiceClient)(nil).List), varargs...) } -// List mocks base method. -func (m *MockNodesServiceClient) List(ctx context.Context, in *Query, opts ...grpc.CallOption) (*Nodes, error) { +// BulkDelete mocks base method +func (m *MockNodesServiceClient) BulkDelete(ctx context.Context, in *Query, opts ...grpc.CallOption) (*BulkDeleteResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "List", varargs...) - ret0, _ := ret[0].(*Nodes) + ret := m.ctrl.Call(m, "BulkDelete", varargs...) + ret0, _ := ret[0].(*BulkDeleteResponse) ret1, _ := ret[1].(error) return ret0, ret1 } -// List indicates an expected call of List. -func (mr *MockNodesServiceClientMockRecorder) List(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// BulkDelete indicates an expected call of BulkDelete +func (mr *MockNodesServiceClientMockRecorder) BulkDelete(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockNodesServiceClient)(nil).List), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BulkDelete", reflect.TypeOf((*MockNodesServiceClient)(nil).BulkDelete), varargs...) } -// Read mocks base method. -func (m *MockNodesServiceClient) Read(ctx context.Context, in *Id, opts ...grpc.CallOption) (*Node, error) { +// BulkCreate mocks base method +func (m *MockNodesServiceClient) BulkCreate(ctx context.Context, in *Nodes, opts ...grpc.CallOption) (*Ids, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "Read", varargs...) - ret0, _ := ret[0].(*Node) + ret := m.ctrl.Call(m, "BulkCreate", varargs...) + ret0, _ := ret[0].(*Ids) ret1, _ := ret[1].(error) return ret0, ret1 } -// Read indicates an expected call of Read. -func (mr *MockNodesServiceClientMockRecorder) Read(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// BulkCreate indicates an expected call of BulkCreate +func (mr *MockNodesServiceClientMockRecorder) BulkCreate(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Read", reflect.TypeOf((*MockNodesServiceClient)(nil).Read), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BulkCreate", reflect.TypeOf((*MockNodesServiceClient)(nil).BulkCreate), varargs...) } -// Update mocks base method. -func (m *MockNodesServiceClient) Update(ctx context.Context, in *Node, opts ...grpc.CallOption) (*emptypb.Empty, error) { +// UpdateNodeDetectInfo mocks base method +func (m *MockNodesServiceClient) UpdateNodeDetectInfo(ctx context.Context, in *NodeDetectJobInfo, opts ...grpc.CallOption) (*emptypb.Empty, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "Update", varargs...) + ret := m.ctrl.Call(m, "UpdateNodeDetectInfo", varargs...) ret0, _ := ret[0].(*emptypb.Empty) ret1, _ := ret[1].(error) return ret0, ret1 } -// Update indicates an expected call of Update. -func (mr *MockNodesServiceClientMockRecorder) Update(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// UpdateNodeDetectInfo indicates an expected call of UpdateNodeDetectInfo +func (mr *MockNodesServiceClientMockRecorder) UpdateNodeDetectInfo(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockNodesServiceClient)(nil).Update), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNodeDetectInfo", reflect.TypeOf((*MockNodesServiceClient)(nil).UpdateNodeDetectInfo), varargs...) } -// UpdateNodeConnectionError mocks base method. +// UpdateNodeConnectionError mocks base method func (m *MockNodesServiceClient) UpdateNodeConnectionError(ctx context.Context, in *NodeError, opts ...grpc.CallOption) (*emptypb.Empty, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} @@ -209,117 +208,102 @@ func (m *MockNodesServiceClient) UpdateNodeConnectionError(ctx context.Context, return ret0, ret1 } -// UpdateNodeConnectionError indicates an expected call of UpdateNodeConnectionError. +// UpdateNodeConnectionError indicates an expected call of UpdateNodeConnectionError func (mr *MockNodesServiceClientMockRecorder) UpdateNodeConnectionError(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNodeConnectionError", reflect.TypeOf((*MockNodesServiceClient)(nil).UpdateNodeConnectionError), varargs...) } -// UpdateNodeDetectInfo mocks base method. -func (m *MockNodesServiceClient) UpdateNodeDetectInfo(ctx context.Context, in *NodeDetectJobInfo, opts ...grpc.CallOption) (*emptypb.Empty, error) { +// BulkDeleteById mocks base method +func (m *MockNodesServiceClient) BulkDeleteById(ctx context.Context, in *Ids, opts ...grpc.CallOption) (*BulkDeleteResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{ctx, in} for _, a := range opts { varargs = append(varargs, a) } - ret := m.ctrl.Call(m, "UpdateNodeDetectInfo", varargs...) - ret0, _ := ret[0].(*emptypb.Empty) + ret := m.ctrl.Call(m, "BulkDeleteById", varargs...) + ret0, _ := ret[0].(*BulkDeleteResponse) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateNodeDetectInfo indicates an expected call of UpdateNodeDetectInfo. -func (mr *MockNodesServiceClientMockRecorder) UpdateNodeDetectInfo(ctx, in interface{}, opts ...interface{}) *gomock.Call { +// BulkDeleteById indicates an expected call of BulkDeleteById +func (mr *MockNodesServiceClientMockRecorder) BulkDeleteById(ctx, in interface{}, opts ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNodeDetectInfo", reflect.TypeOf((*MockNodesServiceClient)(nil).UpdateNodeDetectInfo), varargs...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BulkDeleteById", reflect.TypeOf((*MockNodesServiceClient)(nil).BulkDeleteById), varargs...) } -// MockNodesServiceServer is a mock of NodesServiceServer interface. +// MockNodesServiceServer is a mock of NodesServiceServer interface type MockNodesServiceServer struct { ctrl *gomock.Controller recorder *MockNodesServiceServerMockRecorder } -// MockNodesServiceServerMockRecorder is the mock recorder for MockNodesServiceServer. +// MockNodesServiceServerMockRecorder is the mock recorder for MockNodesServiceServer type MockNodesServiceServerMockRecorder struct { mock *MockNodesServiceServer } -// NewMockNodesServiceServer creates a new mock instance. +// NewMockNodesServiceServer creates a new mock instance func NewMockNodesServiceServer(ctrl *gomock.Controller) *MockNodesServiceServer { mock := &MockNodesServiceServer{ctrl: ctrl} mock.recorder = &MockNodesServiceServerMockRecorder{mock} return mock } -// EXPECT returns an object that allows the caller to indicate expected use. +// EXPECT returns an object that allows the caller to indicate expected use func (m *MockNodesServiceServer) EXPECT() *MockNodesServiceServerMockRecorder { return m.recorder } -// BulkCreate mocks base method. -func (m *MockNodesServiceServer) BulkCreate(arg0 context.Context, arg1 *Nodes) (*Ids, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "BulkCreate", arg0, arg1) - ret0, _ := ret[0].(*Ids) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// BulkCreate indicates an expected call of BulkCreate. -func (mr *MockNodesServiceServerMockRecorder) BulkCreate(arg0, arg1 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BulkCreate", reflect.TypeOf((*MockNodesServiceServer)(nil).BulkCreate), arg0, arg1) -} - -// BulkDelete mocks base method. -func (m *MockNodesServiceServer) BulkDelete(arg0 context.Context, arg1 *Query) (*BulkDeleteResponse, error) { +// Create mocks base method +func (m *MockNodesServiceServer) Create(arg0 context.Context, arg1 *Node) (*Id, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "BulkDelete", arg0, arg1) - ret0, _ := ret[0].(*BulkDeleteResponse) + ret := m.ctrl.Call(m, "Create", arg0, arg1) + ret0, _ := ret[0].(*Id) ret1, _ := ret[1].(error) return ret0, ret1 } -// BulkDelete indicates an expected call of BulkDelete. -func (mr *MockNodesServiceServerMockRecorder) BulkDelete(arg0, arg1 interface{}) *gomock.Call { +// Create indicates an expected call of Create +func (mr *MockNodesServiceServerMockRecorder) Create(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BulkDelete", reflect.TypeOf((*MockNodesServiceServer)(nil).BulkDelete), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockNodesServiceServer)(nil).Create), arg0, arg1) } -// BulkDeleteById mocks base method. -func (m *MockNodesServiceServer) BulkDeleteById(arg0 context.Context, arg1 *Ids) (*BulkDeleteResponse, error) { +// Read mocks base method +func (m *MockNodesServiceServer) Read(arg0 context.Context, arg1 *Id) (*Node, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "BulkDeleteById", arg0, arg1) - ret0, _ := ret[0].(*BulkDeleteResponse) + ret := m.ctrl.Call(m, "Read", arg0, arg1) + ret0, _ := ret[0].(*Node) ret1, _ := ret[1].(error) return ret0, ret1 } -// BulkDeleteById indicates an expected call of BulkDeleteById. -func (mr *MockNodesServiceServerMockRecorder) BulkDeleteById(arg0, arg1 interface{}) *gomock.Call { +// Read indicates an expected call of Read +func (mr *MockNodesServiceServerMockRecorder) Read(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BulkDeleteById", reflect.TypeOf((*MockNodesServiceServer)(nil).BulkDeleteById), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Read", reflect.TypeOf((*MockNodesServiceServer)(nil).Read), arg0, arg1) } -// Create mocks base method. -func (m *MockNodesServiceServer) Create(arg0 context.Context, arg1 *Node) (*Id, error) { +// Update mocks base method +func (m *MockNodesServiceServer) Update(arg0 context.Context, arg1 *Node) (*emptypb.Empty, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Create", arg0, arg1) - ret0, _ := ret[0].(*Id) + ret := m.ctrl.Call(m, "Update", arg0, arg1) + ret0, _ := ret[0].(*emptypb.Empty) ret1, _ := ret[1].(error) return ret0, ret1 } -// Create indicates an expected call of Create. -func (mr *MockNodesServiceServerMockRecorder) Create(arg0, arg1 interface{}) *gomock.Call { +// Update indicates an expected call of Update +func (mr *MockNodesServiceServerMockRecorder) Update(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockNodesServiceServer)(nil).Create), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockNodesServiceServer)(nil).Update), arg0, arg1) } -// Delete mocks base method. +// Delete mocks base method func (m *MockNodesServiceServer) Delete(arg0 context.Context, arg1 *Id) (*emptypb.Empty, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Delete", arg0, arg1) @@ -328,13 +312,13 @@ func (m *MockNodesServiceServer) Delete(arg0 context.Context, arg1 *Id) (*emptyp return ret0, ret1 } -// Delete indicates an expected call of Delete. +// Delete indicates an expected call of Delete func (mr *MockNodesServiceServerMockRecorder) Delete(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockNodesServiceServer)(nil).Delete), arg0, arg1) } -// List mocks base method. +// List mocks base method func (m *MockNodesServiceServer) List(arg0 context.Context, arg1 *Query) (*Nodes, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "List", arg0, arg1) @@ -343,43 +327,58 @@ func (m *MockNodesServiceServer) List(arg0 context.Context, arg1 *Query) (*Nodes return ret0, ret1 } -// List indicates an expected call of List. +// List indicates an expected call of List func (mr *MockNodesServiceServerMockRecorder) List(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockNodesServiceServer)(nil).List), arg0, arg1) } -// Read mocks base method. -func (m *MockNodesServiceServer) Read(arg0 context.Context, arg1 *Id) (*Node, error) { +// BulkDelete mocks base method +func (m *MockNodesServiceServer) BulkDelete(arg0 context.Context, arg1 *Query) (*BulkDeleteResponse, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Read", arg0, arg1) - ret0, _ := ret[0].(*Node) + ret := m.ctrl.Call(m, "BulkDelete", arg0, arg1) + ret0, _ := ret[0].(*BulkDeleteResponse) ret1, _ := ret[1].(error) return ret0, ret1 } -// Read indicates an expected call of Read. -func (mr *MockNodesServiceServerMockRecorder) Read(arg0, arg1 interface{}) *gomock.Call { +// BulkDelete indicates an expected call of BulkDelete +func (mr *MockNodesServiceServerMockRecorder) BulkDelete(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Read", reflect.TypeOf((*MockNodesServiceServer)(nil).Read), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BulkDelete", reflect.TypeOf((*MockNodesServiceServer)(nil).BulkDelete), arg0, arg1) } -// Update mocks base method. -func (m *MockNodesServiceServer) Update(arg0 context.Context, arg1 *Node) (*emptypb.Empty, error) { +// BulkCreate mocks base method +func (m *MockNodesServiceServer) BulkCreate(arg0 context.Context, arg1 *Nodes) (*Ids, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Update", arg0, arg1) + ret := m.ctrl.Call(m, "BulkCreate", arg0, arg1) + ret0, _ := ret[0].(*Ids) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// BulkCreate indicates an expected call of BulkCreate +func (mr *MockNodesServiceServerMockRecorder) BulkCreate(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BulkCreate", reflect.TypeOf((*MockNodesServiceServer)(nil).BulkCreate), arg0, arg1) +} + +// UpdateNodeDetectInfo mocks base method +func (m *MockNodesServiceServer) UpdateNodeDetectInfo(arg0 context.Context, arg1 *NodeDetectJobInfo) (*emptypb.Empty, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "UpdateNodeDetectInfo", arg0, arg1) ret0, _ := ret[0].(*emptypb.Empty) ret1, _ := ret[1].(error) return ret0, ret1 } -// Update indicates an expected call of Update. -func (mr *MockNodesServiceServerMockRecorder) Update(arg0, arg1 interface{}) *gomock.Call { +// UpdateNodeDetectInfo indicates an expected call of UpdateNodeDetectInfo +func (mr *MockNodesServiceServerMockRecorder) UpdateNodeDetectInfo(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockNodesServiceServer)(nil).Update), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNodeDetectInfo", reflect.TypeOf((*MockNodesServiceServer)(nil).UpdateNodeDetectInfo), arg0, arg1) } -// UpdateNodeConnectionError mocks base method. +// UpdateNodeConnectionError mocks base method func (m *MockNodesServiceServer) UpdateNodeConnectionError(arg0 context.Context, arg1 *NodeError) (*emptypb.Empty, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "UpdateNodeConnectionError", arg0, arg1) @@ -388,23 +387,23 @@ func (m *MockNodesServiceServer) UpdateNodeConnectionError(arg0 context.Context, return ret0, ret1 } -// UpdateNodeConnectionError indicates an expected call of UpdateNodeConnectionError. +// UpdateNodeConnectionError indicates an expected call of UpdateNodeConnectionError func (mr *MockNodesServiceServerMockRecorder) UpdateNodeConnectionError(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNodeConnectionError", reflect.TypeOf((*MockNodesServiceServer)(nil).UpdateNodeConnectionError), arg0, arg1) } -// UpdateNodeDetectInfo mocks base method. -func (m *MockNodesServiceServer) UpdateNodeDetectInfo(arg0 context.Context, arg1 *NodeDetectJobInfo) (*emptypb.Empty, error) { +// BulkDeleteById mocks base method +func (m *MockNodesServiceServer) BulkDeleteById(arg0 context.Context, arg1 *Ids) (*BulkDeleteResponse, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateNodeDetectInfo", arg0, arg1) - ret0, _ := ret[0].(*emptypb.Empty) + ret := m.ctrl.Call(m, "BulkDeleteById", arg0, arg1) + ret0, _ := ret[0].(*BulkDeleteResponse) ret1, _ := ret[1].(error) return ret0, ret1 } -// UpdateNodeDetectInfo indicates an expected call of UpdateNodeDetectInfo. -func (mr *MockNodesServiceServerMockRecorder) UpdateNodeDetectInfo(arg0, arg1 interface{}) *gomock.Call { +// BulkDeleteById indicates an expected call of BulkDeleteById +func (mr *MockNodesServiceServerMockRecorder) BulkDeleteById(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNodeDetectInfo", reflect.TypeOf((*MockNodesServiceServer)(nil).UpdateNodeDetectInfo), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BulkDeleteById", reflect.TypeOf((*MockNodesServiceServer)(nil).BulkDeleteById), arg0, arg1) } diff --git a/components/automate-gateway/api/migrations.pb.swagger.go b/components/automate-gateway/api/migrations.pb.swagger.go deleted file mode 100644 index 938d563dc49..00000000000 --- a/components/automate-gateway/api/migrations.pb.swagger.go +++ /dev/null @@ -1,54 +0,0 @@ -package api - -func init() { - Swagger.Add("migrations", `{ - "swagger": "2.0", - "info": { - "title": "external/infra_proxy/migrations.proto", - "version": "version not set" - }, - "consumes": [ - "application/json" - ], - "produces": [ - "application/json" - ], - "paths": {}, - "definitions": { - "google.protobuf.Any": { - "type": "object", - "properties": { - "type_url": { - "type": "string" - }, - "value": { - "type": "string", - "format": "byte" - } - } - }, - "grpc.gateway.runtime.Error": { - "type": "object", - "properties": { - "error": { - "type": "string" - }, - "code": { - "type": "integer", - "format": "int32" - }, - "message": { - "type": "string" - }, - "details": { - "type": "array", - "items": { - "$ref": "#/definitions/google.protobuf.Any" - } - } - } - } - } -} -`) -} diff --git a/components/automate-gateway/gateway/clients.go b/components/automate-gateway/gateway/clients.go index a1c1b36ca0b..5e6bce4cedd 100644 --- a/components/automate-gateway/gateway/clients.go +++ b/components/automate-gateway/gateway/clients.go @@ -25,6 +25,7 @@ import ( "github.com/chef/automate/api/interservice/data_lifecycle" "github.com/chef/automate/api/interservice/deployment" "github.com/chef/automate/api/interservice/event_feed" + infra_proxy_migration "github.com/chef/automate/api/interservice/infra_proxy/migrations/service" infra_proxy "github.com/chef/automate/api/interservice/infra_proxy/service" chef_ingest "github.com/chef/automate/api/interservice/ingest" "github.com/chef/automate/api/interservice/license_control" @@ -121,6 +122,7 @@ type ClientsFactory interface { DatafeedClient() (data_feed.DatafeedServiceClient, error) PurgeClient(service string) (data_lifecycle.PurgeClient, error) InfraProxyClient() (infra_proxy.InfraProxyServiceClient, error) + InfraProxyMigrationClient() (infra_proxy_migration.MigrationDataServiceClient, error) CdsClient() (cds.AutomateCdsServiceClient, error) UserSettingsClient() (user_settings.UserSettingsServiceClient, error) Close() error @@ -451,6 +453,14 @@ func (c *clientsFactory) InfraProxyClient() (infra_proxy.InfraProxyServiceClient return infra_proxy.NewInfraProxyServiceClient(conn), nil } +func (c *clientsFactory) InfraProxyMigrationClient() (infra_proxy_migration.MigrationDataServiceClient, error) { + conn, err := c.connectionByName("infra-proxy-service") + if err != nil { + return nil, err + } + return infra_proxy_migration.NewMigrationDataServiceClient(conn), nil +} + func (c *clientsFactory) UserSettingsClient() (user_settings.UserSettingsServiceClient, error) { conn, err := c.connectionByName("user-settings-service") if err != nil { diff --git a/components/automate-gateway/gateway/server.go b/components/automate-gateway/gateway/server.go index c58f9de3800..89e3ecff1c4 100644 --- a/components/automate-gateway/gateway/server.go +++ b/components/automate-gateway/gateway/server.go @@ -408,6 +408,9 @@ func (s *Server) startHTTPServer() error { // `rpc Create (stream ProfilePostRequest) returns (CheckResult) {};` mux.HandleFunc("/api/v0/compliance/profiles", s.ProfileCreateHandler) + // upload the chef infra server migration zip file + mux.HandleFunc("/api/v0/infra/servers/migrations/upload", s.UploadZipFile) + profileTarHandlerUnlessDELETE := func(w http.ResponseWriter, r *http.Request) { if r.Method != http.MethodDelete { s.ProfileTarHandler(w, r) diff --git a/components/automate-gateway/gateway/services.go b/components/automate-gateway/gateway/services.go index 60ea17dfeb0..12990a49e7e 100644 --- a/components/automate-gateway/gateway/services.go +++ b/components/automate-gateway/gateway/services.go @@ -1,6 +1,7 @@ package gateway import ( + "bufio" "bytes" "context" "encoding/json" @@ -47,6 +48,7 @@ import ( "github.com/chef/automate/api/interservice/compliance/reporting" deploy_api "github.com/chef/automate/api/interservice/deployment" inter_eventfeed_Req "github.com/chef/automate/api/interservice/event_feed" + infra_proxy "github.com/chef/automate/api/interservice/infra_proxy/migrations/request" swagger "github.com/chef/automate/components/automate-gateway/api" pb_deployment "github.com/chef/automate/components/automate-gateway/api/deployment" pb_gateway "github.com/chef/automate/components/automate-gateway/api/gateway" @@ -314,10 +316,13 @@ func (s *Server) RegisterGRPCServices(grpcServer *grpc.Server) error { pb_data_lifecycle.RegisterDataLifecycleServer(grpcServer, dataLifecycleServer) infraProxyClient, err := clients.InfraProxyClient() + + infraProxyMigrationClient, err := clients.InfraProxyMigrationClient() + if err != nil { return errors.Wrap(err, "create client for infra proxy service") } - pb_infra_proxy.RegisterInfraProxyServer(grpcServer, handler_infra_proxy.NewInfraProxyHandler(infraProxyClient)) + pb_infra_proxy.RegisterInfraProxyServer(grpcServer, handler_infra_proxy.NewInfraProxyHandler(infraProxyClient, infraProxyMigrationClient)) userSettingsClient, err := clients.UserSettingsClient() if err != nil { @@ -958,6 +963,118 @@ func (s *Server) DeploymentStatusHandler(w http.ResponseWriter, r *http.Request) } } +// UploadZipFile used to upload the migration zip file of infraProxy +func (s *Server) UploadZipFile(w http.ResponseWriter, r *http.Request) { + var cType, fileName, serverId string + var content bytes.Buffer + file, metaData, err := r.FormFile("file") + serverId = r.FormValue("serverId") + + if err != nil { + http.Error(w, err.Error(), http.StatusBadRequest) + return + } + defer file.Close() // nolint: errcheck + + _, err = io.Copy(&content, file) + if err != nil { + http.Error(w, err.Error(), http.StatusBadRequest) + return + } + contentType := strings.Split(r.Header.Get("Content-Type"), ";") + + if len(contentType) > 0 { + cType = contentType[0] + } + + // validate the content type + if cType != "multipart/form-data" { + http.Error(w, "content type not supported.", http.StatusBadRequest) + return + } + + fileName = metaData.Filename + + const ( + action = "infra:infraServers:sync" + resource = "infra:infraServers" + ) + + ctx, err := s.authRequest(r, resource, action) + if err != nil { + http.Error(w, err.Error(), http.StatusForbidden) + return + } + + infraClient, err := s.clientsFactory.InfraProxyMigrationClient() + if err != nil { + http.Error(w, "grpc service for compliance unavailable", http.StatusServiceUnavailable) + return + } + + request := infra_proxy.UploadZipFileRequest{ + ServerId: serverId, + Meta: &infra_proxy.Metadata{ + ContentType: cType, + Name: fileName, + }, + } + // call the infra proxy service function + stream, err := infraClient.UploadFile(ctx) + if err != nil { + http.Error(w, err.Error(), http.StatusInternalServerError) + return + } + + err = stream.Send(&request) + if err != nil { + http.Error(w, err.Error(), http.StatusBadRequest) + return + } + + // break the data in chunks + reader := bufio.NewReader(&content) + buffer := make([]byte, 1024) + + for { + n, err := reader.Read(buffer) + if err == io.EOF { + break + } + + if err != nil { + log.Fatal("cannot read chunk to buffer: ", err) + } + + request := infra_proxy.UploadZipFileRequest{ + ServerId: serverId, + Chunk: &infra_proxy.Chunk{Data: buffer[:n]}, + Meta: &infra_proxy.Metadata{ + ContentType: cType, + Name: fileName, + }, + } + + err = stream.Send(&request) + if err != nil { + http.Error(w, err.Error(), http.StatusBadRequest) + return + } + } + + reply, err := stream.CloseAndRecv() + if err != nil { + http.Error(w, err.Error(), http.StatusBadRequest) + return + } + data, err := json.Marshal(reply) + if err != nil { + http.Error(w, err.Error(), http.StatusInternalServerError) + return + } + w.Write(data) // nolint: errcheck +} + func init() { // Register streaming methods for introspection. // - Almost all calls of this method are in *.pb.go files, auto-generated from proto files. diff --git a/components/automate-gateway/gateway_mocks/mock_gateway/clients_mock.go b/components/automate-gateway/gateway_mocks/mock_gateway/clients_mock.go index c941a49753b..9f9ef109dd6 100644 --- a/components/automate-gateway/gateway_mocks/mock_gateway/clients_mock.go +++ b/components/automate-gateway/gateway_mocks/mock_gateway/clients_mock.go @@ -21,7 +21,8 @@ import ( data_lifecycle "github.com/chef/automate/api/interservice/data_lifecycle" deployment "github.com/chef/automate/api/interservice/deployment" event_feed "github.com/chef/automate/api/interservice/event_feed" - service1 "github.com/chef/automate/api/interservice/infra_proxy/service" + service1 "github.com/chef/automate/api/interservice/infra_proxy/migrations/service" + service2 "github.com/chef/automate/api/interservice/infra_proxy/service" ingest0 "github.com/chef/automate/api/interservice/ingest" license_control "github.com/chef/automate/api/interservice/license_control" local_user "github.com/chef/automate/api/interservice/local_user" @@ -479,10 +480,10 @@ func (mr *MockClientsFactoryMockRecorder) PurgeClient(service interface{}) *gomo } // InfraProxyClient mocks base method -func (m *MockClientsFactory) InfraProxyClient() (service1.InfraProxyServiceClient, error) { +func (m *MockClientsFactory) InfraProxyClient() (service2.InfraProxyServiceClient, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "InfraProxyClient") - ret0, _ := ret[0].(service1.InfraProxyServiceClient) + ret0, _ := ret[0].(service2.InfraProxyServiceClient) ret1, _ := ret[1].(error) return ret0, ret1 } @@ -493,6 +494,21 @@ func (mr *MockClientsFactoryMockRecorder) InfraProxyClient() *gomock.Call { return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InfraProxyClient", reflect.TypeOf((*MockClientsFactory)(nil).InfraProxyClient)) } +// InfraProxyMigrationClient mocks base method +func (m *MockClientsFactory) InfraProxyMigrationClient() (service1.MigrationDataServiceClient, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "InfraProxyMigrationClient") + ret0, _ := ret[0].(service1.MigrationDataServiceClient) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// InfraProxyMigrationClient indicates an expected call of InfraProxyMigrationClient +func (mr *MockClientsFactoryMockRecorder) InfraProxyMigrationClient() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InfraProxyMigrationClient", reflect.TypeOf((*MockClientsFactory)(nil).InfraProxyMigrationClient)) +} + // CdsClient mocks base method func (m *MockClientsFactory) CdsClient() (service.AutomateCdsServiceClient, error) { m.ctrl.T.Helper() diff --git a/components/automate-gateway/handler/infra_proxy/infra_proxy.go b/components/automate-gateway/handler/infra_proxy/infra_proxy.go index b644290016c..f04a9cf96d2 100644 --- a/components/automate-gateway/handler/infra_proxy/infra_proxy.go +++ b/components/automate-gateway/handler/infra_proxy/infra_proxy.go @@ -1,17 +1,20 @@ package infra_proxy import ( + migrations "github.com/chef/automate/api/interservice/infra_proxy/migrations/service" infra_proxy "github.com/chef/automate/api/interservice/infra_proxy/service" ) // InfraProxyServer stores client type InfraProxyServer struct { - client infra_proxy.InfraProxyServiceClient + client infra_proxy.InfraProxyServiceClient + migrationClient migrations.MigrationDataServiceClient } // NewInfraProxyHandler initializes InfraProxyServer with client -func NewInfraProxyHandler(client infra_proxy.InfraProxyServiceClient) *InfraProxyServer { +func NewInfraProxyHandler(client infra_proxy.InfraProxyServiceClient, migrationClient migrations.MigrationDataServiceClient) *InfraProxyServer { return &InfraProxyServer{ - client: client, + client: client, + migrationClient: migrationClient, } } diff --git a/components/infra-proxy-service/constants/constants.go b/components/infra-proxy-service/constants/constants.go index cbcddd2b2f0..3a334cc7dd9 100644 --- a/components/infra-proxy-service/constants/constants.go +++ b/components/infra-proxy-service/constants/constants.go @@ -18,27 +18,17 @@ const ( ) const ( - StartFileUpload MigrationPhase = iota + 1000 - CompleteFileUpload - FailedFileUpload - StartUnzip - ComplteUnzip - FailedUnzip - StartZipParsing - CompleteZipParsing - FailedZipParsing - StartOrgMigration - CompleteOrgMigration - FailedOrgMigration - StartUserMigration - CompleteUserMigration - FailedUserMigration - StartAssciation - CompleteAssciation - FailedAssciation - StartPermissionMigration - CompletePermissionMigration - FailedPermissionMigration + FileUpload MigrationPhase = iota + 1000 + FileUnzip + OrgsParsing + UsersParsing + UserAssociationParsing + PermissionParsing + CreatePreview + OrgsMigration + UserMigration + UserAssociation + PermissionMigration ) const ( diff --git a/components/infra-proxy-service/server/migrations.go b/components/infra-proxy-service/server/migrations.go new file mode 100644 index 00000000000..6af1a6b03b9 --- /dev/null +++ b/components/infra-proxy-service/server/migrations.go @@ -0,0 +1,133 @@ +package server + +import ( + "bytes" + "context" + "io" + "os" + "path" + + uuid "github.com/chef/automate/lib/uuid4" + log "github.com/sirupsen/logrus" + + "github.com/chef/automate/api/interservice/infra_proxy/migrations/response" + "github.com/chef/automate/api/interservice/infra_proxy/migrations/service" +) + +// Takes up file name from service.MigrationDataService_UploadFileServer.MigrationId and creates the file in the same directory +func saveFile(migrationId string, filename string, fileData bytes.Buffer) error { + folderPath := path.Join("/hab/svc/infra-proxy-service/data", migrationId) + err := os.Mkdir(folderPath, 0777) + if err != nil { + log.WithError(err).Error("Unable to create directory for migration id", migrationId) + return err + } + filePath := path.Join(folderPath, filename) + file, err := os.Create(filePath) + if err != nil { + log.WithError(err).Error("Unable to create zipped file for migration id", migrationId) + return err + } + _, err = fileData.WriteTo(file) + if err != nil { + return err + } + return nil + +} + +// UploadFile Takes the stream of data to upload a file +func (s *Server) UploadFile(stream service.MigrationDataService_UploadFileServer) error { + log.Info("Starting the with the request to upload file") + req, err := stream.Recv() + serverId := req.ServerId + fileName := req.GetMeta().GetName() + ctx := context.Background() + migrationId, err := createMigrationId() + if err != nil { + log.WithError(err).Error("Unable to create migration id") + res := handleErrorForUploadFileAndMigration(err, migrationId, serverId, s, ctx) + stream.SendAndClose(res) + return err + } + log.Info("Starting with migration phase with the upload filei for migration id: ", migrationId) + _, err = s.service.Migration.StartMigration(ctx, migrationId, serverId) + fileData := bytes.Buffer{} + s.service.Migration.StartFileUpload(ctx, migrationId, serverId) + for { + req, err := stream.Recv() + + if err == io.EOF { + break + } + + if err != nil { + log.Errorf("Failed to upload file for migration id %s : %s", migrationId, err.Error()) + res := handleErrorForUploadFileAndMigration(err, migrationId, serverId, s, ctx) + stream.SendAndClose(res) + return err + } + + chunk := req.GetChunk().Data + _, err = fileData.Write(chunk) + if err != nil { + log.Errorf("Failed to upload file for migration id %s : %s", migrationId, err.Error()) + res := handleErrorForUploadFileAndMigration(err, migrationId, serverId, s, ctx) + stream.SendAndClose(res) + return err + } + } + + err = saveFile(migrationId, fileName, fileData) + if err != nil { + log.Errorf("Failed to save uploaded file for migration id %s : %s", migrationId, err.Error()) + res := handleErrorForUploadFileAndMigration(err, migrationId, serverId, s, ctx) + stream.SendAndClose(res) + return err + } + log.Info("File successfully saved in the directory for the requested file for migration id: ", migrationId) + + res := &response.UploadZipFileResponse{ + MigrationId: migrationId, + Success: true, + } + s.service.Migration.CompleteFileUpload(ctx, migrationId, serverId, 0, 0, 0) + log.Info("File successfully uploaded in the directory for the requested file for migration id: ", migrationId) + err = stream.SendAndClose(res) + if err != nil { + handleErrorForUploadFileAndMigration(err, migrationId, serverId, s, ctx) + log.Errorf("Failed to send the response for migration id %s : %s", migrationId, err.Error()) + return err + } + + return nil +} + +func createMigrationId() (string, error) { + uuid, err := uuid.NewV4() + if err != nil { + log.WithError(err).Error("Failed to make a V4 UUID") + return "", err + } + return uuid.String(), nil +} + +//handleErrorForUploadFileAndMigration handles the error for the file upload +func handleErrorForUploadFileAndMigration(err error, migrationId string, serviceId string, s *Server, ctx context.Context) *response.UploadZipFileResponse { + response := createResponseWithErrors(err, migrationId) + s.service.Migration.FailedFileUpload(ctx, migrationId, serviceId, err.Error(), 0, 0, 0) + //ToDo to add the Failed migration status as well + s.service.Migration.FailedMigration(ctx, migrationId, serviceId, err.Error(), 0, 0, 0) + return response + +} + +//createResponseWithErrors created a response with errors +func createResponseWithErrors(err error, migrationId string) *response.UploadZipFileResponse { + errors := []string{err.Error()} + return &response.UploadZipFileResponse{ + Success: false, + MigrationId: migrationId, + Errors: errors, + } +} diff --git a/components/infra-proxy-service/server/server.go b/components/infra-proxy-service/server/server.go index 4e6afdfa2f3..a1fcb4642a4 100644 --- a/components/infra-proxy-service/server/server.go +++ b/components/infra-proxy-service/server/server.go @@ -13,6 +13,7 @@ import ( "net" "net/http" + grpc_migration "github.com/chef/automate/api/interservice/infra_proxy/migrations/service" grpc_s "github.com/chef/automate/api/interservice/infra_proxy/service" "github.com/chef/automate/components/infra-proxy-service/service" "github.com/chef/automate/lib/grpc/health" @@ -84,6 +85,7 @@ func NewGRPCServer(s *service.Service) *grpc.Server { ) health.RegisterHealthServer(g, health.NewService()) grpc_s.RegisterInfraProxyServiceServer(g, NewServer(s)) + grpc_migration.RegisterMigrationDataServiceServer(g, NewServer(s)) reflection.Register(g) return g } diff --git a/components/infra-proxy-service/storage/postgres/migration/sql/07_migration.up.sql b/components/infra-proxy-service/storage/postgres/migration/sql/07_migration.up.sql index 239547bc057..167a71c4525 100644 --- a/components/infra-proxy-service/storage/postgres/migration/sql/07_migration.up.sql +++ b/components/infra-proxy-service/storage/postgres/migration/sql/07_migration.up.sql @@ -29,27 +29,17 @@ CREATE INDEX IF NOT EXISTS migration_migration_id_index ON migration (migration_ -- Insert rows into migration_type INSERT INTO migration_type VALUES (100,'Migration started'), - (1000,'Upload of zip file started'), - (1001,'Upload of zip file completed'), - (1002,'Upload of zip file failed'), - (1003,'Unzip of file started'), - (1004,'Unzip of file completed'), - (1005,'Unzip of file failed'), - (1006,'Parsing of zip file started'), - (1007,'Parsing of zip file completed'), - (1008,'Parsing of zip file failed'), - (1009,'Migration of organization started'), - (1010,'Migration of organization completed'), - (1011,'Migration of organization failed'), - (1012,'Migration of users started'), - (1013,'Migration of users completed'), - (1014,'Migration of users failed'), - (1015,'Association of users to orgs started'), - (1016,'Association of users to orgs completed'), - (1017,'Association of users to orgs failed'), - (1018,'Migrating user permissions started'), - (1019,'Migrating user permissions completed'), - (1020,'Migrating user permissions failed'), + (1000,'Upload of zip file'), + (1001,'Unzip of zip file'), + (1002,'Parsing of orgs file'), + (1003,'Parsing of users file'), + (1004,'Parsing of user association file'), + (1005,'Parsing of user permissions file'), + (1006,'Creating Preview'), + (1007,'Migration of organization'), + (1008,'Migration of users'), + (1009,'Association of users to orgs'), + (1010,'Migrating user permissions'), (5000,'Migration Completed') ; diff --git a/components/infra-proxy-service/storage/postgres/migrations.go b/components/infra-proxy-service/storage/postgres/migrations.go index 312d73014e5..036575a8733 100644 --- a/components/infra-proxy-service/storage/postgres/migrations.go +++ b/components/infra-proxy-service/storage/postgres/migrations.go @@ -15,107 +15,167 @@ func (p *postgres) StartMigration(ctx context.Context, migrationId, serverId str // StartFileUpload Insert a migration entry to the DB for File upload started func (p *postgres) StartFileUpload(ctx context.Context, migrationId, serverId string) (storage.Migration, error) { - return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.StartFileUpload), int64(constants.InProgress), 0, 0, 0) + return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.FileUpload), int64(constants.InProgress), 0, 0, 0) } // CompleteFileUpload Insert a migration entry to the DB for File upload completed func (p *postgres) CompleteFileUpload(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { - return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.CompleteFileUpload), int64(constants.Completed), totalSucceeded, totalSkipped, totalFailed) + return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.FileUpload), int64(constants.Completed), totalSucceeded, totalSkipped, totalFailed) } // CompleteFileUpload Insert a migration entry to the DB for file upload failed func (p *postgres) FailedFileUpload(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { - return p.insertMigration(ctx, migrationId, serverId, message, int64(constants.FailedFileUpload), int64(constants.Failed), totalSucceeded, totalSkipped, totalFailed) + return p.insertMigration(ctx, migrationId, serverId, message, int64(constants.FileUpload), int64(constants.Failed), totalSucceeded, totalSkipped, totalFailed) } // StartUnzip Insert a migration entry to the DB for File unzip started func (p *postgres) StartUnzip(ctx context.Context, migrationId, serverId string) (storage.Migration, error) { - return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.StartUnzip), int64(constants.InProgress), 0, 0, 0) + return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.FileUnzip), int64(constants.InProgress), 0, 0, 0) } -// ComplteUnzip Insert a migration entry to the DB for File unzip completed -func (p *postgres) ComplteUnzip(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { - return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.ComplteUnzip), int64(constants.Completed), totalSucceeded, totalSkipped, totalFailed) +// CompleteUnzip Insert a migration entry to the DB for File unzip completed +func (p *postgres) CompleteUnzip(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { + return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.FileUnzip), int64(constants.Completed), totalSucceeded, totalSkipped, totalFailed) } // FailedUnzip Insert a migration entry to the DB for file unzip failed func (p *postgres) FailedUnzip(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { - return p.insertMigration(ctx, migrationId, serverId, message, int64(constants.FailedUnzip), int64(constants.Failed), totalSucceeded, totalSkipped, totalFailed) + return p.insertMigration(ctx, migrationId, serverId, message, int64(constants.FileUnzip), int64(constants.Failed), totalSucceeded, totalSkipped, totalFailed) } -// StartZipParsing Insert a migration entry to the DB for zip parsing started -func (p *postgres) StartZipParsing(ctx context.Context, migrationId, serverId string) (storage.Migration, error) { - return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.StartZipParsing), int64(constants.InProgress), 0, 0, 0) +// StartOrgsParsing Insert a migration entry to the DB for zip parsing started +func (p *postgres) StartOrgParsing(ctx context.Context, migrationId, serverId string) (storage.Migration, error) { + return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.OrgsParsing), int64(constants.InProgress), 0, 0, 0) } -// CompleteZipParsing Insert a migration entry to the DB for zip parsing completed -func (p *postgres) CompleteZipParsing(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { - return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.CompleteZipParsing), int64(constants.Completed), totalSucceeded, totalSkipped, totalFailed) +// CompleteOrgsParsing Insert a migration entry to the DB for zip parsing completed +func (p *postgres) CompleteOrgParsing(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { + return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.OrgsParsing), int64(constants.Completed), totalSucceeded, totalSkipped, totalFailed) } -// FailedZipParsing Insert a migration entry to the DB for zip parsing failed -func (p *postgres) FailedZipParsing(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { - return p.insertMigration(ctx, migrationId, serverId, message, int64(constants.FailedZipParsing), int64(constants.Failed), totalSucceeded, totalSkipped, totalFailed) +// FailedOrgsParsing Insert a migration entry to the DB for zip parsing failed +func (p *postgres) FailedOrgParsing(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { + return p.insertMigration(ctx, migrationId, serverId, message, int64(constants.OrgsParsing), int64(constants.Failed), totalSucceeded, totalSkipped, totalFailed) } -// StartOrgMigration Insert a migration entry to the DB for org migration started +// StartUsersParsing Insert a migration entry to the DB for zip parsing started +func (p *postgres) StartUsersParsing(ctx context.Context, migrationId, serverId string) (storage.Migration, error) { + return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.UsersParsing), int64(constants.InProgress), 0, 0, 0) +} + +// CompleteUsersParsing Insert a migration entry to the DB for zip parsing completed +func (p *postgres) CompleteUsersParsing(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { + return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.UsersParsing), int64(constants.Completed), totalSucceeded, totalSkipped, totalFailed) +} + +// FailedUsersParsing Insert a migration entry to the DB for zip parsing failed +func (p *postgres) FailedUsersParsing(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { + return p.insertMigration(ctx, migrationId, serverId, message, int64(constants.UsersParsing), int64(constants.Failed), totalSucceeded, totalSkipped, totalFailed) +} + +// StartUserAssociationParsing Insert a migration entry to the DB for zip parsing started +func (p *postgres) StartUserAssociationParsing(ctx context.Context, migrationId, serverId string) (storage.Migration, error) { + return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.UserAssociationParsing), int64(constants.InProgress), 0, 0, 0) +} + +// CompleteUserAssociationParsing Insert a migration entry to the DB for zip parsing completed +func (p *postgres) CompleteUserAssociationParsing(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { + return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.UserAssociationParsing), int64(constants.Completed), totalSucceeded, totalSkipped, totalFailed) +} + +// FailedUserAssociationParsing Insert a migration entry to the DB for zip parsing failed +func (p *postgres) FailedUserAssociationParsing(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { + return p.insertMigration(ctx, migrationId, serverId, message, int64(constants.UserAssociationParsing), int64(constants.Failed), totalSucceeded, totalSkipped, totalFailed) +} + +// StartPermissionParsing Insert a migration entry to the DB for zip parsing started +func (p *postgres) StartPermissionParsing(ctx context.Context, migrationId, serverId string) (storage.Migration, error) { + return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.PermissionParsing), int64(constants.InProgress), 0, 0, 0) +} + +// CompletePermissionParsing Insert a migration entry to the DB for zip parsing completed +func (p *postgres) CompletePermissionParsing(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { + return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.PermissionParsing), int64(constants.Completed), totalSucceeded, totalSkipped, totalFailed) +} + +// FailedPermissionParsing Insert a migration entry to the DB for zip parsing failed +func (p *postgres) FailedPermissionParsing(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { + return p.insertMigration(ctx, migrationId, serverId, message, int64(constants.PermissionParsing), int64(constants.Failed), totalSucceeded, totalSkipped, totalFailed) +} + +// StartCreatePreview Insert a migration entry to the DB for zip parsing started +func (p *postgres) StartCreatePreview(ctx context.Context, migrationId, serverId string) (storage.Migration, error) { + return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.CreatePreview), int64(constants.InProgress), 0, 0, 0) +} + +// CompleteCreatePreview Insert a migration entry to the DB for zip parsing completed +func (p *postgres) CompleteCreatePreview(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { + return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.CreatePreview), int64(constants.Completed), totalSucceeded, totalSkipped, totalFailed) +} + +// FailedCreatePreview Insert a migration entry to the DB for zip parsing failed +func (p *postgres) FailedCreatePreview(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { + return p.insertMigration(ctx, migrationId, serverId, message, int64(constants.CreatePreview), int64(constants.Failed), totalSucceeded, totalSkipped, totalFailed) +} + +// StartOrgsMigration Insert a migration entry to the DB for org migration started func (p *postgres) StartOrgMigration(ctx context.Context, migrationId, serverId string) (storage.Migration, error) { - return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.StartOrgMigration), int64(constants.InProgress), 0, 0, 0) + return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.OrgsMigration), int64(constants.InProgress), 0, 0, 0) } -// CompleteOrgMigration Insert a migration entry to the DB for org migration completed +// CompleteOrgsMigration Insert a migration entry to the DB for org migration completed func (p *postgres) CompleteOrgMigration(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { - return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.CompleteOrgMigration), int64(constants.Completed), totalSucceeded, totalSkipped, totalFailed) + return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.OrgsMigration), int64(constants.Completed), totalSucceeded, totalSkipped, totalFailed) } -// FailedOrgMigration Insert a migration entry to the DB for org migration failed +// FailedOrgsMigration Insert a migration entry to the DB for org migration failed func (p *postgres) FailedOrgMigration(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { - return p.insertMigration(ctx, migrationId, serverId, message, int64(constants.FailedOrgMigration), int64(constants.Failed), totalSucceeded, totalSkipped, totalFailed) + return p.insertMigration(ctx, migrationId, serverId, message, int64(constants.OrgsMigration), int64(constants.Failed), totalSucceeded, totalSkipped, totalFailed) } // StartUserMigration Insert a migration entry to the DB for user migration started func (p *postgres) StartUserMigration(ctx context.Context, migrationId, serverId string) (storage.Migration, error) { - return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.StartUserMigration), int64(constants.InProgress), 0, 0, 0) + return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.UserMigration), int64(constants.InProgress), 0, 0, 0) } // CompleteUserMigration Insert a migration entry to the DB for user migration completed func (p *postgres) CompleteUserMigration(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { - return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.CompleteUserMigration), int64(constants.Completed), totalSucceeded, totalSkipped, totalFailed) + return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.UserMigration), int64(constants.Completed), totalSucceeded, totalSkipped, totalFailed) } // FailedUserMigration Insert a migration entry to the DB for user migration failed func (p *postgres) FailedUserMigration(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { - return p.insertMigration(ctx, migrationId, serverId, message, int64(constants.FailedUserMigration), int64(constants.Failed), totalSucceeded, totalSkipped, totalFailed) + return p.insertMigration(ctx, migrationId, serverId, message, int64(constants.UserMigration), int64(constants.Failed), totalSucceeded, totalSkipped, totalFailed) } -// StartAssciation Insert a migration entry to the DB for Association of users to orgs started -func (p *postgres) StartAssciation(ctx context.Context, migrationId, serverId string) (storage.Migration, error) { - return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.StartAssciation), int64(constants.InProgress), 0, 0, 0) +// StartAssociation Insert a migration entry to the DB for Association of users to orgs started +func (p *postgres) StartAssociation(ctx context.Context, migrationId, serverId string) (storage.Migration, error) { + return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.UserAssociation), int64(constants.InProgress), 0, 0, 0) } -// CompleteAssciation Insert a migration entry to the DB for Association of users to orgs completed -func (p *postgres) CompleteAssciation(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { - return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.CompleteAssciation), int64(constants.Completed), totalSucceeded, totalSkipped, totalFailed) +// CompleteAssociation Insert a migration entry to the DB for Association of users to orgs completed +func (p *postgres) CompleteAssociation(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { + return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.UserAssociation), int64(constants.Completed), totalSucceeded, totalSkipped, totalFailed) } -// FailedAssciation Insert a migration entry to the DB for Association of users to orgs failed -func (p *postgres) FailedAssciation(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { - return p.insertMigration(ctx, migrationId, serverId, message, int64(constants.FailedAssciation), int64(constants.Failed), totalSucceeded, totalSkipped, totalFailed) +// FailedAssociation Insert a migration entry to the DB for Association of users to orgs failed +func (p *postgres) FailedAssociation(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { + return p.insertMigration(ctx, migrationId, serverId, message, int64(constants.UserAssociation), int64(constants.Failed), totalSucceeded, totalSkipped, totalFailed) } // StartPermissionMigration Insert a migration entry to the DB for migrating user permissions started func (p *postgres) StartPermissionMigration(ctx context.Context, migrationId, serverId string) (storage.Migration, error) { - return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.StartPermissionMigration), int64(constants.InProgress), 0, 0, 0) + return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.PermissionMigration), int64(constants.InProgress), 0, 0, 0) } // CompletePermissionMigration Insert a migration entry to the DB for migrating user permissions completed func (p *postgres) CompletePermissionMigration(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { - return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.CompletePermissionMigration), int64(constants.Completed), totalSucceeded, totalSkipped, totalFailed) + return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.PermissionMigration), int64(constants.Completed), totalSucceeded, totalSkipped, totalFailed) } // FailedPermissionMigration Insert a migration entry to the DB for migrating user permissions failed func (p *postgres) FailedPermissionMigration(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { - return p.insertMigration(ctx, migrationId, serverId, message, int64(constants.FailedPermissionMigration), int64(constants.Failed), totalSucceeded, totalSkipped, totalFailed) + return p.insertMigration(ctx, migrationId, serverId, message, int64(constants.PermissionMigration), int64(constants.Failed), totalSucceeded, totalSkipped, totalFailed) } // CompleteMigration Insert a migration entry to the DB for migration completed @@ -123,6 +183,11 @@ func (p *postgres) CompleteMigration(ctx context.Context, migrationId, serverId return p.insertMigration(ctx, migrationId, serverId, "", int64(constants.CompleteMigration), int64(constants.Completed), totalSucceeded, totalSkipped, totalFailed) } +// CompleteMigration Insert a migration entry to the DB for migration completed +func (p *postgres) FailedMigration(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { + return p.insertMigration(ctx, migrationId, serverId, message, int64(constants.CompleteMigration), int64(constants.Failed), totalSucceeded, totalSkipped, totalFailed) +} + //TODO: Optimize the function parameters using variadic //insertMigration Inserts an entry to the DB func (p *postgres) insertMigration(ctx context.Context, migrationId, serverId, message string, migTypeId, migStatusId, totalSucceeded, totalSkipped, totalFailed int64) (storage.Migration, error) { @@ -160,7 +225,7 @@ func (p *postgres) GetActiveMigration(ctx context.Context, serverId string) (sto } //StoreMigrationStage Inserts an entry to the migration_stage -// To use this function, make sure that you should pass the searialized parsed data in []byte +// To use this function, make sure that you should pass the serialized parsed data in []byte func (p *postgres) StoreMigrationStage(ctx context.Context, migrationId string, parsedData interface{}) (storage.MigrationStage, error) { return p.insertMigrationStage(ctx, migrationId, parsedData) } diff --git a/components/infra-proxy-service/storage/storage.go b/components/infra-proxy-service/storage/storage.go index c7193d893e1..b7ec1937029 100644 --- a/components/infra-proxy-service/storage/storage.go +++ b/components/infra-proxy-service/storage/storage.go @@ -35,24 +35,37 @@ type MigrationStorage interface { CompleteFileUpload(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) FailedFileUpload(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) StartUnzip(ctx context.Context, migrationId, serverId string) (Migration, error) - ComplteUnzip(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) + CompleteUnzip(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) FailedUnzip(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) - StartZipParsing(ctx context.Context, migrationId, serverId string) (Migration, error) - CompleteZipParsing(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) - FailedZipParsing(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) + StartOrgParsing(ctx context.Context, migrationId, serverId string) (Migration, error) + CompleteOrgParsing(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) + FailedOrgParsing(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) + StartUsersParsing(ctx context.Context, migrationId, serverId string) (Migration, error) + CompleteUsersParsing(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) + FailedUsersParsing(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) + StartUserAssociationParsing(ctx context.Context, migrationId, serverId string) (Migration, error) + CompleteUserAssociationParsing(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) + FailedUserAssociationParsing(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) + StartPermissionParsing(ctx context.Context, migrationId, serverId string) (Migration, error) + CompletePermissionParsing(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) + FailedPermissionParsing(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) + StartCreatePreview(ctx context.Context, migrationId, serverId string) (Migration, error) + CompleteCreatePreview(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) + FailedCreatePreview(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) StartOrgMigration(ctx context.Context, migrationId, serverId string) (Migration, error) CompleteOrgMigration(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) FailedOrgMigration(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) StartUserMigration(ctx context.Context, migrationId, serverId string) (Migration, error) CompleteUserMigration(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) FailedUserMigration(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) - StartAssciation(ctx context.Context, migrationId, serverId string) (Migration, error) - CompleteAssciation(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) - FailedAssciation(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) + StartAssociation(ctx context.Context, migrationId, serverId string) (Migration, error) + CompleteAssociation(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) + FailedAssociation(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) StartPermissionMigration(ctx context.Context, migrationId, serverId string) (Migration, error) CompletePermissionMigration(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) FailedPermissionMigration(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) CompleteMigration(ctx context.Context, migrationId, serverId string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) + FailedMigration(ctx context.Context, migrationId, serverId, message string, totalSucceeded, totalSkipped, totalFailed int64) (Migration, error) GetActiveMigration(ctx context.Context, serverId string) (ActiveMigration, error) StoreMigrationStage(ctx context.Context, migrationId string, parsedData interface{}) (MigrationStage, error) GetMigrationStage(ctx context.Context, migrationId string) (MigrationStage, error) diff --git a/ec2/Vagrantfile b/ec2/Vagrantfile index 609f4ce015d..407ae9f072d 100644 --- a/ec2/Vagrantfile +++ b/ec2/Vagrantfile @@ -94,13 +94,13 @@ if ['up', 'ssh', 'destroy', 'halt'].include?(ARGV[0]) aws_profile ||= 'chef-engineering' aws_creds_file = "#{ENV['HOME']}/.aws/credentials" - if find_executable('okta_aws') + if find_executable('saml2aws') puts " * okta_aws command detected, using it to refresh the temporary AWS credentials..." - unless File.exist?("#{ENV['HOME']}/.okta_aws.toml") - raise "#{ENV['HOME']}/.okta_aws.toml is not defined, cannot continue. Please read README.md for an example and usage details." + unless File.exist?("#{ENV['HOME']}/.saml2aws") + raise "#{ENV['HOME']}/.saml2aws is not defined, cannot continue. Please read README.md for an example and usage details." end puts " * You might be prompted for your Okta password now..." - `okta_aws "#{aws_profile}"` + `saml2aws login --profile "#{aws_profile}" --skip-prompt` end unless File.exist?(aws_creds_file)