Update _Test header

packages/camera/camera/example/ios/RunnerTests/StreamingTest.m

@@ -9,11 +9,6 @@
 #import <OCMock/OCMock.h>
 #import "CameraTestUtils.h"
 
-@interface FLTImageStreamHandler : NSObject <FlutterStreamHandler>
-- (instancetype)initWithCaptureSessionQueue:(dispatch_queue_t)captureSessionQueue;
-@property FlutterEventSink eventSink;
-@end
-
 @interface StreamingTests : XCTestCase
 @property(readonly, nonatomic) FLTCam *camera;
 @property(readonly, nonatomic) CMSampleBufferRef sampleBuffer;
@@ -27,7 +22,7 @@ - (void)setUp {
   _sampleBuffer = FLTCreateTestSampleBuffer();
 }
 
--(void)tearDown {
+- (void)tearDown {
   CFRelease(_sampleBuffer);
 }
 
@@ -45,6 +40,10 @@ - (void)testExceedMaxStreamingPendingFramesCount {
   id messenger = OCMProtocolMock(@protocol(FlutterBinaryMessenger));
   [_camera startImageStreamWithMessenger:messenger];
 
+  while (!_camera.isStreamingImages) {
+    [NSThread sleepForTimeInterval:0.001];
+  }
+
   streamingExpectation.expectedFulfillmentCount = 4;
   for (int i = 0; i < 10; i++) {
     [_camera captureOutput:nil didOutputSampleBuffer:self.sampleBuffer fromConnection:nil];
@@ -68,6 +67,10 @@ - (void)testReceivedImageStreamData {
   id messenger = OCMProtocolMock(@protocol(FlutterBinaryMessenger));
   [_camera startImageStreamWithMessenger:messenger];
 
+  while (!_camera.isStreamingImages) {
+    [NSThread sleepForTimeInterval:0.001];
+  }
+
   streamingExpectation.expectedFulfillmentCount = 5;
   for (int i = 0; i < 10; i++) {
     [_camera captureOutput:nil didOutputSampleBuffer:self.sampleBuffer fromConnection:nil];

packages/camera/camera/ios/Classes/FLTCam.m

@@ -10,14 +10,6 @@
 @import CoreMotion;
 #import <libkern/OSAtomic.h>
 
-@interface FLTImageStreamHandler : NSObject <FlutterStreamHandler>
-// The queue on which `eventSink` property should be accessed
-@property(nonatomic, strong) dispatch_queue_t captureSessionQueue;
-// `eventSink` property should be accessed on `captureSessionQueue`.
-// The block itself should be invoked on the main queue.
-@property FlutterEventSink eventSink;
-@end
-
 @implementation FLTImageStreamHandler
 
 - (instancetype)initWithCaptureSessionQueue:(dispatch_queue_t)captureSessionQueue {
@@ -68,7 +60,6 @@ @interface FLTCam () <AVCaptureVideoDataOutputSampleBufferDelegate,
 @property(assign, nonatomic) BOOL videoIsDisconnected;
 @property(assign, nonatomic) BOOL audioIsDisconnected;
 @property(assign, nonatomic) BOOL isAudioSetup;
-@property(assign, nonatomic) BOOL isStreamingImages;
 @property(assign, nonatomic) int streamingPendingFramesCount;
 @property(assign, nonatomic) int maxstreamingPendingFramesCount;
 @property(assign, nonatomic) UIDeviceOrientation lockedCaptureOrientation;

packages/camera/camera/ios/Classes/FLTCam_Test.h

@@ -14,6 +14,9 @@
 /// The output for photo capturing. Exposed setter for unit tests.
 @property(strong, nonatomic) AVCapturePhotoOutput *capturePhotoOutput API_AVAILABLE(ios(10));
 
+/// True when images from the camera are being streamed.
+@property(assign, nonatomic) BOOL isStreamingImages;
+
 /// A dictionary to retain all in-progress FLTSavePhotoDelegates. The key of the dictionary is the
 /// AVCapturePhotoSettings's uniqueID for each photo capture operation, and the value is the
 /// FLTSavePhotoDelegate that handles the result of each photo capture operation. Note that photo
@@ -39,3 +42,15 @@
                              error:(NSError **)error;
 
 @end
+
+@interface FLTImageStreamHandler : NSObject <FlutterStreamHandler>
+// The queue on which `eventSink` property should be accessed
+@property(nonatomic, strong) dispatch_queue_t captureSessionQueue;
+// `eventSink` property should be accessed on `captureSessionQueue`.
+// The block itself should be invoked on the main queue.
+@property FlutterEventSink eventSink;
+@end
+
+@interface FLTImageStreamHandler ()
+- (instancetype)initWithCaptureSessionQueue:(dispatch_queue_t)captureSessionQueue;
+@end