environment resource can now also take public_key credential

Author: Adam Bowen

resource_delphix_environment.go

@@ -18,6 +18,7 @@ type Environment struct {
 	address      string
 	toolkitPath  string
 	serverID     string
+	publicKey    bool
 }
 
 func resourceDelphixEnvironment() *schema.Resource {
@@ -36,7 +37,7 @@ func resourceDelphixEnvironment() *schema.Resource {
 			},
 			"user_password": &schema.Schema{
 				Type:      schema.TypeString,
-				Required:  true,
+				Optional:  true,
 				ForceNew:  true,
 				Sensitive: true,
 			},
@@ -62,6 +63,11 @@ func resourceDelphixEnvironment() *schema.Resource {
 				Optional: true,
 				ForceNew: true,
 			},
+			"public_key": &schema.Schema{
+				Type:     schema.TypeBool,
+				Optional: true,
+				ForceNew: true,
+			},
 		},
 	}
 }
@@ -85,18 +91,28 @@ func resourceDelphixEnvironmentCreate(d *schema.ResourceData, meta interface{})
 		userPassword: d.Get("user_password").(string),
 		address:      d.Get("address").(string),
 		toolkitPath:  d.Get("toolkit_path").(string),
+		publicKey:    d.Get("public_key").(bool),
 	}
-	var reference interface{}
+	var reference, thisCrendential interface{}
+
 	client := meta.(*delphix.Client)
+
+	thisCrendential = &delphix.PasswordCredential{
+		Type:     "PasswordCredential",
+		Password: env.userPassword,
+	}
+	if env.publicKey == true {
+		thisCrendential = &delphix.SystemKeyCredential{
+			Type: "SystemKeyCredential",
+		}
+	}
+
 	environmentCreateParams := delphix.HostEnvironmentCreateParameters{
 		Type: "HostEnvironmentCreateParameters",
 		PrimaryUser: &delphix.EnvironmentUser{
-			Type: "EnvironmentUser",
-			Name: env.userName,
-			Credential: &delphix.PasswordCredential{
-				Type:     "PasswordCredential",
-				Password: env.userPassword,
-			},
+			Type:       "EnvironmentUser",
+			Name:       env.userName,
+			Credential: thisCrendential,
 		},
 		HostEnvironment: &delphix.UnixHostEnvironment{
 			Type:        "UnixHostEnvironment",

resource_delphix_environment_test.go

@@ -32,6 +32,27 @@ func TestAccEnvironmentDoesExistBasicCheck(t *testing.T) {
 	})
 }
 
+func TestAccEnvironmentDoesExistBasicCheckPublicKey(t *testing.T) {
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:     func() { testAccPreCheck(t) },
+		Providers:    testAccProviders,
+		CheckDestroy: testEnvironmentCheckDestroy,
+		Steps: []resource.TestStep{
+			resource.TestStep{
+				Config: testAccCheckEnvironmentConfigPublicKey(&testAccDelphixAdminConfig, &testAccEnvironment),
+				Check: resource.ComposeTestCheckFunc(
+					testEnvironmentCheckDoesExist(testAccEnvironment.name),
+					resource.TestCheckResourceAttr(
+						"delphix_environment.test_acc_env", "name", testAccEnvironment.name),
+					resource.TestCheckResourceAttr(
+						"delphix_environment.test_acc_env", "description", testAccEnvironment.description),
+				),
+			},
+		},
+	})
+}
+
 func TestAccEnvironmentDoesExist_UpdatedName(t *testing.T) {
 	updatedEnvironmentName := "UpdatedName"
 	updatedDescription := "UpdatedDescription"
@@ -183,6 +204,31 @@ func testAccCheckEnvironmentConfigImported(c *Config, e *Environment) string {
 	)
 }
 
+func testAccCheckEnvironmentConfigPublicKey(c *Config, e *Environment) string {
+	return fmt.Sprintf(`
+		provider "delphix" {
+		url = "${var.url}"
+		delphix_admin_username = "%s"
+		delphix_admin_password = "%s"
+		}
+
+		resource "delphix_environment" "test_acc_env" {
+		name = "%s"
+		description = "%s"
+		address = "%s"
+		user_name = "%s"
+		toolkit_path = "%s"
+		public_key = true
+		}
+
+		variable "url" {
+		default = "%s"
+		}
+		`,
+		c.username, c.password, e.name, e.description, e.address, e.userName, e.toolkitPath, c.url,
+	)
+}
+
 func testAccCheckEnvironmentConfigUpdatedName(c *Config, e *Environment, n string, d string) string {
 	return fmt.Sprintf(`
 		provider "delphix" {