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
120
121
|
from __future__ import annotations
from typing import TYPE_CHECKING
import pytest
import pykka
from pykka import Actor
if TYPE_CHECKING:
from collections.abc import Iterator
from pykka import ActorProxy
from tests.types import Runtime
class NestedWithNoMarker:
inner = "nested_with_no_marker.inner"
class NestedWithNoMarkerAndSlots:
__slots__ = ["inner"]
def __init__(self) -> None:
self.inner = "nested_with_no_marker_and_slots.inner"
@pykka.traversable
class NestedWithDecoratorMarker:
inner = "nested_with_decorator_marker.inner"
class NestedWithAttrMarker:
pykka_traversable = True
inner = "nested_with_attr_marker.inner"
class NestedWithAttrMarkerAndSlots:
__slots__ = ["inner", "pykka_traversable"]
def __init__(self) -> None:
# Objects using '__slots__' cannot have class attributes.
self.pykka_traversable = True
self.inner = "nested_with_attr_marker_and_slots.inner"
class TraversableObjectsActor(Actor):
nested_with_no_marker = NestedWithNoMarker()
nested_with_function_marker = pykka.traversable(NestedWithNoMarker())
nested_with_decorator_marker = NestedWithDecoratorMarker()
nested_with_attr_marker = NestedWithAttrMarker()
nested_with_attr_marker_and_slots = NestedWithAttrMarkerAndSlots()
@property
def nested_object_property(self) -> NestedWithAttrMarker:
return NestedWithAttrMarker()
@pytest.fixture
def actor_class(runtime: Runtime) -> type[Actor]:
class TraversableObjectsActorImpl(TraversableObjectsActor, runtime.actor_class): # type: ignore[name-defined]
pass
return TraversableObjectsActorImpl
@pytest.fixture
def proxy(
actor_class: type[TraversableObjectsActor],
) -> Iterator[ActorProxy[TraversableObjectsActor]]:
proxy = actor_class.start().proxy()
yield proxy
proxy.stop()
def test_attr_without_marker_cannot_be_traversed(
proxy: ActorProxy[TraversableObjectsActor],
) -> None:
with pytest.raises(AttributeError) as exc_info:
proxy.nested_with_no_marker.inner.get()
assert "object has no attribute 'inner'" in str(exc_info.value)
@pytest.mark.parametrize(
("attr_name", "expected"),
[
("nested_with_function_marker", "nested_with_no_marker.inner"),
("nested_with_decorator_marker", "nested_with_decorator_marker.inner"),
("nested_with_attr_marker", "nested_with_attr_marker.inner"),
(
"nested_with_attr_marker_and_slots",
"nested_with_attr_marker_and_slots.inner",
),
],
)
def test_attr_of_traversable_attr_can_be_read(
proxy: ActorProxy[TraversableObjectsActor],
attr_name: str,
expected: str,
) -> None:
attr = getattr(proxy, attr_name)
assert attr.inner.get() == expected
def test_traversable_object_returned_from_property_is_not_traversed(
proxy: ActorProxy[TraversableObjectsActor],
) -> None:
# In Pykka < 2, it worked like this:
# assert proxy.nested_object_property.inner.get() == 'nested.inner' # noqa: ERA001
# In Pykka >= 2, the property getter always returns a future:
assert proxy.nested_object_property.get().inner == "nested_with_attr_marker.inner"
def test_traversable_cannot_mark_object_using_slots() -> None:
with pytest.raises(Exception, match="cannot be used to mark") as exc_info:
pykka.traversable(NestedWithNoMarkerAndSlots())
assert "cannot be used to mark an object using slots" in str(exc_info.value)
|
