CLOUDP-63103: Implementing e2e test for atlas metric processes (#181)

Author: andreaangiolillo

.evergreen.yml

@@ -1,3 +1,4 @@
+exec_timeout_secs: 3600 # automatically fail any task if it takes longer than 60 minutes to finish.
 stepback: true
 command_type: system
 pre_error_fails_task: true

Makefile

@@ -95,7 +95,7 @@ build: ## Generate a binary in ./bin
 e2e-test: build ## Run E2E tests
 	@echo "==> Running E2E tests..."
 	# the target assumes the MCLI-* environment variables are exported
-	$(E2E_CMD) -v -p 1 -parallel 1 -tags="$(E2E_TAGS)" ./e2e...
+	$(E2E_CMD) -v -p 1 -parallel 1 -timeout 0 -tags="$(E2E_TAGS)" ./e2e...
 
 .PHONY: install
 install: ## Install a binary in $GOPATH/bin

e2e/atlas/clusters_test.go

@@ -22,6 +22,7 @@ import (
 	"os"
 	"os/exec"
 	"path/filepath"
+	"strings"
 	"testing"
 	"time"
 
@@ -73,6 +74,24 @@ func TestAtlasClusters(t *testing.T) {
 		ensureCluster(t, cluster, clusterName, "4.0", 10)
 	})
 
+	t.Run("Watch", func(t *testing.T) {
+		cmd := exec.Command(cliPath,
+			atlasEntity,
+			clustersEntity,
+			"watch",
+			clusterName)
+		cmd.Env = os.Environ()
+		resp, err := cmd.CombinedOutput()
+
+		if err != nil {
+			t.Fatalf("unexpected error: %v, resp: %v", err, string(resp))
+		}
+
+		if !strings.Contains(string(resp), "Cluster available at:") {
+			t.Errorf("got=%#v\nwant=%#v\n", string(resp), "Cluster available at:")
+		}
+	})
+
 	t.Run("List", func(t *testing.T) {
 		cmd := exec.Command(cliPath, atlasEntity, clustersEntity, "ls")
 		cmd.Env = os.Environ()

e2e/atlas/metrics_processes_test.go

@@ -0,0 +1,146 @@
+// 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
+
+package atlas_test
+
+import (
+	"encoding/json"
+	"fmt"
+	"math/rand"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"strconv"
+	"testing"
+	"time"
+
+	"github.com/mongodb/go-client-mongodb-atlas/mongodbatlas"
+)
+
+func TestAtlasMetrics(t *testing.T) {
+	cliPath, err := filepath.Abs("../../bin/mongocli")
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+	_, err = os.Stat(cliPath)
+
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+
+	atlasEntity := "atlas"
+	metricsEntity := "metrics"
+	r := rand.New(rand.NewSource(time.Now().UnixNano()))
+	clusterName := fmt.Sprintf("e2e-cluster-%v", r.Uint32())
+
+	err = deployCluster(cliPath, atlasEntity, clusterName)
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+
+	hostname, err := getHostnameAndPort(cliPath, atlasEntity)
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+
+	t.Run("processes", func(t *testing.T) {
+		cmd := exec.Command(cliPath,
+			atlasEntity,
+			metricsEntity,
+			"processes",
+			hostname,
+			"--granularity=PT30M",
+			"--period=P1DT12H")
+
+		cmd.Env = os.Environ()
+		resp, err := cmd.CombinedOutput()
+
+		metrics := &mongodbatlas.ProcessMeasurements{}
+		err = json.Unmarshal(resp, &metrics)
+
+		if err != nil {
+			t.Fatalf("unexpected error: %v", err)
+		}
+
+		if metrics.Measurements == nil {
+			t.Errorf("there are no measurements")
+		}
+
+		if len(metrics.Measurements) == 0 {
+			t.Errorf("got=%#v\nwant=%#v\n", 0, "len(metrics.Measurements) > 0")
+		}
+	})
+
+	deleteCluster(cliPath, atlasEntity, clusterName)
+}
+
+func getHostnameAndPort(cliPath, atlasEntity string) (string, error) {
+	cmd := exec.Command(cliPath,
+		atlasEntity,
+		"processes",
+		"list")
+
+	cmd.Env = os.Environ()
+	resp, err := cmd.CombinedOutput()
+
+	var processes []*mongodbatlas.Process
+	err = json.Unmarshal(resp, &processes)
+
+	if err != nil {
+		return "", err
+	}
+
+	if len(processes) == 0 {
+		return "", fmt.Errorf("got=%#v\nwant=%#v\n", 0, "len(processes) > 0")
+	}
+
+	// The first element may not be the created cluster but that is fine since
+	// we just need one cluster up and running
+	return processes[0].Hostname + ":" + strconv.Itoa(processes[0].Port), nil
+}
+
+func deployCluster(cliPath, atlasEntity, clusterName string) error {
+	cmd := exec.Command(cliPath,
+		atlasEntity,
+		"clusters",
+		"create",
+		clusterName,
+		"--region=US_EAST_1",
+		"--members=3",
+		"--tier=M10",
+		"--provider=AWS",
+		"--mdbVersion=4.0",
+		"--diskSizeGB=10")
+	cmd.Env = os.Environ()
+	err := cmd.Run()
+
+	if err != nil {
+		return err
+	}
+
+	cmd = exec.Command(cliPath,
+		"atlas",
+		"clusters",
+		"watch",
+		clusterName)
+	cmd.Env = os.Environ()
+	return cmd.Run()
+}
+
+func deleteCluster(cliPath, atlasEntity, clusterName string) error {
+	cmd := exec.Command(cliPath, atlasEntity, "clusters", "delete", clusterName, "--force")
+	cmd.Env = os.Environ()
+	return cmd.Run()
+}