package google

import (
	"fmt"
	"testing"

	"github.com/hashicorp/terraform/helper/acctest"
	"github.com/hashicorp/terraform/helper/resource"
	"github.com/hashicorp/terraform/terraform"
)

func TestAccDnsRecordSet_basic(t *testing.T) {
	resource.Test(t, resource.TestCase{
		PreCheck:     func() { testAccPreCheck(t) },
		Providers:    testAccProviders,
		CheckDestroy: testAccCheckDnsRecordSetDestroy,
		Steps: []resource.TestStep{
			resource.TestStep{
				Config: testAccDnsRecordSet_basic,
				Check: resource.ComposeTestCheckFunc(
					testAccCheckDnsRecordSetExists(
						"google_dns_record_set.foobar"),
				),
			},
		},
	})
}

func testAccCheckDnsRecordSetDestroy(s *terraform.State) error {
	config := testAccProvider.Meta().(*Config)

	for _, rs := range s.RootModule().Resources {
		// Deletion of the managed_zone implies everything is gone
		if rs.Type == "google_dns_managed_zone" {
			_, err := config.clientDns.ManagedZones.Get(
				config.Project, rs.Primary.ID).Do()
			if err == nil {
				return fmt.Errorf("DNS ManagedZone still exists")
			}
		}
	}

	return nil
}

func testAccCheckDnsRecordSetExists(name string) resource.TestCheckFunc {
	return func(s *terraform.State) error {
		rs, ok := s.RootModule().Resources[name]
		if !ok {
			return fmt.Errorf("Not found: %s", name)
		}

		dnsName := rs.Primary.Attributes["name"]
		dnsType := rs.Primary.Attributes["type"]

		if rs.Primary.ID == "" {
			return fmt.Errorf("No ID is set")
		}

		config := testAccProvider.Meta().(*Config)

		resp, err := config.clientDns.ResourceRecordSets.List(
			config.Project, "terraform-test-zone").Name(dnsName).Type(dnsType).Do()
		if err != nil {
			return fmt.Errorf("Error confirming DNS RecordSet existence: %#v", err)
		}
		if len(resp.Rrsets) == 0 {
			// The resource doesn't exist anymore
			return fmt.Errorf("DNS RecordSet not found")
		}

		if len(resp.Rrsets) > 1 {
			return fmt.Errorf("Only expected 1 record set, got %d", len(resp.Rrsets))
		}

		return nil
	}
}

var testAccDnsRecordSet_basic = fmt.Sprintf(`
resource "google_dns_managed_zone" "parent-zone" {
	name = "dnsrecord-test-%s"
	dns_name = "terraform.test."
	description = "Test Description"
}
resource "google_dns_record_set" "foobar" {
	managed_zone = "${google_dns_managed_zone.parent-zone.name}"
	name = "test-record.terraform.test."
	type = "A"
	rrdatas = ["127.0.0.1", "127.0.0.10"]
	ttl = 600
}
`, acctest.RandString(10))