From 6eee79483b5f79854426c7f29daa009b20fadbba Mon Sep 17 00:00:00 2001
From: cnchen <cnchen@google.com>
Date: Mon, 21 Jan 2019 16:04:28 +0800
Subject: [PATCH] [Migration] pass suw extra intent to wallpaper page

Bug: b/117080532
Test: atest && maunally test SUW
Change-Id: I8d6854976a67a01cedf0a26f7d8d37196edbb02d
---
 .../settings/wallpaper/WallpaperSuggestionActivity.java     | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/com/android/settings/wallpaper/WallpaperSuggestionActivity.java b/src/com/android/settings/wallpaper/WallpaperSuggestionActivity.java
index 50819e2f6de..14564b1d575 100644
--- a/src/com/android/settings/wallpaper/WallpaperSuggestionActivity.java
+++ b/src/com/android/settings/wallpaper/WallpaperSuggestionActivity.java
@@ -29,6 +29,8 @@ import androidx.annotation.VisibleForTesting;
 import com.android.settings.R;
 import com.android.settings.core.SubSettingLauncher;
 
+import com.google.android.setupcompat.util.WizardManagerHelper;
+
 public class WallpaperSuggestionActivity extends Activity {
 
     @Override
@@ -39,6 +41,10 @@ public class WallpaperSuggestionActivity extends Activity {
                 .setClassName(getString(R.string.config_wallpaper_picker_package),
                         getString(R.string.config_wallpaper_picker_class))
                 .addFlags(Intent.FLAG_ACTIVITY_FORWARD_RESULT);
+
+        // passing the necessary extra to next page
+        WizardManagerHelper.copyWizardManagerExtras(getIntent(), intent);
+
         if (pm.resolveActivity(intent, 0) != null) {
             startActivity(intent);
         } else {