From c5daf4cf600e9cd72295c566905d1f3c50cc24d0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Petrus=20Nguy=E1=BB=85n=20Th=C3=A1i=20H=E1=BB=8Dc?=
 <hoc081098@gmail.com>
Date: Sat, 29 Jan 2022 16:02:42 +0700
Subject: [PATCH] test(flatMap): fix flatMap tests (#658)

---
 test/transformers/flat_map_test.dart | 42 +++++++++++++++++++---------
 1 file changed, 29 insertions(+), 13 deletions(-)

diff --git a/test/transformers/flat_map_test.dart b/test/transformers/flat_map_test.dart
index a53fa34fc..1b3be182b 100644
--- a/test/transformers/flat_map_test.dart
+++ b/test/transformers/flat_map_test.dart
@@ -149,40 +149,56 @@ void main() {
     const maxConcurrent = 2;
     var activeCount = 0;
 
-    final stream = Stream.fromIterable([1, 2, 3, 4, 5, 6, 7, 8, 9, 10]).flatMap(
+    // 1 -> 500
+    // 2 -> 400
+    // 3 -> 500
+    // 4 -> 200
+    // -----1--4
+    // ----2-----3
+    // ----21--4-3
+    final stream = Stream.fromIterable([1, 2, 3, 4]).flatMap(
       (value) {
         return Rx.defer(() {
           expect(++activeCount, lessThanOrEqualTo(maxConcurrent));
 
-          final ms = value.isOdd ? ((10 - value) * 200) : ((11 - value) * 400);
+          final ms = (value.isOdd ? 5 : 6 - value) * 100;
           return Rx.timer(value, Duration(milliseconds: ms));
         }).doOnDone(() => --activeCount);
       },
       maxConcurrent: maxConcurrent,
     );
 
-    await expectLater(stream,
-        emitsInOrder(<Object>[1, 3, 2, 5, 4, 6, 7, 9, 10, 8, emitsDone]));
+    await expectLater(stream, emitsInOrder(<Object>[2, 1, 4, 3, emitsDone]));
   });
 
   test('Rx.flatMap(maxConcurrent: 3)', () async {
     const maxConcurrent = 3;
     var activeCount = 0;
 
-    final stream = Stream.fromIterable([1, 2, 3, 4, 5, 6, 7, 8, 9, 10]).flatMap(
+    // 1 -> 400
+    // 2 -> 300
+    // 3 -> 200
+    // 4 -> 200
+    // 5 -> 300
+    // 6 -> 400
+    // ----1----6
+    // ---2---5
+    // --3--4
+    // --3214-5-6
+    final stream = Stream.fromIterable([1, 2, 3, 4, 5, 6]).flatMap(
       (value) {
         return Rx.defer(() {
           expect(++activeCount, lessThanOrEqualTo(maxConcurrent));
 
-          final ms = value.isOdd ? ((10 - value) * 240) : ((11 - value) * 360);
+          final ms = (value <= 3 ? 5 - value : value - 2) * 100;
           return Rx.timer(value, Duration(milliseconds: ms));
         }).doOnDone(() => --activeCount);
       },
       maxConcurrent: maxConcurrent,
     );
 
-    await expectLater(stream,
-        emitsInOrder(<Object>[3, 1, 2, 5, 7, 4, 9, 10, 6, 8, emitsDone]));
+    await expectLater(
+        stream, emitsInOrder(<Object>[3, 2, 1, 4, 5, 6, emitsDone]));
   });
 
   test('Rx.flatMap.cancel', () {
@@ -190,35 +206,35 @@ void main() {
         .flatMap(_getOtherStream)
         .listen(expectAsync1((data) {}, count: 0))
         .cancel();
-  }, timeout: const Timeout(Duration(milliseconds: 500)));
+  }, timeout: const Timeout(Duration(milliseconds: 200)));
 
   test('Rx.flatMap(maxConcurrent: 1).cancel', () {
     _getStream()
         .flatMap(_getOtherStream, maxConcurrent: 1)
         .listen(expectAsync1((data) {}, count: 0))
         .cancel();
-  }, timeout: const Timeout(Duration(milliseconds: 500)));
+  }, timeout: const Timeout(Duration(milliseconds: 200)));
 
   test('Rx.flatMap.take.cancel', () {
     _getStream()
         .flatMap(_getOtherStream)
         .take(1)
         .listen(expectAsync1((data) => expect(data, 3), count: 1));
-  }, timeout: const Timeout(Duration(milliseconds: 500)));
+  }, timeout: const Timeout(Duration(milliseconds: 200)));
 
   test('Rx.flatMap(maxConcurrent: 1).take.cancel', () {
     _getStream()
         .flatMap(_getOtherStream, maxConcurrent: 1)
         .take(1)
         .listen(expectAsync1((data) => expect(data, 1), count: 1));
-  }, timeout: const Timeout(Duration(milliseconds: 500)));
+  }, timeout: const Timeout(Duration(milliseconds: 200)));
 
   test('Rx.flatMap(maxConcurrent: 2).take.cancel', () {
     _getStream()
         .flatMap(_getOtherStream, maxConcurrent: 2)
         .take(1)
         .listen(expectAsync1((data) => expect(data, 2), count: 1));
-  }, timeout: const Timeout(Duration(milliseconds: 500)));
+  }, timeout: const Timeout(Duration(milliseconds: 200)));
 }
 
 Stream<int> _getStream() => Stream.fromIterable(const [1, 2, 3]);