test(integration): add tests for deis limits commands

Author: mboersma

tests/apps_test.go

@@ -28,6 +28,9 @@ func TestApps(t *testing.T) {
 	appsInfoTest(t, params)
 	appsRunTest(t, params)
 	appsOpenTest(t, params)
+	limitsSetTest(t, params, 3)
+	appsOpenTest(t, params)
+	limitsUnsetTest(t, params, 5)
 	appsDestroyTest(t, params)
 	appsListTest(t, params, true)
 }

tests/limits_test.go

@@ -0,0 +1,78 @@
+// +build integration
+
+package tests
+
+import (
+	"fmt"
+	"os/exec"
+	"regexp"
+	"strings"
+	"testing"
+
+	"github.com/deis/deis/tests/utils"
+)
+
+var (
+	limitsListCmd     = "limits:list --app={{.AppName}}"
+	limitsSetMemCmd   = "limits:set --app={{.AppName}} web=256M"
+	limitsSetCPUCmd   = "limits:set --app={{.AppName}} -c web=512"
+	limitsUnsetMemCmd = "limits:unset --app={{.AppName}} --memory web"
+	limitsUnsetCPUCmd = "limits:unset --app={{.AppName}} -c web"
+	output1           = `(?s)"CpuShares": 512,.*"Memory": 0,`
+	output2           = `(?s)"CpuShares": 512,.*"Memory": 268435456,`
+	output3           = `(?s)"CpuShares": 0,.*"Memory": 268435456,`
+	output4           = `(?s)"CpuShares": 0,.*"Memory": 0,`
+)
+
+func limitsSetTest(t *testing.T, cfg *utils.DeisTestConfig, ver int) {
+	cpuCmd, memCmd := limitsSetCPUCmd, limitsSetMemCmd
+	if strings.Contains(cfg.ExampleApp, "dockerfile") {
+		cpuCmd = strings.Replace(cpuCmd, "web", "cmd", 1)
+		memCmd = strings.Replace(memCmd, "web", "cmd", 1)
+	}
+	utils.Execute(t, cpuCmd, cfg, false, "512")
+	out := dockerInspect(t, cfg, ver)
+	if _, err := regexp.MatchString(output1, out); err != nil {
+		t.Fatal(err)
+	}
+	utils.Execute(t, memCmd, cfg, false, "256M")
+	out = dockerInspect(t, cfg, ver+1)
+	if _, err := regexp.MatchString(output2, out); err != nil {
+		t.Fatal(err)
+	}
+}
+
+func limitsUnsetTest(t *testing.T, cfg *utils.DeisTestConfig, ver int) {
+	cpuCmd, memCmd := limitsUnsetCPUCmd, limitsUnsetMemCmd
+	if strings.Contains(cfg.ExampleApp, "dockerfile") {
+		cpuCmd = strings.Replace(cpuCmd, "web", "cmd", 1)
+		memCmd = strings.Replace(memCmd, "web", "cmd", 1)
+	}
+	utils.Execute(t, cpuCmd, cfg, false, "Unlimited")
+	out := dockerInspect(t, cfg, ver)
+	if _, err := regexp.MatchString(output3, out); err != nil {
+		t.Fatal(err)
+	}
+	utils.Execute(t, memCmd, cfg, false, "Unlimited")
+	out = dockerInspect(t, cfg, ver+1)
+	if _, err := regexp.MatchString(output4, out); err != nil {
+		t.Fatal(err)
+	}
+}
+
+// dockerInspect creates an SSH session to the Deis controller
+// and runs "docker inspect" on the first app container.
+func dockerInspect(
+	t *testing.T, cfg *utils.DeisTestConfig, ver int) string {
+	cmd := fmt.Sprintf("docker inspect %s_v%d.web.1", cfg.AppName, ver)
+	sshCmd := exec.Command("ssh",
+		"-o", "StrictHostKeyChecking=no",
+		"-o", "UserKnownHostsFile=/dev/null",
+		"-o", "PasswordAuthentication=no",
+		"core@deis."+cfg.Domain, cmd)
+	out, err := sshCmd.Output()
+	if err != nil {
+		t.Fatal(out, err)
+	}
+	return string(out)
+}