def test_delete_transfer_config(self, service_mock):
self.hook.delete_transfer_config(transfer_config_id=TRANSFER_CONFIG_ID,
project_id=PROJECT_ID)
name = DataTransferServiceClient.project_transfer_config_path(
PROJECT_ID, TRANSFER_CONFIG_ID)
service_mock.assert_called_once_with(name=name,
metadata=None,
retry=None,
timeout=None)
def get_conn(self) -> DataTransferServiceClient:
"""
Retrieves connection to Google Bigquery.
:return: Google Bigquery API client
:rtype: google.cloud.bigquery_datatransfer_v1.DataTransferServiceClient
"""
if not self._conn:
self._conn = DataTransferServiceClient(
credentials=self._get_credentials(), client_info=CLIENT_INFO)
return self._conn
def test_start_manual_transfer_runs(self, service_mock):
self.hook.start_manual_transfer_runs(transfer_config_id=TRANSFER_CONFIG_ID, project_id=PROJECT_ID)
parent = DataTransferServiceClient.project_transfer_config_path(PROJECT_ID, TRANSFER_CONFIG_ID)
service_mock.assert_called_once_with(
parent=parent,
requested_time_range=None,
requested_run_time=None,
metadata=None,
retry=None,
timeout=None,
)
def test_create_transfer_config(self, service_mock):
self.hook.create_transfer_config(transfer_config=TRANSFER_CONFIG, project_id=PROJECT_ID)
parent = DataTransferServiceClient.project_path(PROJECT_ID)
expected_config = deepcopy(TRANSFER_CONFIG)
expected_config.schedule_options.disable_auto_scheduling = True
service_mock.assert_called_once_with(
parent=parent,
transfer_config=expected_config,
authorization_code=None,
metadata=None,
retry=None,
timeout=None,
)
