Splitting the SKU assignment out

hashicorp · Sep 5, 2017 · a25d57b · a25d57b
1 parent 5ebf58b
commit a25d57b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/azurerm/resource_arm_postgresql_server.go b/azurerm/resource_arm_postgresql_server.go
@@ -253,8 +253,8 @@ func resourceArmPostgreSQLServerRead(d *schema.ResourceData, meta interface{}) e
 	d.Set("version", string(resp.Version))
 	d.Set("storage_mb", int(*resp.StorageMB))
 	d.Set("ssl_enforcement", string(resp.SslEnforcement))
+	d.Set("sku", flattenPostgreSQLServerSku(resp.Sku))
 
-	flattenAndSetAzureRmPostgreSQLServerSku(d, resp.Sku)
 	flattenAndSetTags(d, resp.Tags)
 
 	// Computed
@@ -295,13 +295,13 @@ func expandAzureRmPostgreSQLServerSku(d *schema.ResourceData, storageMB int) *po
 	}
 }
 
-func flattenAndSetAzureRmPostgreSQLServerSku(d *schema.ResourceData, resp *postgresql.Sku) {
+func flattenPostgreSQLServerSku(d *schema.ResourceData, resp *postgresql.Sku) []interface{} {
 	values := map[string]interface{}{}
 
 	values["name"] = *resp.Name
 	values["capacity"] = int(*resp.Capacity)
 	values["tier"] = string(resp.Tier)
 
 	sku := []interface{}{values}
-	d.Set("sku", sku)
+	return sku
 }