add test on json marshhal for CheckToolUsage operations (#413)

OpsLevel · Jun 7, 2024 · 7271e99 · 7271e99
1 parent fd943cd
commit 7271e99
Showing 1 changed file with 80 additions and 2 deletions.
diff --git a/check_test.go b/check_test.go
@@ -743,7 +743,7 @@ func TestDeleteCheck(t *testing.T) {
 	autopilot.Equals(t, nil, err)
 }
 
-func TestJsonUnmarshalCreateCheck(t *testing.T) {
+func TestJsonUnmarshalCreateCheckManual(t *testing.T) {
 	// Arrange
 	data := `{
 	"name": "Example",
@@ -762,7 +762,46 @@ func TestJsonUnmarshalCreateCheck(t *testing.T) {
 	autopilot.Equals(t, &output, buf1.(*ol.CheckManualCreateInput))
 }
 
-func TestJsonUnmarshalUpdateCheck(t *testing.T) {
+func TestJsonUnmarshalCreateCheckToolUsage(t *testing.T) {
+	// Arrange
+	data := `{
+	"name": "Example",
+	"notes": "Example Notes",
+	"environmentPredicate": {
+    "type": "exists"
+  },
+  "toolNamePredicate": {
+    "type": "contains",
+    "value": "go"
+  },
+  "toolUrlPredicate": {
+    "type": "starts_with",
+    "value": "https"
+  }
+}`
+	output := ol.CheckToolUsageCreateInput{
+		Name:  "Example",
+		Notes: ol.RefOf("Example Notes"),
+		EnvironmentPredicate: &ol.PredicateInput{
+			Type: ol.PredicateTypeEnum("exists"),
+		},
+		ToolNamePredicate: &ol.PredicateInput{
+			Type:  ol.PredicateTypeEnum("contains"),
+			Value: ol.RefOf("go"),
+		},
+		ToolUrlPredicate: &ol.PredicateInput{
+			Type:  ol.PredicateTypeEnum("starts_with"),
+			Value: ol.RefOf("https"),
+		},
+	}
+	// Act
+	buf1, err1 := ol.UnmarshalCheckCreateInput(ol.CheckTypeToolUsage, []byte(data))
+	// Assert
+	autopilot.Ok(t, err1)
+	autopilot.Equals(t, &output, buf1.(*ol.CheckToolUsageCreateInput))
+}
+
+func TestJsonUnmarshalUpdateCheckManual(t *testing.T) {
 	// Arrange
 	data := `{
 	"name": "Example",
@@ -780,3 +819,42 @@ func TestJsonUnmarshalUpdateCheck(t *testing.T) {
 	autopilot.Ok(t, err1)
 	autopilot.Equals(t, &output, buf1.(*ol.CheckManualUpdateInput))
 }
+
+func TestJsonUnmarshalUpdateCheckToolUsage(t *testing.T) {
+	// Arrange
+	data := `{
+	"name": "Example",
+	"notes": "Updated Notes",
+	"environmentPredicate": {
+    "type": "exists"
+  },
+  "toolNamePredicate": {
+    "type": "contains",
+    "value": "go"
+  },
+  "toolUrlPredicate": {
+    "type": "starts_with",
+    "value": "https"
+  }
+}`
+	output := ol.CheckToolUsageUpdateInput{
+		Name:  ol.RefOf("Example"),
+		Notes: ol.RefOf("Updated Notes"),
+		EnvironmentPredicate: &ol.PredicateUpdateInput{
+			Type: ol.RefOf(ol.PredicateTypeEnum("exists")),
+		},
+		ToolNamePredicate: &ol.PredicateUpdateInput{
+			Type:  ol.RefOf(ol.PredicateTypeEnum("contains")),
+			Value: ol.RefOf("go"),
+		},
+		ToolUrlPredicate: &ol.PredicateUpdateInput{
+			Type:  ol.RefOf(ol.PredicateTypeEnum("starts_with")),
+			Value: ol.RefOf("https"),
+		},
+	}
+	// Act
+	buf1, err1 := ol.UnmarshalCheckUpdateInput(ol.CheckTypeToolUsage, []byte(data))
+	// Assert
+	autopilot.Ok(t, err1)
+	autopilot.Equals(t, &output, buf1.(*ol.CheckToolUsageUpdateInput))
+}