atlas

type AdvancedClustersClientMock struct {
	ListFunc     func(projectID string) (*mongodbatlas.AdvancedClustersResponse, *mongodbatlas.Response, error)
	ListRequests map[string]struct{}

	GetFunc     func(projectID string, clusterName string) (*mongodbatlas.AdvancedCluster, *mongodbatlas.Response, error)
	GetRequests map[string]struct{}

	CreateFunc     func(projectID string, cluster *mongodbatlas.AdvancedCluster) (*mongodbatlas.AdvancedCluster, *mongodbatlas.Response, error)
	CreateRequests map[string]*mongodbatlas.AdvancedCluster

	UpdateFunc     func(projectID string, clusterName string, cluster *mongodbatlas.AdvancedCluster) (*mongodbatlas.AdvancedCluster, *mongodbatlas.Response, error)
	UpdateRequests map[string]*mongodbatlas.AdvancedCluster

	DeleteFunc     func(projectID string, clusterName string) (*mongodbatlas.Response, error)
	DeleteRequests map[string]struct{}

	TestFailoverFunc     func(projectID string, clusterName string) (*mongodbatlas.Response, error)
	TestFailoverRequests map[string]struct{}
}

type AlertConfigurationsMock struct {
	CreateFunc     func(projectID string, alertConfig *mongodbatlas.AlertConfiguration) (*mongodbatlas.AlertConfiguration, *mongodbatlas.Response, error)
	CreateRequests map[string]*mongodbatlas.AlertConfiguration

	EnableAnAlertConfigFunc     func(projectID string, alertConfigID string, enabled *bool) (*mongodbatlas.AlertConfiguration, *mongodbatlas.Response, error)
	EnableAnAlertConfigRequests map[string]*bool

	GetAnAlertConfigFunc     func(projectID string, alertConfigID string) (*mongodbatlas.AlertConfiguration, *mongodbatlas.Response, error)
	GetAnAlertConfigRequests map[string]struct{}

	GetOpenAlertsConfigFunc     func(projectID string, alertConfigID string) ([]mongodbatlas.AlertConfiguration, *mongodbatlas.Response, error)
	GetOpenAlertsConfigRequests map[string]struct{}

	ListFunc     func(projectID string) ([]mongodbatlas.AlertConfiguration, *mongodbatlas.Response, error)
	ListRequests map[string]struct{}

	ListMatcherFieldsFunc  func() ([]string, *mongodbatlas.Response, error)
	ListMatcherFieldsCalls int

	UpdateFunc     func(projectID string, alertConfigID string, alertConfig *mongodbatlas.AlertConfiguration) (*mongodbatlas.AlertConfiguration, *mongodbatlas.Response, error)
	UpdateRequests map[string]*mongodbatlas.AlertConfiguration

	DeleteFunc     func(projectID string, alertConfigID string) (*mongodbatlas.Response, error)
	DeleteRequests map[string]struct{}
}

type AuditingClientMock struct {
	GetFunc     func(projectID string) (*mongodbatlas.Auditing, *mongodbatlas.Response, error)
	GetRequests map[string]struct{}

	ConfigureFunc     func(projectID string, auditing *mongodbatlas.Auditing) (*mongodbatlas.Auditing, *mongodbatlas.Response, error)
	ConfigureRequests map[string]*mongodbatlas.Auditing
}

type CloudProviderAccessClientMock struct {
	ListRolesFunc     func(projectID string) (*mongodbatlas.CloudProviderAccessRoles, *mongodbatlas.Response, error)
	ListRolesRequests map[string]struct{}

	GetRoleFunc     func(projectID string, roleID string) (*mongodbatlas.CloudProviderAccessRole, *mongodbatlas.Response, error)
	GetRoleRequests map[string]struct{}

	CreateRoleFunc     func(projectID string, cpa *mongodbatlas.CloudProviderAccessRoleRequest) (*mongodbatlas.CloudProviderAccessRole, *mongodbatlas.Response, error)
	CreateRoleRequests map[string]*mongodbatlas.CloudProviderAccessRoleRequest

	AuthorizeRoleFunc     func(projectID, roleID string, cpa *mongodbatlas.CloudProviderAccessRoleRequest) (*mongodbatlas.CloudProviderAccessRole, *mongodbatlas.Response, error)
	AuthorizeRoleRequests map[string]*mongodbatlas.CloudProviderAccessRoleRequest

	DeauthorizeRoleFunc     func(cpa *mongodbatlas.CloudProviderDeauthorizationRequest) (*mongodbatlas.Response, error)
	DeauthorizeRoleRequests []*mongodbatlas.CloudProviderDeauthorizationRequest
}

