terraform/builtin/providers/heroku/resource_heroku_app_test.go

374 lines
9.0 KiB
Go
Raw Normal View History

2014-07-23 06:03:30 +02:00
package heroku
import (
"context"
2014-07-23 06:03:30 +02:00
"fmt"
"os"
2014-07-23 06:03:30 +02:00
"testing"
"github.com/cyberdelia/heroku-go/v3"
"github.com/hashicorp/terraform/helper/acctest"
2014-07-23 06:03:30 +02:00
"github.com/hashicorp/terraform/helper/resource"
"github.com/hashicorp/terraform/terraform"
)
func TestAccHerokuApp_Basic(t *testing.T) {
var app heroku.AppInfoResult
appName := fmt.Sprintf("tftest-%s", acctest.RandString(10))
2014-07-23 06:03:30 +02:00
resource.Test(t, resource.TestCase{
PreCheck: func() { testAccPreCheck(t) },
Providers: testAccProviders,
CheckDestroy: testAccCheckHerokuAppDestroy,
Steps: []resource.TestStep{
{
Config: testAccCheckHerokuAppConfig_basic(appName),
2014-07-23 06:03:30 +02:00
Check: resource.ComposeTestCheckFunc(
testAccCheckHerokuAppExists("heroku_app.foobar", &app),
testAccCheckHerokuAppAttributes(&app, appName),
2014-07-23 06:03:30 +02:00
resource.TestCheckResourceAttr(
"heroku_app.foobar", "name", appName),
resource.TestCheckResourceAttr(
"heroku_app.foobar", "config_vars.0.FOO", "bar"),
2014-07-23 06:03:30 +02:00
),
},
},
})
}
func TestAccHerokuApp_NameChange(t *testing.T) {
var app heroku.AppInfoResult
appName := fmt.Sprintf("tftest-%s", acctest.RandString(10))
appName2 := fmt.Sprintf("%s-v2", appName)
resource.Test(t, resource.TestCase{
PreCheck: func() { testAccPreCheck(t) },
Providers: testAccProviders,
CheckDestroy: testAccCheckHerokuAppDestroy,
Steps: []resource.TestStep{
{
Config: testAccCheckHerokuAppConfig_basic(appName),
Check: resource.ComposeTestCheckFunc(
testAccCheckHerokuAppExists("heroku_app.foobar", &app),
testAccCheckHerokuAppAttributes(&app, appName),
resource.TestCheckResourceAttr(
"heroku_app.foobar", "name", appName),
resource.TestCheckResourceAttr(
"heroku_app.foobar", "config_vars.0.FOO", "bar"),
),
},
{
Config: testAccCheckHerokuAppConfig_updated(appName2),
Check: resource.ComposeTestCheckFunc(
testAccCheckHerokuAppExists("heroku_app.foobar", &app),
testAccCheckHerokuAppAttributesUpdated(&app, appName2),
resource.TestCheckResourceAttr(
"heroku_app.foobar", "name", appName2),
resource.TestCheckResourceAttr(
"heroku_app.foobar", "config_vars.0.FOO", "bing"),
resource.TestCheckResourceAttr(
"heroku_app.foobar", "config_vars.0.BAZ", "bar"),
),
},
},
})
}
func TestAccHerokuApp_NukeVars(t *testing.T) {
var app heroku.AppInfoResult
appName := fmt.Sprintf("tftest-%s", acctest.RandString(10))
resource.Test(t, resource.TestCase{
PreCheck: func() { testAccPreCheck(t) },
Providers: testAccProviders,
CheckDestroy: testAccCheckHerokuAppDestroy,
Steps: []resource.TestStep{
{
Config: testAccCheckHerokuAppConfig_basic(appName),
Check: resource.ComposeTestCheckFunc(
testAccCheckHerokuAppExists("heroku_app.foobar", &app),
testAccCheckHerokuAppAttributes(&app, appName),
resource.TestCheckResourceAttr(
"heroku_app.foobar", "name", appName),
resource.TestCheckResourceAttr(
"heroku_app.foobar", "config_vars.0.FOO", "bar"),
),
},
{
Config: testAccCheckHerokuAppConfig_no_vars(appName),
Check: resource.ComposeTestCheckFunc(
testAccCheckHerokuAppExists("heroku_app.foobar", &app),
testAccCheckHerokuAppAttributesNoVars(&app, appName),
resource.TestCheckResourceAttr(
"heroku_app.foobar", "name", appName),
resource.TestCheckNoResourceAttr(
"heroku_app.foobar", "config_vars.0.FOO"),
),
},
},
})
}
func TestAccHerokuApp_Organization(t *testing.T) {
var app heroku.OrganizationApp
appName := fmt.Sprintf("tftest-%s", acctest.RandString(10))
org := os.Getenv("HEROKU_ORGANIZATION")
resource.Test(t, resource.TestCase{
PreCheck: func() {
testAccPreCheck(t)
if org == "" {
t.Skip("HEROKU_ORGANIZATION is not set; skipping test.")
}
},
Providers: testAccProviders,
CheckDestroy: testAccCheckHerokuAppDestroy,
Steps: []resource.TestStep{
{
Config: testAccCheckHerokuAppConfig_organization(appName, org),
Check: resource.ComposeTestCheckFunc(
testAccCheckHerokuAppExistsOrg("heroku_app.foobar", &app),
testAccCheckHerokuAppAttributesOrg(&app, appName, org),
),
},
},
})
}
2014-07-23 06:03:30 +02:00
func testAccCheckHerokuAppDestroy(s *terraform.State) error {
client := testAccProvider.Meta().(*heroku.Service)
2014-07-23 06:03:30 +02:00
for _, rs := range s.RootModule().Resources {
2014-07-23 06:03:30 +02:00
if rs.Type != "heroku_app" {
continue
}
_, err := client.AppInfo(context.TODO(), rs.Primary.ID)
2014-07-23 06:03:30 +02:00
if err == nil {
return fmt.Errorf("App still exists")
}
}
return nil
}
func testAccCheckHerokuAppAttributes(app *heroku.AppInfoResult, appName string) resource.TestCheckFunc {
2014-07-23 06:03:30 +02:00
return func(s *terraform.State) error {
client := testAccProvider.Meta().(*heroku.Service)
if app.Region.Name != "us" {
return fmt.Errorf("Bad region: %s", app.Region.Name)
}
if app.Stack.Name != "cedar-14" {
return fmt.Errorf("Bad stack: %s", app.Stack.Name)
}
if app.Name != appName {
return fmt.Errorf("Bad name: %s", app.Name)
}
2014-07-23 06:03:30 +02:00
vars, err := client.ConfigVarInfoForApp(context.TODO(), app.Name)
if err != nil {
return err
}
if vars["FOO"] == nil || *vars["FOO"] != "bar" {
return fmt.Errorf("Bad config vars: %v", vars)
}
2014-07-23 06:03:30 +02:00
return nil
}
}
func testAccCheckHerokuAppAttributesUpdated(app *heroku.AppInfoResult, appName string) resource.TestCheckFunc {
return func(s *terraform.State) error {
client := testAccProvider.Meta().(*heroku.Service)
if app.Name != appName {
return fmt.Errorf("Bad name: %s", app.Name)
}
vars, err := client.ConfigVarInfoForApp(context.TODO(), app.Name)
if err != nil {
return err
}
// Make sure we kept the old one
if vars["FOO"] == nil || *vars["FOO"] != "bing" {
return fmt.Errorf("Bad config vars: %v", vars)
}
if vars["BAZ"] == nil || *vars["BAZ"] != "bar" {
return fmt.Errorf("Bad config vars: %v", vars)
}
return nil
}
}
func testAccCheckHerokuAppAttributesNoVars(app *heroku.AppInfoResult, appName string) resource.TestCheckFunc {
return func(s *terraform.State) error {
client := testAccProvider.Meta().(*heroku.Service)
if app.Name != appName {
return fmt.Errorf("Bad name: %s", app.Name)
}
vars, err := client.ConfigVarInfoForApp(context.TODO(), app.Name)
if err != nil {
return err
}
if len(vars) != 0 {
return fmt.Errorf("vars exist: %v", vars)
}
return nil
}
}
func testAccCheckHerokuAppAttributesOrg(app *heroku.OrganizationApp, appName string, org string) resource.TestCheckFunc {
return func(s *terraform.State) error {
client := testAccProvider.Meta().(*heroku.Service)
if app.Region.Name != "us" {
return fmt.Errorf("Bad region: %s", app.Region.Name)
}
if app.Stack.Name != "cedar-14" {
return fmt.Errorf("Bad stack: %s", app.Stack.Name)
}
if app.Name != appName {
return fmt.Errorf("Bad name: %s", app.Name)
}
if app.Organization == nil || app.Organization.Name != org {
return fmt.Errorf("Bad org: %v", app.Organization)
}
vars, err := client.ConfigVarInfoForApp(context.TODO(), app.Name)
if err != nil {
return err
}
if vars["FOO"] == nil || *vars["FOO"] != "bar" {
return fmt.Errorf("Bad config vars: %v", vars)
}
return nil
}
}
func testAccCheckHerokuAppExists(n string, app *heroku.AppInfoResult) resource.TestCheckFunc {
2014-07-23 06:03:30 +02:00
return func(s *terraform.State) error {
rs, ok := s.RootModule().Resources[n]
2014-07-23 06:03:30 +02:00
if !ok {
return fmt.Errorf("Not found: %s", n)
}
if rs.Primary.ID == "" {
2014-07-23 06:03:30 +02:00
return fmt.Errorf("No App Name is set")
}
client := testAccProvider.Meta().(*heroku.Service)
2014-07-23 06:03:30 +02:00
foundApp, err := client.AppInfo(context.TODO(), rs.Primary.ID)
2014-07-23 06:03:30 +02:00
if err != nil {
return err
}
if foundApp.Name != rs.Primary.ID {
2014-07-23 06:03:30 +02:00
return fmt.Errorf("App not found")
}
*app = *foundApp
2014-07-23 06:03:30 +02:00
return nil
}
}
func testAccCheckHerokuAppExistsOrg(n string, app *heroku.OrganizationApp) 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().(*heroku.Service)
foundApp, err := client.OrganizationAppInfo(context.TODO(), rs.Primary.ID)
if err != nil {
return err
}
if foundApp.Name != rs.Primary.ID {
return fmt.Errorf("App not found")
}
*app = *foundApp
return nil
}
}
func testAccCheckHerokuAppConfig_basic(appName string) string {
return fmt.Sprintf(`
2014-07-23 06:03:30 +02:00
resource "heroku_app" "foobar" {
name = "%s"
region = "us"
config_vars {
FOO = "bar"
}
}`, appName)
}
func testAccCheckHerokuAppConfig_updated(appName string) string {
return fmt.Sprintf(`
resource "heroku_app" "foobar" {
name = "%s"
region = "us"
config_vars {
FOO = "bing"
BAZ = "bar"
}
}`, appName)
}
func testAccCheckHerokuAppConfig_no_vars(appName string) string {
return fmt.Sprintf(`
resource "heroku_app" "foobar" {
name = "%s"
region = "us"
}`, appName)
}
func testAccCheckHerokuAppConfig_organization(appName, org string) string {
return fmt.Sprintf(`
resource "heroku_app" "foobar" {
name = "%s"
region = "us"
organization {
name = "%s"
}
config_vars {
FOO = "bar"
}
}`, appName, org)
}