// +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
	// regression test for https://github.com/deis/deis/issues/1563
	// previously the client would throw a stack trace with empty limits
	utils.Execute(t, limitsListCmd, cfg, false, "Unlimited")
	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, limitsListCmd, cfg, false, "512")
	utils.Execute(t, memCmd, cfg, false, "256M")
	out = dockerInspect(t, cfg, ver+1)
	if _, err := regexp.MatchString(output2, out); err != nil {
		t.Fatal(err)
	}
	utils.Execute(t, limitsListCmd, cfg, false, "256M")
}

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, limitsListCmd, cfg, false, "Unlimited")
	utils.Execute(t, memCmd, cfg, false, "Unlimited")
	out = dockerInspect(t, cfg, ver+1)
	if _, err := regexp.MatchString(output4, out); err != nil {
		t.Fatal(err)
	}
	utils.Execute(t, limitsListCmd, cfg, false, "Unlimited")
}

// 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)
}