File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff 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 ()
You can’t perform that action at this time.
0 commit comments