Skip to content

Commit d7c3753

Browse files
Rename variable in battery status
Rename request_result variable to set_power_result. Signed-off-by: ela-kotulska-frequenz <[email protected]>
1 parent 53d93cb commit d7c3753

File tree

3 files changed

+42
-42
lines changed

3 files changed

+42
-42
lines changed

src/frequenz/sdk/actor/power_distributing/_battery_pool_status.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -94,8 +94,8 @@ def __init__(
9494
self._current_status = BatteryStatus(working=set(), uncertain=set())
9595

9696
# Channel for sending results of requests to the batteries
97-
request_result_channel = Broadcast[SetPowerResult]("battery_request_status")
98-
self._request_result_sender = request_result_channel.new_sender()
97+
set_power_result_channel = Broadcast[SetPowerResult]("battery_request_status")
98+
self._set_power_result_sender = set_power_result_channel.new_sender()
9999

100100
self._batteries: Dict[str, BatteryStatusTracker] = {}
101101

@@ -112,7 +112,7 @@ def __init__(
112112
max_data_age_sec=max_data_age_sec,
113113
max_blocking_duration_sec=max_blocking_duration_sec,
114114
status_sender=channel.sender,
115-
request_result_receiver=request_result_channel.new_receiver(
115+
set_power_result_receiver=set_power_result_channel.new_receiver(
116116
f"battery_{battery_id}_request_status"
117117
),
118118
)
@@ -189,7 +189,7 @@ async def update_status(
189189
succeed_batteries: Batteries that succeed request
190190
failed_batteries: Batteries that failed request
191191
"""
192-
await self._request_result_sender.send(
192+
await self._set_power_result_sender.send(
193193
SetPowerResult(succeed_batteries, failed_batteries)
194194
)
195195

src/frequenz/sdk/actor/power_distributing/_battery_status.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,7 @@ def __init__( # pylint: disable=too-many-arguments
164164
max_data_age_sec: float,
165165
max_blocking_duration_sec: float,
166166
status_sender: Sender[Status],
167-
request_result_receiver: Receiver[SetPowerResult],
167+
set_power_result_receiver: Receiver[SetPowerResult],
168168
) -> None:
169169
"""Create class instance.
170170
@@ -177,7 +177,7 @@ def __init__( # pylint: disable=too-many-arguments
177177
max_blocking_duration_sec: This value tell what should be the maximum
178178
timeout used for blocking failing component.
179179
status_sender: Channel to send status updates.
180-
request_result_receiver: Channel to receive results of the requests to the
180+
set_power_result_receiver: Channel to receive results of the requests to the
181181
components.
182182
183183
Raises:
@@ -204,7 +204,7 @@ def __init__( # pylint: disable=too-many-arguments
204204
self._select = None
205205

206206
self._task = asyncio.create_task(
207-
self._run(status_sender, request_result_receiver)
207+
self._run(status_sender, set_power_result_receiver)
208208
)
209209

210210
@property
@@ -226,15 +226,15 @@ async def stop(self) -> None:
226226
async def _run(
227227
self,
228228
status_sender: Sender[Status],
229-
request_result_receiver: Receiver[SetPowerResult],
229+
set_power_result_receiver: Receiver[SetPowerResult],
230230
) -> None:
231-
"""Process data from the components and request_result_receiver.
231+
"""Process data from the components and set_power_result_receiver.
232232
233233
New status is send only when it change.
234234
235235
Args:
236236
status_sender: Channel to send status updates.
237-
request_result_receiver: Channel to receive results of the requests to the
237+
set_power_result_receiver: Channel to receive results of the requests to the
238238
components.
239239
"""
240240
api_client = get_microgrid().api_client
@@ -247,7 +247,7 @@ async def _run(
247247
battery_timer=self._battery.data_recv_timer,
248248
inverter_timer=self._inverter.data_recv_timer,
249249
inverter=inverter_receiver,
250-
request_result=request_result_receiver,
250+
set_power_result=set_power_result_receiver,
251251
)
252252

253253
while True:
@@ -280,7 +280,7 @@ def _update_status(self, select: Select) -> Optional[Status]:
280280
self._inverter.last_msg_timestamp = msg.inner.timestamp
281281
self._inverter.data_recv_timer.reset()
282282

283-
elif msg := select.request_result:
283+
elif msg := select.set_power_result:
284284
result: SetPowerResult = msg.inner
285285
if self.battery_id in result.succeed:
286286
self._blocking_status.unblock()

tests/actor/test_battery_status.py

Lines changed: 30 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -195,7 +195,7 @@ def __init__( # pylint: disable=too-many-arguments
195195
self,
196196
battery: Optional[BatteryData] = None,
197197
inverter: Optional[InverterData] = None,
198-
request_result: Optional[SetPowerResult] = None,
198+
set_power_result: Optional[SetPowerResult] = None,
199199
battery_timer_flag: bool = False,
200200
inverter_timer_flag: bool = False,
201201
) -> None:
@@ -208,14 +208,14 @@ def __init__( # pylint: disable=too-many-arguments
208208
that no messages have been received for a while.
209209
inverter_timer_flag: If true the inverter data timer will be set to indicate
210210
that no messages have been received for a while.
211-
request_result: Expected SetPowerResult message. Defaults to None.
211+
set_power_result: Expected SetPowerResult message. Defaults to None.
212212
"""
213213
self.battery = None if battery is None else Message(battery)
214214
self.inverter = None if inverter is None else Message(inverter)
215215
self.inverter_timer = inverter_timer_flag
216216
self.battery_timer = battery_timer_flag
217-
self.request_result = (
218-
None if request_result is None else Message(request_result)
217+
self.set_power_result = (
218+
None if set_power_result is None else Message(set_power_result)
219219
)
220220

221221

@@ -255,14 +255,14 @@ async def test_sync_update_status_with_messages(
255255
mock_microgrid: mock_microgrid fixture
256256
"""
257257
status_channel = Broadcast[Status]("battery_status")
258-
request_result_channel = Broadcast[SetPowerResult]("request_result")
258+
set_power_result_channel = Broadcast[SetPowerResult]("set_power_result")
259259

260260
tracker = BatteryStatusTracker(
261261
BATTERY_ID,
262262
max_data_age_sec=5,
263263
max_blocking_duration_sec=30,
264264
status_sender=status_channel.new_sender(),
265-
request_result_receiver=request_result_channel.new_receiver(),
265+
set_power_result_receiver=set_power_result_channel.new_receiver(),
266266
)
267267

268268
assert tracker.battery_id == BATTERY_ID
@@ -402,7 +402,7 @@ async def test_sync_blocking_feature(
402402
"""
403403

404404
status_channel = Broadcast[Status]("battery_status")
405-
request_result_channel = Broadcast[SetPowerResult]("request_result")
405+
set_power_result_channel = Broadcast[SetPowerResult]("set_power_result")
406406

407407
# increase max_data_age_sec for blocking tests.
408408
# Otherwise it will block blocking.
@@ -411,7 +411,7 @@ async def test_sync_blocking_feature(
411411
max_data_age_sec=500,
412412
max_blocking_duration_sec=30,
413413
status_sender=status_channel.new_sender(),
414-
request_result_receiver=request_result_channel.new_receiver(),
414+
set_power_result_receiver=set_power_result_channel.new_receiver(),
415415
)
416416

417417
with time_machine.travel("2022-01-01 00:00 UTC", tick=False) as time:
@@ -428,7 +428,7 @@ async def test_sync_blocking_feature(
428428

429429
# message is not correct, component should not block.
430430
select = FakeSelect(
431-
request_result=SetPowerResult(succeed={1}, failed={106})
431+
set_power_result=SetPowerResult(succeed={1}, failed={106})
432432
)
433433
assert tracker._update_status(select) is None # type: ignore[arg-type]
434434

@@ -440,15 +440,15 @@ async def test_sync_blocking_feature(
440440
for timeout in expected_blocking_timeout:
441441
# message is not correct, component should not block.
442442
select = FakeSelect(
443-
request_result=SetPowerResult(succeed={1}, failed={106})
443+
set_power_result=SetPowerResult(succeed={1}, failed={106})
444444
)
445445
status = tracker._update_status(select) # type: ignore[arg-type]
446446
assert status is Status.UNCERTAIN
447447

448448
# Battery should be still blocked, nothing should happen
449449
time.shift(timeout - 1)
450450
select = FakeSelect(
451-
request_result=SetPowerResult(succeed={1}, failed={106})
451+
set_power_result=SetPowerResult(succeed={1}, failed={106})
452452
)
453453
assert tracker._update_status(select) is None # type: ignore[arg-type]
454454

@@ -462,7 +462,7 @@ async def test_sync_blocking_feature(
462462

463463
# should block for 30 sec
464464
select = FakeSelect(
465-
request_result=SetPowerResult(succeed={1}, failed={106})
465+
set_power_result=SetPowerResult(succeed={1}, failed={106})
466466
)
467467
status = tracker._update_status(select) # type: ignore[arg-type]
468468
assert status is Status.UNCERTAIN
@@ -484,15 +484,15 @@ async def test_sync_blocking_feature(
484484

485485
# should block for 30 sec
486486
select = FakeSelect(
487-
request_result=SetPowerResult(succeed={1}, failed={106})
487+
set_power_result=SetPowerResult(succeed={1}, failed={106})
488488
)
489489
status = tracker._update_status(select) # type: ignore[arg-type]
490490
assert status is Status.UNCERTAIN
491491
time.shift(28)
492492

493493
# If battery succeed, then it should unblock.
494494
select = FakeSelect(
495-
request_result=SetPowerResult(succeed={106}, failed={206})
495+
set_power_result=SetPowerResult(succeed={106}, failed={206})
496496
)
497497
status = tracker._update_status(select) # type: ignore[arg-type]
498498
assert status is Status.WORKING
@@ -512,14 +512,14 @@ async def test_sync_blocking_interrupted_with_with_max_data(
512512
"""
513513

514514
status_channel = Broadcast[Status]("battery_status")
515-
request_result_channel = Broadcast[SetPowerResult]("request_result")
515+
set_power_result_channel = Broadcast[SetPowerResult]("set_power_result")
516516

517517
tracker = BatteryStatusTracker(
518518
BATTERY_ID,
519519
max_data_age_sec=5,
520520
max_blocking_duration_sec=30,
521521
status_sender=status_channel.new_sender(),
522-
request_result_receiver=request_result_channel.new_receiver(),
522+
set_power_result_receiver=set_power_result_channel.new_receiver(),
523523
)
524524

525525
with time_machine.travel("2022-01-01 00:00 UTC", tick=False) as time:
@@ -531,7 +531,7 @@ async def test_sync_blocking_interrupted_with_with_max_data(
531531
assert status is Status.WORKING
532532

533533
select = FakeSelect(
534-
request_result=SetPowerResult(succeed={1}, failed={106})
534+
set_power_result=SetPowerResult(succeed={1}, failed={106})
535535
)
536536
status = tracker._update_status(select) # type: ignore[arg-type]
537537
assert status is Status.UNCERTAIN
@@ -540,7 +540,7 @@ async def test_sync_blocking_interrupted_with_with_max_data(
540540
for timeout in expected_blocking_timeout:
541541
# message is not correct, component should not block.
542542
select = FakeSelect(
543-
request_result=SetPowerResult(succeed={1}, failed={106})
543+
set_power_result=SetPowerResult(succeed={1}, failed={106})
544544
)
545545
assert tracker._update_status(select) is None # type: ignore[arg-type]
546546
time.shift(timeout)
@@ -561,14 +561,14 @@ async def test_sync_blocking_interrupted_with_invalid_message(
561561
"""
562562

563563
status_channel = Broadcast[Status]("battery_status")
564-
request_result_channel = Broadcast[SetPowerResult]("request_result")
564+
set_power_result_channel = Broadcast[SetPowerResult]("set_power_result")
565565

566566
tracker = BatteryStatusTracker(
567567
BATTERY_ID,
568568
max_data_age_sec=5,
569569
max_blocking_duration_sec=30,
570570
status_sender=status_channel.new_sender(),
571-
request_result_receiver=request_result_channel.new_receiver(),
571+
set_power_result_receiver=set_power_result_channel.new_receiver(),
572572
)
573573

574574
select = FakeSelect(inverter=inverter_data(component_id=INVERTER_ID))
@@ -577,7 +577,7 @@ async def test_sync_blocking_interrupted_with_invalid_message(
577577
select = FakeSelect(battery=battery_data(component_id=BATTERY_ID))
578578
assert tracker._update_status(select) is Status.WORKING # type: ignore[arg-type]
579579

580-
select = FakeSelect(request_result=SetPowerResult(succeed={1}, failed={106}))
580+
select = FakeSelect(set_power_result=SetPowerResult(succeed={1}, failed={106}))
581581
assert tracker._update_status(select) is Status.UNCERTAIN # type: ignore[arg-type]
582582

583583
select = FakeSelect(
@@ -588,10 +588,10 @@ async def test_sync_blocking_interrupted_with_invalid_message(
588588
)
589589
assert tracker._update_status(select) is Status.NOT_WORKING # type: ignore[arg-type]
590590

591-
select = FakeSelect(request_result=SetPowerResult(succeed={1}, failed={106}))
591+
select = FakeSelect(set_power_result=SetPowerResult(succeed={1}, failed={106}))
592592
assert tracker._update_status(select) is None # type: ignore[arg-type]
593593

594-
select = FakeSelect(request_result=SetPowerResult(succeed={106}, failed={}))
594+
select = FakeSelect(set_power_result=SetPowerResult(succeed={106}, failed={}))
595595
assert tracker._update_status(select) is None # type: ignore[arg-type]
596596

597597
select = FakeSelect(inverter=inverter_data(component_id=INVERTER_ID))
@@ -613,14 +613,14 @@ async def test_timers(
613613
mocker: pytest mocker instance
614614
"""
615615
status_channel = Broadcast[Status]("battery_status")
616-
request_result_channel = Broadcast[SetPowerResult]("request_result")
616+
set_power_result_channel = Broadcast[SetPowerResult]("set_power_result")
617617

618618
tracker = BatteryStatusTracker(
619619
BATTERY_ID,
620620
max_data_age_sec=5,
621621
max_blocking_duration_sec=30,
622622
status_sender=status_channel.new_sender(),
623-
request_result_receiver=request_result_channel.new_receiver(),
623+
set_power_result_receiver=set_power_result_channel.new_receiver(),
624624
)
625625

626626
battery_timer_spy = mocker.spy(tracker._battery.data_recv_timer, "reset")
@@ -673,17 +673,17 @@ async def test_async_battery_status(
673673
"""
674674

675675
status_channel = Broadcast[Status]("battery_status")
676-
request_result_channel = Broadcast[SetPowerResult]("request_result")
676+
set_power_result_channel = Broadcast[SetPowerResult]("set_power_result")
677677

678678
status_receiver = status_channel.new_receiver()
679-
request_result_sender = request_result_channel.new_sender()
679+
set_power_result_sender = set_power_result_channel.new_sender()
680680

681681
tracker = BatteryStatusTracker(
682682
BATTERY_ID,
683683
max_data_age_sec=5,
684684
max_blocking_duration_sec=30,
685685
status_sender=status_channel.new_sender(),
686-
request_result_receiver=request_result_channel.new_receiver(),
686+
set_power_result_receiver=set_power_result_channel.new_receiver(),
687687
)
688688
await asyncio.sleep(0.01)
689689

@@ -693,7 +693,7 @@ async def test_async_battery_status(
693693
status = await asyncio.wait_for(status_receiver.receive(), timeout=0.1)
694694
assert status is Status.WORKING
695695

696-
assert await request_result_sender.send(
696+
assert await set_power_result_sender.send(
697697
SetPowerResult(succeed={}, failed={BATTERY_ID})
698698
)
699699
status = await asyncio.wait_for(status_receiver.receive(), timeout=0.1)
@@ -714,7 +714,7 @@ async def test_async_battery_status(
714714
status = await asyncio.wait_for(status_receiver.receive(), timeout=0.1)
715715
assert status is Status.NOT_WORKING
716716

717-
assert await request_result_sender.send(
717+
assert await set_power_result_sender.send(
718718
SetPowerResult(succeed={}, failed={BATTERY_ID})
719719
)
720720
await asyncio.sleep(0.3)

0 commit comments

Comments
 (0)