kopia lustrzana https://github.com/corrscope/corrscope
100 wiersze
2.5 KiB
Python
100 wiersze
2.5 KiB
Python
import pytest
|
|
from pytest_cases import fixture
|
|
|
|
from corrscope.gui.model_bind import rgetattr, rsetattr, rhasattr, flatten_attr
|
|
|
|
|
|
class Person(object):
|
|
def __init__(self):
|
|
self.pet = Pet()
|
|
self.residence = Residence()
|
|
|
|
|
|
class Pet(object):
|
|
def __init__(self, name="Fido", species="Dog"):
|
|
self.name = name
|
|
self.species = species
|
|
|
|
|
|
class Residence(object):
|
|
def __init__(self, type="House", sqft=None):
|
|
self.type = type
|
|
self.sqft = sqft
|
|
|
|
|
|
@fixture
|
|
@pytest.mark.parametrize("s", ["__", "."])
|
|
def separator(s: str) -> str:
|
|
return s
|
|
|
|
|
|
def test_rgetattr(separator):
|
|
"""Test to ensure recursive model access works.
|
|
GUI elements are named f"prefix{_}" "recursive{_}attr" and bind to recursive.attr.
|
|
|
|
https://stackoverflow{_}com/a/31174427/
|
|
"""
|
|
p = Person()
|
|
_ = separator
|
|
|
|
# Test rgetattr(present)
|
|
assert rgetattr(p, f"pet{_}species") == "Dog"
|
|
assert rgetattr(p, f"pet{_}species", object()) == "Dog"
|
|
|
|
# Test rgetattr(missing)
|
|
assert rgetattr(p, f"pet{_}ghost{_}species", "calico") == "calico"
|
|
with pytest.raises(AttributeError):
|
|
# Without a default argument, `rgetattr`, like `getattr`, raises
|
|
# AttributeError when the dotted attribute is missing
|
|
print(rgetattr(p, f"pet{_}ghost{_}species"))
|
|
|
|
# Test rsetattr()
|
|
rsetattr(p, f"pet{_}name", "Sparky")
|
|
rsetattr(p, f"residence{_}type", "Apartment")
|
|
assert p.pet.name == "Sparky"
|
|
assert p.residence.type == "Apartment"
|
|
|
|
# Test rhasattr()
|
|
assert rhasattr(p, f"pet")
|
|
assert rhasattr(p, f"pet{_}name")
|
|
|
|
# Test rhasattr(levels of missing)
|
|
assert not rhasattr(p, f"pet{_}ghost")
|
|
assert not rhasattr(p, f"pet{_}ghost{_}species")
|
|
assert not rhasattr(p, f"ghost")
|
|
assert not rhasattr(p, f"ghost{_}species")
|
|
|
|
|
|
def test_flatten_attr(separator):
|
|
p = Person()
|
|
_ = separator
|
|
|
|
# Test nested
|
|
flat, name = flatten_attr(p, f"pet{_}name")
|
|
assert flat is p.pet
|
|
assert name == "name"
|
|
|
|
# Test 1 level
|
|
flat, name = flatten_attr(p, f"pet")
|
|
assert flat is p
|
|
assert name == "pet"
|
|
|
|
|
|
def test_rgetattr_broken(separator):
|
|
"""
|
|
rgetattr(default) fails to short-circuit/return on the first missing attribute.
|
|
I never use rgetattr(default) so I won't bother fixing the bug.
|
|
|
|
Wrong answer:
|
|
- None.foo AKA 1
|
|
- 1.bar AKA 1
|
|
- 1.imag == 0
|
|
|
|
Right answer:
|
|
- None.foo AKA return 1 to caller
|
|
"""
|
|
_ = separator
|
|
|
|
result = rgetattr(object(), f"nothing{_}imag", 1)
|
|
assert result == 1, result
|