1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119
|
from __future__ import annotations
from typing import TYPE_CHECKING, NoReturn
import pytest
from pykka import Actor
if TYPE_CHECKING:
from collections.abc import Iterator
from pykka import ActorProxy
from tests.types import Runtime
class PropertyActor(Actor):
an_attr = "an_attr"
_private_attr = "secret"
@property
def a_ro_property(self) -> str:
return "a_ro_property"
_a_rw_property = "a_rw_property"
@property
def a_rw_property(self) -> str:
return self._a_rw_property
@a_rw_property.setter
def a_rw_property(self, value: str) -> None:
self._a_rw_property = value
@pytest.fixture
def actor_class(runtime: Runtime) -> type[PropertyActor]:
class PropertyActorImpl(PropertyActor, runtime.actor_class): # type: ignore[name-defined]
pass
return PropertyActorImpl
@pytest.fixture
def proxy(
actor_class: type[PropertyActor],
) -> Iterator[ActorProxy[PropertyActor]]:
proxy = actor_class.start().proxy()
yield proxy
proxy.stop()
def test_attr_can_be_read_using_get_postfix(
proxy: ActorProxy[PropertyActor],
) -> None:
assert proxy.an_attr.get() == "an_attr"
def test_attr_can_be_set_using_assignment(
proxy: ActorProxy[PropertyActor],
) -> None:
assert proxy.an_attr.get() == "an_attr"
proxy.an_attr = "an_attr_2"
assert proxy.an_attr.get() == "an_attr_2" # type: ignore[attr-defined]
def test_private_attr_access_raises_exception(
proxy: ActorProxy[PropertyActor],
) -> None:
with pytest.raises(AttributeError) as exc_info:
proxy._private_attr.get() # pyright: ignore[reportUnknownMemberType] # noqa: SLF001
assert "has no attribute '_private_attr'" in str(exc_info.value)
def test_missing_attr_access_raises_exception(
proxy: ActorProxy[PropertyActor],
) -> None:
with pytest.raises(AttributeError) as exc_info:
proxy.missing_attr.get()
assert "has no attribute 'missing_attr'" in str(exc_info.value)
def test_property_can_be_read_using_get_postfix(
proxy: ActorProxy[PropertyActor],
) -> None:
assert proxy.a_ro_property.get() == "a_ro_property"
assert proxy.a_rw_property.get() == "a_rw_property"
def test_property_can_be_set_using_assignment(
proxy: ActorProxy[PropertyActor],
) -> None:
proxy.a_rw_property = "a_rw_property_2"
assert proxy.a_rw_property.get() == "a_rw_property_2" # type: ignore[attr-defined]
def test_read_only_property_cannot_be_set(
proxy: ActorProxy[PropertyActor],
) -> None:
with pytest.raises(AttributeError):
proxy.a_ro_property = "a_ro_property_2"
def test_property_is_not_accessed_when_creating_proxy(runtime: Runtime) -> None:
class ExpensiveSideEffectActor(runtime.actor_class): # type: ignore[name-defined]
@property
def a_property(self) -> NoReturn:
# Imagine code with side effects or heavy resource usage here
raise Exception("Proxy creation accessed property")
actor_ref = ExpensiveSideEffectActor.start()
try:
actor_ref.proxy()
finally:
actor_ref.stop()
|