def test_instance_db_create_missing_project_id(self, mock_hook,
_check_if_db_exists):
_check_if_db_exists.return_value = False
op = CloudSQLCreateInstanceDatabaseOperator(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)
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 = CloudSQLCreateInstanceDatabaseOperator(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()
sql_instance_patch_task = CloudSQLInstancePatchOperator(
project_id=GCP_PROJECT_ID,
body=patch_body,
instance=INSTANCE_NAME,
task_id='sql_instance_patch_task')
sql_instance_patch_task2 = CloudSQLInstancePatchOperator(
body=patch_body,
instance=INSTANCE_NAME,
task_id='sql_instance_patch_task2')
# [END howto_operator_cloudsql_patch]
# [START howto_operator_cloudsql_db_create]
sql_db_create_task = CloudSQLCreateInstanceDatabaseOperator(
project_id=GCP_PROJECT_ID,
body=db_create_body,
instance=INSTANCE_NAME,
task_id='sql_db_create_task')
sql_db_create_task2 = CloudSQLCreateInstanceDatabaseOperator(
body=db_create_body,
instance=INSTANCE_NAME,
task_id='sql_db_create_task2')
# [END howto_operator_cloudsql_db_create]
# [START howto_operator_cloudsql_db_patch]
sql_db_patch_task = CloudSQLPatchInstanceDatabaseOperator(
project_id=GCP_PROJECT_ID,
body=db_patch_body,
instance=INSTANCE_NAME,
database=DB_NAME,
task_id='sql_db_patch_task')