Add more tests

Author: ehmicky

src/return.test.ts

@@ -22,3 +22,37 @@ test('Can return a function from another one', (t) => {
     noop,
   )
 })
+
+test('options.mapReturnValues() can return any type', (t) => {
+  t.is(
+    switchFunctional(0, {
+      mapReturnValues: (condition: number) => -condition,
+    })
+      .case(true, 1)
+      .default(2),
+    -1,
+  )
+})
+
+test('options.mapReturnValues() can return a function', (t) => {
+  t.is(
+    switchFunctional(1, {
+      mapReturnValues: (condition: number) => (value: number) =>
+        condition + value,
+    })
+      .case(true, 2)
+      .default(0),
+    3,
+  )
+})
+
+test('options.mapReturnValues() can take variadic parameters', (t) => {
+  t.is(
+    switchFunctional(0, {
+      mapReturnValues: (one: number, two: number) => one + two,
+    })
+      .case(true, 1, 2)
+      .default(0, 0),
+    3,
+  )
+})