Add Visitor tests

gliptak · gliptak · commit fd9b337abe09 · 2018-08-18T18:34:21.000-04:00
Signed-off-by: Gábor Lipták &lt;gliptak@gmail.com&gt;
diff --git a/core/src/test/java/fj/function/VisitorTest.java b/core/src/test/java/fj/function/VisitorTest.java
@@ -0,0 +1,80 @@
+package fj.function;
+
+import fj.Equal;
+import fj.F;
+import fj.P;
+import fj.P1;
+import fj.data.List;
+import org.junit.Test;
+
+import static fj.data.Option.none;
+import static fj.data.Option.some;
+import static fj.function.Visitor.*;
+import static org.hamcrest.core.Is.is;
+import static org.junit.Assert.assertThat;
+
+public class VisitorTest {
+    @Test
+    public void testFindFirst() {
+        assertThat(findFirst(List.list(none(), some(1), none()), () -> -1), is(1));
+    }
+
+    @Test
+    public void testFindFirstDef() {
+        assertThat(findFirst(List.list(none(), none(), none()), () -> -1), is(-1));
+    }
+
+    @Test
+    public void testNullableFindFirst() {
+        assertThat(nullablefindFirst(List.list(null, 1, null), () -> -1), is(1));
+    }
+
+    @Test
+    public void testNullableFindFirstDef() {
+        assertThat(nullablefindFirst(List.list(null, null, null), () -> -1), is(-1));
+    }
+
+    @Test
+    public void testVisitor() {
+        assertThat(visitor(List.list(i -> some(2 * i)), () -> -1, 10), is(20));
+    }
+
+    @Test
+    public void testVisitorDef() {
+        assertThat(visitor(List.list(i -> none()), () -> "foo", 10), is("foo"));
+    }
+
+    @Test
+    public void testNullableVisitor() {
+        assertThat(nullableVisitor(List.list(i -> 2 * i), () -> -1, 10), is(20));
+    }
+
+    @Test
+    public void testNullableVisitorDef() {
+        assertThat(nullableVisitor(List.list(i -> null), () -> "foo", 10), is("foo"));
+    }
+
+    @Test
+    public void testAssociation() {
+        final F<String, F<Integer, String>> a = association(List.list(P.p(1, "one"), P.p(2, "two")), Equal.intEqual);
+        assertThat(a.f("foo").f(2), is("two"));
+    }
+
+    @Test
+    public void testAssociationDef() {
+        final F<String, F<Integer, String>> a = association(List.list(P.p(1, "one"), P.p(2, "two")), Equal.intEqual);
+        assertThat(a.f("foo").f(3), is("foo"));
+    }
+
+    @Test
+    public void testAssociationLazy() {
+        final F<P1<String>, F<Integer, String>> a = associationLazy(List.list(P.p(1, "one"), P.p(2, "two")), Equal.intEqual);
+        assertThat(a.f(P.p("foo")).f(2), is("two"));
+    }
+
+    @Test
+    public void testAssociationLazyDef() {
+        final F<P1<String>, F<Integer, String>> a = associationLazy(List.list(P.p(1, "one"), P.p(2, "two")), Equal.intEqual);
+        assertThat(a.f(P.p("foo")).f(3), is("foo"));
+    }
+}
