summaryrefslogtreecommitdiff
path: root/src/main/java/au/id/zancanaro
diff options
context:
space:
mode:
Diffstat (limited to 'src/main/java/au/id/zancanaro')
-rw-r--r--src/main/java/au/id/zancanaro/javacheck/statem/Command.java23
-rw-r--r--src/main/java/au/id/zancanaro/javacheck/statem/CommandList.java85
-rw-r--r--src/main/java/au/id/zancanaro/javacheck/statem/CommandListGenerator.java50
-rw-r--r--src/main/java/au/id/zancanaro/javacheck/statem/CommandResult.java31
-rw-r--r--src/main/java/au/id/zancanaro/javacheck/statem/CommandValue.java60
-rw-r--r--src/main/java/au/id/zancanaro/javacheck/statem/GeneratedCommand.java56
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 +
+ '}';
+ }
+}