From b3a601e4c86e9adc302112e45e2830f239c09cd8 Mon Sep 17 00:00:00 2001 From: Emmanuelle Vargas-Gonzalez Date: Fri, 3 May 2019 10:25:11 -0400 Subject: [PATCH] add new files for marking-definition tests --- stix2/test/v20/test_marking_definition.py | 131 ++++++++++++++++++++++ stix2/test/v21/test_marking_definition.py | 131 ++++++++++++++++++++++ 2 files changed, 262 insertions(+) create mode 100644 stix2/test/v20/test_marking_definition.py create mode 100644 stix2/test/v21/test_marking_definition.py diff --git a/stix2/test/v20/test_marking_definition.py b/stix2/test/v20/test_marking_definition.py new file mode 100644 index 0000000..9d03ebe --- /dev/null +++ b/stix2/test/v20/test_marking_definition.py @@ -0,0 +1,131 @@ + +import pytest + +from stix2 import exceptions +from stix2.v20 import ( + TLP_AMBER, TLP_GREEN, TLP_RED, TLP_WHITE, MarkingDefinition, TLPMarking, +) + + +def test_bad_id_marking_tlp_white(): + with pytest.raises(exceptions.TLPMarkingDefinitionError): + MarkingDefinition( + id='marking-definition--4c9faac1-3558-43d2-919e-95c88d3bc332', + definition_type='tlp', + definition=TLPMarking(tlp='white'), + ) + + +def test_bad_id_marking_tlp_green(): + with pytest.raises(exceptions.TLPMarkingDefinitionError): + MarkingDefinition( + id='marking-definition--93023361-d3cf-4666-bca2-8c017948dc3d', + definition_type='tlp', + definition=TLPMarking(tlp='green'), + ) + + +def test_bad_id_marking_tlp_amber(): + with pytest.raises(exceptions.TLPMarkingDefinitionError): + MarkingDefinition( + id='marking-definition--05e32101-a940-42ba-8fe9-39283b999ce4', + definition_type='tlp', + definition=TLPMarking(tlp='amber'), + ) + + +def test_bad_id_marking_tlp_red(): + with pytest.raises(exceptions.TLPMarkingDefinitionError): + MarkingDefinition( + id='marking-definition--9eceb00c-c158-43f4-87f8-1e3648de17e2', + definition_type='tlp', + definition=TLPMarking(tlp='red'), + ) + + +def test_bad_created_marking_tlp_white(): + with pytest.raises(exceptions.TLPMarkingDefinitionError): + MarkingDefinition( + id='marking-definition--613f2e26-407d-48c7-9eca-b8e91df99dc9', + definition_type='tlp', + definition=TLPMarking(tlp='white'), + ) + + +def test_bad_created_marking_tlp_green(): + with pytest.raises(exceptions.TLPMarkingDefinitionError): + MarkingDefinition( + id='marking-definition--34098fce-860f-48ae-8e50-ebd3cc5e41da', + definition_type='tlp', + definition=TLPMarking(tlp='green'), + ) + + +def test_bad_created_marking_tlp_amber(): + with pytest.raises(exceptions.TLPMarkingDefinitionError): + MarkingDefinition( + id='marking-definition--f88d31f6-486f-44da-b317-01333bde0b82', + definition_type='tlp', + definition=TLPMarking(tlp='amber'), + ) + + +def test_bad_created_marking_tlp_red(): + with pytest.raises(exceptions.TLPMarkingDefinitionError): + MarkingDefinition( + id='marking-definition--5e57c739-391a-4eb3-b6be-7d15ca92d5ed', + definition_type='tlp', + definition=TLPMarking(tlp='red'), + ) + + +def test_successful_tlp_white(): + white = MarkingDefinition( + id='marking-definition--613f2e26-407d-48c7-9eca-b8e91df99dc9', + created='2017-01-20T00:00:00.000Z', + definition_type='tlp', + definition=TLPMarking(tlp='white'), + ) + + assert white.serialize() in TLP_WHITE.serialize() + + +def test_successful_tlp_green(): + white = MarkingDefinition( + id='marking-definition--34098fce-860f-48ae-8e50-ebd3cc5e41da', + created='2017-01-20T00:00:00.000Z', + definition_type='tlp', + definition=TLPMarking(tlp='green'), + ) + + assert white.serialize() in TLP_GREEN.serialize() + + +def test_successful_tlp_amber(): + white = MarkingDefinition( + id='marking-definition--f88d31f6-486f-44da-b317-01333bde0b82', + created='2017-01-20T00:00:00.000Z', + definition_type='tlp', + definition=TLPMarking(tlp='amber'), + ) + + assert white.serialize() in TLP_AMBER.serialize() + + +def test_successful_tlp_red(): + white = MarkingDefinition( + id='marking-definition--5e57c739-391a-4eb3-b6be-7d15ca92d5ed', + created='2017-01-20T00:00:00.000Z', + definition_type='tlp', + definition=TLPMarking(tlp='red'), + ) + + assert white.serialize() in TLP_RED.serialize() + + +def test_unknown_tlp_marking(): + with pytest.raises(exceptions.TLPMarkingDefinitionError): + MarkingDefinition( + definition_type='tlp', + definition=TLPMarking(tlp='gray'), + ) diff --git a/stix2/test/v21/test_marking_definition.py b/stix2/test/v21/test_marking_definition.py new file mode 100644 index 0000000..2404ad1 --- /dev/null +++ b/stix2/test/v21/test_marking_definition.py @@ -0,0 +1,131 @@ + +import pytest + +from stix2 import exceptions +from stix2.v21 import ( + TLP_AMBER, TLP_GREEN, TLP_RED, TLP_WHITE, MarkingDefinition, TLPMarking, +) + + +def test_bad_id_marking_tlp_white(): + with pytest.raises(exceptions.TLPMarkingDefinitionError): + MarkingDefinition( + id='marking-definition--4c9faac1-3558-43d2-919e-95c88d3bc332', + definition_type='tlp', + definition=TLPMarking(tlp='white'), + ) + + +def test_bad_id_marking_tlp_green(): + with pytest.raises(exceptions.TLPMarkingDefinitionError): + MarkingDefinition( + id='marking-definition--93023361-d3cf-4666-bca2-8c017948dc3d', + definition_type='tlp', + definition=TLPMarking(tlp='green'), + ) + + +def test_bad_id_marking_tlp_amber(): + with pytest.raises(exceptions.TLPMarkingDefinitionError): + MarkingDefinition( + id='marking-definition--05e32101-a940-42ba-8fe9-39283b999ce4', + definition_type='tlp', + definition=TLPMarking(tlp='amber'), + ) + + +def test_bad_id_marking_tlp_red(): + with pytest.raises(exceptions.TLPMarkingDefinitionError): + MarkingDefinition( + id='marking-definition--9eceb00c-c158-43f4-87f8-1e3648de17e2', + definition_type='tlp', + definition=TLPMarking(tlp='red'), + ) + + +def test_bad_created_marking_tlp_white(): + with pytest.raises(exceptions.TLPMarkingDefinitionError): + MarkingDefinition( + id='marking-definition--613f2e26-407d-48c7-9eca-b8e91df99dc9', + definition_type='tlp', + definition=TLPMarking(tlp='white'), + ) + + +def test_bad_created_marking_tlp_green(): + with pytest.raises(exceptions.TLPMarkingDefinitionError): + MarkingDefinition( + id='marking-definition--34098fce-860f-48ae-8e50-ebd3cc5e41da', + definition_type='tlp', + definition=TLPMarking(tlp='green'), + ) + + +def test_bad_created_marking_tlp_amber(): + with pytest.raises(exceptions.TLPMarkingDefinitionError): + MarkingDefinition( + id='marking-definition--f88d31f6-486f-44da-b317-01333bde0b82', + definition_type='tlp', + definition=TLPMarking(tlp='amber'), + ) + + +def test_bad_created_marking_tlp_red(): + with pytest.raises(exceptions.TLPMarkingDefinitionError): + MarkingDefinition( + id='marking-definition--5e57c739-391a-4eb3-b6be-7d15ca92d5ed', + definition_type='tlp', + definition=TLPMarking(tlp='red'), + ) + + +def test_successful_tlp_white(): + white = MarkingDefinition( + id='marking-definition--613f2e26-407d-48c7-9eca-b8e91df99dc9', + created='2017-01-20T00:00:00.000Z', + definition_type='tlp', + definition=TLPMarking(tlp='white'), + ) + + assert white.serialize() in TLP_WHITE.serialize() + + +def test_successful_tlp_green(): + green = MarkingDefinition( + id='marking-definition--34098fce-860f-48ae-8e50-ebd3cc5e41da', + created='2017-01-20T00:00:00.000Z', + definition_type='tlp', + definition=TLPMarking(tlp='green'), + ) + + assert green.serialize() in TLP_GREEN.serialize() + + +def test_successful_tlp_amber(): + amber = MarkingDefinition( + id='marking-definition--f88d31f6-486f-44da-b317-01333bde0b82', + created='2017-01-20T00:00:00.000Z', + definition_type='tlp', + definition=TLPMarking(tlp='amber'), + ) + + assert amber.serialize() in TLP_AMBER.serialize() + + +def test_successful_tlp_red(): + red = MarkingDefinition( + id='marking-definition--5e57c739-391a-4eb3-b6be-7d15ca92d5ed', + created='2017-01-20T00:00:00.000Z', + definition_type='tlp', + definition=TLPMarking(tlp='red'), + ) + + assert red.serialize() in TLP_RED.serialize() + + +def test_unknown_tlp_marking(): + with pytest.raises(exceptions.TLPMarkingDefinitionError): + MarkingDefinition( + definition_type='tlp', + definition=TLPMarking(tlp='gray'), + )