From 985d8f30f66e549be876f4bccd9f46d2193364e4 Mon Sep 17 00:00:00 2001
From: Chethan UK <chethan.umesha@tessian.com>
Date: Fri, 3 Jun 2022 12:04:27 +0100
Subject: [PATCH] Fix flake8

---
 .../system/providers/snowflake/example_snowflake.py | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/tests/system/providers/snowflake/example_snowflake.py b/tests/system/providers/snowflake/example_snowflake.py
index 012b72a80b8c7f..ae6dca0fe2f71e 100644
--- a/tests/system/providers/snowflake/example_snowflake.py
+++ b/tests/system/providers/snowflake/example_snowflake.py
@@ -54,13 +54,13 @@
 # [START howto_operator_snowflake]
 
 with DAG(
-    ENV_ID,
+    DAG_ID,
     start_date=datetime(2021, 1, 1),
     default_args={'snowflake_conn_id': SNOWFLAKE_CONN_ID},
     tags=['example'],
     catchup=False,
 ) as dag:
-
+    # [START snowflake_example_dag]
     snowflake_op_sql_str = SnowflakeOperator(
         task_id='snowflake_op_sql_str',
         sql=CREATE_TABLE_SQL_STRING,
@@ -80,6 +80,8 @@
         role=SNOWFLAKE_ROLE,
     )
 
+    snowflake_op_sql_list = SnowflakeOperator(task_id='snowflake_op_sql_list', sql=SQL_LIST)
+
     snowflake_op_sql_multiple_stmts = SnowflakeOperator(
         task_id='snowflake_op_sql_multiple_stmts',
         sql=SQL_MULTIPLE_STMTS,
@@ -127,3 +129,10 @@
         ]
         >> slack_report
     )
+    # [END snowflake_example_dag]
+
+
+from tests.system.utils import get_test_run  # noqa: E402
+
+# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
+test_run = get_test_run(dag)