prelude-sdk 2.6.35__tar.gz → 2.6.37__tar.gz

{prelude_sdk-2.6.35 → prelude_sdk-2.6.37}/PKG-INFO

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

{prelude_sdk-2.6.35 → prelude_sdk-2.6.37}/prelude_sdk/controllers/detect_controller.py

@@ -197,7 +197,7 @@ class DetectController(HttpController):
         res = self.post(
             f"{self.account.hq}/detect/threat_hunts/{threat_hunt_id}",
             headers=self.account.headers,
-            timeout=10,
+            timeout=30,
         )
         return res.json()
 

{prelude_sdk-2.6.35 → prelude_sdk-2.6.37}/prelude_sdk/controllers/iam_controller.py

@@ -30,13 +30,21 @@ class IAMAccountController(HttpController):
         return res.json()
 
     @verify_credentials
-    def update_account(self, mode: Mode = None, company: str = None, slug: str = None):
+    def update_account(
+        self,
+        company: str = None,
+        inactivity_timeout: int = None,
+        mode: Mode = None,
+        slug: str = None,
+    ):
         """Update properties on an account"""
         body = dict()
         if mode is not None:
             body["mode"] = mode.name
         if company is not None:
             body["company"] = company
+        if inactivity_timeout is not None:
+            body["inactivity_timeout"] = inactivity_timeout
         if slug is not None:
             body["slug"] = slug
 

{prelude_sdk-2.6.35 → prelude_sdk-2.6.37}/prelude_sdk/controllers/scm_controller.py

@@ -642,23 +642,25 @@ class ScmController(HttpController):
         sort_by: str,
         group_by: str,
         group_limit: int,
-        scopes: str = None,
-        filter: str = None,
+        display_overrides: dict = None,
+        odata_filter: str = None,
+        scopes: dict = None,
     ):
         """Get SCM chart data"""
