Python CloudSqlInstanceDatabaseCreateOperator.execute Examples

Example #1

File: test_gcp_sql_operator.py Project: mboegner/airflow

 def test_instance_db_create_should_abort_and_succeed_if_exists(
         self, mock_hook, _check_if_db_exists):
     _check_if_db_exists.return_value = True
     op = CloudSqlInstanceDatabaseCreateOperator(project_id=PROJECT_ID,
                                                 instance=INSTANCE_NAME,
                                                 body=DATABASE_INSERT_BODY,
                                                 task_id="id")
     result = op.execute(None)
     self.assertTrue(result)
     mock_hook.assert_called_once_with(api_version="v1beta4",
                                       gcp_conn_id="google_cloud_default")
     mock_hook.return_value.create_database.assert_not_called()

Example #2

 def test_instance_db_create_missing_project_id(self, mock_hook,
                                                _check_if_db_exists):
     _check_if_db_exists.return_value = False
     op = CloudSqlInstanceDatabaseCreateOperator(instance=INSTANCE_NAME,
                                                 body=DATABASE_INSERT_BODY,
                                                 task_id="id")
     result = op.execute(None)
     mock_hook.assert_called_once_with(api_version="v1beta4",
                                       gcp_conn_id="google_cloud_default")
     mock_hook.return_value.create_database.assert_called_once_with(
         project_id=None, instance=INSTANCE_NAME, body=DATABASE_INSERT_BODY)
     self.assertTrue(result)

Example #3

File: test_gcp_sql_operator.py Project: alrolorojas/airflow

 def test_instance_db_create_should_abort_and_succeed_if_exists(
         self, mock_hook, _check_if_db_exists):
     _check_if_db_exists.return_value = True
     op = CloudSqlInstanceDatabaseCreateOperator(
         project_id=PROJECT_ID,
         instance=INSTANCE_NAME,
         body=DATABASE_INSERT_BODY,
         task_id="id"
     )
     result = op.execute(None)
     self.assertTrue(result)
     mock_hook.assert_called_once_with(api_version="v1beta4",
                                       gcp_conn_id="google_cloud_default")
     mock_hook.return_value.create_database.assert_not_called()

Example #4

File: test_gcp_sql_operator.py Project: alrolorojas/airflow

 def test_instance_db_create_missing_project_id(self, mock_hook, _check_if_db_exists):
     _check_if_db_exists.return_value = False
     op = CloudSqlInstanceDatabaseCreateOperator(
         instance=INSTANCE_NAME,
         body=DATABASE_INSERT_BODY,
         task_id="id"
     )
     result = op.execute(None)
     mock_hook.assert_called_once_with(api_version="v1beta4",
                                       gcp_conn_id="google_cloud_default")
     mock_hook.return_value.create_database.assert_called_once_with(
         project_id=None,
         instance=INSTANCE_NAME,
         body=DATABASE_INSERT_BODY
     )
     self.assertTrue(result)