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()
