Replace single element slice by next() in Airflow tests

tests/jobs/test_triggerer_job.py

@@ -567,7 +567,7 @@ def test_trigger_failing(session):
         for _ in range(30):
             if job_runner.trigger_runner.failed_triggers:
                 assert len(job_runner.trigger_runner.failed_triggers) == 1
-                trigger_id, exc = list(job_runner.trigger_runner.failed_triggers)[0]
+                trigger_id, exc = next(iter(job_runner.trigger_runner.failed_triggers))
                 assert trigger_id == 1
                 assert isinstance(exc, ValueError)
                 assert exc.args[0] == "Deliberate trigger failure"

tests/providers/amazon/aws/secrets/test_systems_manager.py

@@ -157,11 +157,11 @@ def test_get_variable_non_existent_key(self):
     @mock.patch("airflow.providers.amazon.aws.hooks.base_aws.SessionFactory")
     def test_passing_client_kwargs(self, mock_session_factory):
         backends = initialize_secrets_backends()
-        systems_manager = [
+        systems_manager = next(
             backend
             for backend in backends
             if backend.__class__.__name__ == "SystemsManagerParameterStoreBackend"
-        ][0]
+        )
 
         # Mock SessionFactory, session and client
         mock_session_factory_instance = mock_session_factory.return_value

tests/providers/cncf/kubernetes/executors/test_kubernetes_executor.py

@@ -450,7 +450,7 @@ def test_invalid_executor_config(self, mock_get_kube_client, mock_kubernetes_job
                 ),
             )
 
-            assert list(executor.event_buffer.values())[0][1] == "Invalid executor_config passed"
+            assert next(iter(executor.event_buffer.values()))[1] == "Invalid executor_config passed"
         finally:
             executor.end()
 

tests/providers/google/cloud/operators/test_bigquery.py

@@ -730,7 +730,7 @@ def test_bigquery_operator_extra_serialized_field_when_single_query(
         ]
 
         # Check DeSerialized version of operator link
-        assert isinstance(list(simple_task.operator_extra_links)[0], BigQueryConsoleLink)
+        assert isinstance(next(iter(simple_task.operator_extra_links)), BigQueryConsoleLink)
 
         ti.xcom_push("job_id_path", TEST_FULL_JOB_ID)
 
@@ -768,7 +768,7 @@ def test_bigquery_operator_extra_serialized_field_when_multiple_queries(
         ]
 
         # Check DeSerialized version of operator link
-        assert isinstance(list(simple_task.operator_extra_links)[0], BigQueryConsoleIndexableLink)
+        assert isinstance(next(iter(simple_task.operator_extra_links)), BigQueryConsoleIndexableLink)
 
         ti.xcom_push(key="job_id_path", value=[TEST_FULL_JOB_ID, TEST_FULL_JOB_ID_2])
 

tests/system/providers/amazon/aws/utils/__init__.py

@@ -64,9 +64,9 @@ def _get_test_name() -> str:
     """
     # The exact layer of the stack will depend on if this is called directly
     # or from another helper, but the test will always contain the identifier.
-    test_filename: str = [
+    test_filename: str = next(
         frame.filename for frame in inspect.stack() if TEST_FILE_IDENTIFIER in frame.filename
-    ][0]
+    )
     return splitext(basename(test_filename))[0]
 
 

tests/www/views/test_views_base.py

@@ -251,7 +251,7 @@ def test_views_get(request, url, client, content):
 
 
 def _check_task_stats_json(resp):
-    return set(list(resp.json.items())[0][1][0].keys()) == {"state", "count"}
+    return set(next(iter(resp.json.items()))[1][0]) == {"state", "count"}
 
 
 @pytest.mark.parametrize(