141 lines
3.4 KiB
Go
141 lines
3.4 KiB
Go
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{
|
|
PreCheck: func() { testAccPreCheck(t) },
|
|
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 != "<any>" && out != val {
|
|
return fmt.Errorf("Attribute '%s' value '%s' != '%s'", attr, out, val)
|
|
}
|
|
if val == "<any>" && 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
|
|
}
|
|
}
|
|
`
|