Exemplo n.º 1
0
def test_enable_alerts(cbcsdk_mock):
    """Testing Watchlist.enable_alerts()."""
    api = cbcsdk_mock.api
    id = "watchlistId"
    cbcsdk_mock.mock_request("GET", f"/threathunter/watchlistmgr/v2/watchlist/{id}", WATCHLIST_GET_SPECIFIC_RESP)
    watchlist = Watchlist(api, model_unique_id="watchlistId")
    cbcsdk_mock.mock_request("PUT", f"/threathunter/watchlistmgr/v3/orgs/test/watchlists/{id}/alert", {"alert": True})
    watchlist.enable_alerts()
Exemplo n.º 2
0
def test_enable_alerts_no_id(cbcsdk_mock):
    """Testing Watchlist.enable_alerts() raising InvalidObjectError when ID is missing."""
    api = cbcsdk_mock.api
    watchlist = Watchlist(api, model_unique_id=None)
    with pytest.raises(InvalidObjectError):
        watchlist.enable_alerts()