def test_try_create_search_with_retry__sad(mocker): """ Create an event search with a connection error first, and succesful try after Given: - Event fetch is to be created via the qradar client When: - Search first returns ConnectionError - Search then returns search object Then: - Assert search is created with id and status """ client = QRadarClient("", {}, {"identifier": "*", "password": "******"}) events_query = "" offense = RAW_RESPONSES["fetch-incidents"] max_retries = 0 exception_raised = False mocker.patch.object( client, "search", side_effect=[ConnectionError, RAW_RESPONSES["qradar-searches"]]) try: try_create_search_with_retry(client, events_query, offense, max_retries) except DemistoException: exception_raised = True assert exception_raised
def test_try_create_search_with_retry__semi_happy(mocker): """ Create an event search with a connection error first, and succesful try after Given: - Event fetch is to be created via the qradar client When: - Search first returns ConnectionError - Search then returns search object Then: - Assert search is created with id and status """ client = QRadarClient("", {}, {"identifier": "*", "password": "******"}) events_query = "" offense = RAW_RESPONSES["fetch-incidents"] max_retries = 3 mocker.patch.object( client, "search", side_effect=[ConnectionError, RAW_RESPONSES["qradar-searches"]]) actual_status, actual_id = try_create_search_with_retry( client, events_query, offense, max_retries) assert actual_status == "EXECUTE" assert actual_id == "a135f4cb-c22a-4b3a-aa7d-83058c219d33"