terraform/builtin/providers/rancher/resource_rancher_stack_test.go

312 lines
9.9 KiB
Go
Raw Normal View History

package rancher
import (
"fmt"
"testing"
"github.com/hashicorp/terraform/helper/resource"
"github.com/hashicorp/terraform/terraform"
rancherClient "github.com/rancher/go-rancher/client"
)
func TestAccRancherStack_basic(t *testing.T) {
var stack rancherClient.Environment
resource.Test(t, resource.TestCase{
PreCheck: func() { testAccPreCheck(t) },
Providers: testAccProviders,
CheckDestroy: testAccCheckRancherStackDestroy,
Steps: []resource.TestStep{
resource.TestStep{
Config: testAccRancherStackConfig,
Check: resource.ComposeTestCheckFunc(
testAccCheckRancherStackExists("rancher_stack.foo", &stack),
resource.TestCheckResourceAttr("rancher_stack.foo", "name", "foo"),
resource.TestCheckResourceAttr("rancher_stack.foo", "description", "Terraform acc test group"),
resource.TestCheckResourceAttr("rancher_stack.foo", "catalog_id", ""),
resource.TestCheckResourceAttr("rancher_stack.foo", "docker_compose", ""),
resource.TestCheckResourceAttr("rancher_stack.foo", "rancher_compose", ""),
testAccCheckRancherStackAttributes(&stack, emptyEnvironment, false),
),
},
resource.TestStep{
Config: testAccRancherStackUpdateConfig,
Check: resource.ComposeTestCheckFunc(
testAccCheckRancherStackExists("rancher_stack.foo", &stack),
resource.TestCheckResourceAttr("rancher_stack.foo", "name", "foo2"),
resource.TestCheckResourceAttr("rancher_stack.foo", "description", "Terraform acc test group - updated"),
resource.TestCheckResourceAttr("rancher_stack.foo", "catalog_id", ""),
resource.TestCheckResourceAttr("rancher_stack.foo", "docker_compose", ""),
resource.TestCheckResourceAttr("rancher_stack.foo", "rancher_compose", ""),
testAccCheckRancherStackAttributes(&stack, emptyEnvironment, false),
),
},
},
})
}
func TestAccRancherStack_compose(t *testing.T) {
var stack rancherClient.Environment
resource.Test(t, resource.TestCase{
PreCheck: func() { testAccPreCheck(t) },
Providers: testAccProviders,
CheckDestroy: testAccCheckRancherStackDestroy,
Steps: []resource.TestStep{
resource.TestStep{
Config: testAccRancherStackComposeConfig,
Check: resource.ComposeTestCheckFunc(
testAccCheckRancherStackExists("rancher_stack.compose", &stack),
resource.TestCheckResourceAttr("rancher_stack.compose", "name", "compose"),
resource.TestCheckResourceAttr("rancher_stack.compose", "description", "Terraform acc test group - compose"),
resource.TestCheckResourceAttr("rancher_stack.compose", "catalog_id", ""),
resource.TestCheckResourceAttr("rancher_stack.compose", "docker_compose", "web: { image: nginx }"),
resource.TestCheckResourceAttr("rancher_stack.compose", "rancher_compose", "web: { scale: 1 }"),
testAccCheckRancherStackAttributes(&stack, emptyEnvironment, false),
),
},
},
})
}
//The following tests are run against the Default environment because
//upgrading a stack automatically starts the services which never
//completes if there is no host available
func TestAccRancherStack_catalog(t *testing.T) {
var stack rancherClient.Environment
resource.Test(t, resource.TestCase{
PreCheck: func() { testAccPreCheck(t) },
Providers: testAccProviders,
CheckDestroy: testAccCheckRancherStackDestroy,
Steps: []resource.TestStep{
resource.TestStep{
Config: testAccRancherStackSystemCatalogConfigInitial,
Check: resource.ComposeTestCheckFunc(
testAccCheckRancherStackExists("rancher_stack.catalog", &stack),
resource.TestCheckResourceAttr("rancher_stack.catalog", "name", "catalogInitial"),
resource.TestCheckResourceAttr("rancher_stack.catalog", "description", "Terraform acc test group - catalogInitial"),
resource.TestCheckResourceAttr("rancher_stack.catalog", "catalog_id", "community:janitor:0"),
resource.TestCheckResourceAttr("rancher_stack.catalog", "scope", "system"),
resource.TestCheckResourceAttr("rancher_stack.catalog", "docker_compose", ""),
resource.TestCheckResourceAttr("rancher_stack.catalog", "rancher_compose", ""),
resource.TestCheckResourceAttr("rancher_stack.catalog", "rendered_docker_compose", catalogDockerComposeInitial),
resource.TestCheckResourceAttr("rancher_stack.catalog", "rendered_rancher_compose", catalogRancherComposeInitial),
testAccCheckRancherStackAttributes(&stack, catalogEnvironment, true),
),
},
resource.TestStep{
Config: testAccRancherStackSystemCatalogConfigUpdate,
Check: resource.ComposeTestCheckFunc(
testAccCheckRancherStackExists("rancher_stack.catalog", &stack),
resource.TestCheckResourceAttr("rancher_stack.catalog", "name", "catalogUpdate"),
resource.TestCheckResourceAttr("rancher_stack.catalog", "description", "Terraform acc test group - catalogUpdate"),
resource.TestCheckResourceAttr("rancher_stack.catalog", "catalog_id", "community:janitor:1"),
resource.TestCheckResourceAttr("rancher_stack.catalog", "scope", "user"),
resource.TestCheckResourceAttr("rancher_stack.catalog", "docker_compose", ""),
resource.TestCheckResourceAttr("rancher_stack.catalog", "rancher_compose", ""),
resource.TestCheckResourceAttr("rancher_stack.catalog", "rendered_docker_compose", catalogDockerComposeUpdate),
resource.TestCheckResourceAttr("rancher_stack.catalog", "rendered_rancher_compose", catalogRancherComposeUpdate),
testAccCheckRancherStackAttributes(&stack, catalogEnvironmentUpgrade, true),
),
},
},
})
}
func testAccCheckRancherStackExists(n string, stack *rancherClient.Environment) resource.TestCheckFunc {
return func(s *terraform.State) error {
rs, ok := s.RootModule().Resources[n]
if !ok {
return fmt.Errorf("Not found: %s", n)
}
if rs.Primary.ID == "" {
return fmt.Errorf("No App Name is set")
}
client := testAccProvider.Meta().(*Config)
foundStack, err := client.Environment.ById(rs.Primary.ID)
if err != nil {
return err
}
if foundStack.Resource.Id != rs.Primary.ID {
return fmt.Errorf("Stack not found")
}
*stack = *foundStack
return nil
}
}
func testAccCheckRancherStackAttributes(stack *rancherClient.Environment, environment map[string]string, startOnCreate bool) resource.TestCheckFunc {
return func(s *terraform.State) error {
if len(stack.Environment) != len(environment) {
return fmt.Errorf("Bad environment size: %v should be: %v", len(stack.Environment), environment)
}
for k, v := range stack.Environment {
if environment[k] != v {
return fmt.Errorf("Bad environment value for %s: %s should be: %s", k, environment[k], v)
}
}
if stack.StartOnCreate != startOnCreate {
return fmt.Errorf("Bad startOnCreate: %t should be: %t", stack.StartOnCreate, startOnCreate)
}
return nil
}
}
func testAccCheckRancherStackDestroy(s *terraform.State) error {
client := testAccProvider.Meta().(*Config)
for _, rs := range s.RootModule().Resources {
if rs.Type != "rancher_stack" {
continue
}
stack, err := client.Environment.ById(rs.Primary.ID)
if err == nil {
if stack != nil &&
stack.Resource.Id == rs.Primary.ID &&
stack.State != "removed" {
return fmt.Errorf("Stack still exists")
}
}
return nil
}
return nil
}
const testAccRancherStackConfig = `
resource "rancher_environment" "foo" {
name = "foo"
}
resource "rancher_stack" "foo" {
name = "foo"
description = "Terraform acc test group"
environment_id = "${rancher_environment.foo.id}"
}
`
const testAccRancherStackUpdateConfig = `
resource "rancher_environment" "foo" {
name = "foo"
}
resource "rancher_stack" "foo" {
name = "foo2"
description = "Terraform acc test group - updated"
environment_id = "${rancher_environment.foo.id}"
}
`
const testAccRancherStackComposeConfig = `
resource "rancher_environment" "foo" {
name = "foo"
}
resource "rancher_stack" "compose" {
name = "compose"
description = "Terraform acc test group - compose"
environment_id = "${rancher_environment.foo.id}"
docker_compose = "web: { image: nginx }"
rancher_compose = "web: { scale: 1 }"
}
`
const testAccRancherStackSystemCatalogConfigInitial = `
resource "rancher_stack" "catalog" {
name = "catalogInitial"
description = "Terraform acc test group - catalogInitial"
environment_id = "1a5"
catalog_id = "community:janitor:0"
scope = "system"
start_on_create = true
environment {
EXCLUDE_LABEL = "cleanup=false"
FREQUENCY = "60"
KEEP = "rancher/agent:*"
}
}
`
const testAccRancherStackSystemCatalogConfigUpdate = `
resource "rancher_stack" "catalog" {
name = "catalogUpdate"
description = "Terraform acc test group - catalogUpdate"
environment_id = "1a5"
catalog_id = "community:janitor:1"
scope = "user"
environment {
EXCLUDE_LABEL = "cleanup=false"
FREQUENCY = "60"
KEEP = "rancher/agent:*"
KEEPC = "*:*"
}
}
`
var catalogDockerComposeInitial = `cleanup:
environment:
CLEAN_PERIOD: '60'
DELAY_TIME: '900'
KEEP_IMAGES: rancher/agent:*
labels:
io.rancher.scheduler.global: 'true'
io.rancher.scheduler.affinity:host_label_ne: cleanup=false
tty: true
image: meltwater/docker-cleanup:1.4.0
privileged: true
volumes:
- /var/run/docker.sock:/var/run/docker.sock
- /var/lib/docker:/var/lib/docker
stdin_open: true
`
const catalogRancherComposeInitial = `{}
`
const catalogDockerComposeUpdate = `cleanup:
environment:
CLEAN_PERIOD: '60'
DELAY_TIME: '900'
KEEP_CONTAINERS: '*:*'
KEEP_IMAGES: rancher/agent:*
labels:
io.rancher.scheduler.global: 'true'
io.rancher.scheduler.affinity:host_label_ne: cleanup=false
image: sshipway/docker-cleanup:1.5.2
volumes:
- /var/run/docker.sock:/var/run/docker.sock
- /var/lib/docker:/var/lib/docker
net: none
`
const catalogRancherComposeUpdate = `{}
`
var emptyEnvironment = map[string]string{}
var catalogEnvironment = map[string]string{
"EXCLUDE_LABEL": "cleanup=false",
"FREQUENCY": "60",
"KEEP": "rancher/agent:*",
}
var catalogEnvironmentUpgrade = map[string]string{
"EXCLUDE_LABEL": "cleanup=false",
"FREQUENCY": "60",
"KEEP": "rancher/agent:*",
"KEEPC": "*:*",
}