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)