prelude-sdk-beta 1426__tar.gz → 1428__tar.gz

{prelude_sdk_beta-1426 → prelude_sdk_beta-1428}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prelude-sdk-beta
-Version: 1426
+Version: 1428
 Summary: For interacting with the Prelude API
 Home-page: https://github.com/preludeorg
 Author: Prelude Research

{prelude_sdk_beta-1426 → prelude_sdk_beta-1428}/prelude_sdk_beta/models/codes.py

@@ -420,6 +420,9 @@ class PartnerEvents(Enum, metaclass=MissingItem):
     MISSING_DISK_ENCRYPTION_POLICY = 19
     MISSING_DISK_ENCRYPTION = 20
     NO_REGISTERED_DEVICES = 21
+    MISSING_DEVICE_COMPLIANCE_POLICY = 22
+    NONCOMPLIANT_DEVICE = 23
+    MISSING_ASR_POLICY = 24
 
     @classmethod
     def _missing_(cls, value):
@@ -434,7 +437,9 @@ class PartnerEvents(Enum, metaclass=MissingItem):
                 ControlCategory.IDENTITY,
                 ControlCategory.XDR,
             ],
+            PartnerEvents.MISSING_ASR_POLICY: [ControlCategory.ASSET_MANAGER],
             PartnerEvents.MISSING_AV_POLICY: [ControlCategory.XDR],
+            PartnerEvents.MISSING_DEVICE_COMPLIANCE_POLICY: [ControlCategory.ASSET_MANAGER],
             PartnerEvents.MISSING_DISK_ENCRYPTION: [ControlCategory.ASSET_MANAGER],
             PartnerEvents.MISSING_DISK_ENCRYPTION_POLICY: [
                 ControlCategory.ASSET_MANAGER
@@ -448,6 +453,7 @@ class PartnerEvents(Enum, metaclass=MissingItem):
             PartnerEvents.NO_REGISTERED_DEVICES: [ControlCategory.IDENTITY],
             PartnerEvents.NO_SERVER_MANAGER: [ControlCategory.ASSET_MANAGER],
             PartnerEvents.NO_VULN_MANAGER: [ControlCategory.VULN_MANAGER],
+            PartnerEvents.NONCOMPLIANT_DEVICE: [ControlCategory.ASSET_MANAGER],
             PartnerEvents.OUT_OF_DATE_FIRMWARE: [ControlCategory.NETWORK],
             PartnerEvents.OUT_OF_DATE_SCAN: [ControlCategory.VULN_MANAGER],
             PartnerEvents.REDUCED_FUNCTIONALITY_MODE: [ControlCategory.XDR],
@@ -479,6 +485,9 @@ class AlertTypes(Enum, metaclass=MissingItem):
     NEW_MISSING_DISK_ENCRYPTION_POLICY_ENDPOINTS = 20
     NEW_MISSING_DISK_ENCRYPTION_ENDPOINTS = 21
     NEW_NO_REGISTERED_DEVICES_USERS = 22
+    NEW_MISSING_DEVICE_COMPLIANCE_POLICY_ENDPOINTS = 23
+    NEW_NONCOMPLIANT_DEVICE_ENDPOINTS = 24
+    NEW_MISSING_ASR_POLICY_ENDPOINTS = 25
 
     @classmethod
     def _missing_(cls, value):
@@ -506,6 +515,7 @@ class PolicyType(Enum, metaclass=MissingItem):
     INTEL_TDT = 18
     INTEL_CHIP = 19
     DISK_ENCRYPTION = 20
+    ASR = 21
 
     @classmethod
     def _missing_(cls, value):

{prelude_sdk_beta-1426 → prelude_sdk_beta-1428}/prelude_sdk_beta.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prelude-sdk-beta
-Version: 1426
+Version: 1428
 Summary: For interacting with the Prelude API
 Home-page: https://github.com/preludeorg
 Author: Prelude Research

{prelude_sdk_beta-1426 → prelude_sdk_beta-1428}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = prelude-sdk-beta
-version = 1426
+version = 1428
 author = Prelude Research
 author_email = support@preludesecurity.com
 description = For interacting with the Prelude API

{prelude_sdk_beta-1426 → prelude_sdk_beta-1428}/tests/test_scm.py

@@ -168,6 +168,16 @@ class TestScmPerControl:
                 raise e
 
     def test_evaluation(self, unwrap, control):
+        def _wait_for_policies(control, instance_id, category):
+            timeout = time.time() + 300
+            while time.time() < timeout:
+                evaluation = unwrap(self.scm.evaluation)(self.scm, control, instance_id)
+                evaluation = evaluation[f"{category}_evaluation"]
+                if len(evaluation["policies"]) > 0:
+                    return evaluation
+                time.sleep(5)
+            assert False, "Timed out waiting for policies to show up in evaluation"
+
         instance_id = pytest.controls.get(control.value)
         assert instance_id
         evaluation = unwrap(self.scm.evaluation)(self.scm, control, instance_id)
@@ -178,7 +188,7 @@ class TestScmPerControl:
                 control.control_category == ControlCategory.XDR
                 or control == Control.INTUNE
             ):
-                assert len(evaluation["policies"]) > 0
+                evaluation = _wait_for_policies(control, instance_id, "endpoint")
                 assert {
                     "excepted",
                     "id",
@@ -194,19 +204,14 @@ class TestScmPerControl:
         elif "user_evaluation" in evaluation:
             evaluation = evaluation["user_evaluation"]
             assert {"policies"} == evaluation.keys()
-            assert len(evaluation["policies"]) > 0
-            assert {
-                "excepted",
-                "id",
-                "name",
-                "noncompliant_hostnames",
-                "settings",
-                "user_count",
-            } == evaluation["policies"][0].keys()
+            evaluation = _wait_for_policies(control, instance_id, "user")
+            assert {"excepted", "id", "name", "settings", "user_count"} == evaluation[
+                "policies"
+            ][0].keys()
         elif "inbox_evaluation" in evaluation:
             evaluation = evaluation["inbox_evaluation"]
             assert {"policies"} == evaluation.keys()
-            assert len(evaluation["policies"]) > 0
+            evaluation = _wait_for_policies(control, instance_id, "inbox")
             assert {"excepted", "id", "name", "settings", "inbox_count"} == evaluation[
                 "policies"
             ][0].keys()