Initial commit: only works for plain int typed arguments

1 files changed, 19 insertions, 0 deletions

diff --git a/src/test/java/au/id/zancanaro/SimpleGenerator.java b/src/test/java/au/id/zancanaro/SimpleGenerator.java
new file mode 100644
index 0000000..1b1d570
--- /dev/null
+++ b/src/test/java/au/id/zancanaro/SimpleGenerator.java
@@ -0,0 +1,19 @@
+package au.id.zancanaro;
+
+import org.junit.Assert;
+import org.junit.Test;
+
+import java.util.Random;
+
+public class SimpleGenerator {
+    @Test
+    public void testSimpleGeneratorFlatmap() throws Exception {
+        Generator<Integer> generator = Generator
+                .pure(10)
+                .flatMap((value) ->
+                        Generator.pure(value + 10));
+        Assert.assertEquals(
+                (Object) 20,
+                generator.generate(null, 0).getValue());
+    }
+}