package aws

import (
	"fmt"
	"testing"

	"github.com/hashicorp/terraform/helper/resource"
	"github.com/satori/uuid"
)

func TestAccDataSourceAwsEbsSnapshotIds_basic(t *testing.T) {
	resource.Test(t, resource.TestCase{
		PreCheck:  func() { testAccPreCheck(t) },
		Providers: testAccProviders,
		Steps: []resource.TestStep{
			{
				Config: testAccDataSourceAwsEbsSnapshotIdsConfig_basic,
				Check: resource.ComposeTestCheckFunc(
					testAccCheckAwsEbsSnapshotDataSourceID("data.aws_ebs_snapshot_ids.test"),
				),
			},
		},
	})
}

func TestAccDataSourceAwsEbsSnapshotIds_sorted(t *testing.T) {
	uuid := uuid.NewV4().String()

	resource.Test(t, resource.TestCase{
		PreCheck:  func() { testAccPreCheck(t) },
		Providers: testAccProviders,
		Steps: []resource.TestStep{
			{
				Config: testAccDataSourceAwsEbsSnapshotIdsConfig_sorted1(uuid),
				Check: resource.ComposeTestCheckFunc(
					resource.TestCheckResourceAttrSet("aws_ebs_snapshot.a", "id"),
					resource.TestCheckResourceAttrSet("aws_ebs_snapshot.b", "id"),
				),
			},
			{
				Config: testAccDataSourceAwsEbsSnapshotIdsConfig_sorted2(uuid),
				Check: resource.ComposeTestCheckFunc(
					testAccCheckAwsEbsSnapshotDataSourceID("data.aws_ebs_snapshot_ids.test"),
					resource.TestCheckResourceAttr("data.aws_ebs_snapshot_ids.test", "ids.#", "2"),
					resource.TestCheckResourceAttrPair(
						"data.aws_ebs_snapshot_ids.test", "ids.0",
						"aws_ebs_snapshot.b", "id"),
					resource.TestCheckResourceAttrPair(
						"data.aws_ebs_snapshot_ids.test", "ids.1",
						"aws_ebs_snapshot.a", "id"),
				),
			},
		},
	})
}

func TestAccDataSourceAwsEbsSnapshotIds_empty(t *testing.T) {
	resource.Test(t, resource.TestCase{
		PreCheck:  func() { testAccPreCheck(t) },
		Providers: testAccProviders,
		Steps: []resource.TestStep{
			{
				Config: testAccDataSourceAwsEbsSnapshotIdsConfig_empty,
				Check: resource.ComposeTestCheckFunc(
					testAccCheckAwsEbsSnapshotDataSourceID("data.aws_ebs_snapshot_ids.empty"),
					resource.TestCheckResourceAttr("data.aws_ebs_snapshot_ids.empty", "ids.#", "0"),
				),
			},
		},
	})
}

const testAccDataSourceAwsEbsSnapshotIdsConfig_basic = `
resource "aws_ebs_volume" "test" {
    availability_zone = "us-west-2a"
    size              = 1
}

resource "aws_ebs_snapshot" "test" {
    volume_id = "${aws_ebs_volume.test.id}"
}

data "aws_ebs_snapshot_ids" "test" {
    owners = ["self"]
}
`

func testAccDataSourceAwsEbsSnapshotIdsConfig_sorted1(uuid string) string {
	return fmt.Sprintf(`
resource "aws_ebs_volume" "test" {
    availability_zone = "us-west-2a"
    size              = 1

    count = 2
}

resource "aws_ebs_snapshot" "a" {
    volume_id   = "${aws_ebs_volume.test.*.id[0]}"
    description = "tf-test-%s"
}

resource "aws_ebs_snapshot" "b" {
    volume_id   = "${aws_ebs_volume.test.*.id[1]}"
    description = "tf-test-%s"

    // We want to ensure that 'aws_ebs_snapshot.a.creation_date' is less than
    // 'aws_ebs_snapshot.b.creation_date'/ so that we can ensure that the
    // snapshots are being sorted correctly.
    depends_on = ["aws_ebs_snapshot.a"]
}
`, uuid, uuid)
}

func testAccDataSourceAwsEbsSnapshotIdsConfig_sorted2(uuid string) string {
	return testAccDataSourceAwsEbsSnapshotIdsConfig_sorted1(uuid) + fmt.Sprintf(`
data "aws_ebs_snapshot_ids" "test" {
    owners = ["self"]

    filter {
        name   = "description"
        values = ["tf-test-%s"]
    }
}
`, uuid)
}

const testAccDataSourceAwsEbsSnapshotIdsConfig_empty = `
data "aws_ebs_snapshot_ids" "empty" {
    owners = ["000000000000"]
}
`