Add and update tests for markings API.
Emmanuelle Vargas-Gonzalez
parent
1b7695c4f6
commit
1f258551e1
|
|
@ -0,0 +1,968 @@
|
|||
|
||||
from stix2 import markings
|
||||
|
||||
import pytest
|
||||
|
||||
"""Tests for the Data Markings API."""
|
||||
|
||||
|
||||
def test_add_marking_mark_one_selector_multiple_refs():
|
||||
before = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
}
|
||||
after = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
},
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--2"
|
||||
},
|
||||
]
|
||||
}
|
||||
markings.add_markings(before, ["description"], ["marking-definition--1", "marking-definition--2"])
|
||||
|
||||
for m in before["granular_markings"]:
|
||||
assert m in after["granular_markings"]
|
||||
|
||||
|
||||
def test_add_marking_mark_multiple_selector_one_refs():
|
||||
before = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
}
|
||||
after = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description", "title"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
},
|
||||
]
|
||||
}
|
||||
markings.add_markings(before, ["description", "title"], ["marking-definition--1"])
|
||||
assert before == after
|
||||
|
||||
|
||||
def test_add_marking_mark_multiple_selector_multiple_refs():
|
||||
before = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
}
|
||||
after = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description", "title"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
},
|
||||
{
|
||||
"selectors": ["description", "title"],
|
||||
"marking_ref": "marking-definition--2"
|
||||
},
|
||||
]
|
||||
}
|
||||
markings.add_markings(before, ["description", "title"], ["marking-definition--1", "marking-definition--2"])
|
||||
|
||||
for m in before["granular_markings"]:
|
||||
assert m in after["granular_markings"]
|
||||
|
||||
|
||||
def test_add_marking_mark_another_property_same_marking():
|
||||
before = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
},
|
||||
]
|
||||
}
|
||||
after = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description", "title"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
},
|
||||
]
|
||||
}
|
||||
markings.add_markings(before, ["title"], ["marking-definition--1"])
|
||||
assert before == after
|
||||
|
||||
|
||||
def test_add_marking_mark_same_property_same_marking():
|
||||
before = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
}
|
||||
]
|
||||
}
|
||||
after = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
}
|
||||
]
|
||||
}
|
||||
markings.add_markings(before, ["description"], ["marking-definition--1"])
|
||||
assert before == after
|
||||
|
||||
|
||||
@pytest.mark.parametrize("data,marking", [
|
||||
({"description": "test description"},
|
||||
[["title"], ["marking-definition--1", "marking-definition--2"],
|
||||
"", ["marking-definition--1", "marking-definition--2"],
|
||||
[], ["marking-definition--1", "marking-definition--2"],
|
||||
[""], ["marking-definition--1", "marking-definition--2"],
|
||||
["description"], [""],
|
||||
["description"], [],
|
||||
["description"], ["marking-definition--1", 456]
|
||||
])
|
||||
])
|
||||
def test_add_marking_bad_selector(data, marking):
|
||||
with pytest.raises(AssertionError):
|
||||
markings.add_markings(data, marking[0], marking[1])
|
||||
|
||||
|
||||
GET_MARKINGS_TEST_DATA = \
|
||||
{
|
||||
"a": 333,
|
||||
"b": "value",
|
||||
"c": [
|
||||
17,
|
||||
"list value",
|
||||
{
|
||||
"g": "nested",
|
||||
"h": 45
|
||||
}
|
||||
],
|
||||
"x": {
|
||||
"y": [
|
||||
"hello",
|
||||
88
|
||||
],
|
||||
"z": {
|
||||
"foo1": "bar",
|
||||
"foo2": 65
|
||||
}
|
||||
},
|
||||
"granular_markings": [
|
||||
{
|
||||
"marking_ref": "1",
|
||||
"selectors": ["a"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "2",
|
||||
"selectors": ["c"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "3",
|
||||
"selectors": ["c.[1]"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "4",
|
||||
"selectors": ["c.[2]"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "5",
|
||||
"selectors": ["c.[2].g"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "6",
|
||||
"selectors": ["x"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "7",
|
||||
"selectors": ["x.y"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "8",
|
||||
"selectors": ["x.y.[1]"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "9",
|
||||
"selectors": ["x.z"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "10",
|
||||
"selectors": ["x.z.foo2"]
|
||||
},
|
||||
]
|
||||
}
|
||||
|
||||
|
||||
@pytest.mark.parametrize("data", [GET_MARKINGS_TEST_DATA])
|
||||
def test_get_markings_smoke(data):
|
||||
"""Test get_markings does not fail."""
|
||||
assert len(markings.get_markings(data, "a")) >= 1
|
||||
assert markings.get_markings(data, "a") == ["1"]
|
||||
|
||||
|
||||
@pytest.mark.parametrize("data", [GET_MARKINGS_TEST_DATA])
|
||||
def test_get_markings_not_marked(data):
|
||||
"""Test selector that is not marked returns empty list."""
|
||||
results = markings.get_markings(data, "b")
|
||||
assert len(results) == 0
|
||||
|
||||
|
||||
@pytest.mark.parametrize("data", [GET_MARKINGS_TEST_DATA])
|
||||
def test_get_markings_multiple_selectors(data):
|
||||
"""Test multiple selectors return combination of markings."""
|
||||
total = markings.get_markings(data, ["x.y", "x.z"])
|
||||
xy_markings = markings.get_markings(data, ["x.y"])
|
||||
xz_markings = markings.get_markings(data, ["x.z"])
|
||||
|
||||
assert set(xy_markings).issubset(total)
|
||||
assert set(xz_markings).issubset(total)
|
||||
assert set(xy_markings).union(xz_markings).issuperset(total)
|
||||
|
||||
|
||||
@pytest.mark.parametrize("data,selector", [
|
||||
(GET_MARKINGS_TEST_DATA, "foo"),
|
||||
(GET_MARKINGS_TEST_DATA, ""),
|
||||
(GET_MARKINGS_TEST_DATA, []),
|
||||
(GET_MARKINGS_TEST_DATA, [""]),
|
||||
(GET_MARKINGS_TEST_DATA, "x.z.[-2]"),
|
||||
(GET_MARKINGS_TEST_DATA, "c.f"),
|
||||
(GET_MARKINGS_TEST_DATA, "c.[2].i"),
|
||||
(GET_MARKINGS_TEST_DATA, "c.[3]"),
|
||||
(GET_MARKINGS_TEST_DATA, "d"),
|
||||
(GET_MARKINGS_TEST_DATA, "x.[0]"),
|
||||
(GET_MARKINGS_TEST_DATA, "z.y.w"),
|
||||
(GET_MARKINGS_TEST_DATA, "x.z.[1]"),
|
||||
(GET_MARKINGS_TEST_DATA, "x.z.foo3")
|
||||
])
|
||||
def test_get_markings_bad_selector(data, selector):
|
||||
"""Test bad selectors raise exception"""
|
||||
with pytest.raises(AssertionError):
|
||||
markings.get_markings(data, selector)
|
||||
|
||||
|
||||
@pytest.mark.parametrize("data", [GET_MARKINGS_TEST_DATA])
|
||||
def test_get_markings_positional_arguments_combinations(data):
|
||||
"""Test multiple combinations for inherited and descendant markings."""
|
||||
assert set(markings.get_markings(data, "a", False, False)) == set(["1"])
|
||||
assert set(markings.get_markings(data, "a", True, False)) == set(["1"])
|
||||
assert set(markings.get_markings(data, "a", True, True)) == set(["1"])
|
||||
assert set(markings.get_markings(data, "a", False, True)) == set(["1"])
|
||||
|
||||
assert set(markings.get_markings(data, "b", False, False)) == set([])
|
||||
assert set(markings.get_markings(data, "b", True, False)) == set([])
|
||||
assert set(markings.get_markings(data, "b", True, True)) == set([])
|
||||
assert set(markings.get_markings(data, "b", False, True)) == set([])
|
||||
|
||||
assert set(markings.get_markings(data, "c", False, False)) == set(["2"])
|
||||
assert set(markings.get_markings(data, "c", True, False)) == set(["2"])
|
||||
assert set(markings.get_markings(data, "c", True, True)) == set(["2", "3", "4", "5"])
|
||||
assert set(markings.get_markings(data, "c", False, True)) == set(["2", "3", "4", "5"])
|
||||
|
||||
assert set(markings.get_markings(data, "c.[0]", False, False)) == set([])
|
||||
assert set(markings.get_markings(data, "c.[0]", True, False)) == set(["2"])
|
||||
assert set(markings.get_markings(data, "c.[0]", True, True)) == set(["2"])
|
||||
assert set(markings.get_markings(data, "c.[0]", False, True)) == set([])
|
||||
|
||||
assert set(markings.get_markings(data, "c.[1]", False, False)) == set(["3"])
|
||||
assert set(markings.get_markings(data, "c.[1]", True, False)) == set(["2", "3"])
|
||||
assert set(markings.get_markings(data, "c.[1]", True, True)) == set(["2", "3"])
|
||||
assert set(markings.get_markings(data, "c.[1]", False, True)) == set(["3"])
|
||||
|
||||
assert set(markings.get_markings(data, "c.[2]", False, False)) == set(["4"])
|
||||
assert set(markings.get_markings(data, "c.[2]", True, False)) == set(["2", "4"])
|
||||
assert set(markings.get_markings(data, "c.[2]", True, True)) == set(["2", "4", "5"])
|
||||
assert set(markings.get_markings(data, "c.[2]", False, True)) == set(["4", "5"])
|
||||
|
||||
assert set(markings.get_markings(data, "c.[2].g", False, False)) == set(["5"])
|
||||
assert set(markings.get_markings(data, "c.[2].g", True, False)) == set(["2", "4", "5"])
|
||||
assert set(markings.get_markings(data, "c.[2].g", True, True)) == set(["2", "4", "5"])
|
||||
assert set(markings.get_markings(data, "c.[2].g", False, True)) == set(["5"])
|
||||
|
||||
assert set(markings.get_markings(data, "x", False, False)) == set(["6"])
|
||||
assert set(markings.get_markings(data, "x", True, False)) == set(["6"])
|
||||
assert set(markings.get_markings(data, "x", True, True)) == set(["6", "7", "8", "9", "10"])
|
||||
assert set(markings.get_markings(data, "x", False, True)) == set(["6", "7", "8", "9", "10"])
|
||||
|
||||
assert set(markings.get_markings(data, "x.y", False, False)) == set(["7"])
|
||||
assert set(markings.get_markings(data, "x.y", True, False)) == set(["6", "7"])
|
||||
assert set(markings.get_markings(data, "x.y", True, True)) == set(["6", "7", "8"])
|
||||
assert set(markings.get_markings(data, "x.y", False, True)) == set(["7", "8"])
|
||||
|
||||
assert set(markings.get_markings(data, "x.y.[0]", False, False)) == set([])
|
||||
assert set(markings.get_markings(data, "x.y.[0]", True, False)) == set(["6", "7"])
|
||||
assert set(markings.get_markings(data, "x.y.[0]", True, True)) == set(["6", "7"])
|
||||
assert set(markings.get_markings(data, "x.y.[0]", False, True)) == set([])
|
||||
|
||||
assert set(markings.get_markings(data, "x.y.[1]", False, False)) == set(["8"])
|
||||
assert set(markings.get_markings(data, "x.y.[1]", True, False)) == set(["6", "7", "8"])
|
||||
assert set(markings.get_markings(data, "x.y.[1]", True, True)) == set(["6", "7", "8"])
|
||||
assert set(markings.get_markings(data, "x.y.[1]", False, True)) == set(["8"])
|
||||
|
||||
assert set(markings.get_markings(data, "x.z", False, False)) == set(["9"])
|
||||
assert set(markings.get_markings(data, "x.z", True, False)) == set(["6", "9"])
|
||||
assert set(markings.get_markings(data, "x.z", True, True)) == set(["6", "9", "10"])
|
||||
assert set(markings.get_markings(data, "x.z", False, True)) == set(["9", "10"])
|
||||
|
||||
assert set(markings.get_markings(data, "x.z.foo1", False, False)) == set([])
|
||||
assert set(markings.get_markings(data, "x.z.foo1", True, False)) == set(["6", "9"])
|
||||
assert set(markings.get_markings(data, "x.z.foo1", True, True)) == set(["6", "9"])
|
||||
assert set(markings.get_markings(data, "x.z.foo1", False, True)) == set([])
|
||||
|
||||
assert set(markings.get_markings(data, "x.z.foo2", False, False)) == set(["10"])
|
||||
assert set(markings.get_markings(data, "x.z.foo2", True, False)) == set(["6", "9", "10"])
|
||||
assert set(markings.get_markings(data, "x.z.foo2", True, True)) == set(["6", "9", "10"])
|
||||
assert set(markings.get_markings(data, "x.z.foo2", False, True)) == set(["10"])
|
||||
|
||||
|
||||
def test_remove_marking_remove_one_selector_with_multiple_refs():
|
||||
after = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
}
|
||||
before = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
},
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--2"
|
||||
},
|
||||
]
|
||||
}
|
||||
markings.remove_markings(before, ["description"], ["marking-definition--1", "marking-definition--2"])
|
||||
assert before == after
|
||||
|
||||
|
||||
def test_remove_marking_remove_multiple_selector_one_ref():
|
||||
after = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
}
|
||||
before = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description", "title"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
},
|
||||
]
|
||||
}
|
||||
markings.remove_markings(before, ["description", "title"], ["marking-definition--1"])
|
||||
assert before == after
|
||||
|
||||
|
||||
def test_remove_marking_mark_one_selector_from_multiple_ones():
|
||||
after = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
},
|
||||
]
|
||||
}
|
||||
before = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description", "title"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
},
|
||||
]
|
||||
}
|
||||
markings.remove_markings(before, ["title"], ["marking-definition--1"])
|
||||
assert before == after
|
||||
|
||||
|
||||
def test_remove_marking_mark_one_selector_markings_from_multiple_ones():
|
||||
after = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
},
|
||||
{
|
||||
"selectors": ["description", "title"],
|
||||
"marking_ref": "marking-definition--2"
|
||||
},
|
||||
]
|
||||
}
|
||||
before = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description", "title"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
},
|
||||
{
|
||||
"selectors": ["description", "title"],
|
||||
"marking_ref": "marking-definition--2"
|
||||
},
|
||||
]
|
||||
}
|
||||
markings.remove_markings(before, ["title"], ["marking-definition--1"])
|
||||
|
||||
for m in before["granular_markings"]:
|
||||
assert m in after["granular_markings"]
|
||||
|
||||
|
||||
def test_remove_marking_mark_mutilple_selector_multiple_refs():
|
||||
after = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
}
|
||||
before = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description", "title"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
},
|
||||
{
|
||||
"selectors": ["description", "title"],
|
||||
"marking_ref": "marking-definition--2"
|
||||
},
|
||||
]
|
||||
}
|
||||
markings.remove_markings(before, ["description", "title"], ["marking-definition--1", "marking-definition--2"])
|
||||
assert before == after
|
||||
|
||||
|
||||
def test_remove_marking_mark_another_property_same_marking():
|
||||
after = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
},
|
||||
]
|
||||
}
|
||||
before = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
},
|
||||
{
|
||||
"selectors": ["title"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
}
|
||||
]
|
||||
}
|
||||
markings.remove_markings(before, ["title"], ["marking-definition--1"])
|
||||
assert before == after
|
||||
|
||||
|
||||
def test_remove_marking_mark_same_property_same_marking():
|
||||
after = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
}
|
||||
before = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
}
|
||||
]
|
||||
}
|
||||
markings.remove_markings(before, ["description"], ["marking-definition--1"])
|
||||
assert before == after
|
||||
|
||||
|
||||
def test_remove_marking_bad_selector():
|
||||
before = {
|
||||
"description": "test description",
|
||||
}
|
||||
with pytest.raises(AssertionError):
|
||||
markings.remove_markings(before, ["title"], ["marking-definition--1", "marking-definition--2"])
|
||||
|
||||
|
||||
IS_MARKED_TEST_DATA = \
|
||||
{
|
||||
"title": "test title",
|
||||
"description": "test description",
|
||||
"revision": 2,
|
||||
"type": "test",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
},
|
||||
{
|
||||
"selectors": ["revision", "description"],
|
||||
"marking_ref": "marking-definition--2"
|
||||
},
|
||||
{
|
||||
"selectors": ["revision", "description"],
|
||||
"marking_ref": "marking-definition--3"
|
||||
},
|
||||
]
|
||||
}
|
||||
|
||||
|
||||
@pytest.mark.parametrize("data", [IS_MARKED_TEST_DATA])
|
||||
def test_is_marked_smoke(data):
|
||||
"""Smoke test is_marked call does not fail."""
|
||||
assert markings.is_marked(data, ["description"])
|
||||
assert markings.is_marked(data, ["title"]) is False
|
||||
|
||||
|
||||
@pytest.mark.parametrize("data,selector", [
|
||||
(IS_MARKED_TEST_DATA, "foo"),
|
||||
(IS_MARKED_TEST_DATA, ""),
|
||||
(IS_MARKED_TEST_DATA, []),
|
||||
(IS_MARKED_TEST_DATA, [""]),
|
||||
(IS_MARKED_TEST_DATA, "x.z.[-2]"),
|
||||
(IS_MARKED_TEST_DATA, "c.f"),
|
||||
(IS_MARKED_TEST_DATA, "c.[2].i"),
|
||||
(IS_MARKED_TEST_DATA, "c.[3]"),
|
||||
(IS_MARKED_TEST_DATA, "d"),
|
||||
(IS_MARKED_TEST_DATA, "x.[0]"),
|
||||
(IS_MARKED_TEST_DATA, "z.y.w"),
|
||||
(IS_MARKED_TEST_DATA, "x.z.[1]"),
|
||||
(IS_MARKED_TEST_DATA, "x.z.foo3")
|
||||
])
|
||||
def test_is_marked_invalid_selector(data, selector):
|
||||
"""Test invalid selector raises an error."""
|
||||
with pytest.raises(AssertionError):
|
||||
markings.is_marked(data, selector)
|
||||
|
||||
|
||||
@pytest.mark.parametrize("data", [IS_MARKED_TEST_DATA])
|
||||
def test_is_marked_mix_selector(data):
|
||||
"""Test valid selector, one marked and one not marked returns True."""
|
||||
assert markings.is_marked(data, ["description", "revision"])
|
||||
assert markings.is_marked(data, ["description"])
|
||||
|
||||
|
||||
@pytest.mark.parametrize("data", [IS_MARKED_TEST_DATA])
|
||||
def test_is_marked_valid_selector_no_refs(data):
|
||||
"""Test that a valid selector return True when it has marking refs and False when not."""
|
||||
assert markings.is_marked(data, ["description"])
|
||||
assert markings.is_marked(data, ["description"], ["marking-definition--2", "marking-definition--3"])
|
||||
assert markings.is_marked(data, ["description"], ["marking-definition--2"])
|
||||
assert markings.is_marked(data, ["description"], ["marking-definition--2", "marking-definition--8"]) is False
|
||||
|
||||
|
||||
@pytest.mark.parametrize("data", [IS_MARKED_TEST_DATA])
|
||||
def test_is_marked_valid_selector_and_refs(data):
|
||||
"""Test that a valid selector returns True when marking_refs match."""
|
||||
assert markings.is_marked(data, ["description"], ["marking-definition--1"])
|
||||
assert markings.is_marked(data, ["title"], ["marking-definition--1"]) is False
|
||||
|
||||
|
||||
@pytest.mark.parametrize("data", [IS_MARKED_TEST_DATA])
|
||||
def test_is_marked_valid_selector_multiple_refs(data):
|
||||
"""Test that a valid selector returns True if aall marking_refs match.
|
||||
Otherwise False."""
|
||||
assert markings.is_marked(data, ["revision"], ["marking-definition--2", "marking-definition--3"])
|
||||
assert markings.is_marked(data, ["revision"], ["marking-definition--2", "marking-definition--1"]) is False
|
||||
assert markings.is_marked(data, ["revision"], "marking-definition--2")
|
||||
assert markings.is_marked(data, ["revision"], ["marking-definition--1234"]) is False
|
||||
|
||||
|
||||
@pytest.mark.parametrize("data", [IS_MARKED_TEST_DATA])
|
||||
def test_is_marked_no_marking_refs(data):
|
||||
"""Test that a valid content selector with no marking_refs returns True
|
||||
if there is a granular_marking that asserts that field, False
|
||||
otherwise."""
|
||||
assert markings.is_marked(data, ["type"]) is False
|
||||
assert markings.is_marked(data, ["revision"])
|
||||
|
||||
|
||||
def test_is_marked_positional_arguments_combinations():
|
||||
"""Test multiple combinations for inherited and descendant markings."""
|
||||
test_tlo = \
|
||||
{
|
||||
"a": 333,
|
||||
"b": "value",
|
||||
"c": [
|
||||
17,
|
||||
"list value",
|
||||
{
|
||||
"g": "nested",
|
||||
"h": 45
|
||||
}
|
||||
],
|
||||
"x": {
|
||||
"y": [
|
||||
"hello",
|
||||
88
|
||||
],
|
||||
"z": {
|
||||
"foo1": "bar",
|
||||
"foo2": 65
|
||||
}
|
||||
},
|
||||
"granular_markings": [
|
||||
{
|
||||
"marking_ref": "1",
|
||||
"selectors": ["a"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "2",
|
||||
"selectors": ["c"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "3",
|
||||
"selectors": ["c.[1]"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "4",
|
||||
"selectors": ["c.[2]"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "5",
|
||||
"selectors": ["c.[2].g"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "6",
|
||||
"selectors": ["x"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "7",
|
||||
"selectors": ["x.y"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "8",
|
||||
"selectors": ["x.y.[1]"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "9",
|
||||
"selectors": ["x.z"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "10",
|
||||
"selectors": ["x.z.foo2"]
|
||||
},
|
||||
]
|
||||
}
|
||||
|
||||
assert markings.is_marked(test_tlo, "a", ["1"], False, False)
|
||||
assert markings.is_marked(test_tlo, "a", ["1"], True, False)
|
||||
assert markings.is_marked(test_tlo, "a", ["1"], True, True)
|
||||
assert markings.is_marked(test_tlo, "a", ["1"], False, True)
|
||||
|
||||
assert markings.is_marked(test_tlo, "b", inherited=False, descendants=False) is False
|
||||
assert markings.is_marked(test_tlo, "b", inherited=True, descendants=False) is False
|
||||
assert markings.is_marked(test_tlo, "b", inherited=True, descendants=True) is False
|
||||
assert markings.is_marked(test_tlo, "b", inherited=False, descendants=True) is False
|
||||
|
||||
assert markings.is_marked(test_tlo, "c", ["2"], False, False)
|
||||
assert markings.is_marked(test_tlo, "c", ["2"], True, False)
|
||||
assert markings.is_marked(test_tlo, "c", ["2", "3", "4", "5"], True, True)
|
||||
assert markings.is_marked(test_tlo, "c", ["2", "3", "4", "5"], False, True)
|
||||
|
||||
assert markings.is_marked(test_tlo, "c.[0]", inherited=False, descendants=False) is False
|
||||
assert markings.is_marked(test_tlo, "c.[0]", ["2"], True, False)
|
||||
assert markings.is_marked(test_tlo, "c.[0]", ["2"], True, True)
|
||||
assert markings.is_marked(test_tlo, "c.[0]", inherited=False, descendants=True) is False
|
||||
|
||||
assert markings.is_marked(test_tlo, "c.[1]", ["3"], False, False)
|
||||
assert markings.is_marked(test_tlo, "c.[1]", ["2", "3"], True, False)
|
||||
assert markings.is_marked(test_tlo, "c.[1]", ["2", "3"], True, True)
|
||||
assert markings.is_marked(test_tlo, "c.[1]", ["3"], False, True)
|
||||
|
||||
assert markings.is_marked(test_tlo, "c.[2]", ["4"], False, False)
|
||||
assert markings.is_marked(test_tlo, "c.[2]", ["2", "4"], True, False)
|
||||
assert markings.is_marked(test_tlo, "c.[2]", ["2", "4", "5"], True, True)
|
||||
assert markings.is_marked(test_tlo, "c.[2]", ["4", "5"], False, True)
|
||||
|
||||
assert markings.is_marked(test_tlo, "c.[2].g", ["5"], False, False)
|
||||
assert markings.is_marked(test_tlo, "c.[2].g", ["2", "4", "5"], True, False)
|
||||
assert markings.is_marked(test_tlo, "c.[2].g", ["2", "4", "5"], True, True)
|
||||
assert markings.is_marked(test_tlo, "c.[2].g", ["5"], False, True)
|
||||
|
||||
assert markings.is_marked(test_tlo, "x", ["6"], False, False)
|
||||
assert markings.is_marked(test_tlo, "x", ["6"], True, False)
|
||||
assert markings.is_marked(test_tlo, "x", ["6", "7", "8", "9", "10"], True, True)
|
||||
assert markings.is_marked(test_tlo, "x", ["6", "7", "8", "9", "10"], False, True)
|
||||
|
||||
assert markings.is_marked(test_tlo, "x.y", ["7"], False, False)
|
||||
assert markings.is_marked(test_tlo, "x.y", ["6", "7"], True, False)
|
||||
assert markings.is_marked(test_tlo, "x.y", ["6", "7", "8"], True, True)
|
||||
assert markings.is_marked(test_tlo, "x.y", ["7", "8"], False, True)
|
||||
|
||||
assert markings.is_marked(test_tlo, "x.y.[0]", inherited=False, descendants=False) is False
|
||||
assert markings.is_marked(test_tlo, "x.y.[0]", ["6", "7"], True, False)
|
||||
assert markings.is_marked(test_tlo, "x.y.[0]", ["6", "7"], True, True)
|
||||
assert markings.is_marked(test_tlo, "x.y.[0]", inherited=False, descendants=True) is False
|
||||
|
||||
assert markings.is_marked(test_tlo, "x.y.[1]", ["8"], False, False)
|
||||
assert markings.is_marked(test_tlo, "x.y.[1]", ["6", "7", "8"], True, False)
|
||||
assert markings.is_marked(test_tlo, "x.y.[1]", ["6", "7", "8"], True, True)
|
||||
assert markings.is_marked(test_tlo, "x.y.[1]", ["8"], False, True)
|
||||
|
||||
assert markings.is_marked(test_tlo, "x.z", ["9"], False, False)
|
||||
assert markings.is_marked(test_tlo, "x.z", ["6", "9"], True, False)
|
||||
assert markings.is_marked(test_tlo, "x.z", ["6", "9", "10"], True, True)
|
||||
assert markings.is_marked(test_tlo, "x.z", ["9", "10"], False, True)
|
||||
|
||||
assert markings.is_marked(test_tlo, "x.z.foo1", inherited=False, descendants=False) is False
|
||||
assert markings.is_marked(test_tlo, "x.z.foo1", ["6", "9"], True, False)
|
||||
assert markings.is_marked(test_tlo, "x.z.foo1", ["6", "9"], True, True)
|
||||
assert markings.is_marked(test_tlo, "x.z.foo1", inherited=False, descendants=True) is False
|
||||
|
||||
assert markings.is_marked(test_tlo, "x.z.foo2", ["10"], False, False)
|
||||
assert markings.is_marked(test_tlo, "x.z.foo2", ["6", "9", "10"], True, False)
|
||||
assert markings.is_marked(test_tlo, "x.z.foo2", ["6", "9", "10"], True, True)
|
||||
assert markings.is_marked(test_tlo, "x.z.foo2", ["10"], False, True)
|
||||
|
||||
|
||||
def test_set_marking_mark_one_selector_multiple_refs():
|
||||
before = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
}
|
||||
after = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
},
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--2"
|
||||
},
|
||||
]
|
||||
}
|
||||
markings.set_markings(before, ["description"], ["marking-definition--1", "marking-definition--2"])
|
||||
for m in before["granular_markings"]:
|
||||
assert m in after["granular_markings"]
|
||||
|
||||
|
||||
def test_set_marking_mark_multiple_selector_one_refs():
|
||||
before = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description", "title"],
|
||||
"marking_ref": "marking-definition--3"
|
||||
},
|
||||
]
|
||||
}
|
||||
after = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description", "title"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
},
|
||||
]
|
||||
}
|
||||
markings.set_markings(before, ["description", "title"], ["marking-definition--1"])
|
||||
assert before == after
|
||||
|
||||
|
||||
def test_set_marking_mark_multiple_selector_multiple_refs_from_none():
|
||||
before = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
}
|
||||
after = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description", "title"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
},
|
||||
{
|
||||
"selectors": ["description", "title"],
|
||||
"marking_ref": "marking-definition--2"
|
||||
},
|
||||
]
|
||||
}
|
||||
markings.set_markings(before, ["description", "title"], ["marking-definition--1", "marking-definition--2"])
|
||||
for m in before["granular_markings"]:
|
||||
assert m in after["granular_markings"]
|
||||
|
||||
|
||||
def test_set_marking_mark_another_property_same_marking():
|
||||
before = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
}
|
||||
]
|
||||
}
|
||||
after = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--7"
|
||||
},
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--8"
|
||||
},
|
||||
]
|
||||
}
|
||||
markings.set_markings(before, ["description"], ["marking-definition--7", "marking-definition--8"])
|
||||
|
||||
for m in before["granular_markings"]:
|
||||
assert m in after["granular_markings"]
|
||||
|
||||
|
||||
@pytest.mark.parametrize("marking", [
|
||||
(["foo"], ["marking-definition--7", "marking-definition--8"]),
|
||||
("", ["marking-definition--7", "marking-definition--8"]),
|
||||
([], ["marking-definition--7", "marking-definition--8"]),
|
||||
([""], ["marking-definition--7", "marking-definition--8"])
|
||||
])
|
||||
def test_set_marking_bad_selector(marking):
|
||||
before = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
}
|
||||
]
|
||||
}
|
||||
after = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
},
|
||||
]
|
||||
}
|
||||
|
||||
with pytest.raises(AssertionError):
|
||||
markings.set_markings(before, marking[0], marking[1])
|
||||
|
||||
assert before == after
|
||||
|
||||
|
||||
def test_set_marking_mark_same_property_same_marking():
|
||||
before = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
}
|
||||
]
|
||||
}
|
||||
after = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
}
|
||||
]
|
||||
}
|
||||
markings.set_markings(before, ["description"], ["marking-definition--1"])
|
||||
assert before == after
|
||||
|
||||
|
||||
CLEAR_MARKINGS_TEST_DATA = \
|
||||
{
|
||||
"title": "test title",
|
||||
"description": "test description",
|
||||
"revision": 2,
|
||||
"type": "test",
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--1"
|
||||
},
|
||||
{
|
||||
"selectors": ["revision", "description"],
|
||||
"marking_ref": "marking-definition--2"
|
||||
},
|
||||
{
|
||||
"selectors": ["revision", "description", "type"],
|
||||
"marking_ref": "marking-definition--3"
|
||||
},
|
||||
]
|
||||
}
|
||||
|
||||
|
||||
@pytest.mark.parametrize("data", [CLEAR_MARKINGS_TEST_DATA])
|
||||
def test_clear_marking_smoke(data):
|
||||
"""Test clear_marking call does not fail."""
|
||||
markings.clear_markings(data, "revision")
|
||||
assert markings.is_marked(data, "revision") is False
|
||||
|
||||
|
||||
@pytest.mark.parametrize("data", [CLEAR_MARKINGS_TEST_DATA])
|
||||
def test_clear_marking_multiple_selectors(data):
|
||||
"""Test clearing markings for multiple selectors effectively removes associated markings."""
|
||||
markings.clear_markings(data, ["type", "description"])
|
||||
assert markings.is_marked(data, ["type", "description"]) is False
|
||||
|
||||
|
||||
@pytest.mark.parametrize("data", [CLEAR_MARKINGS_TEST_DATA])
|
||||
def test_clear_marking_one_selector(data):
|
||||
"""Test markings associated with one selector were removed."""
|
||||
markings.clear_markings(data, "description")
|
||||
assert markings.is_marked(data, "description") is False
|
||||
|
||||
|
||||
@pytest.mark.parametrize("data", [CLEAR_MARKINGS_TEST_DATA])
|
||||
def test_clear_marking_all_selectors(data):
|
||||
markings.clear_markings(data, ["description", "type", "revision"])
|
||||
assert markings.is_marked(data, "description") is False
|
||||
assert "granular_markings" not in data
|
||||
|
||||
|
||||
@pytest.mark.parametrize("data,selector", [
|
||||
(CLEAR_MARKINGS_TEST_DATA, "foo"),
|
||||
(CLEAR_MARKINGS_TEST_DATA, ""),
|
||||
(CLEAR_MARKINGS_TEST_DATA, []),
|
||||
(CLEAR_MARKINGS_TEST_DATA, [""]),
|
||||
])
|
||||
def test_clear_marking_bad_selector(data, selector):
|
||||
"""Test bad selector raises exception."""
|
||||
with pytest.raises(AssertionError):
|
||||
markings.clear_markings(data, selector)
|
||||
|
|
@ -0,0 +1,487 @@
|
|||
|
||||
from stix2 import markings
|
||||
|
||||
import pytest
|
||||
|
||||
"""Tests for the Data Markings API."""
|
||||
|
||||
|
||||
def test_add_markings_one_marking():
|
||||
before = {
|
||||
"title": "test title",
|
||||
"description": "test description"
|
||||
}
|
||||
|
||||
after = {
|
||||
"title": "test title",
|
||||
"description": "test description",
|
||||
"object_marking_refs": ["marking-definition--1"]
|
||||
}
|
||||
|
||||
markings.add_markings(before, None, "marking-definition--1")
|
||||
|
||||
assert before == after
|
||||
|
||||
|
||||
def test_add_markings_multiple_marking():
|
||||
before = {
|
||||
"title": "test title",
|
||||
"description": "test description"
|
||||
}
|
||||
|
||||
after = {
|
||||
"title": "test title",
|
||||
"description": "test description",
|
||||
"object_marking_refs": ["marking-definition--1", "marking-definition--2"]
|
||||
}
|
||||
|
||||
markings.add_markings(before, None, ["marking-definition--1", "marking-definition--2"])
|
||||
|
||||
for m in before["object_marking_refs"]:
|
||||
assert m in after["object_marking_refs"]
|
||||
|
||||
|
||||
def test_add_markings_combination():
|
||||
before = {
|
||||
"title": "test title",
|
||||
"description": "test description"
|
||||
}
|
||||
|
||||
after = {
|
||||
"title": "test title",
|
||||
"description": "test description",
|
||||
"object_marking_refs": ["marking-definition--1", "marking-definition--2"],
|
||||
"granular_markings": [
|
||||
{
|
||||
"selectors": ["title"],
|
||||
"marking_ref": "marking-definition--3"
|
||||
},
|
||||
{
|
||||
"selectors": ["description"],
|
||||
"marking_ref": "marking-definition--4"
|
||||
},
|
||||
]
|
||||
}
|
||||
|
||||
markings.add_markings(before, None, "marking-definition--1")
|
||||
markings.add_markings(before, None, "marking-definition--2")
|
||||
markings.add_markings(before, "title", "marking-definition--3")
|
||||
markings.add_markings(before, "description", "marking-definition--4")
|
||||
|
||||
for m in before["granular_markings"]:
|
||||
assert m in after["granular_markings"]
|
||||
|
||||
for m in before["object_marking_refs"]:
|
||||
assert m in after["object_marking_refs"]
|
||||
|
||||
|
||||
@pytest.mark.parametrize("data", [
|
||||
([""]),
|
||||
(""),
|
||||
([]),
|
||||
(["marking-definition--1", 456])
|
||||
])
|
||||
def test_add_markings_bad_markings(data):
|
||||
before = {
|
||||
"title": "test title",
|
||||
"description": "test description"
|
||||
}
|
||||
with pytest.raises(AssertionError):
|
||||
markings.add_markings(before, None, data)
|
||||
|
||||
assert "object_marking_refs" not in before
|
||||
|
||||
|
||||
GET_MARKINGS_TEST_DATA = \
|
||||
{
|
||||
"a": 333,
|
||||
"b": "value",
|
||||
"c": [
|
||||
17,
|
||||
"list value",
|
||||
{
|
||||
"g": "nested",
|
||||
"h": 45
|
||||
}
|
||||
],
|
||||
"x": {
|
||||
"y": [
|
||||
"hello",
|
||||
88
|
||||
],
|
||||
"z": {
|
||||
"foo1": "bar",
|
||||
"foo2": 65
|
||||
}
|
||||
},
|
||||
"object_marking_refs": ["11"],
|
||||
"granular_markings": [
|
||||
{
|
||||
"marking_ref": "1",
|
||||
"selectors": ["a"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "2",
|
||||
"selectors": ["c"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "3",
|
||||
"selectors": ["c.[1]"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "4",
|
||||
"selectors": ["c.[2]"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "5",
|
||||
"selectors": ["c.[2].g"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "6",
|
||||
"selectors": ["x"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "7",
|
||||
"selectors": ["x.y"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "8",
|
||||
"selectors": ["x.y.[1]"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "9",
|
||||
"selectors": ["x.z"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "10",
|
||||
"selectors": ["x.z.foo2"]
|
||||
},
|
||||
]
|
||||
}
|
||||
|
||||
|
||||
@pytest.mark.parametrize("data", [GET_MARKINGS_TEST_DATA])
|
||||
def test_get_markings_object_marking(data):
|
||||
assert set(markings.get_markings(data, None)) == set(["11"])
|
||||
|
||||
|
||||
@pytest.mark.parametrize("data", [GET_MARKINGS_TEST_DATA])
|
||||
def test_get_markings_object_and_granular_combinations(data):
|
||||
"""Test multiple combinations for inherited and descendant markings."""
|
||||
assert set(markings.get_markings(data, "a", False, False)) == set(["1"])
|
||||
assert set(markings.get_markings(data, "a", True, False)) == set(["1", "11"])
|
||||
assert set(markings.get_markings(data, "a", True, True)) == set(["1", "11"])
|
||||
assert set(markings.get_markings(data, "a", False, True)) == set(["1"])
|
||||
|
||||
assert set(markings.get_markings(data, "b", False, False)) == set([])
|
||||
assert set(markings.get_markings(data, "b", True, False)) == set(["11"])
|
||||
assert set(markings.get_markings(data, "b", True, True)) == set(["11"])
|
||||
assert set(markings.get_markings(data, "b", False, True)) == set([])
|
||||
|
||||
assert set(markings.get_markings(data, "c", False, False)) == set(["2"])
|
||||
assert set(markings.get_markings(data, "c", True, False)) == set(["2", "11"])
|
||||
assert set(markings.get_markings(data, "c", True, True)) == set(["2", "3", "4", "5", "11"])
|
||||
assert set(markings.get_markings(data, "c", False, True)) == set(["2", "3", "4", "5"])
|
||||
|
||||
assert set(markings.get_markings(data, "c.[0]", False, False)) == set([])
|
||||
assert set(markings.get_markings(data, "c.[0]", True, False)) == set(["2", "11"])
|
||||
assert set(markings.get_markings(data, "c.[0]", True, True)) == set(["2", "11"])
|
||||
assert set(markings.get_markings(data, "c.[0]", False, True)) == set([])
|
||||
|
||||
assert set(markings.get_markings(data, "c.[1]", False, False)) == set(["3"])
|
||||
assert set(markings.get_markings(data, "c.[1]", True, False)) == set(["2", "3", "11"])
|
||||
assert set(markings.get_markings(data, "c.[1]", True, True)) == set(["2", "3", "11"])
|
||||
assert set(markings.get_markings(data, "c.[1]", False, True)) == set(["3"])
|
||||
|
||||
assert set(markings.get_markings(data, "c.[2]", False, False)) == set(["4"])
|
||||
assert set(markings.get_markings(data, "c.[2]", True, False)) == set(["2", "4", "11"])
|
||||
assert set(markings.get_markings(data, "c.[2]", True, True)) == set(["2", "4", "5", "11"])
|
||||
assert set(markings.get_markings(data, "c.[2]", False, True)) == set(["4", "5"])
|
||||
|
||||
assert set(markings.get_markings(data, "c.[2].g", False, False)) == set(["5"])
|
||||
assert set(markings.get_markings(data, "c.[2].g", True, False)) == set(["2", "4", "5", "11"])
|
||||
assert set(markings.get_markings(data, "c.[2].g", True, True)) == set(["2", "4", "5", "11"])
|
||||
assert set(markings.get_markings(data, "c.[2].g", False, True)) == set(["5"])
|
||||
|
||||
assert set(markings.get_markings(data, "x", False, False)) == set(["6"])
|
||||
assert set(markings.get_markings(data, "x", True, False)) == set(["6", "11"])
|
||||
assert set(markings.get_markings(data, "x", True, True)) == set(["6", "7", "8", "9", "10", "11"])
|
||||
assert set(markings.get_markings(data, "x", False, True)) == set(["6", "7", "8", "9", "10"])
|
||||
|
||||
assert set(markings.get_markings(data, "x.y", False, False)) == set(["7"])
|
||||
assert set(markings.get_markings(data, "x.y", True, False)) == set(["6", "7", "11"])
|
||||
assert set(markings.get_markings(data, "x.y", True, True)) == set(["6", "7", "8", "11"])
|
||||
assert set(markings.get_markings(data, "x.y", False, True)) == set(["7", "8"])
|
||||
|
||||
assert set(markings.get_markings(data, "x.y.[0]", False, False)) == set([])
|
||||
assert set(markings.get_markings(data, "x.y.[0]", True, False)) == set(["6", "7", "11"])
|
||||
assert set(markings.get_markings(data, "x.y.[0]", True, True)) == set(["6", "7", "11"])
|
||||
assert set(markings.get_markings(data, "x.y.[0]", False, True)) == set([])
|
||||
|
||||
assert set(markings.get_markings(data, "x.y.[1]", False, False)) == set(["8"])
|
||||
assert set(markings.get_markings(data, "x.y.[1]", True, False)) == set(["6", "7", "8", "11"])
|
||||
assert set(markings.get_markings(data, "x.y.[1]", True, True)) == set(["6", "7", "8", "11"])
|
||||
assert set(markings.get_markings(data, "x.y.[1]", False, True)) == set(["8"])
|
||||
|
||||
assert set(markings.get_markings(data, "x.z", False, False)) == set(["9"])
|
||||
assert set(markings.get_markings(data, "x.z", True, False)) == set(["6", "9", "11"])
|
||||
assert set(markings.get_markings(data, "x.z", True, True)) == set(["6", "9", "10", "11"])
|
||||
assert set(markings.get_markings(data, "x.z", False, True)) == set(["9", "10"])
|
||||
|
||||
assert set(markings.get_markings(data, "x.z.foo1", False, False)) == set([])
|
||||
assert set(markings.get_markings(data, "x.z.foo1", True, False)) == set(["6", "9", "11"])
|
||||
assert set(markings.get_markings(data, "x.z.foo1", True, True)) == set(["6", "9", "11"])
|
||||
assert set(markings.get_markings(data, "x.z.foo1", False, True)) == set([])
|
||||
|
||||
assert set(markings.get_markings(data, "x.z.foo2", False, False)) == set(["10"])
|
||||
assert set(markings.get_markings(data, "x.z.foo2", True, False)) == set(["6", "9", "10", "11"])
|
||||
assert set(markings.get_markings(data, "x.z.foo2", True, True)) == set(["6", "9", "10", "11"])
|
||||
assert set(markings.get_markings(data, "x.z.foo2", False, True)) == set(["10"])
|
||||
|
||||
|
||||
def test_remove_markings_object_level():
|
||||
after = {
|
||||
"title": "test title",
|
||||
"description": "test description"
|
||||
}
|
||||
|
||||
before = {
|
||||
"title": "test title",
|
||||
"description": "test description",
|
||||
"object_marking_refs": ["marking-definition--1"]
|
||||
}
|
||||
|
||||
markings.remove_markings(before, None, "marking-definition--1")
|
||||
|
||||
assert before == after
|
||||
|
||||
|
||||
def test_remove_markings_multiple():
|
||||
after = {
|
||||
"title": "test title",
|
||||
"description": "test description",
|
||||
"object_marking_refs": ["marking-definition--2"]
|
||||
}
|
||||
|
||||
before = {
|
||||
"title": "test title",
|
||||
"description": "test description",
|
||||
"object_marking_refs": ["marking-definition--1", "marking-definition--2", "marking-definition--3"]
|
||||
}
|
||||
|
||||
markings.remove_markings(before, None, ["marking-definition--1", "marking-definition--3"])
|
||||
|
||||
assert before == after
|
||||
|
||||
|
||||
def test_remove_markings_bad_markings():
|
||||
before = {
|
||||
"title": "test title",
|
||||
"description": "test description",
|
||||
"object_marking_refs": ["marking-definition--1", "marking-definition--2", "marking-definition--3"]
|
||||
}
|
||||
with pytest.raises(AssertionError):
|
||||
markings.remove_markings(before, None, ["marking-definition--5"])
|
||||
|
||||
|
||||
def test_clear_markings():
|
||||
after = {
|
||||
"title": "test title",
|
||||
"description": "test description"
|
||||
}
|
||||
|
||||
before = {
|
||||
"title": "test title",
|
||||
"description": "test description",
|
||||
"object_marking_refs": ["marking-definition--1", "marking-definition--2", "marking-definition--3"]
|
||||
}
|
||||
|
||||
markings.clear_markings(before, None)
|
||||
|
||||
assert before == after
|
||||
|
||||
|
||||
def test_is_marked_object_and_granular_combinations():
|
||||
"""Test multiple combinations for inherited and descendant markings."""
|
||||
test_tlo = \
|
||||
{
|
||||
"a": 333,
|
||||
"b": "value",
|
||||
"c": [
|
||||
17,
|
||||
"list value",
|
||||
{
|
||||
"g": "nested",
|
||||
"h": 45
|
||||
}
|
||||
],
|
||||
"x": {
|
||||
"y": [
|
||||
"hello",
|
||||
88
|
||||
],
|
||||
"z": {
|
||||
"foo1": "bar",
|
||||
"foo2": 65
|
||||
}
|
||||
},
|
||||
"object_marking_refs": "11",
|
||||
"granular_markings": [
|
||||
{
|
||||
"marking_ref": "1",
|
||||
"selectors": ["a"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "2",
|
||||
"selectors": ["c"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "3",
|
||||
"selectors": ["c.[1]"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "4",
|
||||
"selectors": ["c.[2]"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "5",
|
||||
"selectors": ["c.[2].g"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "6",
|
||||
"selectors": ["x"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "7",
|
||||
"selectors": ["x.y"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "8",
|
||||
"selectors": ["x.y.[1]"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "9",
|
||||
"selectors": ["x.z"]
|
||||
},
|
||||
{
|
||||
"marking_ref": "10",
|
||||
"selectors": ["x.z.foo2"]
|
||||
},
|
||||
]
|
||||
}
|
||||
|
||||
assert markings.is_marked(test_tlo, "a", ["1"], False, False)
|
||||
assert markings.is_marked(test_tlo, "a", ["1", "11"], True, False)
|
||||
assert markings.is_marked(test_tlo, "a", ["1", "11"], True, True)
|
||||
assert markings.is_marked(test_tlo, "a", ["1"], False, True)
|
||||
|
||||
assert markings.is_marked(test_tlo, "b", inherited=False, descendants=False) is False
|
||||
assert markings.is_marked(test_tlo, "b", ["11"], True, False)
|
||||
assert markings.is_marked(test_tlo, "b", ["11"], True, True)
|
||||
assert markings.is_marked(test_tlo, "b", inherited=False, descendants=True) is False
|
||||
|
||||
assert markings.is_marked(test_tlo, "c", ["2"], False, False)
|
||||
assert markings.is_marked(test_tlo, "c", ["2", "11"], True, False)
|
||||
assert markings.is_marked(test_tlo, "c", ["2", "3", "4", "5", "11"], True, True)
|
||||
assert markings.is_marked(test_tlo, "c", ["2", "3", "4", "5"], False, True)
|
||||
|
||||
assert markings.is_marked(test_tlo, "c.[0]", inherited=False, descendants=False) is False
|
||||
assert markings.is_marked(test_tlo, "c.[0]", ["2", "11"], True, False)
|
||||
assert markings.is_marked(test_tlo, "c.[0]", ["2", "11"], True, True)
|
||||
assert markings.is_marked(test_tlo, "c.[0]", inherited=False, descendants=True) is False
|
||||
|
||||
assert markings.is_marked(test_tlo, "c.[1]", ["3"], False, False)
|
||||
assert markings.is_marked(test_tlo, "c.[1]", ["2", "3", "11"], True, False)
|
||||
assert markings.is_marked(test_tlo, "c.[1]", ["2", "3", "11"], True, True)
|
||||
assert markings.is_marked(test_tlo, "c.[1]", ["3"], False, True)
|
||||
|
||||
assert markings.is_marked(test_tlo, "c.[2]", ["4"], False, False)
|
||||
assert markings.is_marked(test_tlo, "c.[2]", ["2", "4", "11"], True, False)
|
||||
assert markings.is_marked(test_tlo, "c.[2]", ["2", "4", "5", "11"], True, True)
|
||||
assert markings.is_marked(test_tlo, "c.[2]", ["4", "5"], False, True)
|
||||
|
||||
assert markings.is_marked(test_tlo, "c.[2].g", ["5"], False, False)
|
||||
assert markings.is_marked(test_tlo, "c.[2].g", ["2", "4", "5", "11"], True, False)
|
||||
assert markings.is_marked(test_tlo, "c.[2].g", ["2", "4", "5", "11"], True, True)
|
||||
assert markings.is_marked(test_tlo, "c.[2].g", ["5"], False, True)
|
||||
|
||||
assert markings.is_marked(test_tlo, "x", ["6"], False, False)
|
||||
assert markings.is_marked(test_tlo, "x", ["6", "11"], True, False)
|
||||
assert markings.is_marked(test_tlo, "x", ["6", "7", "8", "9", "10", "11"], True, True)
|
||||
assert markings.is_marked(test_tlo, "x", ["6", "7", "8", "9", "10"], False, True)
|
||||
|
||||
assert markings.is_marked(test_tlo, "x.y", ["7"], False, False)
|
||||
assert markings.is_marked(test_tlo, "x.y", ["6", "7", "11"], True, False)
|
||||
assert markings.is_marked(test_tlo, "x.y", ["6", "7", "8", "11"], True, True)
|
||||
assert markings.is_marked(test_tlo, "x.y", ["7", "8"], False, True)
|
||||
|
||||
assert markings.is_marked(test_tlo, "x.y.[0]", inherited=False, descendants=False) is False
|
||||
assert markings.is_marked(test_tlo, "x.y.[0]", ["6", "7", "11"], True, False)
|
||||
assert markings.is_marked(test_tlo, "x.y.[0]", ["6", "7", "11"], True, True)
|
||||
assert markings.is_marked(test_tlo, "x.y.[0]", inherited=False, descendants=True) is False
|
||||
|
||||
assert markings.is_marked(test_tlo, "x.y.[1]", ["8"], False, False)
|
||||
assert markings.is_marked(test_tlo, "x.y.[1]", ["6", "7", "8", "11"], True, False)
|
||||
assert markings.is_marked(test_tlo, "x.y.[1]", ["6", "7", "8", "11"], True, True)
|
||||
assert markings.is_marked(test_tlo, "x.y.[1]", ["8"], False, True)
|
||||
|
||||
assert markings.is_marked(test_tlo, "x.z", ["9"], False, False)
|
||||
assert markings.is_marked(test_tlo, "x.z", ["6", "9", "11"], True, False)
|
||||
assert markings.is_marked(test_tlo, "x.z", ["6", "9", "10", "11"], True, True)
|
||||
assert markings.is_marked(test_tlo, "x.z", ["9", "10"], False, True)
|
||||
|
||||
assert markings.is_marked(test_tlo, "x.z.foo1", inherited=False, descendants=False) is False
|
||||
assert markings.is_marked(test_tlo, "x.z.foo1", ["6", "9", "11"], True, False)
|
||||
assert markings.is_marked(test_tlo, "x.z.foo1", ["6", "9", "11"], True, True)
|
||||
assert markings.is_marked(test_tlo, "x.z.foo1", inherited=False, descendants=True) is False
|
||||
|
||||
assert markings.is_marked(test_tlo, "x.z.foo2", ["10"], False, False)
|
||||
assert markings.is_marked(test_tlo, "x.z.foo2", ["6", "9", "10", "11"], True, False)
|
||||
assert markings.is_marked(test_tlo, "x.z.foo2", ["6", "9", "10", "11"], True, True)
|
||||
assert markings.is_marked(test_tlo, "x.z.foo2", ["10"], False, True)
|
||||
|
||||
|
||||
def test_set_marking():
|
||||
before = {
|
||||
"title": "test title",
|
||||
"description": "test description",
|
||||
"object_marking_refs": ["marking-definition--1", "marking-definition--2", "marking-definition--3"]
|
||||
}
|
||||
after = {
|
||||
"title": "test title",
|
||||
"description": "test description",
|
||||
"object_marking_refs": ["marking-definition--7", "marking-definition--9"]
|
||||
}
|
||||
|
||||
markings.set_markings(before, None, ["marking-definition--7", "marking-definition--9"])
|
||||
|
||||
for m in before["object_marking_refs"]:
|
||||
assert m in ["marking-definition--7", "marking-definition--9"]
|
||||
|
||||
assert ["marking-definition--1", "marking-definition--2", "marking-definition--3"] not in before["object_marking_refs"]
|
||||
|
||||
for x in before["object_marking_refs"]:
|
||||
assert x in after["object_marking_refs"]
|
||||
|
||||
|
||||
@pytest.mark.parametrize("data", [
|
||||
([]),
|
||||
([""]),
|
||||
(""),
|
||||
(["marking-definition--7", 687])
|
||||
])
|
||||
def test_set_marking_bad_input(data):
|
||||
before = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"object_marking_refs": ["marking-definition--1"]
|
||||
}
|
||||
after = {
|
||||
"description": "test description",
|
||||
"title": "foo",
|
||||
"object_marking_refs": ["marking-definition--1"]
|
||||
}
|
||||
with pytest.raises(AssertionError):
|
||||
markings.set_markings(before, None, data)
|
||||
|
||||
assert before == after
|
||||
Loading…
Reference in New Issue