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"
