kotest shouldBeSuccess() returning value

fork-handles · Sep 22, 2023 · f6c5622 · f6c5622
1 parent a7a2edb
commit f6c5622
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/result4k/kotest/src/main/kotlin/dev/forkhandles/result4k/kotest/matchers.kt b/result4k/kotest/src/main/kotlin/dev/forkhandles/result4k/kotest/matchers.kt
@@ -19,16 +19,16 @@ fun beSuccess(): Matcher<Result<*, *>> = object : Matcher<Result<*, *>> {
 }
 
 @OptIn(ExperimentalContracts::class)
-fun <T> Result<T, *>.shouldBeSuccess() {
+fun <T> Result<T, *>.shouldBeSuccess(): T {
     contract {
-        returns() implies (this@shouldBeSuccess is Success<*>)
+        returns() implies (this@shouldBeSuccess is Success<T>)
     }
     this should beSuccess()
+    return (this as Success).value
 }
 
 infix fun <T> Result<T, *>.shouldBeSuccess(block: (T) -> Unit) {
-    this.shouldBeSuccess()
-    block((this as Success<T>).value)
+    block(this.shouldBeSuccess())
 }
 
 infix fun <T> Result<T, *>.shouldBeSuccess(value: T) = this should beSuccess(value)