"""Test to verify that we can load components.""" import pytest import homeassistant.loader as loader from homeassistant.components import http, hue from homeassistant.components.hue import light as hue_light from tests.common import MockModule, async_mock_service, mock_integration async def test_component_dependencies(hass): """Test if we can get the proper load order of components.""" mock_integration(hass, MockModule('mod1')) mock_integration(hass, MockModule('mod2', ['mod1'])) mock_integration(hass, MockModule('mod3', ['mod2'])) assert {'mod1', 'mod2', 'mod3'} == \ await loader.async_component_dependencies(hass, 'mod3') # Create circular dependency mock_integration(hass, MockModule('mod1', ['mod3'])) with pytest.raises(loader.CircularDependency): print(await loader.async_component_dependencies(hass, 'mod3')) # Depend on non-existing component mock_integration(hass, MockModule('mod1', ['nonexisting'])) with pytest.raises(loader.IntegrationNotFound): print(await loader.async_component_dependencies(hass, 'mod1')) # Try to get dependencies for non-existing component with pytest.raises(loader.IntegrationNotFound): print(await loader.async_component_dependencies(hass, 'nonexisting')) def test_component_loader(hass): """Test loading components.""" components = loader.Components(hass) assert components.http.CONFIG_SCHEMA is http.CONFIG_SCHEMA assert hass.components.http.CONFIG_SCHEMA is http.CONFIG_SCHEMA def test_component_loader_non_existing(hass): """Test loading components.""" components = loader.Components(hass) with pytest.raises(ImportError): components.non_existing async def test_component_wrapper(hass): """Test component wrapper.""" calls = async_mock_service(hass, 'persistent_notification', 'create') components = loader.Components(hass) components.persistent_notification.async_create('message') await hass.async_block_till_done() assert len(calls) == 1 async def test_helpers_wrapper(hass): """Test helpers wrapper.""" helpers = loader.Helpers(hass) result = [] def discovery_callback(service, discovered): """Handle discovery callback.""" result.append(discovered) helpers.discovery.async_listen('service_name', discovery_callback) await helpers.discovery.async_discover('service_name', 'hello', None, {}) await hass.async_block_till_done() assert result == ['hello'] async def test_custom_component_name(hass): """Test the name attribte of custom components.""" integration = await loader.async_get_integration(hass, 'test_standalone') int_comp = integration.get_component() assert int_comp.__name__ == 'custom_components.test_standalone' assert int_comp.__package__ == 'custom_components' comp = hass.components.test_standalone assert comp.__name__ == 'custom_components.test_standalone' assert comp.__package__ == 'custom_components' integration = await loader.async_get_integration(hass, 'test_package') int_comp = integration.get_component() assert int_comp.__name__ == 'custom_components.test_package' assert int_comp.__package__ == 'custom_components.test_package' comp = hass.components.test_package assert comp.__name__ == 'custom_components.test_package' assert comp.__package__ == 'custom_components.test_package' integration = await loader.async_get_integration(hass, 'test') platform = integration.get_platform('light') assert platform.__name__ == 'custom_components.test.light' assert platform.__package__ == 'custom_components.test' # Test custom components is mounted from custom_components.test_package import TEST assert TEST == 5 async def test_log_warning_custom_component(hass, caplog): """Test that we log a warning when loading a custom component.""" hass.components.test_standalone assert 'You are using a custom integration for test_standalone' \ in caplog.text await loader.async_get_integration(hass, 'test') assert 'You are using a custom integration for test ' in caplog.text async def test_get_integration(hass): """Test resolving integration.""" integration = await loader.async_get_integration(hass, 'hue') assert hue == integration.get_component() assert hue_light == integration.get_platform('light') async def test_get_integration_legacy(hass): """Test resolving integration.""" integration = await loader.async_get_integration(hass, 'test_embedded') assert integration.get_component().DOMAIN == 'test_embedded' assert integration.get_platform('switch') is not None async def test_get_integration_custom_component(hass): """Test resolving integration.""" integration = await loader.async_get_integration(hass, 'test_package') print(integration) assert integration.get_component().DOMAIN == 'test_package' assert integration.name == 'Test Package' def test_integration_properties(hass): """Test integration properties.""" integration = loader.Integration( hass, 'homeassistant.components.hue', None, { 'name': 'Philips Hue', 'domain': 'hue', 'dependencies': ['test-dep'], 'requirements': ['test-req==1.0.0'], }) assert integration.name == "Philips Hue" assert integration.domain == 'hue' assert integration.dependencies == ['test-dep'] assert integration.requirements == ['test-req==1.0.0'] async def test_integrations_only_once(hass): """Test that we load integrations only once.""" int_1 = hass.async_create_task( loader.async_get_integration(hass, 'hue')) int_2 = hass.async_create_task( loader.async_get_integration(hass, 'hue')) assert await int_1 is await int_2