55 lines
1.1 KiB
Python
55 lines
1.1 KiB
Python
|
import pytest
|
||
|
|
|
||
|
|
from stix2.properties import Property, IDProperty
|
||
|
|
|
||
|
|
|
||
|
|
def test_property():
|
||
|
|
p = Property()
|
||
|
|
|
||
|
|
assert p.required is False
|
||
|
|
|
||
|
|
|
||
|
|
def test_basic_validate():
|
||
|
|
class Prop(Property):
|
||
|
|
|
||
|
|
def validate(self, value):
|
||
|
|
if value == 42:
|
||
|
|
return value
|
||
|
|
else:
|
||
|
|
raise ValueError("Must be 42")
|
||
|
|
|
||
|
|
p = Prop()
|
||
|
|
|
||
|
|
assert p.validate(42) == 42
|
||
|
|
with pytest.raises(ValueError):
|
||
|
|
p.validate(41)
|
||
|
|
|
||
|
|
|
||
|
|
def test_default_field():
|
||
|
|
class Prop(Property):
|
||
|
|
|
||
|
|
def default(self):
|
||
|
|
return 77
|
||
|
|
|
||
|
|
p = Prop()
|
||
|
|
|
||
|
|
assert p.default() == 77
|
||
|
|
|
||
|
|
|
||
|
|
def test_fixed_property():
|
||
|
|
p = Property(fixed="2.0")
|
||
|
|
|
||
|
|
assert p.validate("2.0") is True
|
||
|
|
assert p.validate("x") is False
|
||
|
|
assert p.validate(2.0) is False
|
||
|
|
|
||
|
|
assert p.default() == "2.0"
|
||
|
|
|
||
|
|
|
||
|
|
def test_id_property():
|
||
|
|
idprop = IDProperty('my-type')
|
||
|
|
|
||
|
|
assert idprop.validate('my-type--90aaca8a-1110-5d32-956d-ac2f34a1bd8c') is True
|
||
|
|
assert idprop.validate('not-my-type--90aaca8a-1110-5d32-956d-ac2f34a1bd8c') is False
|
||
|
|
assert idprop.validate(idprop.default())
|