-        params = {
-            "scm_category": scm_category.name,
-            "sort_by": sort_by,
+        body = {
+            "category": scm_category.name,
+            "display_overrides": display_overrides,
             "group_by": group_by,
             "group_limit": group_limit,
             "scopes": scopes,
+            "sort_by": sort_by,
         }
-        if filter:
-            params["$filter"] = filter
-        res = self.get(
-            f"{self.account.hq}/scm/chart_data",
+        if odata_filter:
+            body["$filter"] = odata_filter
+        res = self.post(
+            f"{self.account.hq}/scm/reports/data",
             headers=self.account.headers,
-            params=params,
+            json=body,
             timeout=30,
         )
         return res.json()

{prelude_sdk-2.6.35 → prelude_sdk-2.6.37}/prelude_sdk/models/codes.py

@@ -533,13 +533,22 @@ class PartnerEvents(Enum, metaclass=MissingItem):
             PartnerEvents.MISSING_SERVER_MANAGER: [ControlCategory.ASSET_MANAGER],
             PartnerEvents.MISSING_VULN_MANAGER: [ControlCategory.VULN_MANAGER],
             PartnerEvents.MISSING_VULN_SCAN: [ControlCategory.VULN_MANAGER],
-            PartnerEvents.NO_ASR_POLICY: [ControlCategory.ASSET_MANAGER],
-            PartnerEvents.NO_AV_POLICY: [ControlCategory.XDR],
+            PartnerEvents.NO_ASR_POLICY: [
+                ControlCategory.ASSET_MANAGER,
+                ControlCategory.XDR,
+            ],
+            PartnerEvents.NO_AV_POLICY: [
+                ControlCategory.ASSET_MANAGER,
+                ControlCategory.XDR,
+            ],
             PartnerEvents.NO_DEVICE_COMPLIANCE_POLICY: [ControlCategory.ASSET_MANAGER],
             PartnerEvents.NO_DISK_ENCRYPTION: [ControlCategory.ASSET_MANAGER],
             PartnerEvents.NO_DISK_ENCRYPTION_POLICY: [ControlCategory.ASSET_MANAGER],
             PartnerEvents.NO_EDR_POLICY: [ControlCategory.XDR],
-            PartnerEvents.NO_HOST_FIREWALL_POLICY: [ControlCategory.ASSET_MANAGER],
+            PartnerEvents.NO_HOST_FIREWALL_POLICY: [
+                ControlCategory.ASSET_MANAGER,
+                ControlCategory.XDR,
+            ],
             PartnerEvents.NO_REGISTERED_DEVICES: [ControlCategory.IDENTITY],
             PartnerEvents.NONCOMPLIANT: [ControlCategory.ASSET_MANAGER],
             PartnerEvents.OUT_OF_DATE_FIRMWARE: [ControlCategory.NETWORK],

{prelude_sdk-2.6.35 → prelude_sdk-2.6.37}/prelude_sdk.egg-info/PKG-INFO

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

{prelude_sdk-2.6.35 → prelude_sdk-2.6.37}/setup.cfg

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

{prelude_sdk-2.6.35 → prelude_sdk-2.6.37}/tests/test_iam.py

@@ -17,7 +17,7 @@ class TestIAM:
     def setup_class(self):
         self.iam_account = IAMAccountController(pytest.account)
         self.iam_user = IAMUserController(pytest.account)
-        self.company = "prelude"
+        self.company = "prelude-sdk-qa"
         self.service_user = "registration"
 
     def test_get_account(self, unwrap):
@@ -29,7 +29,7 @@ class TestIAM:
         res = unwrap(self.iam_user.list_accounts)(self.iam_user)
         assert pytest.expected_account["account_id"] in [a["account_id"] for a in res]
 
-    def test_create_service_user(self, unwrap):
+    def test_create_second_service_user(self, unwrap):
         service_user = unwrap(self.iam_account.create_service_user)(
             self.iam_account, name=self.service_user
         )
@@ -56,7 +56,7 @@ class TestIAM:
         diffs = check_dict_items(pytest.expected_account, res)
         assert not diffs, json.dumps(diffs, indent=2)
 
-    def test_delete_service_user(self, unwrap):
+    def test_delete_second_service_user(self, unwrap):
         unwrap(self.iam_account.delete_service_user)(
             self.iam_account, handle=pytest.second_service_user_handle
         )

{prelude_sdk-2.6.35 → prelude_sdk-2.6.37}/tests/test_partner.py

@@ -498,3 +498,21 @@ class TestPartner:
         assert res["pagination"]["offset"] == 0
         assert res["pagination"]["total"] > 0
         pytest.crowdstrike_advisory_id = res["advisories"][0]["id"]
+
+    def test_delete_endpoint(
+        self,
+        unwrap,
+        host,
+        edr_id,
+        control,
+        os,
+        platform,
+        policy,
+        policy_name,
+        webhook_keys,
+        group_id,
+    ):
+        unwrap(self.detect.delete_endpoint)(self.detect, pytest.endpoint["endpoint_id"])
+        res = unwrap(self.detect.list_endpoints)(self.detect)
+        sorted_res = {r["serial_num"]: r for r in res}
+        assert pytest.endpoint["serial_num"] not in sorted_res

{prelude_sdk-2.6.35 → prelude_sdk-2.6.37}/tests/test_scm.py

@@ -200,6 +200,17 @@ class TestScmPerControl:
                     "endpoint_count",
                     "success_count",
                 } == evaluation["policies"][0].keys()
+            elif control == Control.INTEL_INTUNE:
+                evaluation = _wait_for_policies(control, instance_id, "endpoint")
+                assert {
+                    "active_count",
+                    "excepted",
+                    "id",
+                    "name",
+                    "not_supported_count",
+                    "supported_count",
+                    "techniques",
+                } == evaluation["policies"][0].keys()
             else:
                 assert len(evaluation["policies"]) == 0
         elif "user_evaluation" in evaluation: