def test_instance_export_missing_project_id(self, mock_hook):
mock_hook.return_value.export_instance.return_value = True
op = CloudSQLExportInstanceOperator(instance=INSTANCE_NAME,
body=EXPORT_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.export_instance.assert_called_once_with(
project_id=None, instance=INSTANCE_NAME, body=EXPORT_BODY)
self.assertTrue(result)
def test_instance_export(self, mock_hook):
mock_hook.return_value.export_instance.return_value = True
op = CloudSQLExportInstanceOperator(project_id=PROJECT_ID,
instance=INSTANCE_NAME,
body=EXPORT_BODY,
task_id="id")
result = op.execute(None)
mock_hook.assert_called_once_with(
api_version="v1beta4",
gcp_conn_id="google_cloud_default",
impersonation_chain=None,
)
mock_hook.return_value.export_instance.assert_called_once_with(
project_id=PROJECT_ID, instance=INSTANCE_NAME, body=EXPORT_BODY)
assert result