def test_instance_db_patch_should_throw_ex_when_empty_database(
self, mock_hook):
with self.assertRaises(AirflowException) as cm:
op = CloudSqlInstanceDatabasePatchOperator(
project_id=PROJECT_ID,
instance=INSTANCE_NAME,
database="",
body=DATABASE_INSERT_BODY,
task_id="id")
op.execute(None)
err = cm.exception
self.assertIn("The required parameter 'database' is empty", str(err))
mock_hook.assert_not_called()
mock_hook.return_value.patch_database.assert_not_called()
def test_instance_db_patch_should_throw_ex_when_empty_database(self, mock_hook):
with self.assertRaises(AirflowException) as cm:
op = CloudSqlInstanceDatabasePatchOperator(
project_id=PROJECT_ID,
instance=INSTANCE_NAME,
database="",
body=DATABASE_INSERT_BODY,
task_id="id"
)
op.execute(None)
err = cm.exception
self.assertIn("The required parameter 'database' is empty", str(err))
mock_hook.assert_not_called()
mock_hook.return_value.patch_database.assert_not_called()
def test_instance_db_patch_should_throw_ex_if_not_exists(
self, mock_hook, _check_if_db_exists):
_check_if_db_exists.return_value = False
with self.assertRaises(AirflowException) as cm:
op = CloudSqlInstanceDatabasePatchOperator(
project_id=PROJECT_ID,
instance=INSTANCE_NAME,
database=DB_NAME,
body=DATABASE_PATCH_BODY,
task_id="id")
op.execute(None)
err = cm.exception
self.assertIn("Cloud SQL instance with ID", str(err))
self.assertIn("does not contain database", str(err))
mock_hook.assert_called_once_with(api_version="v1beta4",
gcp_conn_id="google_cloud_default")
mock_hook.return_value.patch_database.assert_not_called()
def test_instance_db_patch_should_throw_ex_if_not_exists(
self, mock_hook, _check_if_db_exists):
_check_if_db_exists.return_value = False
with self.assertRaises(AirflowException) as cm:
op = CloudSqlInstanceDatabasePatchOperator(
project_id=PROJECT_ID,
instance=INSTANCE_NAME,
database=DB_NAME,
body=DATABASE_PATCH_BODY,
task_id="id"
)
op.execute(None)
err = cm.exception
self.assertIn("Cloud SQL instance with ID", str(err))
self.assertIn("does not contain database", str(err))
mock_hook.assert_called_once_with(api_version="v1beta4",
gcp_conn_id="google_cloud_default")
mock_hook.return_value.patch_database.assert_not_called()
def test_instance_db_patch(self, mock_hook, _check_if_db_exists):
_check_if_db_exists.return_value = True
op = CloudSqlInstanceDatabasePatchOperator(project_id=PROJECT_ID,
instance=INSTANCE_NAME,
database=DB_NAME,
body=DATABASE_PATCH_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.patch_database.assert_called_once_with(
PROJECT_ID, INSTANCE_NAME, DB_NAME, DATABASE_PATCH_BODY)
self.assertTrue(result)
def test_instance_db_patch(self, mock_hook, _check_if_db_exists):
_check_if_db_exists.return_value = True
op = CloudSqlInstanceDatabasePatchOperator(
project_id=PROJECT_ID,
instance=INSTANCE_NAME,
database=DB_NAME,
body=DATABASE_PATCH_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.patch_database.assert_called_once_with(
PROJECT_ID, INSTANCE_NAME, DB_NAME, DATABASE_PATCH_BODY
)
self.assertTrue(result)
