Photo Picker dialog: Add a test.

chrome/android/java/src/org/chromium/chrome/browser/photo_picker/PickerBitmapView.java

 // Copyright 2017 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 package org.chromium.chrome.browser.photo_picker;
 
 import android.content.Context;
 import android.content.res.Resources;
 import android.graphics.Bitmap;
 import android.graphics.PorterDuff;
 import android.graphics.drawable.Drawable;
+import android.support.annotation.IntDef;
 import android.support.annotation.Nullable;
 import android.support.graphics.drawable.VectorDrawableCompat;
 import android.util.AttributeSet;
 import android.view.View;
 import android.view.ViewGroup;
 import android.view.animation.Animation;
 import android.view.animation.ScaleAnimation;
 import android.widget.ImageView;
 import android.widget.TextView;
 
 import org.chromium.base.ApiCompatibilityUtils;
+import org.chromium.base.VisibleForTesting;
 import org.chromium.chrome.R;
 import org.chromium.chrome.browser.widget.selection.SelectableItemView;
 import org.chromium.chrome.browser.widget.selection.SelectionDelegate;
 
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
 import java.util.List;
 
 /**
  * A container class for a view showing a photo in the Photo Picker.
  */
 public class PickerBitmapView extends SelectableItemView<PickerBitmap> {
     // The length of the image selection animation (in ms).
     private static final int ANIMATION_DURATION = 100;
 
     // The length of the fade in animation (in ms).
     private static final int IMAGE_FADE_IN_DURATION = 200;
 
+    // The possible stages of loading an image.
+    @IntDef({PENDING, LOADING, LOADED})
+    @Retention(RetentionPolicy.SOURCE)
+    public @interface LoadingStage {}
+    public static final int PENDING = 0;
+    public static final int LOADING = 1;
+    public static final int LOADED = 2;
+
     // Our context.
     private Context mContext;
 
     // Our parent category.
     private PickerCategoryView mCategoryView;
 
     // Our selection delegate.
     private SelectionDelegate<PickerBitmap> mSelectionDelegate;
 
     // The request details (meta-data) for the bitmap shown.
@@ skipping 14 matching lines @@
 
     // The camera/gallery special tile (with icon as drawable).
     private View mSpecialTile;
 
     // The camera/gallery icon.
     public ImageView mSpecialTileIcon;
 
     // The label under the special tile.
     public TextView mSpecialTileLabel;
 
-    // Whether the image has been loaded already.
-    private boolean mImageLoaded;
+    // The loading stage for this tile.
+    @LoadingStage
+    private int mLoadingStage = PENDING;
 
     // The amount to use for the border.
     private int mBorder;
 
     /**
      * Constructor for inflating from XML.
      */
     public PickerBitmapView(Context context, AttributeSet attrs) {
         super(context, attrs);
         mContext = context;
@@ skipping 114 matching lines @@
      * @param placeholder Whether the image given is a placeholder or the actual image.
      */
     public void initialize(
             PickerBitmap bitmapDetails, @Nullable Bitmap thumbnail, boolean placeholder) {
         resetTile();
 
         mBitmapDetails = bitmapDetails;
         setItem(bitmapDetails);
         if (isCameraTile() || isGalleryTile()) {
             initializeSpecialTile(mBitmapDetails);
-            mImageLoaded = true;
+            mLoadingStage = LOADED;
         } else {
             setThumbnailBitmap(thumbnail);
-            mImageLoaded = !placeholder;
+            mLoadingStage = placeholder ? LOADING : LOADED;
         }
 
         updateSelectionState();
     }
 
     /**
      * Initialization for the special tiles (camera/gallery icon).
      * @param bitmapDetails The details about the bitmap represented by this PickerBitmapView.
      */
     public void initializeSpecialTile(PickerBitmap bitmapDetails) {
@@ skipping 29 matching lines @@
         mIconView.setImageBitmap(thumbnail);
 
         // If the tile has been selected before the bitmap has loaded, make sure it shows up with
         // a selection border on load.
         if (super.isChecked()) {
             mIconView.getLayoutParams().height = imageSizeWithBorders();
             mIconView.getLayoutParams().width = imageSizeWithBorders();
             addPaddingToParent(mIconView, mBorder);
         }
 
-        boolean noImageWasLoaded = !mImageLoaded;
-        mImageLoaded = true;
+        boolean noImageWasLoaded = mLoadingStage != LOADED;
+        mLoadingStage = LOADED;
         updateSelectionState();
 
         return noImageWasLoaded;
     }
 
     /** Returns the size of the image plus the pre-determined border on each side. */
     private int imageSizeWithBorders() {
         return mCategoryView.getImageSize() - 2 * mBorder;
     }
 
@@ skipping 57 matching lines @@
             Drawable drawable = mSpecialTileIcon.getDrawable();
             int color = ApiCompatibilityUtils.getColor(resources, fgColorId);
             drawable.setColorFilter(color, PorterDuff.Mode.SRC_IN);
             mSpecialTileIcon.invalidate();
         }
 
         setBackgroundColor(ApiCompatibilityUtils.getColor(resources, bgColorId));
 
         // The visibility of the unselected image is a little more complex because we don't want
         // to show it when nothing is selected and also not on a blank canvas.
+        boolean imageLoaded = mLoadingStage == LOADED;
         mSelectedView.setVisibility(!special && checked ? View.VISIBLE : View.GONE);
         mUnselectedView.setVisibility(
-                !special && !checked && anySelection && mImageLoaded ? View.VISIBLE : View.GONE);
+                !special && !checked && anySelection && imageLoaded ? View.VISIBLE : View.GONE);
         mScrim.setVisibility(
-                !special && !checked && anySelection && mImageLoaded ? View.VISIBLE : View.GONE);
+                !special && !checked && anySelection && imageLoaded ? View.VISIBLE : View.GONE);
     }
 
     private boolean isGalleryTile() {
         return mBitmapDetails.type() == PickerBitmap.GALLERY;
     }
 
     private boolean isCameraTile() {
         return mBitmapDetails.type() == PickerBitmap.CAMERA;
     }
 
     private boolean isPictureTile() {
         return mBitmapDetails.type() == PickerBitmap.PICTURE;
     }
+
+    @VisibleForTesting
+    public boolean getImageLoadingForTesting() {
+        return mLoadingStage == LOADING;
+    }
 }