Merge pull request #1197 from jlu-cloudflare/master

Add CheckDisks for disk encryption posture rule
cloudflare · Mar 8, 2023 · d2d0904 · d2d0904
2 parents ce804ed + dd14318
commit d2d0904
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 20 deletions.
diff --git a/.changelog/1197.txt b/.changelog/1197.txt
@@ -0,0 +1,3 @@
+```release-note:enhancement
+teams_account: add support for `check_disks`
+```
diff --git a/device_posture_rule.go b/device_posture_rule.go
@@ -162,25 +162,26 @@ type DevicePostureRuleMatch struct {
 
 // DevicePostureRuleInput represents the value to be checked against.
 type DevicePostureRuleInput struct {
-	ID               string `json:"id,omitempty"`
-	Path             string `json:"path,omitempty"`
-	Exists           bool   `json:"exists,omitempty"`
-	Thumbprint       string `json:"thumbprint,omitempty"`
-	Sha256           string `json:"sha256,omitempty"`
-	Running          bool   `json:"running,omitempty"`
-	RequireAll       bool   `json:"requireAll,omitempty"`
-	Enabled          bool   `json:"enabled,omitempty"`
-	Version          string `json:"version,omitempty"`
-	VersionOperator  string `json:"versionOperator,omitempty"`
-	Overall          string `json:"overall,omitempty"`
-	SensorConfig     string `json:"sensor_config,omitempty"`
-	Os               string `json:"os,omitempty"`
-	OsDistroName     string `json:"os_distro_name,omitempty"`
-	OsDistroRevision string `json:"os_distro_revision,omitempty"`
-	Operator         string `json:"operator,omitempty"`
-	Domain           string `json:"domain,omitempty"`
-	ComplianceStatus string `json:"compliance_status,omitempty"`
-	ConnectionID     string `json:"connection_id,omitempty"`
+	ID               string   `json:"id,omitempty"`
+	Path             string   `json:"path,omitempty"`
+	Exists           bool     `json:"exists,omitempty"`
+	Thumbprint       string   `json:"thumbprint,omitempty"`
+	Sha256           string   `json:"sha256,omitempty"`
+	Running          bool     `json:"running,omitempty"`
+	RequireAll       bool     `json:"requireAll,omitempty"`
+	CheckDisks       []string `json:"checkDisks,omitempty"`
+	Enabled          bool     `json:"enabled,omitempty"`
+	Version          string   `json:"version,omitempty"`
+	VersionOperator  string   `json:"versionOperator,omitempty"`
+	Overall          string   `json:"overall,omitempty"`
+	SensorConfig     string   `json:"sensor_config,omitempty"`
+	Os               string   `json:"os,omitempty"`
+	OsDistroName     string   `json:"os_distro_name,omitempty"`
+	OsDistroRevision string   `json:"os_distro_revision,omitempty"`
+	Operator         string   `json:"operator,omitempty"`
+	Domain           string   `json:"domain,omitempty"`
+	ComplianceStatus string   `json:"compliance_status,omitempty"`
+	ConnectionID     string   `json:"connection_id,omitempty"`
 }
 
 // DevicePostureRuleListResponse represents the response from the list

diff --git a/device_posture_rule_test.go b/device_posture_rule_test.go
@@ -402,7 +402,8 @@ func TestDevicePostureDiskEncryptionRule(t *testing.T) {
 					}
 				],
 				"input": {
-					"requireAll": true
+					"requireAll": true,
+					"checkDisks": ["C", "D"]
 				}
 			}
 		}
@@ -419,6 +420,7 @@ func TestDevicePostureDiskEncryptionRule(t *testing.T) {
 		Match:       []DevicePostureRuleMatch{{Platform: "ios"}},
 		Input: DevicePostureRuleInput{
 			RequireAll: true,
+			CheckDisks: []string{"C", "D"},
 		},
 	}