Fix crash when workspace preview arrives later than expected. am: 0c26b0e979

Original change: https://googleplex-android-review.googlesource.com/c/platform/packages/apps/WallpaperPicker2/+/15048682

Change-Id: I1172a78e8a306d61388959308359d7fc99f4754e
diff --git a/src/com/android/wallpaper/picker/WorkspaceSurfaceHolderCallback.java b/src/com/android/wallpaper/picker/WorkspaceSurfaceHolderCallback.java
index c469820..2661923 100644
--- a/src/com/android/wallpaper/picker/WorkspaceSurfaceHolderCallback.java
+++ b/src/com/android/wallpaper/picker/WorkspaceSurfaceHolderCallback.java
@@ -164,6 +164,12 @@
 
     protected void requestPreview(SurfaceView workspaceSurface,
             PreviewUtils.WorkspacePreviewCallback callback) {
+        if (workspaceSurface.getDisplay() == null) {
+            Log.w(TAG,
+                    "No display ID, avoiding asking for workspace preview, lest WallpaperPicker "
+                            + "crash");
+            return;
+        }
         Bundle request = SurfaceViewUtils.createSurfaceViewRequest(workspaceSurface);
         if (mWallpaperColors != null) {
             request.putParcelable(KEY_WALLPAPER_COLORS, mWallpaperColors);