Add infix operators for mapOk/flatMapOk (#27)

* Add infix operators for mapOk/flatMapOk * Fix infix operators and use in tests * Add test for infix operations * Pretty up >>= test * Remove temp code used to fail tests
RationalJS · Nov 29, 2018 · 44ed243 · 44ed243
1 parent c2b0899
commit 44ed243
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 0 deletions.
diff --git a/src/Future.re b/src/Future.re
@@ -99,3 +99,6 @@ let tapError = (future, f) => future |. tap(r => switch(r) {
   | Belt.Result.Error(v) => f(v) |. ignore
   | Ok(_) => ()
 });
+
+let (>>=) = flatMapOk;
+let (<$>) = mapOk;
diff --git a/tests/TestFuture.re b/tests/TestFuture.re
@@ -118,6 +118,7 @@ describe("Future", () => {
 
 
 describe("Future Belt.Result", () => {
+  let ((>>=), (<$>)) = Future.((>>=), (<$>));
 
   test("mapOk", () => {
     Belt.Result.Ok("two")
@@ -262,4 +263,42 @@ describe("Future Belt.Result", () => {
     });
   });
 
+  test("<$>", () => {
+    Future.value(Belt.Result.Ok("infix ops"))
+    <$> ((++)(" "))
+    <$> ((++)("rock!"))
+    |. Future.get(r =>
+      r
+      |. Belt.Result.getExn
+      |. equals("infix ops rock!")
+    );
+
+    Future.value(Belt.Result.Error("infix ops"))
+    <$> ((++)(" "))
+    <$> ((++)("suck!"))
+    |. Future.get(r => switch(r) {
+      | Ok(_) => raise(TestError("shouldn't be possible"))
+      | Error(e) => e |. equals("infix ops")
+    });
+  });
+
+  test(">>=", () => {
+    let appendToString = (appendedString, s) =>
+      (s ++ appendedString) |. Belt.Result.Ok |. Future.value;
+
+    Future.value(Belt.Result.Ok("infix ops"))
+    >>= appendToString(" still rock!")
+    |. Future.get(r =>
+      r
+      |. Belt.Result.getExn
+      |. equals("infix ops still rock!")
+    );
+
+    Future.value(Belt.Result.Error("infix ops"))
+    >>= appendToString(" still sucks!")
+    |. Future.get(r => switch (r) {
+      | Ok(_) => raise(TestError("shouldn't be possible"))
+      | Error(e) => e |. equals("infix ops");
+    });
+  });
 });