updated Google DV360 Hook to fix SDF issue (#13703)

Co-authored-by: Sara Hamilton <[email protected]>
apache · Jan 17, 2021 · 7ec858c · 7ec858c
1 parent 2f79fb9
commit 7ec858c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/airflow/providers/google/marketing_platform/hooks/display_video.py b/airflow/providers/google/marketing_platform/hooks/display_video.py
@@ -228,7 +228,7 @@ def get_sdf_download_operation(self, operation_name: str):
  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
@@ -314,7 +314,7 @@ def test_get_sdf_download_tasks_called_with_params(self, get_conn_to_display_vid
  # 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 @@ def test_get_sdf_download_tasks_called_once(self, get_conn_to_display_video):
  # 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 @@ def get_sdf_download_tasks_return_equal_values(self, get_conn_to_display_video):
  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
  )