Black

tests/components/python_script/test_init.py

@@ -11,44 +11,46 @@ from homeassistant.components.python_script import execute
 def test_setup(hass):
     """Test we can discover scripts."""
     scripts = [
-        '/some/config/dir/python_scripts/hello.py',
-        '/some/config/dir/python_scripts/world_beer.py'
+        "/some/config/dir/python_scripts/hello.py",
+        "/some/config/dir/python_scripts/world_beer.py",
     ]
-    with patch('homeassistant.components.python_script.os.path.isdir',
-               return_value=True), \
-            patch('homeassistant.components.python_script.glob.iglob',
-                  return_value=scripts):
-        res = yield from async_setup_component(hass, 'python_script', {})
+    with patch(
+        "homeassistant.components.python_script.os.path.isdir", return_value=True
+    ), patch("homeassistant.components.python_script.glob.iglob", return_value=scripts):
+        res = yield from async_setup_component(hass, "python_script", {})
 
     assert res
-    assert hass.services.has_service('python_script', 'hello')
-    assert hass.services.has_service('python_script', 'world_beer')
+    assert hass.services.has_service("python_script", "hello")
+    assert hass.services.has_service("python_script", "world_beer")
 
-    with patch('homeassistant.components.python_script.open',
-               mock_open(read_data='fake source'), create=True), \
-            patch('homeassistant.components.python_script.execute') as mock_ex:
+    with patch(
+        "homeassistant.components.python_script.open",
+        mock_open(read_data="fake source"),
+        create=True,
+    ), patch("homeassistant.components.python_script.execute") as mock_ex:
         yield from hass.services.async_call(
-            'python_script', 'hello', {'some': 'data'}, blocking=True)
+            "python_script", "hello", {"some": "data"}, blocking=True
+        )
 
     assert len(mock_ex.mock_calls) == 1
     hass, script, source, data = mock_ex.mock_calls[0][1]
 
     assert hass is hass
-    assert script == 'hello.py'
-    assert source == 'fake source'
-    assert data == {'some': 'data'}
+    assert script == "hello.py"
+    assert source == "fake source"
+    assert data == {"some": "data"}
 
 
 @asyncio.coroutine
 def test_setup_fails_on_no_dir(hass, caplog):
     """Test we fail setup when no dir found."""
-    with patch('homeassistant.components.python_script.os.path.isdir',
-               return_value=False):
-        res = yield from async_setup_component(hass, 'python_script', {})
+    with patch(
+        "homeassistant.components.python_script.os.path.isdir", return_value=False
+    ):
+        res = yield from async_setup_component(hass, "python_script", {})
 
     assert not res
-    assert 'Folder python_scripts not found in configuration folder' in \
-           caplog.text
+    assert "Folder python_scripts not found in configuration folder" in caplog.text
 
 
 @asyncio.coroutine
@@ -59,13 +61,13 @@ def test_execute_with_data(hass, caplog):
 hass.states.set('test.entity', data.get('name', 'not set'))
     """
 
-    hass.async_add_job(execute, hass, 'test.py', source, {'name': 'paulus'})
+    hass.async_add_job(execute, hass, "test.py", source, {"name": "paulus"})
     yield from hass.async_block_till_done()
 
-    assert hass.states.is_state('test.entity', 'paulus')
+    assert hass.states.is_state("test.entity", "paulus")
 
     # No errors logged = good
-    assert caplog.text == ''
+    assert caplog.text == ""
 
 
 @asyncio.coroutine
@@ -76,7 +78,7 @@ def test_execute_warns_print(hass, caplog):
 print("This triggers warning.")
     """
 
-    hass.async_add_job(execute, hass, 'test.py', source, {})
+    hass.async_add_job(execute, hass, "test.py", source, {})
     yield from hass.async_block_till_done()
 
     assert "Don't use print() inside scripts." in caplog.text
@@ -90,7 +92,7 @@ def test_execute_logging(hass, caplog):
 logger.info('Logging from inside script')
     """
 
-    hass.async_add_job(execute, hass, 'test.py', source, {})
+    hass.async_add_job(execute, hass, "test.py", source, {})
     yield from hass.async_block_till_done()
 
     assert "Logging from inside script" in caplog.text
@@ -104,7 +106,7 @@ def test_execute_compile_error(hass, caplog):
 this is not valid Python
     """
 
