Fix mock return value

llucax · llucax · commit d6f843a7c0e3 · 2024-06-10T09:33:52.000+02:00
The `set_power()` method is from the microgrid client, not the protobuf
stub, so it should really return `None` instead of `Empty`.

Signed-off-by: Leandro Lucarella &lt;luca-frequenz@llucax.com&gt;
diff --git a/tests/utils/mock_microgrid_client.py b/tests/utils/mock_microgrid_client.py
@@ -18,7 +18,6 @@
     Location,
     MeterData,
 )
-from google.protobuf.empty_pb2 import Empty  # pylint: disable=no-name-in-module
 from pytest_mock import MockerFixture
 
 from frequenz.sdk._internal._constants import RECEIVER_MAX_SIZE
@@ -282,7 +281,7 @@ def _create_mock_api(
         )
 
         # Can be override in the future
-        api.set_power = AsyncMock(return_value=Empty)
+        api.set_power = AsyncMock(return_value=None)
         return api
 
     def _get_battery_receiver(

Original file line number	Diff line number	Diff line change
`@@ -18,7 +18,6 @@`
`18`	`18`	`Location,`
`19`	`19`	`MeterData,`
`20`	`20`	`)`
`21`		`-from google.protobuf.empty_pb2 import Empty # pylint: disable=no-name-in-module`
`22`	`21`	`from pytest_mock import MockerFixture`
`23`	`22`
`24`	`23`	`from frequenz.sdk._internal._constants import RECEIVER_MAX_SIZE`
`@@ -282,7 +281,7 @@ def _create_mock_api(`
`282`	`281`	`)`
`283`	`282`
`284`	`283`	`# Can be override in the future`
`285`		`- api.set_power = AsyncMock(return_value=Empty)`
	`284`	`+ api.set_power = AsyncMock(return_value=None)`
`286`	`285`	`return api`
`287`	`286`
`288`	`287`	`def _get_battery_receiver(`