Update python_script tests to use async_add_executor_job (home-assistant#41553)

Author: bdraco

tests/components/python_script/test_init.py

@@ -60,7 +60,7 @@ async 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_executor_job(execute, hass, "test.py", source, {"name": "paulus"})
     await hass.async_block_till_done()
 
     assert hass.states.is_state("test.entity", "paulus")
@@ -76,7 +76,7 @@ async def test_execute_warns_print(hass, caplog):
 print("This triggers warning.")
     """
 
-    hass.async_add_job(execute, hass, "test.py", source, {})
+    hass.async_add_executor_job(execute, hass, "test.py", source, {})
     await hass.async_block_till_done()
 
     assert "Don't use print() inside scripts." in caplog.text
@@ -89,7 +89,7 @@ async def test_execute_logging(hass, caplog):
 logger.info('Logging from inside script')
     """
 
-    hass.async_add_job(execute, hass, "test.py", source, {})
+    hass.async_add_executor_job(execute, hass, "test.py", source, {})
     await hass.async_block_till_done()
 
     assert "Logging from inside script" in caplog.text
@@ -102,7 +102,7 @@ async def test_execute_compile_error(hass, caplog):
 this is not valid Python
     """
 
-    hass.async_add_job(execute, hass, "test.py", source, {})
+    hass.async_add_executor_job(execute, hass, "test.py", source, {})
     await hass.async_block_till_done()
 
     assert "Error loading script test.py" in caplog.text
@@ -115,7 +115,7 @@ async def test_execute_runtime_error(hass, caplog):
 raise Exception('boom')
     """
 
-    hass.async_add_job(execute, hass, "test.py", source, {})
+    hass.async_add_executor_job(execute, hass, "test.py", source, {})
     await hass.async_block_till_done()
 
     assert "Error executing script: boom" in caplog.text
@@ -128,7 +128,7 @@ async def test_accessing_async_methods(hass, caplog):
 hass.async_stop()
     """
 
-    hass.async_add_job(execute, hass, "test.py", source, {})
+    hass.async_add_executor_job(execute, hass, "test.py", source, {})
     await hass.async_block_till_done()
 
     assert "Not allowed to access async methods" in caplog.text
@@ -143,7 +143,7 @@ async def test_using_complex_structures(hass, caplog):
 logger.info('Logging from inside script: %s %s' % (mydict["a"], mylist[2]))
     """
 
-    hass.async_add_job(execute, hass, "test.py", source, {})
+    hass.async_add_executor_job(execute, hass, "test.py", source, {})
     await hass.async_block_till_done()
 
     assert "Logging from inside script: 1 3" in caplog.text
@@ -160,7 +160,7 @@ async def test_accessing_forbidden_methods(hass, caplog):
         "time.tzset()": "TimeWrapper.tzset",
     }.items():
         caplog.records.clear()
-        hass.async_add_job(execute, hass, "test.py", source, {})
+        hass.async_add_executor_job(execute, hass, "test.py", source, {})
         await hass.async_block_till_done()
         assert f"Not allowed to access {name}" in caplog.text
 
@@ -172,7 +172,7 @@ async def test_iterating(hass):
     hass.states.set('hello.{}'.format(i), 'world')
     """
 
-    hass.async_add_job(execute, hass, "test.py", source, {})
+    hass.async_add_executor_job(execute, hass, "test.py", source, {})
     await hass.async_block_till_done()
 
     assert hass.states.is_state("hello.1", "world")
@@ -190,7 +190,7 @@ async def test_unpacking_sequence(hass, caplog):
 hass.states.set('hello.ab_list', '{}'.format(ab_list))
 """
 
-    hass.async_add_job(execute, hass, "test.py", source, {})
+    hass.async_add_executor_job(execute, hass, "test.py", source, {})
     await hass.async_block_till_done()
 
     assert hass.states.is_state("hello.a", "1")
@@ -211,7 +211,7 @@ async def test_execute_sorted(hass, caplog):
 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_executor_job(execute, hass, "test.py", source, {})
     await hass.async_block_till_done()
 
     assert hass.states.is_state("hello.a", "1")
@@ -232,7 +232,7 @@ async def test_exposed_modules(hass, caplog):
                 datetime.timedelta(minutes=1).total_seconds())
 """
 
-    hass.async_add_job(execute, hass, "test.py", source, {})
+    hass.async_add_executor_job(execute, hass, "test.py", source, {})
     await hass.async_block_till_done()
 
     assert hass.states.is_state("module.time", "1986")
@@ -256,7 +256,7 @@ def b():
 
 b()
 """
-    hass.async_add_job(execute, hass, "test.py", source, {})
+    hass.async_add_executor_job(execute, hass, "test.py", source, {})
     await hass.async_block_till_done()
 
     assert hass.states.is_state("hello.a", "one")
@@ -400,7 +400,7 @@ async def test_sleep_warns_one(hass, caplog):
 """
 
     with patch("homeassistant.components.python_script.time.sleep"):
-        hass.async_add_job(execute, hass, "test.py", source, {})
+        hass.async_add_executor_job(execute, hass, "test.py", source, {})
         await hass.async_block_till_done()
 
     assert caplog.text.count("time.sleep") == 1