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, )