import datetime as dt import pytest import pytz import stix2 from .constants import ( FAKE_TIME, INDICATOR_ID, MALWARE_ID, RELATIONSHIP_ID, RELATIONSHIP_KWARGS, ) EXPECTED_RELATIONSHIP = """{ "type": "relationship", "id": "relationship--df7c87eb-75d2-4948-af81-9d49d246f301", "created": "2016-04-06T20:06:37.000Z", "modified": "2016-04-06T20:06:37.000Z", "relationship_type": "indicates", "source_ref": "indicator--a740531e-63ff-4e49-a9e1-a0a3eed0e3e7", "target_ref": "malware--9c4638ec-f1de-4ddb-abf4-1b760417654e" }""" def test_relationship_all_required_properties(): now = dt.datetime(2016, 4, 6, 20, 6, 37, tzinfo=pytz.utc) rel = stix2.v20.Relationship( type='relationship', id=RELATIONSHIP_ID, created=now, modified=now, relationship_type='indicates', source_ref=INDICATOR_ID, target_ref=MALWARE_ID, ) assert str(rel) == EXPECTED_RELATIONSHIP def test_relationship_autogenerated_properties(relationship): assert relationship.type == 'relationship' assert relationship.id == 'relationship--00000000-0000-4000-8000-000000000001' assert relationship.created == FAKE_TIME assert relationship.modified == FAKE_TIME assert relationship.relationship_type == 'indicates' assert relationship.source_ref == INDICATOR_ID assert relationship.target_ref == MALWARE_ID assert relationship['type'] == 'relationship' assert relationship['id'] == 'relationship--00000000-0000-4000-8000-000000000001' assert relationship['created'] == FAKE_TIME assert relationship['modified'] == FAKE_TIME assert relationship['relationship_type'] == 'indicates' assert relationship['source_ref'] == INDICATOR_ID assert relationship['target_ref'] == MALWARE_ID def test_relationship_type_must_be_relationship(): with pytest.raises(stix2.exceptions.InvalidValueError) as excinfo: stix2.v20.Relationship(type='xxx', **RELATIONSHIP_KWARGS) assert excinfo.value.cls == stix2.v20.Relationship assert excinfo.value.prop_name == "type" assert excinfo.value.reason == "must equal 'relationship'." assert str(excinfo.value) == "Invalid value for Relationship 'type': must equal 'relationship'." def test_relationship_id_must_start_with_relationship(): with pytest.raises(stix2.exceptions.InvalidValueError) as excinfo: stix2.v20.Relationship(id='my-prefix--', **RELATIONSHIP_KWARGS) assert excinfo.value.cls == stix2.v20.Relationship assert excinfo.value.prop_name == "id" assert excinfo.value.reason == "must start with 'relationship--'." assert str(excinfo.value) == "Invalid value for Relationship 'id': must start with 'relationship--'." def test_relationship_required_property_relationship_type(): with pytest.raises(stix2.exceptions.MissingPropertiesError) as excinfo: stix2.v20.Relationship() assert excinfo.value.cls == stix2.v20.Relationship assert excinfo.value.properties == ["relationship_type", "source_ref", "target_ref"] def test_relationship_missing_some_required_properties(): with pytest.raises(stix2.exceptions.MissingPropertiesError) as excinfo: stix2.v20.Relationship(relationship_type='indicates') assert excinfo.value.cls == stix2.v20.Relationship assert excinfo.value.properties == ["source_ref", "target_ref"] def test_relationship_required_properties_target_ref(): with pytest.raises(stix2.exceptions.MissingPropertiesError) as excinfo: stix2.v20.Relationship( relationship_type='indicates', source_ref=INDICATOR_ID, ) assert excinfo.value.cls == stix2.v20.Relationship assert excinfo.value.properties == ["target_ref"] def test_cannot_assign_to_relationship_attributes(relationship): with pytest.raises(stix2.exceptions.ImmutableError) as excinfo: relationship.relationship_type = "derived-from" assert str(excinfo.value) == "Cannot modify 'relationship_type' property in 'Relationship' after creation." def test_invalid_kwarg_to_relationship(): with pytest.raises(stix2.exceptions.ExtraPropertiesError) as excinfo: stix2.v20.Relationship(my_custom_property="foo", **RELATIONSHIP_KWARGS) assert excinfo.value.cls == stix2.v20.Relationship assert excinfo.value.properties == ['my_custom_property'] assert str(excinfo.value) == "Unexpected properties for Relationship: (my_custom_property)." def test_create_relationship_from_objects_rather_than_ids(indicator, malware): rel = stix2.v20.Relationship( relationship_type="indicates", source_ref=indicator, target_ref=malware, ) assert rel.relationship_type == 'indicates' assert rel.source_ref == 'indicator--00000000-0000-4000-8000-000000000001' assert rel.target_ref == 'malware--00000000-0000-4000-8000-000000000003' assert rel.id == 'relationship--00000000-0000-4000-8000-000000000005' def test_create_relationship_with_positional_args(indicator, malware): rel = stix2.v20.Relationship(indicator, 'indicates', malware) assert rel.relationship_type == 'indicates' assert rel.source_ref == 'indicator--00000000-0000-4000-8000-000000000001' assert rel.target_ref == 'malware--00000000-0000-4000-8000-000000000003' assert rel.id == 'relationship--00000000-0000-4000-8000-000000000005' @pytest.mark.parametrize( "data", [ EXPECTED_RELATIONSHIP, { "created": "2016-04-06T20:06:37Z", "id": RELATIONSHIP_ID, "modified": "2016-04-06T20:06:37Z", "type": "relationship", "relationship_type": "indicates", "source_ref": INDICATOR_ID, "target_ref": MALWARE_ID, }, ], ) def test_parse_relationship(data): rel = stix2.parse(data, version="2.0") assert rel.type == 'relationship' assert rel.id == RELATIONSHIP_ID assert rel.created == dt.datetime(2016, 4, 6, 20, 6, 37, tzinfo=pytz.utc) assert rel.modified == dt.datetime(2016, 4, 6, 20, 6, 37, tzinfo=pytz.utc) assert rel.relationship_type == "indicates" assert rel.source_ref == INDICATOR_ID assert rel.target_ref == MALWARE_ID