CLOUDP-272605 Remove IDPs on cleanup script (#3320)

mongodb · Oct 11, 2024 · 71ad153 · 71ad153
1 parent c510096
commit 71ad153
Show file tree

Hide file tree

Showing 3 changed files with 62 additions and 1 deletion.
diff --git a/build/ci/evergreen.yml b/build/ci/evergreen.yml
@@ -1707,13 +1707,15 @@ buildvariants:
       - name: ".e2e .backup .atlas"
   - name: e2e_cleanup
     display_name: "E2E Cleanup Resources"
+    allowed_requesters: ["patch", "ad_hoc", "github_pr"]
+    tags:
+      - cleanup
     run_on:
       - rhel80-small
     expansions:
       <<: *go_linux_version
     tasks:
       - name: ".e2e .cleanup"
-        cron: "0 2 * * *" # 2am daily
   - name: e2e_atlas_deployments
     display_name: "E2E Atlas Deployments Tests"
     run_on:
@@ -1867,6 +1869,9 @@ patch_aliases:
   - alias: "packer"
     variant_tags: ["packer"]
     task: ".*"
+  - alias: "cleanup"
+    variant_tags: ["cleanup"]
+    task: ".*"
 github_pr_aliases:
   - variant: "code_health"
     task_tags: ["code_health"]

diff --git a/test/e2e/atlas/cleanup_test.go b/test/e2e/atlas/cleanup_test.go
@@ -40,6 +40,13 @@ func TestCleanup(t *testing.T) {
 		t.Parallel()
 		deleteOrgTeams(t, cliPath)
 	})
+	t.Run("trying to delete all org idps", func(t *testing.T) {
+		if IsGov() {
+			t.Skip("idps are not available on gov")
+		}
+		t.Parallel()
+		deleteAllIDPs(t, cliPath)
+	})
 	args := []string{projectEntity,
 		"list",
 		"--limit=500",

diff --git a/test/e2e/atlas/helper_test.go b/test/e2e/atlas/helper_test.go
@@ -1265,3 +1265,52 @@ func installExamplePlugin(t *testing.T, cliPath string, version string) {
 	resp, err := e2e.RunAndGetStdOut(cmd)
 	require.NoError(t, err, string(resp))
 }
+
+func getFedSettingsID(t *testing.T, cliPath string) string {
+	t.Helper()
+	args := []string{federatedAuthenticationEntity,
+		federationSettingsEntity,
+		"describe",
+		"-o=json",
+	}
+	if orgID, set := os.LookupEnv("MCLI_ORG_ID"); set {
+		args = append(args, "--orgId", orgID)
+	}
+	cmd := exec.Command(cliPath, args...)
+	cmd.Env = os.Environ()
+	resp, err := e2e.RunAndGetStdOut(cmd)
+	require.NoError(t, err, string(resp))
+	var settings *atlasv2.OrgFederationSettings
+	require.NoError(t, json.Unmarshal(resp, &settings))
+	require.NotNil(t, settings.Id)
+
+	return *settings.Id
+}
+
+func listIDPs(t *testing.T, cliPath string, fedSettingsID string) *atlasv2.PaginatedFederationIdentityProvider {
+	t.Helper()
+	cmd := exec.Command(cliPath, "federatedAuthentication", "federationSettings", "identityProvider", "list", "--federationSettingsId", fedSettingsID, "-o", "json")
+	cmd.Env = os.Environ()
+	resp, err := e2e.RunAndGetStdOut(cmd)
+	require.NoError(t, err, string(resp))
+	var idps *atlasv2.PaginatedFederationIdentityProvider
+	require.NoError(t, json.Unmarshal(resp, &idps))
+	return idps
+}
+
+func deleteIDP(t *testing.T, cliPath string, id string, fedSettingsID string) {
+	t.Helper()
+	cmd := exec.Command(cliPath, federatedAuthenticationEntity, federationSettingsEntity, "identityProvider", "delete", id, "--federationSettingsId", fedSettingsID, "--force")
+	cmd.Env = os.Environ()
+	resp, err := e2e.RunAndGetStdOut(cmd)
+	require.NoError(t, err, string(resp))
+}
+
+func deleteAllIDPs(t *testing.T, cliPath string) {
+	t.Helper()
+	fedSettingsID := getFedSettingsID(t, cliPath)
+	idps := listIDPs(t, cliPath, fedSettingsID)
+	for _, idp := range *idps.Results {
+		deleteIDP(t, cliPath, idp.Id, fedSettingsID)
+	}
+}