From f27d1500a98896fb9acf590e6d549d8aea2a138f Mon Sep 17 00:00:00 2001
From: Jan Christoph Uhde <Jan@UhdeJC.com>
Date: Mon, 19 Mar 2018 13:27:32 +0100
Subject: [PATCH] Adds ssl support for scaling test

---
 tests/scale_test.go | 77 +++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 74 insertions(+), 3 deletions(-)

diff --git a/tests/scale_test.go b/tests/scale_test.go
index 3004b229d..d6f140d13 100644
--- a/tests/scale_test.go
+++ b/tests/scale_test.go
@@ -13,16 +13,87 @@ import (
 
 // TestScaleCluster tests scaling up/down the number of DBServers & coordinators
 // of a cluster.
-func TestScaleCluster(t *testing.T) {
+func TestScaleClusterNonTLS(t *testing.T) {
 	longOrSkip(t)
 	c := client.MustNewInCluster()
 	kubecli := mustNewKubeClient(t)
 	ns := getNamespace(t)
 
 	// Prepare deployment config
-	depl := newDeployment("test-scale-" + uniuri.NewLen(4))
+	depl := newDeployment("test-scale-non-tls" + uniuri.NewLen(4))
 	depl.Spec.Mode = api.DeploymentModeCluster
-	depl.Spec.SetDefaults(depl.GetName())
+	depl.Spec.TLS = api.TLSSpec{"None", nil, 50}
+	depl.Spec.SetDefaults(depl.GetName()) // this must be last
+
+	// Create deployment
+	apiObject, err := c.DatabaseV1alpha().ArangoDeployments(ns).Create(depl)
+	if err != nil {
+		t.Fatalf("Create deployment failed: %v", err)
+	}
+
+	// Wait for deployment to be ready
+	if _, err := waitUntilDeployment(c, depl.GetName(), ns, deploymentHasState(api.DeploymentStateRunning)); err != nil {
+		t.Fatalf("Deployment not running in time: %v", err)
+	}
+
+	// Create a database client
+	ctx := context.Background()
+	client := mustNewArangodDatabaseClient(ctx, kubecli, apiObject, t)
+
+	// Wait for cluster to be completely ready
+	if err := waitUntilClusterHealth(client, func(h driver.ClusterHealth) error {
+		return clusterHealthEqualsSpec(h, apiObject.Spec)
+	}); err != nil {
+		t.Fatalf("Cluster not running in expected health in time: %v", err)
+	}
+
+	// Add 2 DBServers, 1 coordinator
+	updated, err := updateDeployment(c, depl.GetName(), ns, func(spec *api.DeploymentSpec) {
+		spec.DBServers.Count = 5
+		spec.Coordinators.Count = 4
+	})
+	if err != nil {
+		t.Fatalf("Failed to update deployment: %v", err)
+	}
+
+	// Wait for cluster to reach new size
+	if err := waitUntilClusterHealth(client, func(h driver.ClusterHealth) error {
+		return clusterHealthEqualsSpec(h, updated.Spec)
+	}); err != nil {
+		t.Fatalf("Cluster not running, after scale-up, in expected health in time: %v", err)
+	}
+
+	// Remove 3 DBServers, 2 coordinator
+	updated, err = updateDeployment(c, depl.GetName(), ns, func(spec *api.DeploymentSpec) {
+		spec.DBServers.Count = 3
+		spec.Coordinators.Count = 2
+	})
+	if err != nil {
+		t.Fatalf("Failed to update deployment: %v", err)
+	}
+
+	// Wait for cluster to reach new size
+	if err := waitUntilClusterHealth(client, func(h driver.ClusterHealth) error {
+		return clusterHealthEqualsSpec(h, updated.Spec)
+	}); err != nil {
+		t.Fatalf("Cluster not running, after scale-down, in expected health in time: %v", err)
+	}
+
+	// Cleanup
+	removeDeployment(c, depl.GetName(), ns)
+}
+
+func TestScaleClusterTLS(t *testing.T) {
+	longOrSkip(t)
+	c := client.MustNewInCluster()
+	kubecli := mustNewKubeClient(t)
+	ns := getNamespace(t)
+
+	// Prepare deployment config
+	depl := newDeployment("test-scale-tls" + uniuri.NewLen(4))
+	depl.Spec.Mode = api.DeploymentModeCluster
+	depl.Spec.TLS = api.TLSSpec{}         // should auto-generate cert
+	depl.Spec.SetDefaults(depl.GetName()) // this must be last
 
 	// Create deployment
 	apiObject, err := c.DatabaseV1alpha().ArangoDeployments(ns).Create(depl)