Leveraging MonadErrorAssert in EitherTTest

jnape · jnape · commit 0c5f7e2de8d9 · 2020-02-08T16:16:13.000-06:00
diff --git a/src/test/java/com/jnape/palatable/lambda/monad/transformer/builtin/EitherTTest.java b/src/test/java/com/jnape/palatable/lambda/monad/transformer/builtin/EitherTTest.java
@@ -2,7 +2,6 @@
 
 import com.jnape.palatable.lambda.adt.Either;
 import com.jnape.palatable.lambda.adt.Unit;
-import com.jnape.palatable.lambda.functions.specialized.Pure;
 import com.jnape.palatable.lambda.functor.builtin.Identity;
 import com.jnape.palatable.lambda.io.IO;
 import com.jnape.palatable.traitor.annotations.TestTraits;
@@ -28,9 +27,9 @@
 import static com.jnape.palatable.lambda.io.IO.io;
 import static com.jnape.palatable.lambda.monad.transformer.builtin.EitherT.eitherT;
 import static com.jnape.palatable.lambda.monad.transformer.builtin.EitherT.liftEitherT;
-import static com.jnape.palatable.lambda.monad.transformer.builtin.EitherT.pureEitherT;
 import static com.jnape.palatable.traitor.framework.Subjects.subjects;
 import static org.junit.Assert.assertEquals;
+import static testsupport.assertion.MonadErrorAssert.assertLaws;
 
 @RunWith(Traits.class)
 public class EitherTTest {
@@ -78,12 +77,9 @@ public void staticPure() {
 
     @Test
     public void monadError() {
-        Pure<EitherT<Identity<?>, String, ?>> pure = pureEitherT(pureIdentity());
-
-        assertEquals(eitherT(new Identity<>(left("Hello"))),
-                     pure.<Integer, EitherT<Identity<?>, String, Integer>>apply(1).throwError("Hello"));
-        assertEquals(pure.apply(5),
-                     EitherT.<Identity<?>, String, Integer>eitherT(new Identity<>(left("Hello")))
-                             .catchError(s -> pure.apply(s.length())));
+        assertLaws(subjects(eitherT(new Identity<>(right(1))),
+                            eitherT(new Identity<>(left("")))),
+                   "bar",
+                   str -> eitherT(new Identity<>(right(str.length()))));
     }
 }
