Added warehouse_type parameter to databricks_sql_endpoint to support PRO SKU (#1728)

Author: alexott

docs/resources/sql_endpoint.md

@@ -39,6 +39,7 @@ The following arguments are supported:
 * `enable_serverless_compute` - Whether this SQL endpoint is a Serverless endpoint. To use a Serverless SQL endpoint, you must enable Serverless SQL endpoints for the workspace.
 * `channel` block, consisting of following fields:
   * `name` - Name of the Databricks SQL release channel. Possible values are: `CHANNEL_NAME_PREVIEW` and `CHANNEL_NAME_CURRENT`. Default is `CHANNEL_NAME_CURRENT`.
+* `warehouse_type` - [SQL Warehouse Type](https://docs.databricks.com/sql/admin/sql-endpoints.html#switch-the-sql-warehouse-type-pro-classic-or-serverless): `PRO` (default) or `CLASSIC`.  If Serverless SQL is enabled, you can only specify `PRO`.
 
 ## Attribute Reference
 

sql/resource_sql_endpoint.go

@@ -37,6 +37,7 @@ type SQLEndpoint struct {
 	Tags                    *Tags           `json:"tags,omitempty" tf:"suppress_diff"`
 	SpotInstancePolicy      string          `json:"spot_instance_policy,omitempty" tf:"default:COST_OPTIMIZED"`
 	Channel                 *ReleaseChannel `json:"channel,omitempty" tf:"suppress_diff"`
+	WarehouseType           string          `json:"warehouse_type,omitempty" tf:"default:PRO"`
 
 	// The data source ID is not part of the endpoint API response.
 	// We manually resolve it by retrieving the list of data sources
@@ -76,7 +77,6 @@ type Tag struct {
 //
 // Note: this object returns more fields than contained in this struct,
 // but we only list the ones that are in use here.
-//
 type DataSource struct {
 	ID         string `json:"id"`
 	EndpointID string `json:"endpoint_id"`
@@ -198,6 +198,8 @@ func ResourceSqlEndpoint() *schema.Resource {
 			validation.StringInSlice(ClusterSizes, false))
 		m["max_num_clusters"].ValidateDiagFunc = validation.ToDiagFunc(
 			validation.IntBetween(1, MaxNumClusters))
+		m["warehouse_type"].ValidateDiagFunc = validation.ToDiagFunc(
+			validation.StringInSlice([]string{"PRO", "CLASSIC"}, false))
 		return m
 	})
 	return common.Resource{

sql/resource_sql_endpoint_test.go

@@ -81,6 +81,7 @@ func TestResourceSQLEndpointCreate(t *testing.T) {
 					NumClusters:        1,
 					EnablePhoton:       true,
 					SpotInstancePolicy: "COST_OPTIMIZED",
+					WarehouseType:      "PRO",
 				},
 				Response: SQLEndpoint{
 					ID: "abc",
@@ -128,6 +129,7 @@ func TestResourceSQLEndpointCreateNoAutoTermination(t *testing.T) {
 					NumClusters:        1,
 					EnablePhoton:       true,
 					SpotInstancePolicy: "COST_OPTIMIZED",
+					WarehouseType:      "PRO",
 				},
 				Response: SQLEndpoint{
 					ID: "abc",
@@ -228,6 +230,7 @@ func TestResourceSQLEndpointUpdate(t *testing.T) {
 					NumClusters:        1,
 					EnablePhoton:       true,
 					SpotInstancePolicy: "COST_OPTIMIZED",
+					WarehouseType:      "PRO",
 				},
 			},
 			{