packages/image_picker/example/android/app/src/test/java/io/flutter/plugins/imagepicker/ImagePickerCacheTest.java

// Copyright 2019 The Flutter 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 io.flutter.plugins.imagepicker;

import static io.flutter.plugins.imagepicker.ImagePickerCache.MAP_KEY_IMAGE_QUALITY;
import static io.flutter.plugins.imagepicker.ImagePickerCache.SHARED_PREFERENCES_NAME;
import static org.hamcrest.core.IsEqual.equalTo;
import static org.junit.Assert.assertThat;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import android.app.Activity;
import android.content.Context;
import android.content.SharedPreferences;
import android.content.pm.PackageManager;
import io.flutter.plugin.common.MethodCall;
import java.util.HashMap;
import java.util.Map;
import org.junit.Before;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;

public class ImagePickerCacheTest {
  private static final int IMAGE_QUALITY = 90;

  @Mock Activity mockActivity;
  @Mock SharedPreferences mockPreference;
  @Mock SharedPreferences.Editor mockEditor;
  @Mock MethodCall mockMethodCall;

  static Map<String, Object> preferenceStorage;

  @Before
  public void setUp() {
    MockitoAnnotations.initMocks(this);

    preferenceStorage = new HashMap();
    when(mockActivity.getPackageName()).thenReturn("com.example.test");
    when(mockActivity.getPackageManager()).thenReturn(mock(PackageManager.class));
    when(mockActivity.getSharedPreferences(SHARED_PREFERENCES_NAME, Context.MODE_PRIVATE))
        .thenReturn(mockPreference);
    when(mockPreference.edit()).thenReturn(mockEditor);
    when(mockEditor.putInt(any(String.class), any(int.class)))
        .then(
            i -> {
              preferenceStorage.put(i.getArgument(0), i.getArgument(1));
              return mockEditor;
            });
    when(mockEditor.putLong(any(String.class), any(long.class)))
        .then(
            i -> {
              preferenceStorage.put(i.getArgument(0), i.getArgument(1));
              return mockEditor;
            });
    when(mockEditor.putString(any(String.class), any(String.class)))
        .then(
            i -> {
              preferenceStorage.put(i.getArgument(0), i.getArgument(1));
              return mockEditor;
            });

    when(mockPreference.getInt(any(String.class), any(int.class)))
        .then(
            i -> {
              int result =
                  (int)
                      ((preferenceStorage.get(i.getArgument(0)) != null)
                          ? preferenceStorage.get(i.getArgument(0))
                          : i.getArgument(1));
              return result;
            });
    when(mockPreference.getLong(any(String.class), any(long.class)))
        .then(
            i -> {
              long result =
                  (long)
                      ((preferenceStorage.get(i.getArgument(0)) != null)
                          ? preferenceStorage.get(i.getArgument(0))
                          : i.getArgument(1));
              return result;
            });
    when(mockPreference.getString(any(String.class), any(String.class)))
        .then(
            i -> {
              String result =
                  (String)
                      ((preferenceStorage.get(i.getArgument(0)) != null)
                          ? preferenceStorage.get(i.getArgument(0))
                          : i.getArgument(1));
              return result;
            });

    when(mockPreference.contains(any(String.class))).thenReturn(true);
  }

  @Test
  public void ImageCache_ShouldBeAbleToSetAndGetQuality() {
    when(mockMethodCall.argument(MAP_KEY_IMAGE_QUALITY)).thenReturn(IMAGE_QUALITY);
    ImagePickerCache cache = new ImagePickerCache(mockActivity);
    cache.saveDimensionWithMethodCall(mockMethodCall);
    Map<String, Object> resultMap = cache.getCacheMap();
    int imageQuality = (int) resultMap.get(cache.MAP_KEY_IMAGE_QUALITY);
    assertThat(imageQuality, equalTo(IMAGE_QUALITY));

    when(mockMethodCall.argument(MAP_KEY_IMAGE_QUALITY)).thenReturn(null);
    cache.saveDimensionWithMethodCall(mockMethodCall);
    Map<String, Object> resultMapWithDefaultQuality = cache.getCacheMap();
    int defaultImageQuality = (int) resultMapWithDefaultQuality.get(cache.MAP_KEY_IMAGE_QUALITY);
    assertThat(defaultImageQuality, equalTo(100));
  }
}