-    hass.async_add_job(execute, hass, 'test.py', source, {})
+    hass.async_add_job(execute, hass, "test.py", source, {})
     yield from hass.async_block_till_done()
 
     assert "Error loading script test.py" in caplog.text
@@ -118,7 +120,7 @@ def test_execute_runtime_error(hass, caplog):
 raise Exception('boom')
     """
 
-    hass.async_add_job(execute, hass, 'test.py', source, {})
+    hass.async_add_job(execute, hass, "test.py", source, {})
     yield from hass.async_block_till_done()
 
     assert "Error executing script: boom" in caplog.text
@@ -132,7 +134,7 @@ def test_accessing_async_methods(hass, caplog):
 hass.async_stop()
     """
 
-    hass.async_add_job(execute, hass, 'test.py', source, {})
+    hass.async_add_job(execute, hass, "test.py", source, {})
     yield from hass.async_block_till_done()
 
     assert "Not allowed to access async methods" in caplog.text
@@ -148,7 +150,7 @@ mylist = [1, 2, 3, 4]
 logger.info('Logging from inside script: %s %s' % (mydict["a"], mylist[2]))
     """
 
-    hass.async_add_job(execute, hass, 'test.py', source, {})
+    hass.async_add_job(execute, hass, "test.py", source, {})
     yield from hass.async_block_till_done()
 
     assert "Logging from inside script: 1 3" in caplog.text
@@ -160,13 +162,13 @@ def test_accessing_forbidden_methods(hass, caplog):
     caplog.set_level(logging.ERROR)
 
     for source, name in {
-        'hass.stop()': 'HomeAssistant.stop',
-        'dt_util.set_default_time_zone()': 'module.set_default_time_zone',
-        'datetime.non_existing': 'module.non_existing',
-        'time.tzset()': 'TimeWrapper.tzset',
+        "hass.stop()": "HomeAssistant.stop",
+        "dt_util.set_default_time_zone()": "module.set_default_time_zone",
+        "datetime.non_existing": "module.non_existing",
+        "time.tzset()": "TimeWrapper.tzset",
     }.items():
         caplog.records.clear()
-        hass.async_add_job(execute, hass, 'test.py', source, {})
+        hass.async_add_job(execute, hass, "test.py", source, {})
         yield from hass.async_block_till_done()
         assert "Not allowed to access {}".format(name) in caplog.text
 
@@ -179,11 +181,11 @@ for i in [1, 2]:
     hass.states.set('hello.{}'.format(i), 'world')
     """
 
-    hass.async_add_job(execute, hass, 'test.py', source, {})
+    hass.async_add_job(execute, hass, "test.py", source, {})
     yield from hass.async_block_till_done()
 
-    assert hass.states.is_state('hello.1', 'world')
-    assert hass.states.is_state('hello.2', 'world')
+    assert hass.states.is_state("hello.1", "world")
+    assert hass.states.is_state("hello.2", "world")
 
 
 @asyncio.coroutine
@@ -198,15 +200,15 @@ hass.states.set('hello.b', b)
 hass.states.set('hello.ab_list', '{}'.format(ab_list))
 """
 
-    hass.async_add_job(execute, hass, 'test.py', source, {})
+    hass.async_add_job(execute, hass, "test.py", source, {})
     yield from hass.async_block_till_done()
 
-    assert hass.states.is_state('hello.a', '1')
-    assert hass.states.is_state('hello.b', '2')
-    assert hass.states.is_state('hello.ab_list', '[(1, 2), (3, 4)]')
+    assert hass.states.is_state("hello.a", "1")
+    assert hass.states.is_state("hello.b", "2")
+    assert hass.states.is_state("hello.ab_list", "[(1, 2), (3, 4)]")
 
     # No errors logged = good
-    assert caplog.text == ''
+    assert caplog.text == ""
 
 
 @asyncio.coroutine
@@ -220,14 +222,14 @@ hass.states.set('hello.a', a[0])
 hass.states.set('hello.b', a[1])
 hass.states.set('hello.c', a[2])
 """
-    hass.async_add_job(execute, hass, 'test.py', source, {})
+    hass.async_add_job(execute, hass, "test.py", source, {})
     yield from hass.async_block_till_done()
 
