def test_put_process_cve_fail(mock_gremlin, mock_pv):
"""Test the CVEPut.process() success."""
mock_pv.return_value = [], True, {}
mock_gremlin.side_effect = [RequestsMockResponse({}, 200),
RequestsMockResponse({}, 200),
RequestsMockResponse({}, 500),
RequestsMockResponse({}, 200)]
cve = CVEPut(valid_put_input)
cve.process()
def test_snyk_put_process_cve_fail(mock_gremlin, mock_pv):
"""Test the SnykCVEPut.process() success."""
mock_pv.return_value = [], True, {}
mock_gremlin.side_effect = [RequestsMockResponse({}, 200),
RequestsMockResponse({}, 200),
RequestsMockResponse({}, 500),
RequestsMockResponse({}, 200)]
cve = SnykCVEPut(valid_snyk_put_input)
try:
cve.process()
except InternalServerError as e:
assert "Snyk CVEIngestionError - While creating CVE edges." in str(e)
def test_bad_gremlin_call(mocker):
"""Test utils.call_gremlin()."""
mocker.return_value = RequestsMockResponse({}, 500)
with pytest.raises(ValueError):
call_gremlin({'dummy': 'payload'})
def test_gremlin_call(mocker):
"""Test utils.call_gremlin()."""
mocker.return_value = RequestsMockResponse({}, 200)
assert call_gremlin({'dummy': 'payload'}) == {}
