def test_snyk_cve_delete_prepare_payload(): """Test SnykCVEDelete.prepare_payload().""" cve = SnykCVEDelete(valid_snyk_delete_input) json_payload = cve.prepare_payload() assert 'gremlin' in json_payload assert 'bindings' in json_payload bindings = json_payload['bindings'] assert 'snyk_vuln_id' in bindings assert bindings['snyk_vuln_id']
def snyk_cves_put_delete(): """Put or delete Snyk CVE nodes. Missing EPVs will be created. """ payload = request.get_json(silent=True) try: if request.method == 'PUT': cve = SnykCVEPut(payload) elif request.method == 'DELETE': cve = SnykCVEDelete(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_snyk_cve_delete_creation(): """Test SnykCVEDelete input validation.""" assert SnykCVEDelete(valid_snyk_delete_input) with pytest.raises(ValueError): SnykCVEDelete(invalid_snyk_delete_input)