Update the logic for featured collections

Bug: 188589807
Test: manual
Change-Id: Ib20aa0c2d5097e3640896b34238c8524e2be7e34
diff --git a/src/com/android/wallpaper/picker/CategorySelectorFragment.java b/src/com/android/wallpaper/picker/CategorySelectorFragment.java
index 0455f47..a506bca 100644
--- a/src/com/android/wallpaper/picker/CategorySelectorFragment.java
+++ b/src/com/android/wallpaper/picker/CategorySelectorFragment.java
@@ -122,6 +122,7 @@
     private ArrayList<Category> mCategories = new ArrayList<>();
     private Point mTileSizePx;
     private boolean mAwaitingCategories;
+    private boolean mIsFeaturedCollectionAvailable;
 
     public CategorySelectorFragment() {
         mAdapter = new CategoryAdapter(mCategories);
@@ -250,6 +251,8 @@
             mAdapter.notifyItemRemoved(mAdapter.getItemCount() - 1);
             mAwaitingCategories = false;
         }
+
+        mIsFeaturedCollectionAvailable = mCategoryProvider.isFeaturedCollectionAvailable();
     }
 
     void notifyDataSetChanged() {
@@ -443,7 +446,7 @@
                 return ITEM_VIEW_TYPE_MY_PHOTOS;
             }
 
-            if (mCategoryProvider.isFeaturedCategory(mCategories.get(position))) {
+            if (mIsFeaturedCollectionAvailable && (position == 1 || position == 2)) {
                 return ITEM_VIEW_TYPE_FEATURED_CATEGORY;
             }