Skip to content

Commit cc6ee78

Browse files
elena-kolevskaLorenzo Curcio
authored andcommitted
Revert "fixes try_add_state"
This reverts commit 254ad17. Signed-off-by: Lorenzo Curcio <Lorenzo.curcio@valuecuberesearch.com>
1 parent e67659d commit cc6ee78

File tree

4 files changed

+3
-116
lines changed

4 files changed

+3
-116
lines changed

dapr/actor/runtime/mock_state_manager.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ async def try_add_state(self, state_name: str, value: T) -> bool:
5151
return True
5252
return False
5353
existed = state_name in self._mock_state
54-
if existed:
54+
if not existed:
5555
return False
5656
self._default_state_change_tracker[state_name] = StateMetadata(value, StateChangeKind.add)
5757
self._mock_state[state_name] = value

dapr/actor/runtime/state_manager.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ async def try_add_state(self, state_name: str, value: T) -> bool:
9090
existed = await self._actor.runtime_ctx.state_provider.contains_state(
9191
self._type_name, self._actor.id.id, state_name
9292
)
93-
if existed:
93+
if not existed:
9494
return False
9595

9696
state_change_tracker[state_name] = StateMetadata(value, StateChangeKind.add)

tests/actor/test_mock_state_manager.py

Lines changed: 0 additions & 99 deletions
This file was deleted.

tests/actor/test_state_manager.py

Lines changed: 1 addition & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ def setUp(self):
4646

4747
@mock.patch(
4848
'tests.actor.fake_client.FakeDaprActorClient.get_state',
49-
new=_async_mock(),
49+
new=_async_mock(return_value=base64.b64encode(b'"value1"')),
5050
)
5151
@mock.patch(
5252
'tests.actor.fake_client.FakeDaprActorClient.save_state_transactionally', new=_async_mock()
@@ -67,20 +67,6 @@ def test_add_state(self):
6767
added = _run(state_manager.try_add_state('state1', 'value1'))
6868
self.assertFalse(added)
6969

70-
@mock.patch(
71-
'tests.actor.fake_client.FakeDaprActorClient.get_state',
72-
new=_async_mock(return_value=base64.b64encode(b'"value1"')),
73-
)
74-
@mock.patch(
75-
'tests.actor.fake_client.FakeDaprActorClient.save_state_transactionally', new=_async_mock()
76-
)
77-
def test_add_state_with_existing_state(self):
78-
state_manager = ActorStateManager(self._fake_actor)
79-
80-
# Add first 'state1'
81-
added = _run(state_manager.try_add_state('state1', 'value1'))
82-
self.assertFalse(added)
83-
8470
@mock.patch('tests.actor.fake_client.FakeDaprActorClient.get_state', new=_async_mock())
8571
def test_get_state_for_no_state(self):
8672
state_manager = ActorStateManager(self._fake_actor)

0 commit comments

Comments
 (0)