izzy lyseggen
75 lines
1.9 KiB
Python
75 lines
1.9 KiB
Python
import pytest
|
|
from specklepy.objects import Base
|
|
from specklepy.api import operations
|
|
from contextlib import ExitStack as does_not_raise
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"invalid_prop_name",
|
|
[
|
|
(""),
|
|
("@"),
|
|
("@@wow"),
|
|
("this.is.bad"),
|
|
("super/bad"),
|
|
],
|
|
)
|
|
def test_empty_prop_names(invalid_prop_name: str) -> None:
|
|
base = Base()
|
|
with pytest.raises(ValueError):
|
|
base[invalid_prop_name] = "🐛️"
|
|
|
|
|
|
class FakeModel(Base):
|
|
"""Just a test class type."""
|
|
|
|
foo: str = ""
|
|
|
|
|
|
def test_new_type_registration() -> None:
|
|
"""Test if a new subclass is registered into the type register."""
|
|
assert Base.get_registered_type("FakeModel") == FakeModel
|
|
assert Base.get_registered_type("🐺️") is None
|
|
|
|
|
|
def test_fake_base_serialization() -> None:
|
|
fake_model = FakeModel(foo="bar")
|
|
|
|
serialized = operations.serialize(fake_model)
|
|
deserialized = operations.deserialize(serialized)
|
|
|
|
assert fake_model.get_id() == deserialized.get_id()
|
|
|
|
|
|
def test_duplicate_speckle_type_raises_error():
|
|
with pytest.raises(ValueError):
|
|
|
|
class NaughtyClass(Base, speckle_type="Base"):
|
|
"""This class has a speckle_type that is already taken."""
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"forbidden_attribute_name, expectation",
|
|
[
|
|
("", pytest.raises(ValueError)),
|
|
("@", pytest.raises(ValueError)),
|
|
("@@", pytest.raises(ValueError)),
|
|
("im.cheeky", pytest.raises(ValueError)),
|
|
("im.cheeky", pytest.raises(ValueError)),
|
|
("imgood", does_not_raise()),
|
|
],
|
|
)
|
|
def test_attribute_name_validation(
|
|
forbidden_attribute_name: str,
|
|
expectation,
|
|
base: Base,
|
|
):
|
|
with expectation:
|
|
base[forbidden_attribute_name] = None
|
|
|
|
|
|
def test_speckle_type_cannot_be_set(base: Base) -> None:
|
|
assert base.speckle_type == "Base"
|
|
base.speckle_type = "unset"
|
|
assert base.speckle_type == "Base"
|