chore(nhi): add vault_path and vault_path_count attributes to PoliyBreak model

Author: amascia-gg

pygitguardian/models.py

@@ -234,6 +234,8 @@ class PolicyBreakSchema(BaseSchema):
         DiffKind, by_value=True, required=False, load_default=None, dump_default=None
     )
     is_vaulted = fields.Boolean(required=False, load_default=False, dump_default=False)
+    vault_path = fields.String(required=False, load_default=None, dump_default=None)
+    vault_path_count = fields.Int(required=False, load_default=None, dump_default=None)
 
     @post_load
     def make_policy_break(self, data: Dict[str, Any], **kwargs: Any) -> "PolicyBreak":
@@ -265,6 +267,8 @@ def __init__(
         exclude_reason: Optional[str] = None,
         diff_kind: Optional[DiffKind] = None,
         is_vaulted: bool = False,
+        vault_path: Optional[str] = None,
+        vault_path_count: Optional[int] = None,
         **kwargs: Any,
     ) -> None:
         super().__init__()
@@ -281,6 +285,8 @@ def __init__(
         self.exclude_reason = exclude_reason
         self.diff_kind = diff_kind
         self.is_vaulted = is_vaulted
+        self.vault_path = vault_path
+        self.vault_path_count = vault_path_count
 
     @property
     def is_secret(self) -> bool:

tests/test_models.py

@@ -171,6 +171,8 @@ def test_document_handle_surrogates(self):
                     "exclude_reason": None,
                     "diff_kind": None,
                     "is_vaulted": False,
+                    "vault_path": None,
+                    "vault_path_count": None,
                 },
             ),
             (
@@ -190,6 +192,8 @@ def test_document_handle_surrogates(self):
                     "exclude_reason": None,
                     "diff_kind": "addition",
                     "is_vaulted": True,
+                    "vault_path": "hashicorpvault:my-vault:my-secret",
+                    "vault_path_count": 3,
                 },
             ),
             (