Skip to content

Commit 58dee40

Browse files
committed
fixup add microgrid info tests
1 parent b4204d0 commit 58dee40

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/test_microgrid_info.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -313,18 +313,18 @@ def test_microgrid_info_from_proto(
313313
assert info.name is None
314314

315315
# Verify mock calls
316-
mock_to_datetime.assert_called_once()
317-
mock_enum_from_proto.assert_called_once()
316+
mock_to_datetime.assert_called_once_with(proto.create_timestamp)
317+
mock_enum_from_proto.assert_called_once_with(proto.status, MicrogridStatus)
318318

319319
if case.has_delivery_area:
320-
mock_delivery_area_from_proto.assert_called_once()
320+
mock_delivery_area_from_proto.assert_called_once_with(proto.delivery_area)
321321
assert info.delivery_area == mock_delivery_area
322322
else:
323323
mock_delivery_area_from_proto.assert_not_called()
324324
assert info.delivery_area is None
325325

326326
if case.has_location:
327-
mock_location_from_proto.assert_called_once()
327+
mock_location_from_proto.assert_called_once_with(proto.location)
328328
assert info.location == mock_location
329329
else:
330330
mock_location_from_proto.assert_not_called()

0 commit comments

Comments
 (0)