2023-04-22 16:33:28 +12:00
|
|
|
defmodule AshStateMachineTest do
|
2023-04-22 05:43:36 +12:00
|
|
|
use ExUnit.Case
|
2023-04-22 16:33:28 +12:00
|
|
|
doctest AshStateMachine
|
2023-04-22 05:43:36 +12:00
|
|
|
|
2023-04-23 13:38:33 +12:00
|
|
|
defmodule Order do
|
|
|
|
# leaving out data layer configuration for brevity
|
|
|
|
use Ash.Resource,
|
|
|
|
extensions: [AshStateMachine]
|
|
|
|
|
|
|
|
state_machine do
|
|
|
|
initial_states [:pending]
|
|
|
|
default_initial_state :pending
|
|
|
|
|
|
|
|
transitions do
|
|
|
|
transition :confirm, from: :pending, to: :confirmed
|
|
|
|
transition :begin_delivery, from: :confirmed, to: :on_its_way
|
|
|
|
transition :package_arrived, from: :on_its_way, to: :arrived
|
|
|
|
transition :error, from: [:pending, :confirmed, :on_its_way], to: :error
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
actions do
|
|
|
|
# create sets the st
|
|
|
|
defaults [:create, :read]
|
|
|
|
|
|
|
|
update :confirm do
|
|
|
|
# accept [...] you can change other attributes
|
|
|
|
# or do anything else an action can normally do
|
|
|
|
# this transition will be validated according to
|
|
|
|
# the state machine rules above
|
|
|
|
change transition_state(:confirmed)
|
|
|
|
end
|
|
|
|
|
|
|
|
update :begin_delivery do
|
|
|
|
# accept [...]
|
|
|
|
change transition_state(:on_its_way)
|
|
|
|
end
|
|
|
|
|
|
|
|
update :package_arrived do
|
|
|
|
# accept [...]
|
|
|
|
change transition_state(:arrived)
|
|
|
|
end
|
|
|
|
|
|
|
|
update :error do
|
|
|
|
accept [:error_state, :error]
|
|
|
|
change transition_state(:error)
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
changes do
|
|
|
|
# any failures should be captured and transitioned to the error state
|
|
|
|
change after_transaction(fn
|
|
|
|
changeset, {:ok, result} ->
|
|
|
|
{:ok, result}
|
|
|
|
|
|
|
|
changeset, {:error, error} ->
|
|
|
|
message = Exception.message(error)
|
|
|
|
|
|
|
|
changeset.data
|
|
|
|
|> Ash.Changeset.for_update(:error, %{
|
|
|
|
message: message,
|
|
|
|
error_state: changeset.data.state
|
|
|
|
})
|
|
|
|
end),
|
|
|
|
on: [:update]
|
|
|
|
end
|
|
|
|
|
|
|
|
attributes do
|
|
|
|
uuid_primary_key :id
|
|
|
|
# ...attributes like address/delivery options would go here
|
|
|
|
attribute :error, :string
|
|
|
|
attribute :error_state, :string
|
|
|
|
# :state attribute is added for you by `state_machine`
|
|
|
|
# however, you can add it yourself, and you will be guided by
|
|
|
|
# compile errors on what states need to be allowed by your type.
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2023-04-22 07:25:39 +12:00
|
|
|
defmodule ThreeStates do
|
2023-04-22 05:43:36 +12:00
|
|
|
use Ash.Resource,
|
|
|
|
data_layer: Ash.DataLayer.Ets,
|
2023-04-22 16:33:28 +12:00
|
|
|
extensions: [AshStateMachine]
|
2023-04-22 05:43:36 +12:00
|
|
|
|
2023-04-22 16:33:28 +12:00
|
|
|
state_machine do
|
2023-04-23 12:39:29 +12:00
|
|
|
initial_states [:pending]
|
2023-04-22 07:25:39 +12:00
|
|
|
default_initial_state :pending
|
2023-04-22 05:43:36 +12:00
|
|
|
|
2023-04-22 16:33:28 +12:00
|
|
|
transitions do
|
|
|
|
transition(:begin, from: :pending, to: :executing)
|
|
|
|
transition(:complete, from: :executing, to: :complete)
|
2023-04-22 07:25:39 +12:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
actions do
|
|
|
|
defaults [:create]
|
|
|
|
|
|
|
|
update :begin do
|
|
|
|
change transition_state(:executing)
|
|
|
|
end
|
|
|
|
|
|
|
|
update :complete do
|
|
|
|
change transition_state(:complete)
|
2023-04-22 05:43:36 +12:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
ets do
|
|
|
|
private? true
|
|
|
|
end
|
|
|
|
|
|
|
|
attributes do
|
|
|
|
uuid_primary_key :id
|
|
|
|
end
|
2023-04-22 07:25:39 +12:00
|
|
|
|
|
|
|
code_interface do
|
2023-04-22 16:33:28 +12:00
|
|
|
define_for AshStateMachineTest.Api
|
2023-04-22 07:25:39 +12:00
|
|
|
define :create
|
|
|
|
define :begin
|
|
|
|
define :complete
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
defmodule Api do
|
|
|
|
use Ash.Api
|
|
|
|
|
|
|
|
resources do
|
|
|
|
allow_unregistered? true
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
describe "transformers" do
|
|
|
|
test "infers all states" do
|
2023-04-22 16:33:28 +12:00
|
|
|
assert Enum.sort(AshStateMachine.Info.state_machine_all_states(ThreeStates)) ==
|
2023-04-22 07:25:39 +12:00
|
|
|
Enum.sort([:executing, :pending, :complete])
|
|
|
|
end
|
2023-04-22 05:43:36 +12:00
|
|
|
end
|
|
|
|
|
2023-04-22 07:25:39 +12:00
|
|
|
describe "behavior" do
|
|
|
|
test "begins in the appropriate state" do
|
|
|
|
assert ThreeStates.create!().state == :pending
|
|
|
|
end
|
|
|
|
|
|
|
|
test "it transitions to the appropriate state" do
|
2023-04-22 16:33:28 +12:00
|
|
|
state_machine = ThreeStates.create!()
|
2023-04-22 07:25:39 +12:00
|
|
|
|
2023-04-22 16:33:28 +12:00
|
|
|
assert ThreeStates.begin!(state_machine).state == :executing
|
2023-04-22 07:25:39 +12:00
|
|
|
end
|
|
|
|
|
|
|
|
test "it transitions again to the appropriate state" do
|
2023-04-22 16:33:28 +12:00
|
|
|
state_machine = ThreeStates.create!() |> ThreeStates.begin!()
|
2023-04-22 07:25:39 +12:00
|
|
|
|
2023-04-22 16:33:28 +12:00
|
|
|
assert ThreeStates.complete!(state_machine).state == :complete
|
2023-04-22 07:25:39 +12:00
|
|
|
end
|
2023-04-22 05:43:36 +12:00
|
|
|
end
|
2023-04-23 10:50:23 +12:00
|
|
|
|
|
|
|
describe "charts" do
|
|
|
|
test "it generates the appropriate chart" do
|
2023-04-23 13:38:33 +12:00
|
|
|
AshStateMachine.Charts.mermaid_flowchart(Order) |> IO.puts()
|
|
|
|
|
2023-04-23 10:50:23 +12:00
|
|
|
assert AshStateMachine.Charts.mermaid_flowchart(ThreeStates) ==
|
|
|
|
"""
|
|
|
|
flowchart TD
|
|
|
|
pending --> |begin| executing
|
|
|
|
executing --> |complete| complete
|
|
|
|
"""
|
|
|
|
|> String.trim_trailing()
|
|
|
|
end
|
|
|
|
end
|
2023-04-22 05:43:36 +12:00
|
|
|
end
|