assert Camera2Interop.Extender is constructed

Author: dasyad00

packages/camera/camera_android_camerax/android/src/test/java/io/flutter/plugins/camerax/ImageAnalysisTest.java

@@ -11,21 +11,26 @@
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
 
+import android.hardware.camera2.CaptureRequest;
+import android.util.Range;
 import android.view.Surface;
+import androidx.camera.camera2.interop.Camera2Interop;
 import androidx.camera.core.ImageAnalysis;
 import androidx.camera.core.ImageAnalysis.Analyzer;
 import androidx.camera.core.resolutionselector.ResolutionSelector;
 import androidx.core.content.ContextCompat;
 import java.util.concurrent.Executor;
 import org.junit.Test;
 import org.junit.runner.RunWith;
+import org.mockito.MockedConstruction;
 import org.mockito.MockedStatic;
 import org.mockito.Mockito;
 import org.mockito.stubbing.Answer;
 import org.robolectric.RobolectricTestRunner;
 
 @RunWith(RobolectricTestRunner.class)
 public class ImageAnalysisTest {
+  @SuppressWarnings({"unchecked", "rawtypes"})
   @Test
   public void pigeon_defaultConstructor_createsExpectedImageAnalysisInstance() {
     final PigeonApiImageAnalysis api = new TestProxyApiRegistrar().getPigeonApiImageAnalysis();
@@ -34,13 +39,25 @@ public void pigeon_defaultConstructor_createsExpectedImageAnalysisInstance() {
     final long targetResolution = Surface.ROTATION_0;
     final long targetFps = 30;
     final long outputImageFormat = ImageAnalysis.OUTPUT_IMAGE_FORMAT_NV21;
-    final ImageAnalysis imageAnalysis =
-        api.pigeon_defaultConstructor(
-            mockResolutionSelector, targetResolution, targetFps, outputImageFormat);
 
-    assertEquals(imageAnalysis.getResolutionSelector(), mockResolutionSelector);
-    assertEquals(imageAnalysis.getTargetRotation(), Surface.ROTATION_0);
-    assertEquals(imageAnalysis.getOutputImageFormat(), ImageAnalysis.OUTPUT_IMAGE_FORMAT_NV21);
+    try (MockedConstruction<Camera2Interop.Extender> mockCamera2InteropExtender =
+        Mockito.mockConstruction(
+            Camera2Interop.Extender.class,
+            (mock, context) -> {
+              when(mock.setCaptureRequestOption(
+                      CaptureRequest.CONTROL_AE_TARGET_FPS_RANGE,
+                      new Range<>(targetFps, targetFps)))
+                  .thenReturn(mock);
+            })) {
+      final ImageAnalysis imageAnalysis =
+          api.pigeon_defaultConstructor(
+              mockResolutionSelector, targetResolution, targetFps, outputImageFormat);
+
+      assertEquals(mockResolutionSelector, imageAnalysis.getResolutionSelector());
+      assertEquals(Surface.ROTATION_0, imageAnalysis.getTargetRotation());
+      assertEquals(1, mockCamera2InteropExtender.constructed().size());
+      assertEquals(ImageAnalysis.OUTPUT_IMAGE_FORMAT_NV21, imageAnalysis.getOutputImageFormat());
+    }
   }
 
   @Test

packages/camera/camera_android_camerax/android/src/test/java/io/flutter/plugins/camerax/PreviewTest.java

@@ -13,9 +13,12 @@
 import static org.mockito.Mockito.verifyNoMoreInteractions;
 import static org.mockito.Mockito.when;
 
+import android.hardware.camera2.CaptureRequest;
+import android.util.Range;
 import android.util.Size;
 import android.view.Surface;
 import androidx.annotation.NonNull;
+import androidx.camera.camera2.interop.Camera2Interop;
 import androidx.camera.core.Preview;
 import androidx.camera.core.ResolutionInfo;
 import androidx.camera.core.SurfaceRequest;
@@ -26,22 +29,37 @@
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.mockito.ArgumentCaptor;
+import org.mockito.MockedConstruction;
+import org.mockito.Mockito;
 import org.robolectric.RobolectricTestRunner;
 
 @RunWith(RobolectricTestRunner.class)
 public class PreviewTest {
+  @SuppressWarnings({"unchecked", "rawtypes"})
   @Test
   public void pigeon_defaultConstructor_createsPreviewWithCorrectConfiguration() {
     final PigeonApiPreview api = new TestProxyApiRegistrar().getPigeonApiPreview();
 
     final ResolutionSelector mockResolutionSelector = new ResolutionSelector.Builder().build();
     final long targetResolution = Surface.ROTATION_0;
     final long targetFps = 30;
-    final Preview instance =
-        api.pigeon_defaultConstructor(mockResolutionSelector, targetResolution, targetFps);
 
-    assertEquals(instance.getResolutionSelector(), mockResolutionSelector);
-    assertEquals(instance.getTargetRotation(), Surface.ROTATION_0);
+    try (MockedConstruction<Camera2Interop.Extender> mockCamera2InteropExtender =
+        Mockito.mockConstruction(
+            Camera2Interop.Extender.class,
+            (mock, context) -> {
+              when(mock.setCaptureRequestOption(
+                      CaptureRequest.CONTROL_AE_TARGET_FPS_RANGE,
+                      new Range<>(targetFps, targetFps)))
+                  .thenReturn(mock);
+            })) {
+      final Preview instance =
+          api.pigeon_defaultConstructor(mockResolutionSelector, targetResolution, targetFps);
+
+      assertEquals(mockResolutionSelector, instance.getResolutionSelector());
+      assertEquals(Surface.ROTATION_0, instance.getTargetRotation());
+      assertEquals(1, mockCamera2InteropExtender.constructed().size());
+    }
   }
 
   @Test

packages/camera/camera_android_camerax/android/src/test/java/io/flutter/plugins/camerax/VideoCaptureTest.java

@@ -9,10 +9,15 @@
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
 
+import android.hardware.camera2.CaptureRequest;
+import android.util.Range;
+import androidx.camera.camera2.interop.Camera2Interop;
 import androidx.camera.video.VideoCapture;
 import androidx.camera.video.VideoOutput;
 import org.junit.Test;
 import org.junit.runner.RunWith;
+import org.mockito.MockedConstruction;
+import org.mockito.Mockito;
 import org.robolectric.RobolectricTestRunner;
 
 @RunWith(RobolectricTestRunner.class)
@@ -25,8 +30,20 @@ public void withOutput_createsVideoCaptureWithVideoOutput() {
     final VideoOutput videoOutput = mock(VideoOutput.class);
     final long targetFps = 30;
 
-    final VideoCapture videoCapture = api.withOutput(videoOutput, targetFps);
-    assertEquals(videoCapture.getOutput(), videoOutput);
+    try (MockedConstruction<Camera2Interop.Extender> mockCamera2InteropExtender =
+        Mockito.mockConstruction(
+            Camera2Interop.Extender.class,
+            (mock, context) -> {
+              when(mock.setCaptureRequestOption(
+                      CaptureRequest.CONTROL_AE_TARGET_FPS_RANGE,
+                      new Range<>(targetFps, targetFps)))
+                  .thenReturn(mock);
+            })) {
+      final VideoCapture videoCapture = api.withOutput(videoOutput, targetFps);
+
+      assertEquals(1, mockCamera2InteropExtender.constructed().size());
+      assertEquals(videoOutput, videoCapture.getOutput());
+    }
   }
 
   @SuppressWarnings("unchecked")