fn: fix UnwrapOrFail semantics

This commit is contained in:
Keagan McClelland committed 2024-11-06 15:55:18 -07:00
1 parent 4778b146cc
commit 7a7f3bdb2c
4 files changed
+35 -14

No files matched your search

+7 -8
View File
@@ -1,6 +1,10 @@
package fn package fn
import "testing" import (
"testing"
"github.com/stretchr/testify/require"
)
// Option[A] represents a value which may or may not be there. This is very // Option[A] represents a value which may or may not be there. This is very
// often preferable to nil-able pointers. // often preferable to nil-able pointers.
@@ -61,14 +65,9 @@ func (o Option[A]) UnwrapOrFunc(f func() A) A {
func (o Option[A]) UnwrapOrFail(t *testing.T) A { func (o Option[A]) UnwrapOrFail(t *testing.T) A {
t.Helper() t.Helper()
if o.isSome { require.True(t, o.isSome, "Option[%T] was None()", o.some)
return o.some
}
t.Fatalf("Option[%T] was None()", o.some) return o.some
var zero A
return zero
} }
// UnwrapOrErr is used to extract a value from an option, if the option is // UnwrapOrErr is used to extract a value from an option, if the option is
+11
View File
@@ -0,0 +1,11 @@
package fn
import (
"testing"
"github.com/stretchr/testify/require"
)
func TestOptionUnwrapOrFail(t *testing.T) {
require.Equal(t, Some(1).UnwrapOrFail(t), 1)
}
+6 -6
View File
@@ -3,6 +3,8 @@ package fn
import ( import (
"fmt" "fmt"
"testing" "testing"
"github.com/stretchr/testify/require"
) )
// Result represents a value that can either be a success (T) or an error. // Result represents a value that can either be a success (T) or an error.
@@ -114,13 +116,11 @@ func (r Result[T]) UnwrapOrElse(f func() T) T {
func (r Result[T]) UnwrapOrFail(t *testing.T) T { func (r Result[T]) UnwrapOrFail(t *testing.T) T {
t.Helper() t.Helper()
if r.IsErr() { require.True(
t.Fatalf("Result[%T] contained error: %v", r.left, r.right) t, r.IsOk(), "Result[%T] contained error: %v", r.left, r.right,
} )
var zero T return r.left
return zero
} }
// FlatMap applies a function that returns a Result to the success value if it // FlatMap applies a function that returns a Result to the success value if it
+11
View File
@@ -0,0 +1,11 @@
package fn
import (
"testing"
"github.com/stretchr/testify/require"
)
func TestResultUnwrapOrFail(t *testing.T) {
require.Equal(t, Ok(1).UnwrapOrFail(t), 1)
}