From b9e3c485ea2a308ea532b29ca15af0c04aee39ed Mon Sep 17 00:00:00 2001
From: Ioannis Foukarakis <ioannis.foukarakis@mattermost.com>
Date: Thu, 29 Feb 2024 15:15:55 +0200
Subject: [PATCH] Fix failing job (#1454)

---
 airflow/dags/mattermost_dags/extract/push_proxy_new.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/airflow/dags/mattermost_dags/extract/push_proxy_new.py b/airflow/dags/mattermost_dags/extract/push_proxy_new.py
index 324862cb8..2287b08e2 100644
--- a/airflow/dags/mattermost_dags/extract/push_proxy_new.py
+++ b/airflow/dags/mattermost_dags/extract/push_proxy_new.py
@@ -11,6 +11,7 @@
     SNOWFLAKE_ACCOUNT,
     SNOWFLAKE_LOAD_DATABASE,
     SNOWFLAKE_LOAD_WAREHOUSE,
+    SNOWFLAKE_LOAD_ROLE,
 )
 
 
@@ -38,12 +39,14 @@ def push_proxy_loader():
             SNOWFLAKE_ACCOUNT,
             SNOWFLAKE_LOAD_DATABASE,
             SNOWFLAKE_LOAD_WAREHOUSE,
+            SNOWFLAKE_LOAD_ROLE,
         ],
         env_vars={},
         arguments=[
             "push_proxy PUSH_PROXY_LOGS_US_NEW LOGS_US_NEW "
             " --prefix /AWSLogs/{{ var.value.push_proxy_aws_account_id }}/elasticloadbalancing/{{ var.value.push_proxy_aws_region }}"
             " -s {{ var.value.push_proxy_target_schema }}"
+            " -a ${SNOWFLAKE_ACCOUNT}"
             " -d ${SNOWFLAKE_LOAD_DATABASE}"
             " -w ${SNOWFLAKE_LOAD_WAREHOUSE}"
             " -r ${SNOWFLAKE_LOAD_ROLE}"
@@ -63,12 +66,14 @@ def push_proxy_loader():
             SNOWFLAKE_ACCOUNT,
             SNOWFLAKE_LOAD_DATABASE,
             SNOWFLAKE_LOAD_WAREHOUSE,
+            SNOWFLAKE_LOAD_ROLE,
         ],
         env_vars={},
         arguments=[
             "push_proxy PUSH_PROXY_LOGS_TEST_NEW LOGS_TEST_NEW "
             " --prefix /AWSLogs/{{ var.value.push_proxy_aws_account_id }}/elasticloadbalancing/{{ var.value.push_proxy_aws_region }}"
             " -s {{ var.value.push_proxy_target_schema }}"
+            " -a ${SNOWFLAKE_ACCOUNT}"
             " -d ${SNOWFLAKE_LOAD_DATABASE}"
             " -w ${SNOWFLAKE_LOAD_WAREHOUSE}"
             " -r ${SNOWFLAKE_LOAD_ROLE}"