CLOUDP-73546: Add e2e test for atlas customDns (#456)

Author: andreaangiolillo

e2e/atlas/custom_dns_test.go

@@ -0,0 +1,115 @@
+// Copyright 2020 MongoDB Inc
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+// +build e2e atlas,generic
+
+package atlas_test
+
+import (
+	"encoding/json"
+	"fmt"
+	"os"
+	"os/exec"
+	"testing"
+
+	"github.com/mongodb/mongocli/e2e"
+	"github.com/stretchr/testify/assert"
+	"go.mongodb.org/atlas/mongodbatlas"
+)
+
+func TestCustomDNS(t *testing.T) {
+	n, err := e2e.RandInt(255)
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+
+	cliPath, err := e2e.Bin()
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+
+	projectName := fmt.Sprintf("e2e-integration-custom-dns-%v", n)
+	projectID, err := createProject(projectName)
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+
+	defer func() {
+		if e := deleteProject(projectID); e != nil {
+			t.Errorf("error deleting project: %v", e)
+		}
+	}()
+
+	t.Run("Enable", func(t *testing.T) {
+		cmd := exec.Command(cliPath,
+			atlasEntity,
+			customDNSEntity,
+			awsEntity,
+			"enable",
+			"--projectId",
+			projectID,
+			"-o=json")
+		cmd.Env = os.Environ()
+		resp, err := cmd.CombinedOutput()
+
+		a := assert.New(t)
+		a.NoError(err, string(resp))
+
+		var dns mongodbatlas.AWSCustomDNSSetting
+		if err := json.Unmarshal(resp, &dns); a.NoError(err) {
+			a.True(dns.Enabled)
+		}
+	})
+
+	t.Run("Describe", func(t *testing.T) {
+		cmd := exec.Command(cliPath,
+			atlasEntity,
+			customDNSEntity,
+			awsEntity,
+			"describe",
+			"--projectId",
+			projectID,
+			"-o=json")
+		cmd.Env = os.Environ()
+		resp, err := cmd.CombinedOutput()
+
+		a := assert.New(t)
+		a.NoError(err, string(resp))
+
+		var dns mongodbatlas.AWSCustomDNSSetting
+		if err := json.Unmarshal(resp, &dns); a.NoError(err) {
+			a.True(dns.Enabled)
+		}
+	})
+
+	t.Run("Disable", func(t *testing.T) {
+		cmd := exec.Command(cliPath,
+			atlasEntity,
+			customDNSEntity,
+			awsEntity,
+			"disable",
+			"--projectId",
+			projectID,
+			"-o=json")
+		cmd.Env = os.Environ()
+		resp, err := cmd.CombinedOutput()
+
+		a := assert.New(t)
+		a.NoError(err, string(resp))
+
+		var dns mongodbatlas.AWSCustomDNSSetting
+		if err := json.Unmarshal(resp, &dns); a.NoError(err) {
+			a.False(dns.Enabled)
+		}
+	})
+}

e2e/atlas/helper_test.go

@@ -47,6 +47,8 @@ const (
 	integrationsEntity     = "integrations"
 	securityEntity         = "security"
 	ldapEntity             = "ldap"
+	awsEntity              = "aws"
+	customDNSEntity        = "customDns"
 )
 
 func getHostnameAndPort() (string, error) {