def test_alerts_bulk_threat_error(monkeypatch): """Test error raise from bulk threat update status""" api = CBCloudAPI(url="https://example.com", token="ABCD/1234", org_key="Z100", ssl_verify=True) with pytest.raises(ApiError): api.bulk_threat_dismiss([123], "Fixed", "Yessir")
def test_alerts_bulk_dismiss_threat(monkeypatch): _was_called = False def _do_dismiss(url, body, **kwargs): nonlocal _was_called assert url == "/appservices/v6/orgs/Z100/threat/workflow/_criteria" assert body == {"threat_id": ["B0RG", "F3R3NG1"], "state": "DISMISSED", "remediation_state": "Fixed", "comment": "Yessir"} _was_called = True return StubResponse({"request_id": "497ABX"}) api = CBCloudAPI(url="https://example.com", token="ABCD/1234", org_key="Z100", ssl_verify=True) patch_cbapi(monkeypatch, api, POST=_do_dismiss) reqid = api.bulk_threat_dismiss(["B0RG", "F3R3NG1"], "Fixed", "Yessir") assert _was_called assert reqid == "497ABX"