Add table checks and tests (#92)

simple-machines · May 9, 2024 · fdc2887 · fdc2887
2 parents c19ef40 + 580932d
commit fdc2887
Show file tree

Hide file tree

Showing 6 changed files with 481 additions and 120 deletions.
diff --git a/client/data_source_user.go b/client/data_source_user.go
@@ -34,7 +34,7 @@ func DataSourceUser() *schema.Resource {
 				Type:     schema.TypeList,
 				Computed: true,
 				Elem: &schema.Schema{
-					Type:         schema.TypeString,
+					Type: schema.TypeString,
 				},
 			},
 		},

diff --git a/client/models.go b/client/models.go
@@ -67,11 +67,35 @@ type ColumnKind struct {
 	Units *string `json:"units,omitempty"`
 }
 
+const (
+	ColumnConstraint_NOT_NULL            = "notnull"
+	ColumnConstraint_UNIQUE              = "uniqueness"
+	ColumnConstraint_NOT_CONSTANT        = "notconstant"
+	ColumnConstraint_ACCEPTED_VALUES     = "acceptedvalues"
+	ColumnConstraint_IN_RANGE            = "inrange"
+	ColumnConstraint_STATISTICS_IN_RANGE = "statisticsinrange"
+	ColumnConstraint_ROW_CHECK           = "rowcheck"
+	ColumnConstraint_AGGREGATE_CHECK     = "aggregatecheck"
+)
+
+type ColumnConstraint struct {
+	Type         string               `json:"adt_type"`
+	Name         *string              `json:"name,omitempty"`
+	Threshold    *float64             `json:"threshold"`
+	Min          *string              `json:"min"`
+	Max          *string              `json:"max"`
+	Expression   *SQLExpression       `json:"expression"`
+	Aggregation  *AggregateExpression `json:"aggregation"`
+	PerPartition *bool                `json:"perPartition,omitempty"`
+	Acceptable   []string             `json:"ok,omitempty"`
+}
+
 // ColumnInfo ..
 type ColumnInfo struct {
 	Description string                `json:"description"`
 	Column      *ColumnRepresentation `json:"column"`
 	Kind        *ColumnKind           `json:"kind,omitempty"`
+	Constraints []ColumnConstraint    `json:"constraints,omitempty"`
 }
 
 // Table ...
@@ -542,6 +566,7 @@ type MonitoringPlan struct {
 	Type     string `json:"adt_type"`
 	Tables   []int  `json:"tables,omitempty"`
 	Excluded []int  `json:"excluded"`
+	FullScan *bool  `json:"fullScan"`
 }
 
 // TableMonitoring ...

diff --git a/client/resource_metrics_set.go b/client/resource_metrics_set.go
@@ -376,7 +376,7 @@ func buildDimensions(d *schema.ResourceData) ([]Dimension, error) {
 		name := value["name"].(string)
 		expression := value["expression"].(string)
 		var filter *string
-		if filterRaw, _ := value["filter"].(string); filterRaw != ""  {
+		if filterRaw, _ := value["filter"].(string); filterRaw != "" {
 			filter = &filterRaw
 		}
 		Dimension := Dimension{