You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@airflow.apache.org by po...@apache.org on 2021/01/17 12:47:48 UTC

[airflow] branch master updated: updated Google DV360 Hook to fix SDF issue (#13703)

This is an automated email from the ASF dual-hosted git repository.

potiuk pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/airflow.git


The following commit(s) were added to refs/heads/master by this push:
     new 7ec858c  updated Google DV360 Hook to fix SDF issue (#13703)
7ec858c is described below

commit 7ec858c4523b24e7a3d6dd1d49e3813e6eee7dff
Author: Sara Hamilton <sa...@yale.edu>
AuthorDate: Sun Jan 17 06:47:35 2021 -0600

    updated Google DV360 Hook to fix SDF issue (#13703)
    
    Co-authored-by: Sara Hamilton <sa...@google.com>
---
 airflow/providers/google/marketing_platform/hooks/display_video.py  | 2 +-
 .../providers/google/marketing_platform/hooks/test_display_video.py | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/airflow/providers/google/marketing_platform/hooks/display_video.py b/airflow/providers/google/marketing_platform/hooks/display_video.py
index 8789633..4cb8c36 100644
--- a/airflow/providers/google/marketing_platform/hooks/display_video.py
+++ b/airflow/providers/google/marketing_platform/hooks/display_video.py
@@ -228,7 +228,7 @@ class GoogleDisplayVideo360Hook(GoogleBaseHook):
         result = (
             self.get_conn_to_display_video()  # pylint: disable=no-member
             .sdfdownloadtasks()
-            .operation()
+            .operations()
             .get(name=operation_name)
             .execute(num_retries=self.num_retries)
         )
diff --git a/tests/providers/google/marketing_platform/hooks/test_display_video.py b/tests/providers/google/marketing_platform/hooks/test_display_video.py
index 2a14ac5..1cd0a2e 100644
--- a/tests/providers/google/marketing_platform/hooks/test_display_video.py
+++ b/tests/providers/google/marketing_platform/hooks/test_display_video.py
@@ -314,7 +314,7 @@ class TestGoogleDisplayVideo360Hook(TestCase):
         # fmt: off
         get_conn_to_display_video.return_value. \
             sdfdownloadtasks.return_value. \
-            operation.return_value. \
+            operations.return_value. \
             get.assert_called_once_with(name=operation_name)
         # fmt: on
 
@@ -328,7 +328,7 @@ class TestGoogleDisplayVideo360Hook(TestCase):
         # fmt: off
         get_conn_to_display_video.return_value. \
             sdfdownloadtasks.return_value. \
-            operation.return_value. \
+            operations.return_value. \
             get.assert_called_once()
         # fmt: on
 
@@ -340,7 +340,7 @@ class TestGoogleDisplayVideo360Hook(TestCase):
         operation_name = "operation"
         response = "reposonse"
 
-        get_conn_to_display_video.return_value.sdfdownloadtasks.return_value.operation.return_value.get = (
+        get_conn_to_display_video.return_value.sdfdownloadtasks.return_value.operations.return_value.get = (
             response
         )