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 cves_put_delete(): """Put or delete CVE nodes. Missing EPVs will be created. """ payload = request.get_json(silent=True) try: if request.method == 'PUT': cve = CVEPut(payload) elif request.method == 'DELETE': cve = CVEDelete(payload) else: # this should never happen return flask.jsonify({'error': 'method not allowed'}), 405 except ValueError as e: return flask.jsonify({'error': str(e)}), 400 try: cve.process() except ValueError as e: return flask.jsonify({'error': str(e)}), 500 return flask.jsonify({}), 200
def test_put_process_epv_fail(mock_pv): """Test the CVEPut.process() fail.""" mock_pv.return_value = [], False, {} cve = CVEPut(valid_put_input) cve.process()