type CloudProviderSnapshotBackupPoliciesClientMock struct {
	GetFunc     func(projectID string, clusterName string) (*mongodbatlas.CloudProviderSnapshotBackupPolicy, *mongodbatlas.Response, error)
	GetRequests map[string]struct{}

	UpdateFunc     func(projectID string, clusterName string, backup *mongodbatlas.CloudProviderSnapshotBackupPolicy) (*mongodbatlas.CloudProviderSnapshotBackupPolicy, *mongodbatlas.Response, error)
	UpdateRequests map[string]*mongodbatlas.CloudProviderSnapshotBackupPolicy

	DeleteFunc     func(projectID string, clusterName string) (*mongodbatlas.CloudProviderSnapshotBackupPolicy, *mongodbatlas.Response, error)
	DeleteRequests map[string]struct{}
}

type ContainerClientMock struct {
	ListFunc     func(projectID string) ([]mongodbatlas.Container, *mongodbatlas.Response, error)
	ListRequests map[string]struct{}

	ListAllFunc     func(projectID string) ([]mongodbatlas.Container, *mongodbatlas.Response, error)
	ListAllRequests map[string]struct{}

	GetFunc     func(projectID string, containerID string) (*mongodbatlas.Container, *mongodbatlas.Response, error)
	GetRequests map[string]struct{}

	CreateFunc     func(projectID string, container *mongodbatlas.Container) (*mongodbatlas.Container, *mongodbatlas.Response, error)
	CreateRequests map[string]*mongodbatlas.Container

	UpdateFunc     func(projectID string, containerID string, container *mongodbatlas.Container) (*mongodbatlas.Container, *mongodbatlas.Response, error)
	UpdateRequests map[string]*mongodbatlas.Container

	DeleteFunc     func(projectID string, containerID string) (*mongodbatlas.Response, error)
	DeleteRequests map[string]struct{}
}

type CustomRolesClientMock struct {
	ListFunc     func(projectID string) (*[]mongodbatlas.CustomDBRole, *mongodbatlas.Response, error)
	ListRequests map[string]struct{}

	GetFunc     func(projectID string, customRoleID string) (*mongodbatlas.CustomDBRole, *mongodbatlas.Response, error)
	GetRequests map[string]struct{}

	CreateFunc     func(projectID string, customRole *mongodbatlas.CustomDBRole) (*mongodbatlas.CustomDBRole, *mongodbatlas.Response, error)
	CreateRequests map[string]*mongodbatlas.CustomDBRole

	UpdateFunc     func(projectID string, customRoleID string, customRole *mongodbatlas.CustomDBRole) (*mongodbatlas.CustomDBRole, *mongodbatlas.Response, error)
	UpdateRequests map[string]*mongodbatlas.CustomDBRole

	DeleteFunc     func(projectID string, customRoleID string) (*mongodbatlas.Response, error)
	DeleteRequests map[string]struct{}
}

type EncryptionAtRestClientMock struct {
	CreateFunc     func(ear *mongodbatlas.EncryptionAtRest) (*mongodbatlas.EncryptionAtRest, *mongodbatlas.Response, error)
	CreateRequests []*mongodbatlas.EncryptionAtRest

	GetFunc     func(projectID string) (*mongodbatlas.EncryptionAtRest, *mongodbatlas.Response, error)
	GetRequests map[string]struct{}

	DeleteFunc     func(projectID string) (*mongodbatlas.Response, error)
	DeleteRequests map[string]struct{}
}

type GlobalClustersClientMock struct {
	GetFunc     func(projectID string, clusterName string) (*mongodbatlas.GlobalCluster, *mongodbatlas.Response, error)
	GetRequests map[string]struct{}

	AddManagedNamespaceFunc     func(projectID string, clusterName string, namespace *mongodbatlas.ManagedNamespace) (*mongodbatlas.GlobalCluster, *mongodbatlas.Response, error)
	AddManagedNamespaceRequests map[string]*mongodbatlas.ManagedNamespace

	DeleteManagedNamespaceFunc     func(projectID string, clusterName string, namespace *mongodbatlas.ManagedNamespace) (*mongodbatlas.GlobalCluster, *mongodbatlas.Response, error)
	DeleteManagedNamespaceRequests map[string]*mongodbatlas.ManagedNamespace

	AddCustomZoneMappingsFunc     func(projectID string, clusterName string, request *mongodbatlas.CustomZoneMappingsRequest) (*mongodbatlas.GlobalCluster, *mongodbatlas.Response, error)
	AddCustomZoneMappingsRequests map[string]*mongodbatlas.CustomZoneMappingsRequest

	DeleteCustomZoneMappingsFunc     func(projectID string, clusterName string) (*mongodbatlas.GlobalCluster, *mongodbatlas.Response, error)
	DeleteCustomZoneMappingsRequests map[string]struct{}
}

