docker-compose/tests/aci-e2e/e2e-aci.go

package main

import (
	"context"
	"log"

	"github.com/Azure/azure-sdk-for-go/profiles/2019-03-01/resources/mgmt/resources"
	"github.com/docker/api/azure"
	. "github.com/docker/api/tests/framework"
	. "github.com/onsi/gomega"
)

const resourceGroupName = "resourceGroupTest"

var location = "westeurope"

const contextName = "acitest"

func main() {
	SetupTest()

	It("ensures context command includes azure-login and aci-create", func() {
		output := NewDockerCommand("context", "create", "--help").ExecOrDie()
		Expect(output).To(ContainSubstring("docker context create CONTEXT BACKEND [OPTIONS] [flags]"))
		Expect(output).To(ContainSubstring("--aci-location"))
		Expect(output).To(ContainSubstring("--aci-subscription-id"))
		Expect(output).To(ContainSubstring("--aci-resource-group"))
	})

	It("should be initialized with default context", func() {
		_, err := NewCommand("docker", "context", "rm", "-f", contextName).Exec()
		if err == nil {
			log.Println("Cleaning existing test context")
		}

		NewCommand("docker", "context", "use", "default").ExecOrDie()
		output := NewCommand("docker", "context", "ls").ExecOrDie()
		Expect(output).To(Not(ContainSubstring(contextName)))
		Expect(output).To(ContainSubstring("default *"))
	})

	It("creates a new aci context for tests", func() {
		setupTestResourecGroup(resourceGroupName)
		subscriptionID, err := azure.GetSubscriptionID(context.TODO())
		Expect(err).To(BeNil())

		NewDockerCommand("context", "create", contextName, "aci", "--aci-subscription-id", subscriptionID, "--aci-resource-group", resourceGroupName, "--aci-location", location).ExecOrDie()
		//Expect(output).To(ContainSubstring("ACI context acitest created"))
	})

	defer deleteResourceGroup(resourceGroupName)

	It("uses the aci context", func() {
		currentContext := NewCommand("docker", "context", "use", contextName).ExecOrDie()
		Expect(currentContext).To(ContainSubstring(contextName))
		output := NewCommand("docker", "context", "ls").ExecOrDie()
		Expect(output).To(ContainSubstring("acitest *"))
	})

	It("ensures no container is running initially", func() {
		output := NewDockerCommand("ps").ExecOrDie()
		Expect(len(Lines(output))).To(Equal(1))
	})

	/*
		var nginxID string
		It("runs nginx on port 80", func() {
			NewDockerCommand("run", "nginx", "-p", "80:80").ExecOrDie()
			output := NewDockerCommand("ps").ExecOrDie()
			Lines := Lines(output)
			Expect(len(Lines)).To(Equal(2))

			containerFields := Columns(Lines[1])
			nginxID = containerFields[0]
			Expect(containerFields[1]).To(Equal("nginx"))
			Expect(containerFields[2]).To(Equal("Running"))
			exposedIP := containerFields[3]
			Expect(exposedIP).To(ContainSubstring(":80->80/TCP"))

			url := strings.ReplaceAll(exposedIP, "->80/TCP", "")
			output = NewCommand("curl", url).ExecOrDie()
			Expect(output).To(ContainSubstring("Welcome to nginx!"))
		})

		It("removes container nginx", func() {
			output := NewDockerCommand("rm", nginxID).ExecOrDie()
			Expect(Lines(output)[0]).To(Equal(nginxID))
		})
	*/

	It("deploys a compose app", func() {
		NewDockerCommand("compose", "up", "-f", "./composefiles/aci-demo/aci_demo_port.yaml", "--name", "acidemo").ExecOrDie()
		//Expect(output).To(ContainSubstring("Successfully deployed"))
		output := NewDockerCommand("ps").ExecOrDie()
		Lines := Lines(output)
		Expect(len(Lines)).To(Equal(4))
		for _, line := range Lines[1:] {
			Expect(line).To(ContainSubstring("Running"))
		}
		/*
				if strings.Contains(line, "acicompose_web") {
					webChecked = true
					containerFields := Columns(line)
					exposedIP := containerFields[3]
					Expect(exposedIP).To(ContainSubstring(":80->80/TCP"))

					url := strings.ReplaceAll(exposedIP, "->80/TCP", "")
					output = NewCommand("curl", url).ExecOrDie()
					Expect(output).To(ContainSubstring("Docker Compose demo"))
					output = NewCommand("curl", url+"/words/noun").ExecOrDie()
					Expect(output).To(ContainSubstring("\"word\":"))
				}
			}
			Expect(webChecked).To(BeTrue())
		*/
	})

	It("get logs from web service", func() {
		output := NewDockerCommand("logs", "acidemo_web").ExecOrDie()
		Expect(output).To(ContainSubstring("Listening on port 80"))
	})

	It("shutdown compose app", func() {
		NewDockerCommand("compose", "down", "-f", "./composefiles/aci-demo/aci_demo_port.yaml", "--name", "acidemo").ExecOrDie()
	})
	It("switches back to default context", func() {
		output := NewCommand("docker", "context", "use", "default").ExecOrDie()
		Expect(output).To(ContainSubstring("default"))
	})

	It("deletes test context", func() {
		output := NewCommand("docker", "context", "rm", contextName).ExecOrDie()
		Expect(output).To(ContainSubstring(contextName))
	})
}

func setupTestResourecGroup(groupName string) {
	log.Println("Creating resource group " + resourceGroupName)
	ctx := context.TODO()
	subscriptionID, err := azure.GetSubscriptionID(ctx)
	Expect(err).To(BeNil())
	gc := azure.GetGroupsClient(subscriptionID)
	_, err = gc.CreateOrUpdate(ctx, groupName, resources.Group{
		Location: &location,
	})
	Expect(err).To(BeNil())
}

func deleteResourceGroup(groupName string) {
	log.Println("Deleting resource group " + resourceGroupName)
	ctx := context.TODO()
	subscriptionID, err := azure.GetSubscriptionID(ctx)
	Expect(err).To(BeNil())
	gc := azure.GetGroupsClient(subscriptionID)
	_, err = gc.Delete(ctx, groupName)
	Expect(err).To(BeNil())
}