add sqlAssertion rule (#11043) (#18559)

[upstream:32ed20cf417ff8c1ee86a713b03d39147d9c2da2]

Signed-off-by: Modular Magician <[email protected]>

Author: modular-magician

.changelog/11043.txt

@@ -0,0 +1,3 @@
+```release-note:enhancement
+dataplex: added `sql_assertion` field to `google_dataplex_datascan` resource
+```

google/services/dataplex/resource_dataplex_datascan.go

@@ -403,6 +403,21 @@ Only relevant if a minValue has been defined. Default = false.`,
 											},
 										},
 									},
+									"sql_assertion": {
+										Type:        schema.TypeList,
+										Optional:    true,
+										Description: `Table rule which evaluates whether any row matches invalid state.`,
+										MaxItems:    1,
+										Elem: &schema.Resource{
+											Schema: map[string]*schema.Schema{
+												"sql_statement": {
+													Type:        schema.TypeString,
+													Required:    true,
+													Description: `The SQL Statement.`,
+												},
+											},
+										},
+									},
 									"statistic_range_expectation": {
 										Type:        schema.TypeList,
 										Optional:    true,
@@ -1223,6 +1238,7 @@ func flattenDataplexDatascanDataQualitySpecRules(v interface{}, d *schema.Resour
 			"statistic_range_expectation": flattenDataplexDatascanDataQualitySpecRulesStatisticRangeExpectation(original["statisticRangeExpectation"], d, config),
 			"row_condition_expectation":   flattenDataplexDatascanDataQualitySpecRulesRowConditionExpectation(original["rowConditionExpectation"], d, config),
 			"table_condition_expectation": flattenDataplexDatascanDataQualitySpecRulesTableConditionExpectation(original["tableConditionExpectation"], d, config),
+			"sql_assertion":               flattenDataplexDatascanDataQualitySpecRulesSqlAssertion(original["sqlAssertion"], d, config),
 		})
 	}
 	return transformed
@@ -1411,6 +1427,23 @@ func flattenDataplexDatascanDataQualitySpecRulesTableConditionExpectationSqlExpr
 	return v
 }
 
+func flattenDataplexDatascanDataQualitySpecRulesSqlAssertion(v interface{}, d *schema.ResourceData, config *transport_tpg.Config) interface{} {
+	if v == nil {
+		return nil
+	}
+	original := v.(map[string]interface{})
+	if len(original) == 0 {
+		return nil
+	}
+	transformed := make(map[string]interface{})
+	transformed["sql_statement"] =
+		flattenDataplexDatascanDataQualitySpecRulesSqlAssertionSqlStatement(original["sqlStatement"], d, config)
+	return []interface{}{transformed}
+}
+func flattenDataplexDatascanDataQualitySpecRulesSqlAssertionSqlStatement(v interface{}, d *schema.ResourceData, config *transport_tpg.Config) interface{} {
+	return v
+}
+
 func flattenDataplexDatascanDataProfileSpec(v interface{}, d *schema.ResourceData, config *transport_tpg.Config) interface{} {
 	if v == nil {
 		return nil
@@ -1854,6 +1887,13 @@ func expandDataplexDatascanDataQualitySpecRules(v interface{}, d tpgresource.Ter
 			transformed["tableConditionExpectation"] = transformedTableConditionExpectation
 		}
 
+		transformedSqlAssertion, err := expandDataplexDatascanDataQualitySpecRulesSqlAssertion(original["sql_assertion"], d, config)
+		if err != nil {
+			return nil, err
+		} else if val := reflect.ValueOf(transformedSqlAssertion); val.IsValid() && !tpgresource.IsEmptyValue(val) {
+			transformed["sqlAssertion"] = transformedSqlAssertion
+		}
+
 		req = append(req, transformed)
 	}
 	return req, nil
@@ -2128,6 +2168,29 @@ func expandDataplexDatascanDataQualitySpecRulesTableConditionExpectationSqlExpre
 	return v, nil
 }
 
+func expandDataplexDatascanDataQualitySpecRulesSqlAssertion(v interface{}, d tpgresource.TerraformResourceData, config *transport_tpg.Config) (interface{}, error) {
+	l := v.([]interface{})
+	if len(l) == 0 || l[0] == nil {
+		return nil, nil
+	}
+	raw := l[0]
+	original := raw.(map[string]interface{})
+	transformed := make(map[string]interface{})
+
+	transformedSqlStatement, err := expandDataplexDatascanDataQualitySpecRulesSqlAssertionSqlStatement(original["sql_statement"], d, config)
+	if err != nil {
+		return nil, err
+	} else if val := reflect.ValueOf(transformedSqlStatement); val.IsValid() && !tpgresource.IsEmptyValue(val) {
+		transformed["sqlStatement"] = transformedSqlStatement
+	}
+
+	return transformed, nil
+}
+
+func expandDataplexDatascanDataQualitySpecRulesSqlAssertionSqlStatement(v interface{}, d tpgresource.TerraformResourceData, config *transport_tpg.Config) (interface{}, error) {
+	return v, nil
+}
+
 func expandDataplexDatascanDataProfileSpec(v interface{}, d tpgresource.TerraformResourceData, config *transport_tpg.Config) (interface{}, error) {
 	l := v.([]interface{})
 	if len(l) == 0 {

google/services/dataplex/resource_dataplex_datascan_generated_test.go

@@ -344,6 +344,13 @@ resource "google_dataplex_datascan" "full_quality" {
         sql_expression = "COUNT(*) > 0"
       }
     }
+    
+    rules {
+      dimension = "VALIDITY"
+      sql_assertion {
+        sql_statement = "select * from bigquery-public-data.austin_bikeshare.bikeshare_stations where station_id is null"
+      }
+    }
   }
 
 

website/docs/r/dataplex_datascan.html.markdown

@@ -239,6 +239,13 @@ resource "google_dataplex_datascan" "full_quality" {
         sql_expression = "COUNT(*) > 0"
       }
     }
+    
+    rules {
+      dimension = "VALIDITY"
+      sql_assertion {
+        sql_statement = "select * from bigquery-public-data.austin_bikeshare.bikeshare_stations where station_id is null"
+      }
+    }
   }
 
 
@@ -450,6 +457,11 @@ The following arguments are supported:
   Table rule which evaluates whether the provided expression is true.
   Structure is [documented below](#nested_table_condition_expectation).
 
+* `sql_assertion` -
+  (Optional)
+  Table rule which evaluates whether any row matches invalid state.
+  Structure is [documented below](#nested_sql_assertion).
+
 
 <a name="nested_range_expectation"></a>The `range_expectation` block supports:
 
@@ -522,6 +534,12 @@ The following arguments are supported:
   (Required)
   The SQL expression.
 
+<a name="nested_sql_assertion"></a>The `sql_assertion` block supports:
+
+* `sql_statement` -
+  (Required)
+  The SQL Statement.
+
 <a name="nested_data_profile_spec"></a>The `data_profile_spec` block supports:
 
 * `sampling_percent` -