Move fixtures to conftest.py

stix2.1
Greg Back 2017-04-07 15:36:42 -05:00
parent 57869205ad
commit 1517eb899b
7 changed files with 15 additions and 18 deletions

View File

@ -2,8 +2,6 @@ import pytest
import stix2 import stix2
from .fixtures import clock, uuid4, indicator, malware, relationship # noqa: F401
EXPECTED_BUNDLE = """{ EXPECTED_BUNDLE = """{
"id": "bundle--00000000-0000-0000-0000-000000000004", "id": "bundle--00000000-0000-0000-0000-000000000004",
"objects": [ "objects": [
@ -73,13 +71,13 @@ def test_bundle_with_wrong_spec_version():
assert str(excinfo.value) == "Invalid value for Bundle 'spec_version': must equal '2.0'." assert str(excinfo.value) == "Invalid value for Bundle 'spec_version': must equal '2.0'."
def test_create_bundle(indicator, malware, relationship): # noqa: F811 def test_create_bundle(indicator, malware, relationship):
bundle = stix2.Bundle(objects=[indicator, malware, relationship]) bundle = stix2.Bundle(objects=[indicator, malware, relationship])
assert str(bundle) == EXPECTED_BUNDLE assert str(bundle) == EXPECTED_BUNDLE
def test_create_bundle_with_positional_args(indicator, malware, relationship): # noqa: F811 def test_create_bundle_with_positional_args(indicator, malware, relationship):
bundle = stix2.Bundle(indicator, malware, relationship) bundle = stix2.Bundle(indicator, malware, relationship)
assert str(bundle) == EXPECTED_BUNDLE assert str(bundle) == EXPECTED_BUNDLE

View File

@ -2,14 +2,13 @@ import datetime as dt
import uuid import uuid
from .constants import FAKE_TIME from .constants import FAKE_TIME
from .fixtures import clock, uuid4 # noqa: F401
def test_clock(clock): # noqa: F811 def test_clock(clock):
assert dt.datetime.now() == FAKE_TIME assert dt.datetime.now() == FAKE_TIME
def test_my_uuid4_fixture(uuid4): # noqa: F811 def test_my_uuid4_fixture(uuid4):
assert uuid.uuid4() == "00000000-0000-0000-0000-000000000001" assert uuid.uuid4() == "00000000-0000-0000-0000-000000000001"
assert uuid.uuid4() == "00000000-0000-0000-0000-000000000002" assert uuid.uuid4() == "00000000-0000-0000-0000-000000000002"
assert uuid.uuid4() == "00000000-0000-0000-0000-000000000003" assert uuid.uuid4() == "00000000-0000-0000-0000-000000000003"

View File

@ -6,7 +6,6 @@ import pytz
import stix2 import stix2
from .constants import FAKE_TIME, INDICATOR_ID, INDICATOR_KWARGS from .constants import FAKE_TIME, INDICATOR_ID, INDICATOR_KWARGS
from .fixtures import clock, uuid4, indicator # noqa: F401
EXPECTED_INDICATOR = """{ EXPECTED_INDICATOR = """{
"created": "2017-01-01T00:00:01Z", "created": "2017-01-01T00:00:01Z",
@ -49,7 +48,7 @@ def test_indicator_with_all_required_fields():
assert repr(ind) == EXPECTED_INDICATOR_REPR assert repr(ind) == EXPECTED_INDICATOR_REPR
def test_indicator_autogenerated_fields(indicator): # noqa: F811 def test_indicator_autogenerated_fields(indicator):
assert indicator.type == 'indicator' assert indicator.type == 'indicator'
assert indicator.id == 'indicator--00000000-0000-0000-0000-000000000001' assert indicator.id == 'indicator--00000000-0000-0000-0000-000000000001'
assert indicator.created == FAKE_TIME assert indicator.created == FAKE_TIME
@ -105,7 +104,7 @@ def test_indicator_revoked_invalid():
assert str(excinfo.value) == "Invalid value for Indicator 'revoked': must be a boolean value." assert str(excinfo.value) == "Invalid value for Indicator 'revoked': must be a boolean value."
def test_cannot_assign_to_indicator_attributes(indicator): # noqa: F811 def test_cannot_assign_to_indicator_attributes(indicator):
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError) as excinfo:
indicator.valid_from = dt.datetime.now() indicator.valid_from = dt.datetime.now()

View File

@ -6,7 +6,6 @@ import pytz
import stix2 import stix2
from .constants import FAKE_TIME, MALWARE_ID, MALWARE_KWARGS from .constants import FAKE_TIME, MALWARE_ID, MALWARE_KWARGS
from .fixtures import clock, uuid4, malware # noqa: F401
EXPECTED_MALWARE = """{ EXPECTED_MALWARE = """{
"created": "2016-05-12T08:17:27Z", "created": "2016-05-12T08:17:27Z",
@ -35,7 +34,7 @@ def test_malware_with_all_required_fields():
assert str(mal) == EXPECTED_MALWARE assert str(mal) == EXPECTED_MALWARE
def test_malware_autogenerated_fields(malware): # noqa: F811 def test_malware_autogenerated_fields(malware):
assert malware.type == 'malware' assert malware.type == 'malware'
assert malware.id == 'malware--00000000-0000-0000-0000-000000000001' assert malware.id == 'malware--00000000-0000-0000-0000-000000000001'
assert malware.created == FAKE_TIME assert malware.created == FAKE_TIME
@ -77,7 +76,7 @@ def test_malware_required_field_name():
assert str(excinfo.value) == "Missing required field(s) for Malware: (name)." assert str(excinfo.value) == "Missing required field(s) for Malware: (name)."
def test_cannot_assign_to_malware_attributes(malware): # noqa: F811 def test_cannot_assign_to_malware_attributes(malware):
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError) as excinfo:
malware.name = "Cryptolocker II" malware.name = "Cryptolocker II"

View File

@ -7,7 +7,6 @@ import stix2
from .constants import FAKE_TIME, INDICATOR_ID, MALWARE_ID, RELATIONSHIP_ID from .constants import FAKE_TIME, INDICATOR_ID, MALWARE_ID, RELATIONSHIP_ID
from .constants import RELATIONSHIP_KWARGS from .constants import RELATIONSHIP_KWARGS
from .fixtures import clock, uuid4, indicator, malware, relationship # noqa: F401
EXPECTED_RELATIONSHIP = """{ EXPECTED_RELATIONSHIP = """{
@ -36,7 +35,7 @@ def test_relationship_all_required_fields():
assert str(rel) == EXPECTED_RELATIONSHIP assert str(rel) == EXPECTED_RELATIONSHIP
def test_relationship_autogenerated_fields(relationship): # noqa: F811 def test_relationship_autogenerated_fields(relationship):
assert relationship.type == 'relationship' assert relationship.type == 'relationship'
assert relationship.id == 'relationship--00000000-0000-0000-0000-000000000001' assert relationship.id == 'relationship--00000000-0000-0000-0000-000000000001'
assert relationship.created == FAKE_TIME assert relationship.created == FAKE_TIME
@ -89,7 +88,7 @@ def test_relationship_required_field_target_ref():
assert str(excinfo.value) == "Missing required field(s) for Relationship: (target_ref)." assert str(excinfo.value) == "Missing required field(s) for Relationship: (target_ref)."
def test_cannot_assign_to_relationship_attributes(relationship): # noqa: F811 def test_cannot_assign_to_relationship_attributes(relationship):
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError) as excinfo:
relationship.relationship_type = "derived-from" relationship.relationship_type = "derived-from"
@ -102,7 +101,7 @@ def test_invalid_kwarg_to_relationship():
assert str(excinfo.value) == "unexpected keyword arguments: ['my_custom_property']" in str(excinfo) assert str(excinfo.value) == "unexpected keyword arguments: ['my_custom_property']" in str(excinfo)
def test_create_relationship_from_objects_rather_than_ids(indicator, malware): # noqa: F811 def test_create_relationship_from_objects_rather_than_ids(indicator, malware):
rel = stix2.Relationship( rel = stix2.Relationship(
relationship_type="indicates", relationship_type="indicates",
source_ref=indicator, source_ref=indicator,
@ -115,7 +114,7 @@ def test_create_relationship_from_objects_rather_than_ids(indicator, malware):
assert rel.id == 'relationship--00000000-0000-0000-0000-000000000003' assert rel.id == 'relationship--00000000-0000-0000-0000-000000000003'
def test_create_relationship_with_positional_args(indicator, malware): # noqa: F811 def test_create_relationship_with_positional_args(indicator, malware):
rel = stix2.Relationship(indicator, 'indicates', malware) rel = stix2.Relationship(indicator, 'indicates', malware)
assert rel.relationship_type == 'indicates' assert rel.relationship_type == 'indicates'

View File

@ -17,6 +17,9 @@ commands =
ignore= ignore=
max-line-length=160 max-line-length=160
[flake8]
max-line-length=160
[travis] [travis]
python = python =
2.6: py26 2.6: py26