From 2a6723c48a5f1ed46298fc8a215a13d57287a244 Mon Sep 17 00:00:00 2001 From: Jarrett Spiker Date: Wed, 6 Mar 2024 14:24:29 -0500 Subject: [PATCH] Update mocks --- mocks/organization_mocks.go | 19 +++++++++++++++++-- mocks/workspace_mocks.go | 19 +++++++++++++++++-- 2 files changed, 34 insertions(+), 4 deletions(-) diff --git a/mocks/organization_mocks.go b/mocks/organization_mocks.go index e00c2a177..8b0748167 100644 --- a/mocks/organization_mocks.go +++ b/mocks/organization_mocks.go @@ -124,10 +124,10 @@ func (mr *MockOrganizationsMockRecorder) ReadCapacity(ctx, organization interfac } // ReadDataRetentionPolicy mocks base method. -func (m *MockOrganizations) ReadDataRetentionPolicy(ctx context.Context, organization string) (*tfe.DataRetentionPolicyChoice, error) { +func (m *MockOrganizations) ReadDataRetentionPolicy(ctx context.Context, organization string) (*tfe.DataRetentionPolicy, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ReadDataRetentionPolicy", ctx, organization) - ret0, _ := ret[0].(*tfe.DataRetentionPolicyChoice) + ret0, _ := ret[0].(*tfe.DataRetentionPolicy) ret1, _ := ret[1].(error) return ret0, ret1 } @@ -138,6 +138,21 @@ func (mr *MockOrganizationsMockRecorder) ReadDataRetentionPolicy(ctx, organizati return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadDataRetentionPolicy", reflect.TypeOf((*MockOrganizations)(nil).ReadDataRetentionPolicy), ctx, organization) } +// ReadDataRetentionPolicyV2 mocks base method. +func (m *MockOrganizations) ReadDataRetentionPolicyV2(ctx context.Context, organization string) (*tfe.DataRetentionPolicyChoice, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadDataRetentionPolicyV2", ctx, organization) + ret0, _ := ret[0].(*tfe.DataRetentionPolicyChoice) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ReadDataRetentionPolicyV2 indicates an expected call of ReadDataRetentionPolicyV2. +func (mr *MockOrganizationsMockRecorder) ReadDataRetentionPolicyV2(ctx, organization interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadDataRetentionPolicyV2", reflect.TypeOf((*MockOrganizations)(nil).ReadDataRetentionPolicyV2), ctx, organization) +} + // ReadEntitlements mocks base method. func (m *MockOrganizations) ReadEntitlements(ctx context.Context, organization string) (*tfe.Entitlements, error) { m.ctrl.T.Helper() diff --git a/mocks/workspace_mocks.go b/mocks/workspace_mocks.go index a8d00ed81..3da66d42a 100644 --- a/mocks/workspace_mocks.go +++ b/mocks/workspace_mocks.go @@ -257,10 +257,10 @@ func (mr *MockWorkspacesMockRecorder) ReadByIDWithOptions(ctx, workspaceID, opti } // ReadDataRetentionPolicy mocks base method. -func (m *MockWorkspaces) ReadDataRetentionPolicy(ctx context.Context, workspaceID string) (*tfe.DataRetentionPolicyChoice, error) { +func (m *MockWorkspaces) ReadDataRetentionPolicy(ctx context.Context, workspaceID string) (*tfe.DataRetentionPolicy, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ReadDataRetentionPolicy", ctx, workspaceID) - ret0, _ := ret[0].(*tfe.DataRetentionPolicyChoice) + ret0, _ := ret[0].(*tfe.DataRetentionPolicy) ret1, _ := ret[1].(error) return ret0, ret1 } @@ -271,6 +271,21 @@ func (mr *MockWorkspacesMockRecorder) ReadDataRetentionPolicy(ctx, workspaceID i return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadDataRetentionPolicy", reflect.TypeOf((*MockWorkspaces)(nil).ReadDataRetentionPolicy), ctx, workspaceID) } +// ReadDataRetentionPolicyV2 mocks base method. +func (m *MockWorkspaces) ReadDataRetentionPolicyV2(ctx context.Context, workspaceID string) (*tfe.DataRetentionPolicyChoice, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadDataRetentionPolicyV2", ctx, workspaceID) + ret0, _ := ret[0].(*tfe.DataRetentionPolicyChoice) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ReadDataRetentionPolicyV2 indicates an expected call of ReadDataRetentionPolicyV2. +func (mr *MockWorkspacesMockRecorder) ReadDataRetentionPolicyV2(ctx, workspaceID interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadDataRetentionPolicyV2", reflect.TypeOf((*MockWorkspaces)(nil).ReadDataRetentionPolicyV2), ctx, workspaceID) +} + // ReadWithOptions mocks base method. func (m *MockWorkspaces) ReadWithOptions(ctx context.Context, organization, workspace string, options *tfe.WorkspaceReadOptions) (*tfe.Workspace, error) { m.ctrl.T.Helper()