Skip to content

Commit c6a1142

Browse files
committed
fix tests
1 parent 01ed402 commit c6a1142

File tree

2 files changed

+16
-16
lines changed

2 files changed

+16
-16
lines changed

tests/integration/test_actor_lifecycle.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ async def test_actor_init_and_double_init_prevention(
1515
async def main() -> None:
1616
my_actor = Actor
1717
await my_actor.init()
18-
assert my_actor._is_initialized is True
18+
assert my_actor.active is True
1919
double_init = False
2020
try:
2121
await my_actor.init()
@@ -33,7 +33,7 @@ async def main() -> None:
3333
raise
3434
await my_actor.exit()
3535
assert double_init is False
36-
assert my_actor._is_initialized is False
36+
assert my_actor.active is False
3737

3838
actor = await make_actor(label='actor-init', main_func=main)
3939
run_result = await run_actor(actor)
@@ -49,8 +49,8 @@ async def main() -> None:
4949
import apify._actor
5050

5151
async with Actor:
52-
assert apify._actor.Actor._is_initialized
53-
assert apify._actor.Actor._is_initialized is False
52+
assert apify._actor.Actor.active
53+
assert apify._actor.Actor.active is False
5454

5555
actor = await make_actor(label='with-actor-init', main_func=main)
5656
run_result = await run_actor(actor)

tests/unit/actor/test_actor_lifecycle.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -21,37 +21,37 @@
2121
async def test_actor_properly_init_with_async() -> None:
2222
async with Actor:
2323
assert cast(Proxy, apify._actor.Actor).__wrapped__ is not None
24-
assert cast(Proxy, apify._actor.Actor).__wrapped__._is_initialized
25-
assert not cast(Proxy, apify._actor.Actor).__wrapped__._is_initialized
24+
assert cast(Proxy, apify._actor.Actor).__wrapped__.active
25+
assert not cast(Proxy, apify._actor.Actor).__wrapped__.active
2626

2727

2828
async def test_actor_init() -> None:
2929
my_actor = _ActorType()
3030

3131
await my_actor.init()
32-
assert my_actor._is_initialized is True
32+
assert my_actor.active is True
3333

3434
await my_actor.exit()
35-
assert my_actor._is_initialized is False
35+
assert my_actor.active is False
3636

3737

3838
async def test_double_init_raises_error(prepare_test_env: Callable) -> None:
3939
async with Actor:
40-
assert Actor._is_initialized
40+
assert Actor.active
4141
with pytest.raises(RuntimeError):
4242
await Actor.init()
4343

4444
prepare_test_env()
4545

4646
async with Actor() as actor:
47-
assert actor._is_initialized
47+
assert actor.active
4848
with pytest.raises(RuntimeError):
4949
await actor.init()
5050

5151
prepare_test_env()
5252

5353
async with _ActorType() as actor:
54-
assert actor._is_initialized
54+
assert actor.active
5555
with pytest.raises(RuntimeError):
5656
await actor.init()
5757

@@ -73,7 +73,7 @@ def on_event(event_type: Event) -> Callable:
7373

7474
my_actor = _ActorType()
7575
async with my_actor:
76-
assert my_actor._is_initialized
76+
assert my_actor.active
7777
my_actor.on(Event.PERSIST_STATE, on_event(Event.PERSIST_STATE))
7878
my_actor.on(Event.SYSTEM_INFO, on_event(Event.SYSTEM_INFO))
7979
await asyncio.sleep(1)
@@ -96,21 +96,21 @@ async def test_exit_without_init_raises_error() -> None:
9696

9797
async def test_actor_fails_cleanly() -> None:
9898
async with _ActorType() as my_actor:
99-
assert my_actor._is_initialized
99+
assert my_actor.active
100100
await my_actor.fail()
101-
assert my_actor._is_initialized is False
101+
assert my_actor.active is False
102102

103103

104104
async def test_actor_handles_failure_gracefully() -> None:
105105
my_actor = None
106106

107107
with contextlib.suppress(Exception):
108108
async with _ActorType() as my_actor:
109-
assert my_actor._is_initialized
109+
assert my_actor.active
110110
raise Exception('Failed') # noqa: TRY002
111111

112112
assert my_actor is not None
113-
assert my_actor._is_initialized is False
113+
assert my_actor.active is False
114114

115115

116116
async def test_fail_without_init_raises_error() -> None:

0 commit comments

Comments
 (0)