-    assert hass.states.is_state('hello.a', '1')
-    assert hass.states.is_state('hello.b', '2')
-    assert hass.states.is_state('hello.c', '3')
+    assert hass.states.is_state("hello.a", "1")
+    assert hass.states.is_state("hello.b", "2")
+    assert hass.states.is_state("hello.c", "3")
     # No errors logged = good
-    assert caplog.text == ''
+    assert caplog.text == ""
 
 
 @asyncio.coroutine
@@ -242,50 +244,49 @@ hass.states.set('module.datetime',
                 datetime.timedelta(minutes=1).total_seconds())
 """
 
-    hass.async_add_job(execute, hass, 'test.py', source, {})
+    hass.async_add_job(execute, hass, "test.py", source, {})
     yield from hass.async_block_till_done()
 
-    assert hass.states.is_state('module.time', '1986')
-    assert hass.states.is_state('module.time_strptime', '12:34')
-    assert hass.states.is_state('module.datetime', '60.0')
+    assert hass.states.is_state("module.time", "1986")
+    assert hass.states.is_state("module.time_strptime", "12:34")
+    assert hass.states.is_state("module.datetime", "60.0")
 
     # No errors logged = good
-    assert caplog.text == ''
+    assert caplog.text == ""
 
 
 @asyncio.coroutine
 def test_reload(hass):
     """Test we can re-discover scripts."""
     scripts = [
-        '/some/config/dir/python_scripts/hello.py',
-        '/some/config/dir/python_scripts/world_beer.py'
+        "/some/config/dir/python_scripts/hello.py",
+        "/some/config/dir/python_scripts/world_beer.py",
     ]
-    with patch('homeassistant.components.python_script.os.path.isdir',
-               return_value=True), \
-            patch('homeassistant.components.python_script.glob.iglob',
-                  return_value=scripts):
-        res = yield from async_setup_component(hass, 'python_script', {})
+    with patch(
+        "homeassistant.components.python_script.os.path.isdir", return_value=True
+    ), patch("homeassistant.components.python_script.glob.iglob", return_value=scripts):
+        res = yield from async_setup_component(hass, "python_script", {})
 
     assert res
-    assert hass.services.has_service('python_script', 'hello')
-    assert hass.services.has_service('python_script', 'world_beer')
-    assert hass.services.has_service('python_script', 'reload')
+    assert hass.services.has_service("python_script", "hello")
+    assert hass.services.has_service("python_script", "world_beer")
+    assert hass.services.has_service("python_script", "reload")
 
     scripts = [
-        '/some/config/dir/python_scripts/hello2.py',
-        '/some/config/dir/python_scripts/world_beer.py'
+        "/some/config/dir/python_scripts/hello2.py",
+        "/some/config/dir/python_scripts/world_beer.py",
     ]
-    with patch('homeassistant.components.python_script.os.path.isdir',
-               return_value=True), \
-            patch('homeassistant.components.python_script.glob.iglob',
-                  return_value=scripts):
+    with patch(
+        "homeassistant.components.python_script.os.path.isdir", return_value=True
+    ), patch("homeassistant.components.python_script.glob.iglob", return_value=scripts):
         yield from hass.services.async_call(
-            'python_script', 'reload', {}, blocking=True)
+            "python_script", "reload", {}, blocking=True
+        )
 
-    assert not hass.services.has_service('python_script', 'hello')
-    assert hass.services.has_service('python_script', 'hello2')
-    assert hass.services.has_service('python_script', 'world_beer')
-    assert hass.services.has_service('python_script', 'reload')
+    assert not hass.services.has_service("python_script", "hello")
+    assert hass.services.has_service("python_script", "hello2")
+    assert hass.services.has_service("python_script", "world_beer")
+    assert hass.services.has_service("python_script", "reload")
 
 
 @asyncio.coroutine
@@ -297,8 +298,8 @@ time.sleep(2)
 time.sleep(5)
 """
 
-    with patch('homeassistant.components.python_script.time.sleep'):
-        hass.async_add_job(execute, hass, 'test.py', source, {})
+    with patch("homeassistant.components.python_script.time.sleep"):
+        hass.async_add_job(execute, hass, "test.py", source, {})
         yield from hass.async_block_till_done()
 
-    assert caplog.text.count('time.sleep') == 1
+    assert caplog.text.count("time.sleep") == 1