def test_instance_query_dml(self, mock_hook):
mock_hook.return_value.execute_dml.return_value = None
op = SpannerQueryDatabaseInstanceOperator(project_id=PROJECT_ID,
instance_id=INSTANCE_ID,
database_id=DB_ID,
query=INSERT_QUERY,
task_id="id")
op.execute(None)
mock_hook.assert_called_once_with(gcp_conn_id="google_cloud_default")
mock_hook.return_value.execute_dml.assert_called_once_with(
project_id=PROJECT_ID,
instance_id=INSTANCE_ID,
database_id=DB_ID,
queries=[INSERT_QUERY])
def test_instance_query_missing_project_id(self, mock_hook):
mock_hook.return_value.execute_sql.return_value = None
op = SpannerQueryDatabaseInstanceOperator(instance_id=INSTANCE_ID,
database_id=DB_ID,
query=INSERT_QUERY,
task_id="id")
result = op.execute(None) # pylint: disable=assignment-from-no-return
mock_hook.assert_called_once_with(gcp_conn_id="google_cloud_default")
mock_hook.return_value.execute_dml.assert_called_once_with(
project_id=None,
instance_id=INSTANCE_ID,
database_id=DB_ID,
queries=[INSERT_QUERY])
self.assertIsNone(result)
def test_instance_query(self, mock_hook):
mock_hook.return_value.execute_sql.return_value = None
op = SpannerQueryDatabaseInstanceOperator(
project_id=PROJECT_ID,
instance_id=INSTANCE_ID,
database_id=DB_ID,
query=INSERT_QUERY,
task_id="id",
)
result = op.execute(None) # pylint: disable=assignment-from-no-return
mock_hook.assert_called_once_with(
gcp_conn_id="google_cloud_default",
impersonation_chain=None,
)
mock_hook.return_value.execute_dml.assert_called_once_with(
project_id=PROJECT_ID,
instance_id=INSTANCE_ID,
database_id=DB_ID,
queries=[INSERT_QUERY])
assert result is None
