def test_database_update_ex_if_database_not_exist(self, mock_hook):
mock_hook.return_value.get_database.return_value = None
with self.assertRaises(AirflowException) as cm:
op = SpannerUpdateDatabaseInstanceOperator(
project_id=PROJECT_ID,
instance_id=INSTANCE_ID,
database_id=DB_ID,
ddl_statements=DDL_STATEMENTS,
task_id="id")
op.execute(None)
err = cm.exception
self.assertIn(
"The Cloud Spanner database 'db1' in project 'project-id' and "
"instance 'instance-id' is missing", str(err))
mock_hook.assert_called_once_with(gcp_conn_id="google_cloud_default")
def test_database_update_missing_project_id(self, mock_hook):
mock_hook.return_value.get_database.return_value = {"name": DB_ID}
op = SpannerUpdateDatabaseInstanceOperator(
instance_id=INSTANCE_ID,
database_id=DB_ID,
ddl_statements=DDL_STATEMENTS,
task_id="id"
)
result = op.execute(None)
mock_hook.assert_called_once_with(gcp_conn_id="google_cloud_default")
mock_hook.return_value.update_database.assert_called_once_with(
project_id=None, instance_id=INSTANCE_ID, database_id=DB_ID,
ddl_statements=DDL_STATEMENTS, operation_id=None
)
self.assertTrue(result)
def test_database_update(self, mock_hook):
mock_hook.return_value.get_database.return_value = {"name": DB_ID}
op = SpannerUpdateDatabaseInstanceOperator(
project_id=PROJECT_ID,
instance_id=INSTANCE_ID,
database_id=DB_ID,
ddl_statements=DDL_STATEMENTS,
task_id="id",
)
result = op.execute(None)
mock_hook.assert_called_once_with(
gcp_conn_id="google_cloud_default",
impersonation_chain=None,
)
mock_hook.return_value.update_database.assert_called_once_with(
project_id=PROJECT_ID,
instance_id=INSTANCE_ID,
database_id=DB_ID,
ddl_statements=DDL_STATEMENTS,
operation_id=None,
)
assert result