type IntegrationsMock struct {
	CreateFunc  func(ctx context.Context, projectID string, integrationType string, integration *mongodbatlas.ThirdPartyIntegration) (*mongodbatlas.ThirdPartyIntegrations, *mongodbatlas.Response, error)
	ReplaceFunc func(ctx context.Context, projectID string, integrationType string, integration *mongodbatlas.ThirdPartyIntegration) (*mongodbatlas.ThirdPartyIntegrations, *mongodbatlas.Response, error)
	DeleteFunc  func(ctx context.Context, projectID string, integrationType string) (*mongodbatlas.Response, error)
	GetFunc     func(ctx context.Context, projectID string, integrationType string) (*mongodbatlas.ThirdPartyIntegration, *mongodbatlas.Response, error)
	ListFunc    func(ctx context.Context, projectID string) (*mongodbatlas.ThirdPartyIntegrations, *mongodbatlas.Response, error)
}

type MaintenanceWindowClientMock struct {
	GetFunc     func(projectID string) (*mongodbatlas.MaintenanceWindow, *mongodbatlas.Response, error)
	GetRequests map[string]struct{}

	UpdateFunc     func(projectID string, maintenance *mongodbatlas.MaintenanceWindow) (*mongodbatlas.Response, error)
	UpdateRequests map[string]*mongodbatlas.MaintenanceWindow

	DeferFunc     func(projectID string) (*mongodbatlas.Response, error)
	DeferRequests map[string]struct{}

	AutoDeferFunc     func(projectID string) (*mongodbatlas.Response, error)
	AutoDeferRequests map[string]struct{}

	ResetFunc     func(projectID string) (*mongodbatlas.Response, error)
	ResetRequests map[string]struct{}
}

type NetworkPeeringClientMock struct {
	ListFunc     func(projectID string) ([]mongodbatlas.Peer, *mongodbatlas.Response, error)
	ListRequests map[string]struct{}

	GetFunc     func(projectID string, peerID string) (*mongodbatlas.Peer, *mongodbatlas.Response, error)
	GetRequests map[string]struct{}

	CreateFunc     func(projectID string, peer *mongodbatlas.Peer) (*mongodbatlas.Peer, *mongodbatlas.Response, error)
	CreateRequests map[string]*mongodbatlas.Peer

	UpdateFunc     func(projectID string, peerID string, peer *mongodbatlas.Peer) (*mongodbatlas.Peer, *mongodbatlas.Response, error)
	UpdateRequests map[string]*mongodbatlas.Peer

	DeleteFunc     func(projectID string, peerID string) (*mongodbatlas.Response, error)
	DeleteRequests map[string]struct{}
}

type PrivateEndpointsClientMock struct {
	CreateFunc     func(projectID string, endpoint *mongodbatlas.PrivateEndpointConnection) (*mongodbatlas.PrivateEndpointConnection, *mongodbatlas.Response, error)
	CreateRequests map[string]*mongodbatlas.PrivateEndpointConnection

	GetFunc     func(projectID string, cloudProvider string, endpointServiceID string) (*mongodbatlas.PrivateEndpointConnection, *mongodbatlas.Response, error)
	GetRequests map[string]struct{}

	ListFunc     func(projectID, providerName string) ([]mongodbatlas.PrivateEndpointConnection, *mongodbatlas.Response, error)
	ListRequests map[string]struct{}

	DeleteFunc     func(projectID string, cloudProvider string, endpointServiceID string) (*mongodbatlas.Response, error)
	DeleteRequests map[string]struct{}

	AddOnePrivateEndpointFunc     func(projectID string, cloudProvider string, endpointServiceID string, endpoint *mongodbatlas.InterfaceEndpointConnection) (*mongodbatlas.InterfaceEndpointConnection, *mongodbatlas.Response, error)
	AddOnePrivateEndpointRequests map[string]*mongodbatlas.InterfaceEndpointConnection

	GetOnePrivateEndpointFunc     func(projectID string, cloudProvider string, endpointServiceID string, privateEndpointID string) (*mongodbatlas.InterfaceEndpointConnection, *mongodbatlas.Response, error)
	GetOnePrivateEndpointRequests map[string]struct{}

	DeleteOnePrivateEndpointFunc     func(projectID string, cloudProvider string, endpointServiceID string, privateEndpointID string) (*mongodbatlas.Response, error)
	DeleteOnePrivateEndpointRequests map[string]struct{}

	UpdateRegionalizedPrivateEndpointSettingFunc     func(projectID string, enabled bool) (*mongodbatlas.RegionalizedPrivateEndpointSetting, *mongodbatlas.Response, error)
	UpdateRegionalizedPrivateEndpointSettingRequests map[string]bool

	GetRegionalizedPrivateEndpointSettingFunc     func(projectID string) (*mongodbatlas.RegionalizedPrivateEndpointSetting, *mongodbatlas.Response, error)
	GetRegionalizedPrivateEndpointSettingRequests map[string]struct{}
}

