def test_validate_extra_object(): # arrange verdict = Verdict().set_malware_family("Eicar") # act verdict.add_extra("new_key", {"other_key": "string_value"}) # assert assert Verdict.validate(json.loads(verdict.json()))
def test_add_extra_object(): # arrange verdict = Verdict().set_malware_family("Eicar") # act verdict.add_extra("new_key", {"other_key": "string_value"}) # assert k, v = verdict.extra[0] assert k == 'new_key' assert v == {"other_key": "string_value"}
def test_add_extra_array(): # arrange verdict = Verdict().set_malware_family("Eicar") # act verdict.add_extra("new_key", ["string_value"]) # assert k, v = verdict.extra[0] assert k == 'new_key' assert v == ["string_value"]