First cut of a stateful testing framework (it's pretty hacky at the moment, but all the pieces are there)

6 files changed, 305 insertions, 0 deletions

diff --git a/src/main/java/au/id/zancanaro/javacheck/statem/Command.java b/src/main/java/au/id/zancanaro/javacheck/statem/Command.java
new file mode 100644
index 0000000..a741f0a
--- /dev/null
+++ b/src/main/java/au/id/zancanaro/javacheck/statem/Command.java
@@ -0,0 +1,23 @@
+package au.id.zancanaro.javacheck.statem;
+
+import au.id.zancanaro.javacheck.Generator;
+
+public abstract class Command<State,Args,Result> {
+    public Generator<Args> argsGenerator(State state) {
+        return Generator.pure(null);
+    }
+
+    public boolean preCondition(State state, Args args) {
+        return true;
+    }
+
+    public abstract Result runCommand(Args args);
+
+    public State nextState(State state, Args args, CommandValue<Result> result) {
+        return state;
+    }
+
+    public boolean postCondition(State oldState, State newState, Args args, Result result) {
+        return true;
+    }
+}
diff --git a/src/main/java/au/id/zancanaro/javacheck/statem/CommandList.java b/src/main/java/au/id/zancanaro/javacheck/statem/CommandList.java
new file mode 100644
index 0000000..9e8948a
--- /dev/null
+++ b/src/main/java/au/id/zancanaro/javacheck/statem/CommandList.java
@@ -0,0 +1,85 @@
+package au.id.zancanaro.javacheck.statem;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+public class CommandList<State> {
+    private final List<GeneratedCommand<State, ?, ?>> commands;
+
+    public CommandList(List<GeneratedCommand<State, ?, ?>> commands) {
+        this.commands = new ArrayList<>(commands);
+    }
+
+    public CommandResult<State> run(State initialState) {
+        Map<Integer, Object> values = new HashMap<>();
+        CommandResult<State> result = CommandResult.success(initialState);
+        for (GeneratedCommand<State, ?, ?> generated : commands) {
+            result = runRealCommand(generated, result.getState(), values);
+            if (result.isFailed()) {
+                break;
+            }
+        }
+        return result;
+    }
+
+    private static <State, Args, Result> CommandResult<State> runRealCommand(
+            GeneratedCommand<State, Args, Result> generated,
+            State state,
+            Map<Integer, Object> values) {
+        int id = generated.getId();
+        Command<State, Args, Result> command = generated.getCommand();
+        Args args = generated.getArgs();
+        try {
+            if (!command.preCondition(state, args)) {
+                return CommandResult.fail(state, new Error("Precondition failed"));
+            }
+            Result result = command.runCommand(args);
+            values.put(id, result);
+            State oldState = state;
+            state = command.nextState(state, args, new CommandValue.ConcreteValue<>(id, values));
+            if (!command.postCondition(oldState, state, args, result)) {
+                return CommandResult.fail(state, new Error("Postcondition failed"));
+            }
+            return CommandResult.success(state);
+        } catch (Throwable ex) {
+            return CommandResult.fail(state, ex);
+        }
+    }
+
+    private static <State, Args, Result> CommandResult<State> runAbstractCommand(
+            GeneratedCommand<State, Args, Result> generated,
+            State state) {
+        int id = generated.getId();
+        Command<State, Args, Result> command = generated.getCommand();
+        Args args = generated.getArgs();
+        try {
+            if (!command.preCondition(state, args)) {
+                return CommandResult.fail(state, new Error("Precondition failed"));
+            }
+            state = command.nextState(state, args, new CommandValue.AbstractValue<>(id));
+            return CommandResult.success(state);
+        } catch (Throwable ex) {
+            return CommandResult.fail(state, ex);
+        }
+    }
+
+    public boolean isValid() {
+        CommandResult<State> result = CommandResult.success(null);
+        for (GeneratedCommand<State, ?, ?> generated : commands) {
+            result = runAbstractCommand(generated, result.getState());
+            if (result.isFailed()) {
+                break;
+            }
+        }
+        return !result.isFailed();
+    }
+
+    @Override
+    public String toString() {
+        return "CommandList{" +
+                "commands=" + commands +
+                '}';
+    }
+}
diff --git a/src/main/java/au/id/zancanaro/javacheck/statem/CommandListGenerator.java b/src/main/java/au/id/zancanaro/javacheck/statem/CommandListGenerator.java
new file mode 100644
index 0000000..a0df66f
--- /dev/null
+++ b/src/main/java/au/id/zancanaro/javacheck/statem/CommandListGenerator.java
@@ -0,0 +1,50 @@
+package au.id.zancanaro.javacheck.statem;
+
+import au.id.zancanaro.javacheck.Generator;
+import au.id.zancanaro.javacheck.ShrinkTree;
+
+import java.util.Random;
+import java.util.function.Function;
+
+import static au.id.zancanaro.javacheck.Generator.pure;
+import static au.id.zancanaro.javacheck.Generators.noShrink;
+
+public class CommandListGenerator<State> implements Generator<CommandList<State>> {
+    private final Function<State, Generator<Command<State, ?, ?>>> generateCommand;
+
+    public CommandListGenerator(Function<State, Generator<Command<State, ?, ?>>> generateCommand) {
+        this.generateCommand = generateCommand;
+    }
+
+    public Generator<GeneratedCommand<State, ?, ?>> commandGenerator(int id, State state) {
+        return noShrink(generateCommand.apply(state))
+                .flatMap(command -> generateSingleCommand(id, command, state));
+    }
+
+    public <Args, Result> Generator<GeneratedCommand<State, ?, ?>> generateSingleCommand(int id, Command<State, Args, Result> command, State state) {
+        return command.argsGenerator(state).flatMap(generatedArgs ->
+                command.preCondition(state, generatedArgs) ?
+                        pure(new GeneratedCommand<>(id, command, generatedArgs)) :
+                        commandGenerator(id, state));
+    }
+
+    public <Args, Result> State nextState(int id, GeneratedCommand<State, Args, Result> generatedCommand, State state) {
+        return generatedCommand.getCommand().nextState(state, generatedCommand.getArgs(), new CommandValue.AbstractValue<>(id));
+    }
+
+    @Override
+    public ShrinkTree<CommandList<State>> generate(Random random, int size) {
+        int count = random.nextInt(size);
+        @SuppressWarnings("unchecked")
+        ShrinkTree<GeneratedCommand<State, ?, ?>>[] commandTrees = (ShrinkTree<GeneratedCommand<State, ?, ?>>[]) new ShrinkTree[count];
+        State state = null;
+        for (int i = 0; i < count; ++i) {
+            commandTrees[i] = commandGenerator(i, state).generate(random, size);
+            GeneratedCommand<State, ?, ?> generatedCommand = commandTrees[i].getValue();
+            state = nextState(i, generatedCommand, state);
+        }
+        return ShrinkTree.combine(commandTrees, ShrinkTree::removeAndPromoteChildren)
+                .map(list -> new CommandList<>(list))
+                .filter(CommandList::isValid);
+    }
+}
diff --git a/src/main/java/au/id/zancanaro/javacheck/statem/CommandResult.java b/src/main/java/au/id/zancanaro/javacheck/statem/CommandResult.java
new file mode 100644
index 0000000..dc5b085
--- /dev/null
+++ b/src/main/java/au/id/zancanaro/javacheck/statem/CommandResult.java
@@ -0,0 +1,31 @@
+package au.id.zancanaro.javacheck.statem;
+
+public class CommandResult<State> {
+    private final State state;
+    private final Throwable thrown;
+
+    private CommandResult(State state, Throwable thrown) {
+        this.state = state;
+        this.thrown = thrown;
+    }
+
+    public State getState() {
+        return state;
+    }
+
+    public boolean isFailed() {
+        return thrown != null;
+    }
+
+    public Throwable getThrown() {
+        return thrown;
+    }
+
+    public static <State> CommandResult<State> success(State state) {
+        return new CommandResult<>(state, null);
+    }
+
+    public static <State> CommandResult<State> fail(State state, Throwable ex) {
+        return new CommandResult<>(state, ex);
+    }
+}
diff --git a/src/main/java/au/id/zancanaro/javacheck/statem/CommandValue.java b/src/main/java/au/id/zancanaro/javacheck/statem/CommandValue.java
new file mode 100644
index 0000000..0a1fb61
--- /dev/null
+++ b/src/main/java/au/id/zancanaro/javacheck/statem/CommandValue.java
@@ -0,0 +1,60 @@
+package au.id.zancanaro.javacheck.statem;
+
+import java.util.Map;
+import java.util.NoSuchElementException;
+
+public abstract class CommandValue<T> {
+    private final int id;
+
+    public CommandValue(int id) {
+        this.id = id;
+    }
+
+    public abstract boolean isAbstract();
+
+    public abstract T get();
+
+    public int getId() {
+        return id;
+    }
+
+    static class AbstractValue<T> extends CommandValue<T> {
+        public AbstractValue(int id) {
+            super(id);
+        }
+
+        @Override
+        public boolean isAbstract() {
+            return true;
+        }
+
+        @Override
+        public T get() {
+            throw new NoSuchElementException("Abstract values cannot be supplied");
+        }
+    }
+
+    static class ConcreteValue<T> extends CommandValue<T> {
+        private final Map<Integer, Object> values;
+
+        public ConcreteValue(int id, Map<Integer, Object> values) {
+            super(id);
+            this.values = values;
+        }
+
+        @Override
+        public boolean isAbstract() {
+            return true;
+        }
+
+        @Override
+        @SuppressWarnings("unchecked")
+        public T get() {
+            if (values.containsKey(getId())) {
+                return (T) values.get(getId());
+            } else {
+                throw new NoSuchElementException("Concrete values cannot be supplied prior to being calculated");
+            }
+        }
+    }
+}
diff --git a/src/main/java/au/id/zancanaro/javacheck/statem/GeneratedCommand.java b/src/main/java/au/id/zancanaro/javacheck/statem/GeneratedCommand.java
new file mode 100644
index 0000000..5e576b4
--- /dev/null
+++ b/src/main/java/au/id/zancanaro/javacheck/statem/GeneratedCommand.java
@@ -0,0 +1,56 @@
+package au.id.zancanaro.javacheck.statem;
+
+public class GeneratedCommand<State, Args, Result> {
+    private final int id;
+    private final Command<State, Args, Result> command;
+    private final Args args;
+
+    public GeneratedCommand(int id, Command<State, Args, Result> command, Args args) {
+        this.id = id;
+        this.command = command;
+        this.args = args;
+    }
+
+    public int getId() {
+        return id;
+    }
+
+    public Command<State, Args, Result> getCommand() {
+        return command;
+    }
+
+    public Args getArgs() {
+        return args;
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+
+        GeneratedCommand that = (GeneratedCommand) o;
+
+        if (args != null ? !args.equals(that.args) : that.args != null)
+            return false;
+        if (command != null ? !command.equals(that.command) : that.command != null)
+            return false;
+
+        return true;
+    }
+
+    @Override
+    public int hashCode() {
+        int result = command != null ? command.hashCode() : 0;
+        result = 31 * result + (args != null ? args.hashCode() : 0);
+        return result;
+    }
+
+    @Override
+    public String toString() {
+        return "GeneratedCommand{" +
+                "id=" + id +
+                ", command=" + command +
+                ", args=" + args +
+                '}';
+    }
+}