Black

tests/components/spc/test_init.py

@@ -3,36 +3,29 @@ from unittest.mock import patch, PropertyMock, Mock
 
 from homeassistant.bootstrap import async_setup_component
 from homeassistant.components.spc import DATA_API
-from homeassistant.const import (STATE_ALARM_ARMED_AWAY, STATE_ALARM_DISARMED)
+from homeassistant.const import STATE_ALARM_ARMED_AWAY, STATE_ALARM_DISARMED
 
 from tests.common import mock_coro
 
 
 async def test_valid_device_config(hass, monkeypatch):
     """Test valid device config."""
-    config = {
-        'spc': {
-            'api_url': 'http://localhost/',
-            'ws_url': 'ws://localhost/'
-        }
-    }
+    config = {"spc": {"api_url": "http://localhost/", "ws_url": "ws://localhost/"}}
 
-    with patch('pyspcwebgw.SpcWebGateway.async_load_parameters',
-               return_value=mock_coro(True)):
-        assert await async_setup_component(hass, 'spc', config) is True
+    with patch(
+        "pyspcwebgw.SpcWebGateway.async_load_parameters", return_value=mock_coro(True)
+    ):
+        assert await async_setup_component(hass, "spc", config) is True
 
 
 async def test_invalid_device_config(hass, monkeypatch):
     """Test valid device config."""
-    config = {
-        'spc': {
-            'api_url': 'http://localhost/'
-        }
-    }
+    config = {"spc": {"api_url": "http://localhost/"}}
 
-    with patch('pyspcwebgw.SpcWebGateway.async_load_parameters',
-               return_value=mock_coro(True)):
-        assert await async_setup_component(hass, 'spc', config) is False
+    with patch(
+        "pyspcwebgw.SpcWebGateway.async_load_parameters", return_value=mock_coro(True)
+    ):
+        assert await async_setup_component(hass, "spc", config) is False
 
 
 async def test_update_alarm_device(hass):
@@ -40,36 +33,38 @@ async def test_update_alarm_device(hass):
     import pyspcwebgw
     from pyspcwebgw.const import AreaMode
 
-    config = {
-        'spc': {
-            'api_url': 'http://localhost/',
-            'ws_url': 'ws://localhost/'
-        }
-    }
+    config = {"spc": {"api_url": "http://localhost/", "ws_url": "ws://localhost/"}}
 
-    area_mock = Mock(spec=pyspcwebgw.area.Area, id='1',
-                     mode=AreaMode.FULL_SET, last_changed_by='Sven')
-    area_mock.name = 'House'
+    area_mock = Mock(
+        spec=pyspcwebgw.area.Area,
+        id="1",
+        mode=AreaMode.FULL_SET,
+        last_changed_by="Sven",
+    )
+    area_mock.name = "House"
     area_mock.verified_alarm = False
 
-    with patch('pyspcwebgw.SpcWebGateway.areas',
-               new_callable=PropertyMock) as mock_areas:
-        mock_areas.return_value = {'1': area_mock}
-        with patch('pyspcwebgw.SpcWebGateway.async_load_parameters',
-                   return_value=mock_coro(True)):
-            assert await async_setup_component(hass, 'spc', config) is True
+    with patch(
+        "pyspcwebgw.SpcWebGateway.areas", new_callable=PropertyMock
+    ) as mock_areas:
+        mock_areas.return_value = {"1": area_mock}
+        with patch(
+            "pyspcwebgw.SpcWebGateway.async_load_parameters",
+            return_value=mock_coro(True),
+        ):
+            assert await async_setup_component(hass, "spc", config) is True
 
         await hass.async_block_till_done()
 
-    entity_id = 'alarm_control_panel.house'
+    entity_id = "alarm_control_panel.house"
 
     assert hass.states.get(entity_id).state == STATE_ALARM_ARMED_AWAY
-    assert hass.states.get(entity_id).attributes['changed_by'] == 'Sven'
+    assert hass.states.get(entity_id).attributes["changed_by"] == "Sven"
 
     area_mock.mode = AreaMode.UNSET
-    area_mock.last_changed_by = 'Anna'
+    area_mock.last_changed_by = "Anna"
     await hass.data[DATA_API]._async_callback(area_mock)
     await hass.async_block_till_done()
 
     assert hass.states.get(entity_id).state == STATE_ALARM_DISARMED
-    assert hass.states.get(entity_id).attributes['changed_by'] == 'Anna'
+    assert hass.states.get(entity_id).attributes["changed_by"] == "Anna"