provider/google: SQL user resource, documentation & tests

This commit is contained in:
Lars Wander 2016-01-13 16:33:08 -05:00
parent 39ae38c627
commit 68351ed44d
3 changed files with 326 additions and 0 deletions

View File

@ -70,6 +70,7 @@ func Provider() terraform.ResourceProvider {
"google_dns_record_set": resourceDnsRecordSet(),
"google_sql_database": resourceSqlDatabase(),
"google_sql_database_instance": resourceSqlDatabaseInstance(),
"google_sql_user": resourceSqlUser(),
"google_pubsub_topic": resourcePubsubTopic(),
"google_pubsub_subscription": resourcePubsubSubscription(),
"google_storage_bucket": resourceStorageBucket(),

183
resource_sql_user.go Normal file
View File

@ -0,0 +1,183 @@
package google
import (
"fmt"
"log"
"github.com/hashicorp/terraform/helper/schema"
"google.golang.org/api/googleapi"
"google.golang.org/api/sqladmin/v1beta4"
)
func resourceSqlUser() *schema.Resource {
return &schema.Resource{
Create: resourceSqlUserCreate,
Read: resourceSqlUserRead,
Update: resourceSqlUserUpdate,
Delete: resourceSqlUserDelete,
Schema: map[string]*schema.Schema{
"name": &schema.Schema{
Type: schema.TypeString,
Required: true,
ForceNew: true,
},
"password": &schema.Schema{
Type: schema.TypeString,
Required: true,
},
"host": &schema.Schema{
Type: schema.TypeString,
Required: true,
ForceNew: true,
},
"instance": &schema.Schema{
Type: schema.TypeString,
Required: true,
ForceNew: true,
},
},
}
}
func resourceSqlUserCreate(d *schema.ResourceData, meta interface{}) error {
config := meta.(*Config)
name := d.Get("name").(string)
instance := d.Get("instance").(string)
password := d.Get("password").(string)
host := d.Get("host").(string)
project := config.Project
user := &sqladmin.User{
Name: name,
Instance: instance,
Password: password,
Host: host,
}
op, err := config.clientSqlAdmin.Users.Insert(project, instance,
user).Do()
if err != nil {
return fmt.Errorf("Error, failed to insert "+
"user %s into instance %s: %s", name, instance, err)
}
err = sqladminOperationWait(config, op, "Insert User")
if err != nil {
return fmt.Errorf("Error, failure waiting for insertion of %s "+
"into %s: %s", name, instance, err)
}
return resourceSqlUserRead(d, meta)
}
func resourceSqlUserRead(d *schema.ResourceData, meta interface{}) error {
config := meta.(*Config)
name := d.Get("name").(string)
instance := d.Get("instance").(string)
project := config.Project
users, err := config.clientSqlAdmin.Users.List(project, instance).Do()
if err != nil {
if gerr, ok := err.(*googleapi.Error); ok && gerr.Code == 404 {
log.Printf("[WARN] Removing SQL User %q because it's gone", d.Get("name").(string))
d.SetId("")
return nil
}
return fmt.Errorf("Error, failed to get user %s in instance %s: %s", name, instance, err)
}
found := false
for _, user := range users.Items {
if user.Name == name {
found = true
break
}
}
if !found {
log.Printf("[WARN] Removing SQL User %q because it's gone", d.Get("name").(string))
d.SetId("")
return nil
}
d.SetId(name)
return nil
}
func resourceSqlUserUpdate(d *schema.ResourceData, meta interface{}) error {
config := meta.(*Config)
if d.HasChange("password") {
name := d.Get("name").(string)
instance := d.Get("instance").(string)
host := d.Get("host").(string)
password := d.Get("password").(string)
project := config.Project
user := &sqladmin.User{
Name: name,
Instance: instance,
Password: password,
Host: host,
}
op, err := config.clientSqlAdmin.Users.Update(project, instance, host, name,
user).Do()
if err != nil {
return fmt.Errorf("Error, failed to update"+
"user %s into user %s: %s", name, instance, err)
}
err = sqladminOperationWait(config, op, "Insert User")
if err != nil {
return fmt.Errorf("Error, failure waiting for update of %s "+
"in %s: %s", name, instance, err)
}
return resourceSqlUserRead(d, meta)
}
return nil
}
func resourceSqlUserDelete(d *schema.ResourceData, meta interface{}) error {
config := meta.(*Config)
name := d.Get("name").(string)
instance := d.Get("instance").(string)
host := d.Get("host").(string)
project := config.Project
op, err := config.clientSqlAdmin.Users.Delete(project, instance, host, name).Do()
if err != nil {
return fmt.Errorf("Error, failed to delete"+
"user %s in instance %s: %s", name,
instance, err)
}
err = sqladminOperationWait(config, op, "Delete User")
if err != nil {
return fmt.Errorf("Error, failure waiting for deletion of %s "+
"in %s: %s", name, instance, err)
}
return nil
}

142
resource_sql_user_test.go Normal file
View File

@ -0,0 +1,142 @@
package google
import (
"fmt"
"testing"
"github.com/hashicorp/terraform/helper/acctest"
"github.com/hashicorp/terraform/helper/resource"
"github.com/hashicorp/terraform/terraform"
)
func TestAccGoogleSqlUser_basic(t *testing.T) {
user := acctest.RandString(10)
instance := acctest.RandString(10)
resource.Test(t, resource.TestCase{
PreCheck: func() { testAccPreCheck(t) },
Providers: testAccProviders,
CheckDestroy: testAccGoogleSqlUserDestroy,
Steps: []resource.TestStep{
resource.TestStep{
Config: testGoogleSqlUser_basic(instance, user),
Check: resource.ComposeTestCheckFunc(
testAccCheckGoogleSqlUserExists("google_sql_user.user"),
),
},
},
})
}
func TestAccGoogleSqlUser_update(t *testing.T) {
user := acctest.RandString(10)
instance := acctest.RandString(10)
resource.Test(t, resource.TestCase{
PreCheck: func() { testAccPreCheck(t) },
Providers: testAccProviders,
CheckDestroy: testAccGoogleSqlUserDestroy,
Steps: []resource.TestStep{
resource.TestStep{
Config: testGoogleSqlUser_basic(instance, user),
Check: resource.ComposeTestCheckFunc(
testAccCheckGoogleSqlUserExists("google_sql_user.user"),
),
},
resource.TestStep{
Config: testGoogleSqlUser_basic2(instance, user),
Check: resource.ComposeTestCheckFunc(
testAccCheckGoogleSqlUserExists("google_sql_user.user"),
),
},
},
})
}
func testAccCheckGoogleSqlUserExists(n string) resource.TestCheckFunc {
return func(s *terraform.State) error {
config := testAccProvider.Meta().(*Config)
rs, ok := s.RootModule().Resources[n]
if !ok {
return fmt.Errorf("Resource not found: %s", n)
}
name := rs.Primary.Attributes["name"]
instance := rs.Primary.Attributes["instance"]
host := rs.Primary.Attributes["host"]
users, err := config.clientSqlAdmin.Users.List(config.Project,
instance).Do()
for _, user := range users.Items {
if user.Name == name && user.Host == host {
return nil
}
}
return fmt.Errorf("Not found: %s: %s", n, err)
}
}
func testAccGoogleSqlUserDestroy(s *terraform.State) error {
for _, rs := range s.RootModule().Resources {
config := testAccProvider.Meta().(*Config)
if rs.Type != "google_sql_database" {
continue
}
name := rs.Primary.Attributes["name"]
instance := rs.Primary.Attributes["instance"]
host := rs.Primary.Attributes["host"]
users, err := config.clientSqlAdmin.Users.List(config.Project,
instance).Do()
for _, user := range users.Items {
if user.Name == name && user.Host == host {
return fmt.Errorf("User still %s exists %s", name, err)
}
}
return nil
}
return nil
}
func testGoogleSqlUser_basic(instance, user string) string {
return fmt.Sprintf(`
resource "google_sql_database_instance" "instance" {
name = "i%s"
region = "us-central"
settings {
tier = "D0"
}
}
resource "google_sql_user" "user" {
name = "user%s"
instance = "${google_sql_database_instance.instance.name}"
host = "google.com"
password = "hunter2"
}
`, instance, user)
}
func testGoogleSqlUser_basic2(instance, user string) string {
return fmt.Sprintf(`
resource "google_sql_database_instance" "instance" {
name = "i%s"
region = "us-central"
settings {
tier = "D0"
}
}
resource "google_sql_user" "user" {
name = "user%s"
instance = "${google_sql_database_instance.instance.name}"
host = "google.com"
password = "oops"
}
`, instance, user)
}