def test_malware_attributes(): """Tests that a created Malware has all needed attributes.""" allitems = Malware.list() for malware in allitems: assert hasattr(malware, 'family') assert isinstance(malware.family, list)
def test_malware_delete(): malware = Malware(name='asd', labels=['label1']).save() malware.update({'name': 'dsa'}) assert len(Malware.list()) == 1 malware.delete(all_versions=True) assert not Malware.list()