Esempio n. 1
0
 def test_operation_cancel_should_throw_ex_when_name_none(self, mock_hook):
     with self.assertRaises(AirflowException) as cm:
         op = GcpTransferServiceOperationCancelOperator(operation_name="",
                                                        task_id=TASK_ID)
         op.execute(None)
     err = cm.exception
     self.assertIn(
         "The required parameter 'operation_name' is empty or None",
         str(err))
     mock_hook.assert_not_called()
Esempio n. 2
0
 def test_operation_cancel(self, mock_hook):
     op = GcpTransferServiceOperationCancelOperator(
         operation_name=OPERATION_NAME, task_id=TASK_ID)
     result = op.execute(None)  # pylint:disable=assignment-from-no-return
     mock_hook.assert_called_once_with(api_version='v1',
                                       gcp_conn_id='google_cloud_default')
     mock_hook.return_value.cancel_transfer_operation.assert_called_once_with(
         operation_name=OPERATION_NAME)
     self.assertIsNone(result)