def test_workbench_get_all_vulnerabilities(): vuln = Vulnerability(id=VULNERABILITY_ID, **VULNERABILITY_KWARGS) save(vuln) resp = vulnerabilities() assert len(resp) == 1 assert resp[0].id == VULNERABILITY_ID
def test_workbench_add_to_bundle(): vuln = Vulnerability(**constants.VULNERABILITY_KWARGS) bundle = Bundle(vuln) assert bundle.objects[0].name == 'Heartbleed'