type ProjectsClientMock struct {
	GetAllProjectsFunc  func() (*mongodbatlas.Projects, *mongodbatlas.Response, error)
	GetAllProjectsCalls int

	GetOneProjectFunc     func(projectID string) (*mongodbatlas.Project, *mongodbatlas.Response, error)
	GetOneProjectRequests map[string]struct{}

	GetOneProjectByNameFunc     func(projectName string) (*mongodbatlas.Project, *mongodbatlas.Response, error)
	GetOneProjectByNameRequests map[string]struct{}

	CreateFunc     func(project *mongodbatlas.Project) (*mongodbatlas.Project, *mongodbatlas.Response, error)
	CreateRequests []*mongodbatlas.Project

	UpdateFunc     func(projectID string, project *mongodbatlas.ProjectUpdateRequest) (*mongodbatlas.Project, *mongodbatlas.Response, error)
	UpdateRequests map[string]*mongodbatlas.ProjectUpdateRequest

	DeleteFunc     func(projectID string) (*mongodbatlas.Response, error)
	DeleteRequests map[string]struct{}

	GetProjectTeamsAssignedFunc     func(projectID string) (*mongodbatlas.TeamsAssigned, *mongodbatlas.Response, error)
	GetProjectTeamsAssignedRequests map[string]struct{}

	AddTeamsToProjectFunc     func(projectId string, teams []*mongodbatlas.ProjectTeam) (*mongodbatlas.TeamsAssigned, *mongodbatlas.Response, error)
	AddTeamsToProjectRequests map[string][]*mongodbatlas.ProjectTeam

	RemoveUserFromProjectFunc     func(projectID string, userID string) (*mongodbatlas.Response, error)
	RemoveUserFromProjectRequests map[string]struct{}

	InvitationsFunc     func(projectID string, invitations *mongodbatlas.InvitationOptions) ([]*mongodbatlas.Invitation, *mongodbatlas.Response, error)
	InvitationsRequests map[string]*mongodbatlas.InvitationOptions

	InvitationFunc     func(projectID string, invitationID string) (*mongodbatlas.Invitation, *mongodbatlas.Response, error)
	InvitationRequests map[string]struct{}

	InviteUserFunc     func(projectID string, invitation *mongodbatlas.Invitation) (*mongodbatlas.Invitation, *mongodbatlas.Response, error)
	InviteUserRequests map[string]*mongodbatlas.Invitation

	UpdateInvitationFunc     func(projectID string, invitation *mongodbatlas.Invitation) (*mongodbatlas.Invitation, *mongodbatlas.Response, error)
	UpdateInvitationRequests map[string]*mongodbatlas.Invitation

	UpdateInvitationByIDFunc     func(projectID string, invitationID string, invitation *mongodbatlas.Invitation) (*mongodbatlas.Invitation, *mongodbatlas.Response, error)
	UpdateInvitationByIDRequests map[string]*mongodbatlas.Invitation

	DeleteInvitationFunc     func(projectID string, invitationID string) (*mongodbatlas.Response, error)
	DeleteInvitationRequests map[string]struct{}

	GetProjectSettingsFunc     func(projectID string) (*mongodbatlas.ProjectSettings, *mongodbatlas.Response, error)
	GetProjectSettingsRequests map[string]struct{}

	UpdateProjectSettingsFunc     func(projectID string, settings *mongodbatlas.ProjectSettings) (*mongodbatlas.ProjectSettings, *mongodbatlas.Response, error)
	UpdateProjectSettingsRequests map[string]*mongodbatlas.ProjectSettings
}

