terraform/builtin/providers/mysql/resource_database_test.go

92 lines
2.2 KiB
Go
Raw Normal View History

package mysql
import (
"fmt"
"strings"
"testing"
mysqlc "github.com/ziutek/mymysql/mysql"
"github.com/hashicorp/terraform/helper/resource"
"github.com/hashicorp/terraform/terraform"
)
func TestAccDatabase(t *testing.T) {
var dbName string
resource.Test(t, resource.TestCase{
PreCheck: func() { testAccPreCheck(t) },
Providers: testAccProviders,
CheckDestroy: testAccDatabaseCheckDestroy(dbName),
Steps: []resource.TestStep{
resource.TestStep{
Config: testAccDatabaseConfig_basic,
Check: testAccDatabaseCheck(
"mysql_database.test", &dbName,
),
},
},
})
}
func testAccDatabaseCheck(rn string, name *string) resource.TestCheckFunc {
return func(s *terraform.State) error {
rs, ok := s.RootModule().Resources[rn]
if !ok {
return fmt.Errorf("resource not found: %s", rn)
}
if rs.Primary.ID == "" {
return fmt.Errorf("database id not set")
}
conn := testAccProvider.Meta().(mysqlc.Conn)
rows, _, err := conn.Query("SHOW CREATE DATABASE terraform_acceptance_test")
if err != nil {
return fmt.Errorf("error reading database: %s", err)
}
if len(rows) != 1 {
return fmt.Errorf("expected 1 row reading database but got %d", len(rows))
}
row := rows[0]
createSQL := string(row[1].([]byte))
if strings.Index(createSQL, "CHARACTER SET utf8") == -1 {
return fmt.Errorf("database default charset isn't utf8")
}
if strings.Index(createSQL, "COLLATE utf8_bin") == -1 {
return fmt.Errorf("database default collation isn't utf8_bin")
}
*name = rs.Primary.ID
return nil
}
}
func testAccDatabaseCheckDestroy(name string) resource.TestCheckFunc {
return func(s *terraform.State) error {
conn := testAccProvider.Meta().(mysqlc.Conn)
_, _, err := conn.Query("SHOW CREATE DATABASE terraform_acceptance_test")
if err == nil {
return fmt.Errorf("database still exists after destroy")
}
if mysqlErr, ok := err.(*mysqlc.Error); ok {
if mysqlErr.Code == mysqlc.ER_BAD_DB_ERROR {
return nil
}
}
return fmt.Errorf("got unexpected error: %s", err)
}
}
const testAccDatabaseConfig_basic = `
resource "mysql_database" "test" {
name = "terraform_acceptance_test"
default_character_set = "utf8"
default_collation = "utf8_bin"
}
`