def test_validate_fetch_incidents_params_when_first_fetch_is_invalid(self):
"""Test case scenario when argument named first_fetch is invalid."""
from Flashpoint import validate_fetch_incidents_params
with pytest.raises(ValueError) as err:
validate_fetch_incidents_params({"first_fetch": "abc"}, {})
assert str(err.value) == INVALID_DATE_MESSAGE
with pytest.raises(ValueError) as err:
validate_fetch_incidents_params({"first_fetch": None}, {})
assert str(err.value) == MESSAGES['INVALID_FIRST_FETCH']
def test_validate_fetch_incidents_params_when_max_fetch_is_invalid(self):
"""Test case scenario when argument named max_fetch is invalid."""
from Flashpoint import validate_fetch_incidents_params
with pytest.raises(ValueError) as err:
validate_fetch_incidents_params({"max_fetch": "abc"}, {})
assert str(err.value) == '"abc" is not a valid number'
with pytest.raises(ValueError) as err:
validate_fetch_incidents_params({"max_fetch": ""}, {})
assert str(err.value) == MESSAGES['INVALID_MAX_FETCH'].format('None')
def test_validate_fetch_incidents_params_when_valid_params_are_provided(
self):
"""Test case scenario when the arguments provided are valid."""
from Flashpoint import validate_fetch_incidents_params
params = {
'fetch_type': 'Alerts',
'first_fetch': START_DATE,
'max_fetch': '20',
'is_fresh_compromised_credentials': False
}
fetch_params = {
'size': 20,
'since': START_DATE,
}
expected_params = {
'fetch_type': 'Alerts',
'start_time': START_DATE,
'fetch_params': fetch_params
}
assert validate_fetch_incidents_params(params, {}) == expected_params
del params['fetch_type']
start_time = '2021-08-04T10:10:00Z'
last_run = {
'fetch_count': 1,
'end_time': '2021-08-05T03:43:52Z',
'start_time': start_time,
'fetch_sum': 20
}
fetch_params = {
'limit':
20,
'query':
'+basetypes:(credential-sighting) +header_.indexed_at: [1628071800'
' TO 1628135032]',
'skip':
20,
'sort':
'header_.indexed_at:asc'
}
expected_params = {
'fetch_type': 'Compromised Credentials',
'start_time': start_time,
'fetch_params': fetch_params
}
assert validate_fetch_incidents_params(params,
last_run) == expected_params