cti-python-stix2/stix2/test/test_properties.py

296 lines
6.9 KiB
Python
Raw Normal View History

2017-02-24 16:28:53 +01:00
import pytest
from stix2 import TCPExt
2017-05-18 17:24:43 +02:00
from stix2.exceptions import AtLeastOnePropertyError, DictionaryKeyError
from stix2.observables import EmailMIMEComponent, ExtensionsProperty
from stix2.properties import (BinaryProperty, BooleanProperty,
DictionaryProperty, EmbeddedObjectProperty,
2017-07-14 22:02:29 +02:00
EnumProperty, HashesProperty, HexProperty,
IDProperty, IntegerProperty, ListProperty,
Property, ReferenceProperty, StringProperty,
TimestampProperty, TypeProperty)
2017-04-11 18:10:55 +02:00
from .constants import FAKE_TIME
2017-02-24 16:28:53 +01:00
def test_property():
p = Property()
assert p.required is False
2017-04-17 21:13:11 +02:00
def test_basic_clean():
2017-02-24 16:28:53 +01:00
class Prop(Property):
2017-04-17 21:13:11 +02:00
def clean(self, value):
2017-02-24 16:28:53 +01:00
if value == 42:
return value
else:
raise ValueError("Must be 42")
p = Prop()
2017-04-17 21:13:11 +02:00
assert p.clean(42) == 42
2017-02-24 16:28:53 +01:00
with pytest.raises(ValueError):
2017-04-17 21:13:11 +02:00
p.clean(41)
2017-02-24 16:28:53 +01:00
def test_property_default():
2017-02-24 16:28:53 +01:00
class Prop(Property):
def default(self):
return 77
p = Prop()
assert p.default() == 77
def test_fixed_property():
p = Property(fixed="2.0")
2017-04-17 21:13:11 +02:00
assert p.clean("2.0")
with pytest.raises(ValueError):
2017-04-17 21:13:11 +02:00
assert p.clean("x") is False
with pytest.raises(ValueError):
2017-04-17 21:13:11 +02:00
assert p.clean(2.0) is False
2017-02-24 16:28:53 +01:00
assert p.default() == "2.0"
2017-04-17 21:13:11 +02:00
assert p.clean(p.default())
def test_list_property():
p = ListProperty(StringProperty)
2017-04-17 21:13:11 +02:00
assert p.clean(['abc', 'xyz'])
with pytest.raises(ValueError):
2017-04-17 21:13:11 +02:00
p.clean([])
def test_string_property():
prop = StringProperty()
2017-04-17 21:13:11 +02:00
assert prop.clean('foobar')
assert prop.clean(1)
assert prop.clean([1, 2, 3])
def test_type_property():
prop = TypeProperty('my-type')
2017-04-17 21:13:11 +02:00
assert prop.clean('my-type')
with pytest.raises(ValueError):
2017-04-17 21:13:11 +02:00
prop.clean('not-my-type')
assert prop.clean(prop.default())
2017-02-24 16:28:53 +01:00
def test_id_property():
idprop = IDProperty('my-type')
2017-04-17 21:13:11 +02:00
assert idprop.clean('my-type--90aaca8a-1110-5d32-956d-ac2f34a1bd8c')
2017-04-06 19:08:48 +02:00
with pytest.raises(ValueError) as excinfo:
2017-04-17 21:13:11 +02:00
idprop.clean('not-my-type--90aaca8a-1110-5d32-956d-ac2f34a1bd8c')
2017-04-06 19:08:48 +02:00
assert str(excinfo.value) == "must start with 'my-type--'."
with pytest.raises(ValueError) as excinfo:
2017-04-17 21:13:11 +02:00
idprop.clean('my-type--foo')
2017-04-25 16:03:37 +02:00
assert str(excinfo.value) == "must have a valid UUID after the prefix."
2017-04-06 19:08:48 +02:00
2017-04-17 21:13:11 +02:00
assert idprop.clean(idprop.default())
2017-04-18 15:19:38 +02:00
@pytest.mark.parametrize("value", [
2,
-1,
3.14,
False,
])
def test_integer_property_valid(value):
int_prop = IntegerProperty()
assert int_prop.clean(value) is not None
@pytest.mark.parametrize("value", [
"something",
StringProperty(),
])
def test_integer_property_invalid(value):
int_prop = IntegerProperty()
with pytest.raises(ValueError):
int_prop.clean(value)
2017-04-10 16:18:54 +02:00
@pytest.mark.parametrize("value", [
True,
False,
'True',
'False',
'true',
'false',
'TRUE',
'FALSE',
'T',
'F',
't',
'f',
1,
0,
])
def test_boolean_property_valid(value):
bool_prop = BooleanProperty()
2017-04-17 21:13:11 +02:00
assert bool_prop.clean(value) is not None
2017-04-10 16:18:54 +02:00
@pytest.mark.parametrize("value", [
'abc',
['false'],
{'true': 'true'},
2,
-1,
])
def test_boolean_property_invalid(value):
bool_prop = BooleanProperty()
with pytest.raises(ValueError):
2017-04-17 21:13:11 +02:00
bool_prop.clean(value)
def test_reference_property():
ref_prop = ReferenceProperty()
2017-04-17 21:13:11 +02:00
assert ref_prop.clean("my-type--3a331bfe-0566-55e1-a4a0-9a2cd355a300")
with pytest.raises(ValueError):
2017-04-17 21:13:11 +02:00
ref_prop.clean("foo")
2017-04-11 18:10:55 +02:00
@pytest.mark.parametrize("value", [
'2017-01-01T12:34:56Z',
'2017-01-01 12:34:56',
'Jan 1 2017 12:34:56',
])
def test_timestamp_property_valid(value):
ts_prop = TimestampProperty()
2017-04-17 21:13:11 +02:00
assert ts_prop.clean(value) == FAKE_TIME
2017-04-11 18:10:55 +02:00
def test_timestamp_property_invalid():
ts_prop = TimestampProperty()
with pytest.raises(ValueError):
2017-04-17 21:13:11 +02:00
ts_prop.clean(1)
2017-04-11 18:10:55 +02:00
with pytest.raises(ValueError):
2017-04-17 21:13:11 +02:00
ts_prop.clean("someday sometime")
def test_binary_property():
bin_prop = BinaryProperty()
assert bin_prop.clean("TG9yZW0gSXBzdW0=")
with pytest.raises(ValueError):
bin_prop.clean("foobar")
def test_hex_property():
hex_prop = HexProperty()
assert hex_prop.clean("4c6f72656d20497073756d")
with pytest.raises(ValueError):
hex_prop.clean("foobar")
@pytest.mark.parametrize("d", [
{'description': 'something'},
[('abc', 1), ('bcd', 2), ('cde', 3)],
])
def test_dictionary_property_valid(d):
dict_prop = DictionaryProperty()
assert dict_prop.clean(d)
@pytest.mark.parametrize("d", [
{'a': 'something'},
{'a'*300: 'something'},
{'Hey!': 'something'},
])
def test_dictionary_property_invalid(d):
dict_prop = DictionaryProperty()
with pytest.raises(DictionaryKeyError):
dict_prop.clean(d)
@pytest.mark.parametrize("value", [
{"sha256": "6db12788c37247f2316052e142f42f4b259d6561751e5f401a1ae2a6df9c674b"},
[('MD5', '2dfb1bcc980200c6706feee399d41b3f'), ('RIPEMD-160', 'b3a8cd8a27c90af79b3c81754f267780f443dfef')],
])
def test_hashes_property_valid(value):
hash_prop = HashesProperty()
assert hash_prop.clean(value)
@pytest.mark.parametrize("value", [
{"MD5": "a"},
{"SHA-256": "2dfb1bcc980200c6706feee399d41b3f"},
])
def test_hashes_property_invalid(value):
hash_prop = HashesProperty()
with pytest.raises(ValueError):
hash_prop.clean(value)
def test_embedded_property():
emb_prop = EmbeddedObjectProperty(type=EmailMIMEComponent)
mime = EmailMIMEComponent(
content_type="text/plain; charset=utf-8",
content_disposition="inline",
body="Cats are funny!"
)
assert emb_prop.clean(mime)
with pytest.raises(ValueError):
emb_prop.clean("string")
def test_enum_property():
enum_prop = EnumProperty(['a', 'b', 'c'])
assert enum_prop.clean('b')
with pytest.raises(ValueError):
enum_prop.clean('z')
def test_extension_property_valid():
ext_prop = ExtensionsProperty(enclosing_type='file')
assert ext_prop({
'windows-pebinary-ext': {
'pe_type': 'exe'
},
})
@pytest.mark.parametrize("data", [
1,
{'foobar-ext': {
'pe_type': 'exe'
}},
])
def test_extension_property_invalid(data):
ext_prop = ExtensionsProperty(enclosing_type='file')
with pytest.raises(ValueError):
ext_prop.clean(data)
def test_extension_property_invalid_type():
ext_prop = ExtensionsProperty(enclosing_type='indicator')
with pytest.raises(ValueError) as excinfo:
ext_prop.clean({
'windows-pebinary-ext': {
'pe_type': 'exe'
}}
)
assert 'no extensions defined' in str(excinfo.value)
2017-05-18 17:24:43 +02:00
def test_extension_at_least_one_property_constraint():
with pytest.raises(AtLeastOnePropertyError):
TCPExt()