Test: EitherStream.collectRight

raquo · Mar 13, 2024 · 1059443 · 1059443
1 parent d717403
commit 1059443
Showing 1 changed file with 97 additions and 0 deletions.
diff --git a/src/test/scala/com/raquo/airstream/extensions/EitherObservableSpec.scala b/src/test/scala/com/raquo/airstream/extensions/EitherObservableSpec.scala
@@ -251,4 +251,101 @@ class EitherObservableSpec extends UnitSpec {
     )
     effects.clear()
   }
+
+  it("EitherStream: collectRight") {
+    val owner: TestableOwner = new TestableOwner
+
+    val bus = new EventBus[Either[Int, String]]
+
+    val effects = mutable.Buffer[Effect[_]]()
+
+    bus
+      .events
+      .collectRight
+      .foreach(v => effects += Effect("obs", v))(owner)
+
+    effects shouldBe mutable.Buffer()
+
+    // --
+
+    bus.emit(Left(1))
+
+    effects.toList shouldBe Nil
+
+    // --
+
+    bus.emit(Right("a"))
+
+    effects.toList shouldBe List(
+      Effect("obs", "a")
+    )
+    effects.clear()
+
+    // --
+
+    bus.emit(Right("b"))
+
+    effects.toList shouldBe List(
+      Effect("obs", "b")
+    )
+    effects.clear()
+
+    // --
+
+    bus.emit(Left(2))
+
+    effects.toList shouldBe Nil
+  }
+
+  it("EitherStream: collectRight()") {
+    val owner: TestableOwner = new TestableOwner
+
+    val bus = new EventBus[Either[Int, String]]
+
+    val effects = mutable.Buffer[Effect[_]]()
+
+    bus
+      .events
+      .collectRight { case "a" => true }
+      .foreach(v => effects += Effect("obs", v))(owner)
+
+    effects shouldBe mutable.Buffer()
+
+    // --
+
+    bus.emit(Left(1))
+
+    effects.toList shouldBe Nil
+
+    // --
+
+    bus.emit(Right("a"))
+
+    effects.toList shouldBe List(
+      Effect("obs", true)
+    )
+    effects.clear()
+
+    // --
+
+    bus.emit(Right("b"))
+
+    effects.toList shouldBe Nil
+    effects.clear()
+
+    // --
+
+    bus.emit(Right("a"))
+
+    effects.toList shouldBe List(
+      Effect("obs", true)
+    )
+    effects.clear()
+
+    // --
+
+    bus.emit(Left(2))
+
+    effects.toList shouldBe Nil
+  }
 }