86 lines
2.8 KiB
Python
86 lines
2.8 KiB
Python
|
import datetime as dt
|
||
|
|
import re
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
import pytz
|
||
|
|
|
||
|
|
import stix2
|
||
|
|
|
||
|
|
from .constants import OPINION_ID
|
||
|
|
|
||
|
|
DESCRIPTION = ('This doesn\'t seem like it is feasible. We\'ve seen how '
|
||
|
|
'PandaCat has attacked Spanish infrastructure over the '
|
||
|
|
'last 3 years, so this change in targeting seems too great'
|
||
|
|
' to be viable. The methods used are more commonly '
|
||
|
|
'associated with the FlameDragonCrew.')
|
||
|
|
|
||
|
|
EXPECTED_OPINION = """{
|
||
|
|
"type": "opinion",
|
||
|
|
"spec_version": "2.1",
|
||
|
|
"id": "opinion--b01efc25-77b4-4003-b18b-f6e24b5cd9f7",
|
||
|
|
"created": "2016-05-12T08:17:27.000Z",
|
||
|
|
"modified": "2016-05-12T08:17:27.000Z",
|
||
|
|
"description": "%s",
|
||
|
|
"object_refs": [
|
||
|
|
"relationship--16d2358f-3b0d-4c88-b047-0da2f7ed4471"
|
||
|
|
],
|
||
|
|
"opinion": "strongly-disagree"
|
||
|
|
}""" % DESCRIPTION
|
||
|
|
|
||
|
|
EXPECTED_OPINION_REPR = "Opinion(" + " ".join(("""
|
||
|
|
type='opinion',
|
||
|
|
spec_version='2.1',
|
||
|
|
id='opinion--b01efc25-77b4-4003-b18b-f6e24b5cd9f7',
|
||
|
|
created='2016-05-12T08:17:27.000Z',
|
||
|
|
modified='2016-05-12T08:17:27.000Z',
|
||
|
|
description="%s",
|
||
|
|
object_refs=['relationship--16d2358f-3b0d-4c88-b047-0da2f7ed4471'],
|
||
|
|
opinion='strongly-disagree'""" % DESCRIPTION).split()) + ")"
|
||
|
|
|
||
|
|
|
||
|
|
def test_opinion_with_required_properties():
|
||
|
|
now = dt.datetime(2016, 5, 12, 8, 17, 27, tzinfo=pytz.utc)
|
||
|
|
|
||
|
|
opi = stix2.Opinion(
|
||
|
|
type='opinion',
|
||
|
|
id=OPINION_ID,
|
||
|
|
created=now,
|
||
|
|
modified=now,
|
||
|
|
object_refs=['relationship--16d2358f-3b0d-4c88-b047-0da2f7ed4471'],
|
||
|
|
opinion='strongly-disagree',
|
||
|
|
description=DESCRIPTION
|
||
|
|
)
|
||
|
|
|
||
|
|
assert str(opi) == EXPECTED_OPINION
|
||
|
|
rep = re.sub(r"(\[|=| )u('|\"|\\\'|\\\")", r"\g<1>\g<2>", repr(opi))
|
||
|
|
assert rep == EXPECTED_OPINION_REPR
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("data", [
|
||
|
|
EXPECTED_OPINION,
|
||
|
|
{
|
||
|
|
"type": "opinion",
|
||
|
|
"spec_version": "2.1",
|
||
|
|
"id": "opinion--b01efc25-77b4-4003-b18b-f6e24b5cd9f7",
|
||
|
|
"created": "2016-05-12T08:17:27.000Z",
|
||
|
|
"modified": "2016-05-12T08:17:27.000Z",
|
||
|
|
"description": DESCRIPTION,
|
||
|
|
"object_refs": [
|
||
|
|
"relationship--16d2358f-3b0d-4c88-b047-0da2f7ed4471"
|
||
|
|
],
|
||
|
|
"opinion": "strongly-disagree"
|
||
|
|
}
|
||
|
|
])
|
||
|
|
def test_parse_opinion(data):
|
||
|
|
opinion = stix2.parse(data)
|
||
|
|
|
||
|
|
assert opinion.type == 'opinion'
|
||
|
|
assert opinion.id == OPINION_ID
|
||
|
|
assert opinion.created == dt.datetime(2016, 5, 12, 8, 17, 27, tzinfo=pytz.utc)
|
||
|
|
assert opinion.modified == dt.datetime(2016, 5, 12, 8, 17, 27, tzinfo=pytz.utc)
|
||
|
|
assert opinion.opinion == 'strongly-disagree'
|
||
|
|
assert opinion.object_refs[0] == 'relationship--16d2358f-3b0d-4c88-b047-0da2f7ed4471'
|
||
|
|
assert opinion.description == DESCRIPTION
|
||
|
|
rep = re.sub(r"(\[|=| )u('|\"|\\\'|\\\")", r"\g<1>\g<2>", repr(opinion))
|
||
|
|
assert rep == EXPECTED_OPINION_REPR
|