fix: don't catch exception when headOption is called

Author: geminicaprograms

core/src/main/scala/ox/channels/SourceOps.scala

@@ -5,7 +5,6 @@ import ox.*
 import java.util.concurrent.{CountDownLatch, Semaphore}
 import scala.collection.{IterableOnce, mutable}
 import scala.concurrent.duration.FiniteDuration
-import scala.util.Try
 
 trait SourceOps[+T] { this: Source[T] =>
   // view ops (lazy)
@@ -515,11 +514,13 @@ trait SourceOps[+T] { this: Source[T] =>
     }
     c
 
-  /** Returns the first element from this source wrapped in `Some` or `None` when the source is empty or fails during the receive operation.
-    * Note that `headOption` is not an idempotent operation on source as it receives elements from it.
+  /** Returns the first element from this source wrapped in `Some` or `None` when the source is empty. Note that `headOption` is not an
+    * idempotent operation on source as it receives elements from it.
     *
     * @return
-    *   A `Some(first element)` if source is not empty or None` otherwise.
+    *   A `Some(first element)` if source is not empty or `None` otherwise.
+    * @throws ChannelClosedException.Error
+    *   When `receive()` fails.
     * @example
     *   {{{
     *   import ox.*
@@ -533,7 +534,13 @@ trait SourceOps[+T] { this: Source[T] =>
     *   }
     *   }}}
     */
-  def headOption(): Option[T] = Try(head()).toOption
+  def headOption(): Option[T] =
+    supervised {
+      receive() match
+        case ChannelClosed.Done     => None
+        case e: ChannelClosed.Error => throw e.toThrowable
+        case t: T @unchecked        => Some(t)
+    }
 
   /** Returns the first element from this source or throws `NoSuchElementException` when the source is empty. In case when the `receive()`
     * operation fails with exception then `ChannelClosedException.Error`` thrown. Note that `headOption` is not an idempotent operation on
@@ -544,7 +551,7 @@ trait SourceOps[+T] { this: Source[T] =>
     * @throws NoSuchElementException
     *   When source is empty or `receive()` failed without error.
     * @throws ChannelClosedException.Error
-    *   When `receive()` fails then this exception is thrown.
+    *   When `receive()` fails.
     * @example
     *   {{{
     *   import ox.*
@@ -558,13 +565,7 @@ trait SourceOps[+T] { this: Source[T] =>
     *   }
     *   }}}
     */
-  def head(): T =
-    supervised {
-      receive() match
-        case ChannelClosed.Done     => throw new NoSuchElementException("cannot obtain head from an empty source")
-        case e: ChannelClosed.Error => throw e.toThrowable
-        case t: T @unchecked        => t
-    }
+  def head(): T = headOption().getOrElse(throw new NoSuchElementException("cannot obtain head from an empty source"))
 
   /** Returns the last element from this source wrapped in `Some` or `None` when the source is empty. Note that `lastOption` is a terminal
     * operation leaving the source in `ChannelClosed.Done` state.

core/src/test/scala/ox/channels/SourceOpsHeadOptionTest.scala

@@ -12,10 +12,18 @@ class SourceOpsHeadOptionTest extends AnyFlatSpec with Matchers with OptionValue
     Source.empty[Int].headOption() shouldBe None
   }
 
-  it should "return None for the failed source" in supervised {
-    Source
-      .failed(new RuntimeException("source is broken"))
-      .headOption() shouldBe None
+  it should "throw ChannelClosedException.Error with exception and message that was thrown during retrieval" in supervised {
+    the[ChannelClosedException.Error] thrownBy {
+      Source
+        .failed(new RuntimeException("source is broken"))
+        .headOption()
+    } should have message "java.lang.RuntimeException: source is broken"
+  }
+
+  it should "throw ChannelClosedException.Error for source failed without exception" in supervised {
+    the[ChannelClosedException.Error] thrownBy {
+      Source.failedWithoutReason[Int]().headOption()
+    }
   }
 
   it should "return Some element for the non-empty source" in supervised {