package consul import ( "fmt" "testing" consulapi "github.com/hashicorp/consul/api" "github.com/hashicorp/terraform/helper/resource" "github.com/hashicorp/terraform/terraform" ) func TestAccConsulKeys_basic(t *testing.T) { resource.Test(t, resource.TestCase{ Providers: testAccProviders, CheckDestroy: testAccCheckConsulKeysDestroy, Steps: []resource.TestStep{ resource.TestStep{ Config: testAccConsulKeysConfig, Check: resource.ComposeTestCheckFunc( testAccCheckConsulKeysExists(), testAccCheckConsulKeysValue("consul_keys.app", "enabled", "true"), testAccCheckConsulKeysValue("consul_keys.app", "set", "acceptance"), testAccCheckConsulKeysValue("consul_keys.app", "remove_one", "hello"), ), }, resource.TestStep{ Config: testAccConsulKeysConfig_Update, Check: resource.ComposeTestCheckFunc( testAccCheckConsulKeysExists(), testAccCheckConsulKeysValue("consul_keys.app", "enabled", "true"), testAccCheckConsulKeysValue("consul_keys.app", "set", "acceptanceUpdated"), testAccCheckConsulKeysRemoved("consul_keys.app", "remove_one"), ), }, }, }) } func testAccCheckConsulKeysDestroy(s *terraform.State) error { kv := testAccProvider.Meta().(*consulapi.Client).KV() opts := &consulapi.QueryOptions{Datacenter: "dc1"} pair, _, err := kv.Get("test/set", opts) if err != nil { return err } if pair != nil { return fmt.Errorf("Key still exists: %#v", pair) } return nil } func testAccCheckConsulKeysExists() resource.TestCheckFunc { return func(s *terraform.State) error { kv := testAccProvider.Meta().(*consulapi.Client).KV() opts := &consulapi.QueryOptions{Datacenter: "dc1"} pair, _, err := kv.Get("test/set", opts) if err != nil { return err } if pair == nil { return fmt.Errorf("Key 'test/set' does not exist") } return nil } } func testAccCheckConsulKeysValue(n, attr, val string) resource.TestCheckFunc { return func(s *terraform.State) error { rn, ok := s.RootModule().Resources[n] if !ok { return fmt.Errorf("Resource not found") } out, ok := rn.Primary.Attributes["var."+attr] if !ok { return fmt.Errorf("Attribute '%s' not found: %#v", attr, rn.Primary.Attributes) } if val != "" && out != val { return fmt.Errorf("Attribute '%s' value '%s' != '%s'", attr, out, val) } if val == "" && out == "" { return fmt.Errorf("Attribute '%s' value '%s'", attr, out) } return nil } } func testAccCheckConsulKeysRemoved(n, attr string) resource.TestCheckFunc { return func(s *terraform.State) error { rn, ok := s.RootModule().Resources[n] if !ok { return fmt.Errorf("Resource not found") } _, ok = rn.Primary.Attributes["var."+attr] if ok { return fmt.Errorf("Attribute '%s' still present: %#v", attr, rn.Primary.Attributes) } return nil } } const testAccConsulKeysConfig = ` resource "consul_keys" "app" { datacenter = "dc1" key { name = "enabled" path = "test/enabled" default = "true" } key { name = "set" path = "test/set" value = "acceptance" delete = true } key { name = "remove_one" path = "test/remove_one" value = "hello" delete = true } } ` const testAccConsulKeysConfig_Update = ` resource "consul_keys" "app" { datacenter = "dc1" key { name = "enabled" path = "test/enabled" default = "true" } key { name = "set" path = "test/set" value = "acceptanceUpdated" delete = true } } `