Fix tests

tests/common.py

@@ -696,13 +696,15 @@ def assert_setup_component(count, domain=None):
     config = {}
 
     @ha.callback
-    def mock_psc(hass, config_input, domain):
+    def mock_psc(hass, config_input, domain_input):
         """Mock the prepare_setup_component to capture config."""
         res = async_process_component_config(
-            hass, config_input, domain)
-        config[domain] = None if res is None else res.get(domain)
+            hass, config_input, domain_input)
+        config[domain_input] = None if res is None else res.get(domain_input)
         _LOGGER.debug("Configuration for %s, Validated: %s, Original %s",
-                      domain, config[domain], config_input.get(domain))
+                      domain_input,
+                      config[domain_input],
+                      config_input.get(domain_input))
         return res
 
     assert isinstance(config, dict)

tests/test_config_entries.py

@@ -162,7 +162,7 @@ async def test_remove_entry(hass, manager):
     """Test that we can remove an entry."""
     async def mock_setup_entry(hass, entry):
         """Mock setting up entry."""
-        hass.loop.create_task(hass.config_entries.async_forward_entry_setup(
+        hass.async_create_task(hass.config_entries.async_forward_entry_setup(
             entry, 'light'))
         return True