Add tests for mergePreferred

typelevel · Jul 11, 2023 · 83fcff2 · 83fcff2
1 parent b1b9aab
commit 83fcff2
Showing 1 changed file with 33 additions and 0 deletions.
diff --git a/core/shared/src/test/scala/fs2/StreamMergeSuite.scala b/core/shared/src/test/scala/fs2/StreamMergeSuite.scala
@@ -242,4 +242,37 @@ class StreamMergeSuite extends Fs2Suite {
         }
     }
   }
+
+  test("mergePreferred prefers") {
+    forAllF { (leftStream: Stream[Pure, Int], rightStream: Stream[Pure, Int]) =>
+      val leftTagged = leftStream.covary[IO]
+      val rightTagged = rightStream.covary[IO].delayBy(10.milli)
+      leftTagged
+        .mergePreferred(rightTagged)
+        .assertEmitsSameAs(leftStream ++ rightStream)
+    }
+  }
+
+  test("mergePreferred fully consumes this") {
+    forAllF { (stream: Stream[Pure, Int]) =>
+      stream.covary[IO].mergePreferred(Stream.empty.covary[IO]).assertEmitsSameAs(stream)
+    }
+  }
+
+  test("mergePreferred fully consumes that") {
+    forAllF { (stream: Stream[Pure, Int]) =>
+      Stream.empty.covary[IO].mergePreferred(stream.covary[IO]).assertEmitsSameAs(stream)
+    }
+  }
+
+  test("mergePreferred fully consumes both") {
+    forAllF { (leftStream: Stream[Pure, Int], rightStream: Stream[Pure, Int]) =>
+      val leftTagged = leftStream.covary[IO]
+      val rightTagged = rightStream.covary[IO]
+      leftTagged
+        .mergePreferred(rightTagged)
+        .assertEmitsUnorderedSameAs(leftStream ++ rightStream)
+    }
+  }
+
 }