//go:build integ
// +build integ

// Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License"). You may
// not use this file except in compliance with the License. A copy of the
// License is located at
//
//	http://aws.amazon.com/apache2.0/
//
// or in the "license" file accompanying this file. This file 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.

package e2e

import (
	"io/ioutil"
	"os"
	"testing"

	"github.com/aws/amazon-ecs-cli/ecs-cli/integ/ecs"

	"github.com/stretchr/testify/require"

	"github.com/aws/amazon-ecs-cli/ecs-cli/integ/cmd"
)

// TestCreateClusterWithEC2Task runs the sequence of ecs-cli commands from
// the EC2 tutorial: https://docs.aws.amazon.com/AmazonECS/latest/developerguide/ecs-cli-tutorial-ec2.html
func TestCreateClusterWithEC2Task(t *testing.T) {
	// Create the cluster
	conf := cmd.TestEC2TutorialConfig(t)

	// Ensure cluster cleanup
	defer cmd.TestDown(t, conf)

	cmd.TestUp(t,
		conf,
		cmd.WithCapabilityIAM(),
		cmd.WithInstanceType("t2.medium"),
		cmd.WithSize(2),
		cmd.WithIMDSv2(),
	)

	// TODO add waiter/retry?
	ecs.TestClusterSize(t, conf.ClusterName, 2)

	// Create the files for a task definition
	project := cmd.NewProject("e2e-ec2-tutorial-taskdef", conf.ConfigName)
	project.ComposeFileName = createEC2TutorialComposeFile(t)
	defer os.Remove(project.ComposeFileName)

	// Ensure cleanup of task
	defer ecs.TestListTasks(t, conf.ClusterName, 0)
	defer cmd.TestTaskDown(t, project)

	// Create a new task with 2 containers.
	cmd.TestTaskUp(t, project)
	ecs.TestListTasks(t, conf.ClusterName, 1)
	cmd.TestPsRunning(t, project, 2)

	// Add an additional task to the cluster, so we expect 4 running containers.
	cmd.TestTaskScale(t, project, 2)
	ecs.TestListTasks(t, conf.ClusterName, 2)
	cmd.TestPsRunning(t, project, 4)
}

func createEC2TutorialComposeFile(t *testing.T) string {
	content := `
version: '2'
services:
  wordpress:
    image: wordpress
    cpu_shares: 100
    mem_limit: 524288000
    ports:
      - "80:80"
    links:
      - mysql
    stop_grace_period: "1m30s"
  mysql:
    image: mysql:5.7
    cpu_shares: 100
    mem_limit: 524288000
    environment:
      MYSQL_ROOT_PASSWORD: password`

	tmpfile, err := ioutil.TempFile("", "docker-compose-*.yml")
	require.NoError(t, err, "Failed to create docker-compose.yml")

	_, err = tmpfile.Write([]byte(content))
	require.NoErrorf(t, err, "Failed to write to %s", tmpfile.Name())

	err = tmpfile.Close()
	require.NoErrorf(t, err, "Failed to close %s", tmpfile.Name())

	t.Logf("Created %s successfully", tmpfile.Name())
	return tmpfile.Name()
}