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'}) == {}