From 498cd0d33e9861b237d16a503be8e55c03c1bd95 Mon Sep 17 00:00:00 2001
From: IB <baitcode@baitbook.local>
Date: Mon, 27 Nov 2023 19:56:56 +0000
Subject: [PATCH] test

---
 src/main/kotlin/Task.kt            |  5 +++-
 src/test/kotlin/TaskManagerTest.kt | 37 ++++++++++++++++++++++++++++++
 2 files changed, 41 insertions(+), 1 deletion(-)

diff --git a/src/main/kotlin/Task.kt b/src/main/kotlin/Task.kt
index a37e1ff..8312725 100644
--- a/src/main/kotlin/Task.kt
+++ b/src/main/kotlin/Task.kt
@@ -130,7 +130,10 @@ class Task<T : Any> @PublishedApi internal constructor(
                 }
 
                 for ((exception, handler) in manager.taskErrorHandlers) {
-                    if (e::class == exception) handler.invoke(logger, inputStr)
+                    if (e::class.java == exception) {
+                        handler.invoke(logger, inputStr)
+                        return
+                    }
                 }
 
                 if (getRetryPolicy(manager).shouldRetry(params)) {
diff --git a/src/test/kotlin/TaskManagerTest.kt b/src/test/kotlin/TaskManagerTest.kt
index 6bd76a2..a8beefd 100644
--- a/src/test/kotlin/TaskManagerTest.kt
+++ b/src/test/kotlin/TaskManagerTest.kt
@@ -13,6 +13,9 @@ import io.kotest.framework.concurrency.eventually
 import io.kotest.framework.concurrency.until
 import io.kotest.matchers.shouldBe
 import io.kotest.matchers.shouldNotBe
+import io.mockk.every
+import io.mockk.mockk
+import io.mockk.verify
 import kotlinx.coroutines.launch
 import kotlinx.datetime.Clock
 import org.testcontainers.containers.wait.strategy.Wait
@@ -212,3 +215,37 @@ fun taskManagerTest(taskManager: TaskManager) = funSpec {
     }
 
 }
+
+
+
+
+class TaskManagerErrorHandling: FunSpec({
+    class UnhadledError : Exception()
+
+    val errorHandler = mockk<TaskErrorHandler>()
+    every { errorHandler.invoke(any(), any()) } returns Unit
+
+    val tm = TaskManager(
+        LocalBroker(),
+        taskErrorHandlers = listOf(
+            UnhadledError::class.java to errorHandler
+        )
+    )
+
+    val testTask1 =
+        Task.create("failing-task-${randomSuffix()}",) { ctx, input: TaskTrackExecutionWithContextCountInput ->
+            throw UnhadledError()
+        }
+
+    test("test error") {
+        tm.startWorkers(testTask1)
+
+        TaskTrackExecutionWithContextCountInput.new().let {
+            testTask1.callLater(it)
+            eventually(1000) {
+                verify(exactly = 1) { errorHandler.invoke(any(), any()) }
+            }
+
+        }
+    }
+})
\ No newline at end of file