Minor style changes.

stix2.1
Emmanuelle Vargas-Gonzalez 2017-08-23 13:12:40 -04:00
parent f33427328b
commit 8d4c1d55b5
3 changed files with 6 additions and 6 deletions

View File

@ -1,7 +1,7 @@
import pytest
from stix2 import Malware, exceptions, markings
from stix2 import Malware, markings
from .constants import FAKE_TIME, MALWARE_ID, MARKING_IDS
from .constants import MALWARE_KWARGS as MALWARE_KWARGS_CONST

View File

@ -155,8 +155,8 @@ def test_campaign_with_granular_markings_example():
marking_ref="marking-definition--613f2e26-407d-48c7-9eca-b8e91df99dc9",
selectors=["description"])
])
print (markings.get_markings(campaign, None))
print (markings.add_markings(campaign, None, "marking-definition--00000000-0000-0000-0000-000000000000"))
print(markings.get_markings(campaign, None))
print(markings.add_markings(campaign, None, "marking-definition--00000000-0000-0000-0000-000000000000"))
assert str(campaign) == EXPECTED_CAMPAIGN_WITH_GRANULAR_MARKINGS

View File

@ -23,7 +23,7 @@ def test_add_markings_one_marking():
after = Malware(
object_marking_refs=[MARKING_IDS[0]],
**MALWARE_KWARGS,
**MALWARE_KWARGS
)
before = markings.add_markings(before, None, MARKING_IDS[0])
@ -39,7 +39,7 @@ def test_add_markings_multiple_marking():
after = Malware(
object_marking_refs=[MARKING_IDS[0], MARKING_IDS[1]],
**MALWARE_KWARGS,
**MALWARE_KWARGS
)
before = markings.add_markings(before, None, [MARKING_IDS[0], MARKING_IDS[1]])
@ -62,7 +62,7 @@ def test_add_markings_combination():
{
"selectors": ["name"],
"marking_ref": MARKING_IDS[3]
},
}
],
**MALWARE_KWARGS
)