add and use flag accuratePump

fzyzcjy · fzyzcjy · commit e0b5882b87bc · 2022-09-29T11:46:56.000+08:00
diff --git a/packages/flutter_test/lib/src/binding.dart b/packages/flutter_test/lib/src/binding.dart
@@ -1077,6 +1077,10 @@ class AutomatedTestWidgetsFlutterBinding extends TestWidgetsFlutterBinding {
   @override
   int get microtaskCount => _currentFakeAsync!.microtaskCount;
 
+  /// In order to let animation get accurate time information during [pump],
+  /// set the [accurate] flag to true. Default is false for compatibility.
+  bool accuratePump = false;
+
   @override
   Future<void> pump([ Duration? duration, EnginePhase newPhase = EnginePhase.sendSemanticsUpdate ]) {
     return TestAsyncUtils.guard<void>(() {
@@ -1089,9 +1093,11 @@ class AutomatedTestWidgetsFlutterBinding extends TestWidgetsFlutterBinding {
       if (hasScheduledFrame) {
         addTime(const Duration(milliseconds: 500));
         _currentFakeAsync!.flushMicrotasks();
-        handleBeginFrame(Duration(
-          microseconds: _clock!.now().microsecondsSinceEpoch,
-        ));
+        handleBeginFrame(
+          accuratePump
+              ? Duration(microseconds: _clock!.now().microsecondsSinceEpoch)
+              : Duration(milliseconds: _clock!.now().millisecondsSinceEpoch)
+        );
         _currentFakeAsync!.flushMicrotasks();
         handleDrawFrame();
       }
diff --git a/packages/flutter_test/test/bindings_test.dart b/packages/flutter_test/test/bindings_test.dart
@@ -57,15 +57,41 @@ void main() {
     order += 1;
   });
 
-  testWidgets('timeStamp should be accurate', (WidgetTester tester) async {
-    final WidgetsBinding widgetsBinding = WidgetsFlutterBinding.ensureInitialized();
+  group('"accurate" argument in pump', () {
+    testWidgets(
+        'when accurate=true, timeStamp should be truncated for compatibility',
+        (WidgetTester tester) async {
+      final WidgetsBinding widgetsBinding =
+          WidgetsFlutterBinding.ensureInitialized();
+
+      await tester.pumpWidget(const CircularProgressIndicator());
+
+      final Duration timeStampBefore =
+          widgetsBinding.currentSystemFrameTimeStamp;
+      await tester.pump(const Duration(microseconds: 12345));
+      final Duration timeStampAfter =
+          widgetsBinding.currentSystemFrameTimeStamp;
+
+      expect(timeStampAfter - timeStampBefore,
+          const Duration(microseconds: 12000));
+    });
+
+    testWidgets('when accurate=true, timeStamp should be accurate',
+        (WidgetTester tester) async {
+      final AutomatedTestWidgetsFlutterBinding widgetsBinding =
+          AutomatedTestWidgetsFlutterBinding.ensureInitialized();
+      widgetsBinding.accuratePump = true;
 
-    await tester.pumpWidget(const CircularProgressIndicator());
+      await tester.pumpWidget(const CircularProgressIndicator());
 
-    final Duration timeStampBefore = widgetsBinding.currentSystemFrameTimeStamp;
-    await tester.pump(const Duration(microseconds: 12345));
-    final Duration timeStampAfter = widgetsBinding.currentSystemFrameTimeStamp;
+      final Duration timeStampBefore =
+          widgetsBinding.currentSystemFrameTimeStamp;
+      await tester.pump(const Duration(microseconds: 12345));
+      final Duration timeStampAfter =
+          widgetsBinding.currentSystemFrameTimeStamp;
 
-    expect(timeStampAfter - timeStampBefore, const Duration(microseconds: 12345));
+      expect(timeStampAfter - timeStampBefore,
+          const Duration(microseconds: 12345));
+    });
   });
 }