type ServerlessInstancesClientMock struct {
	ListFunc     func(projectID string) (*mongodbatlas.ClustersResponse, *mongodbatlas.Response, error)
	ListRequests map[string]struct{}

	GetFunc     func(projectID string, name string) (*mongodbatlas.Cluster, *mongodbatlas.Response, error)
	GetRequests map[string]struct{}

	CreateFunc     func(projectID string, instance *mongodbatlas.ServerlessCreateRequestParams) (*mongodbatlas.Cluster, *mongodbatlas.Response, error)
	CreateRequests map[string]*mongodbatlas.ServerlessCreateRequestParams

	UpdateFunc     func(projectID string, name string, instance *mongodbatlas.ServerlessUpdateRequestParams) (*mongodbatlas.Cluster, *mongodbatlas.Response, error)
	UpdateRequests map[string]*mongodbatlas.ServerlessUpdateRequestParams

	DeleteFunc     func(projectID string, name string) (*mongodbatlas.Response, error)
	DeleteRequests map[string]struct{}
}

type TeamsClientMock struct {
	ListFunc     func(orgID string) ([]mongodbatlas.Team, *mongodbatlas.Response, error)
	ListRequests map[string]struct{}

	GetFunc     func(orgID string, teamID string) (*mongodbatlas.Team, *mongodbatlas.Response, error)
	GetRequests map[string]struct{}

	GetOneTeamByNameFunc     func(orgID string, name string) (*mongodbatlas.Team, *mongodbatlas.Response, error)
	GetOneTeamByNameRequests map[string]struct{}

	GetTeamUsersAssignedFunc     func(orgID string, teamID string) ([]mongodbatlas.AtlasUser, *mongodbatlas.Response, error)
	GetTeamUsersAssignedRequests map[string]struct{}

	CreateFunc     func(orgID string, team *mongodbatlas.Team) (*mongodbatlas.Team, *mongodbatlas.Response, error)
	CreateRequests map[string]*mongodbatlas.Team

	RenameFunc     func(orgID string, teamID string, name string) (*mongodbatlas.Team, *mongodbatlas.Response, error)
	RenameRequests map[string]struct{}

	UpdateTeamRolesFunc     func(orgID string, teamID string, roles *mongodbatlas.TeamUpdateRoles) ([]mongodbatlas.TeamRoles, *mongodbatlas.Response, error)
	UpdateTeamRolesRequests map[string]*mongodbatlas.TeamUpdateRoles

	AddUsersToTeamFunc     func(orgID string, teamID string, userIDs []string) ([]mongodbatlas.AtlasUser, *mongodbatlas.Response, error)
	AddUsersToTeamRequests map[string][]string

	RemoveUserToTeamFunc     func(orgID string, teamID string, userID string) (*mongodbatlas.Response, error)
	RemoveUserToTeamRequests map[string]struct{}

	RemoveTeamFromOrganizationFunc     func(orgID string, teamID string) (*mongodbatlas.Response, error)
	RemoveTeamFromOrganizationRequests map[string]struct{}

	RemoveTeamFromProjectFunc     func(projectID string, teamID string) (*mongodbatlas.Response, error)
	RemoveTeamFromProjectRequests map[string]struct{}
}

type TestProvider struct {
	ClientFunc       func(ctx context.Context, creds *atlas.Credentials, log *zap.SugaredLogger) (*mongodbatlas.Client, error)
	SdkClientSetFunc func(ctx context.Context, creds *atlas.Credentials, log *zap.SugaredLogger) (*atlas.ClientSet, error)
	IsCloudGovFunc   func() bool
	IsSupportedFunc  func() bool
}

type ThirdPartyIntegrationsClientMock struct {
	CreateFunc     func(projectID string, integrationType string, integration *mongodbatlas.ThirdPartyIntegration) (*mongodbatlas.ThirdPartyIntegrations, *mongodbatlas.Response, error)
	CreateRequests map[string]*mongodbatlas.ThirdPartyIntegration

	ReplaceFunc     func(projectID string, integrationType string, integration *mongodbatlas.ThirdPartyIntegration) (*mongodbatlas.ThirdPartyIntegrations, *mongodbatlas.Response, error)
	ReplaceRequests map[string]*mongodbatlas.ThirdPartyIntegration

	DeleteFunc     func(projectID string, integrationType string) (*mongodbatlas.Response, error)
	DeleteRequests map[string]struct{}

	GetFunc     func(projectID string, integrationType string) (*mongodbatlas.ThirdPartyIntegration, *mongodbatlas.Response, error)
	GetRequests map[string]struct{}

	ListFunc     func(projectID string) (*mongodbatlas.ThirdPartyIntegrations, *mongodbatlas.Response, error)
	ListRequests map[string]struct{}
}