Fix zha test RuntimeWarnings (#97733)

This commit is contained in:
Marc Mueller 2023-08-04 01:07:11 +02:00 committed by GitHub
parent 282ae80cc2
commit d1ad1c47e6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 4 deletions

View File

@ -80,6 +80,9 @@ def mock_app():
"can_rewrite_custom_eui64": False, "can_rewrite_custom_eui64": False,
} }
} }
mock_app.add_listener = MagicMock()
mock_app.groups = MagicMock()
mock_app.devices = MagicMock()
with patch( with patch(
"zigpy.application.ControllerApplication.new", AsyncMock(return_value=mock_app) "zigpy.application.ControllerApplication.new", AsyncMock(return_value=mock_app)

View File

@ -324,12 +324,15 @@ async def test_gateway_initialize_bellows_thread(
zha_gateway._config.setdefault("zigpy_config", {}).update(config_override) zha_gateway._config.setdefault("zigpy_config", {}).update(config_override)
with patch( with patch(
"bellows.zigbee.application.ControllerApplication.new", "bellows.zigbee.application.ControllerApplication.new"
new=AsyncMock(), ) as controller_app_mock:
) as mock_new: mock = AsyncMock()
mock.add_listener = MagicMock()
mock.groups = MagicMock()
controller_app_mock.return_value = mock
await zha_gateway.async_initialize() await zha_gateway.async_initialize()
assert mock_new.mock_calls[0].args[0]["use_thread"] is thread_state assert controller_app_mock.mock_calls[0].args[0]["use_thread"] is thread_state
@pytest.mark.parametrize( @pytest.mark.parametrize(