add unit tests

2026-11-12 21:20:34 +00:00 · 2026-03-14 16:22:35 -07:00
parent b082630ec1
commit 6ac7664d4f
4 changed files with 243 additions and 61 deletions
@@ -330,3 +330,66 @@ pub struct ApplyPatchApprovalRequestEvent {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub grant_root: Option<PathBuf>,
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use pretty_assertions::assert_eq;
+
+    #[test]
+    fn command_prompts_advertise_override_decision() {
+        let command = vec!["echo".to_string(), "hi".to_string()];
+
+        let decisions =
+            ExecApprovalRequestEvent::default_available_decisions(&command, None, None, None, None);
+
+        assert_eq!(
+            decisions,
+            vec![
+                ReviewDecision::Approved,
+                ReviewDecision::ApprovedOverrideCommand {
+                    command: command.clone(),
+                },
+                ReviewDecision::Abort,
+            ]
+        );
+    }
+
+    #[test]
+    fn network_only_prompts_do_not_advertise_override_decision() {
+        let decisions = ExecApprovalRequestEvent::default_available_decisions(
+            &["echo".to_string(), "hi".to_string()],
+            Some(&NetworkApprovalContext {
+                host: "example.com".to_string(),
+                protocol: NetworkApprovalProtocol::Https,
+            }),
+            None,
+            Some(&[
+                NetworkPolicyAmendment {
+                    host: "example.com".to_string(),
+                    action: NetworkPolicyRuleAction::Allow,
+                },
+                NetworkPolicyAmendment {
+                    host: "example.com".to_string(),
+                    action: NetworkPolicyRuleAction::Deny,
+                },
+            ]),
+            None,
+        );
+
+        assert_eq!(
+            decisions,
+            vec![
+                ReviewDecision::Approved,
+                ReviewDecision::ApprovedForSession,
+                ReviewDecision::NetworkPolicyAmendment {
+                    network_policy_amendment: NetworkPolicyAmendment {
+                        host: "example.com".to_string(),
+                        action: NetworkPolicyRuleAction::Allow,
+                    },
+                },
+                ReviewDecision::Abort,
+            ]
+        );
+    }
+}