2017-07-12 17:36:15 +02:00
|
|
|
import stix2
|
|
|
|
|
|
|
|
from .constants import IDENTITY_ID, IDENTITY_KWARGS, INDICATOR_KWARGS
|
|
|
|
|
|
|
|
|
|
|
|
def test_object_factory_created_by_ref_str():
|
|
|
|
factory = stix2.ObjectFactory(created_by=IDENTITY_ID)
|
|
|
|
ind = factory.create(stix2.Indicator, **INDICATOR_KWARGS)
|
|
|
|
assert ind.created_by_ref == IDENTITY_ID
|
|
|
|
|
|
|
|
|
|
|
|
def test_object_factory_created_by_ref_obj():
|
|
|
|
id_obj = stix2.Identity(id=IDENTITY_ID, **IDENTITY_KWARGS)
|
|
|
|
factory = stix2.ObjectFactory(created_by=id_obj)
|
|
|
|
ind = factory.create(stix2.Indicator, **INDICATOR_KWARGS)
|
|
|
|
assert ind.created_by_ref == IDENTITY_ID
|
2017-07-12 20:44:52 +02:00
|
|
|
|
|
|
|
|
|
|
|
def test_object_factory_obj_markings():
|
|
|
|
stmt_marking = stix2.StatementMarking("Copyright 2016, Example Corp")
|
|
|
|
mark_def = stix2.MarkingDefinition(definition_type="statement",
|
|
|
|
definition=stmt_marking)
|
|
|
|
factory = stix2.ObjectFactory(markings=[mark_def, stix2.TLP_AMBER])
|
|
|
|
ind = factory.create(stix2.Indicator, **INDICATOR_KWARGS)
|
|
|
|
assert mark_def.id in ind.object_marking_refs
|
|
|
|
assert stix2.TLP_AMBER.id in ind.object_marking_refs
|
|
|
|
|
|
|
|
factory = stix2.ObjectFactory(markings=stix2.TLP_RED)
|
|
|
|
ind = factory.create(stix2.Indicator, **INDICATOR_KWARGS)
|
|
|
|
assert stix2.TLP_RED.id in ind.object_marking_refs
|