rename @__struct__ to @___struct___ to use with typespec

lib/machinist.ex

@@ -12,7 +12,7 @@ defmodule Machinist do
  @doc false
  defmacro __using__(_) do
  quote do
- @__attr__ :state
+ @___attr___ :state
 
  @behaviour Machinist.Transition
 
@@ -34,7 +34,7 @@ defmodule Machinist do
  """
  defmacro transitions(do: block) do
  quote do
- @__struct__ __MODULE__
+ @___struct___ __MODULE__
 
  unquote(block)
  end
@@ -84,16 +84,16 @@ defmodule Machinist do
 
  defmacro transitions([attr: attr], do: block) do
  quote do
- @__attr__ unquote(attr)
- @__struct__ __MODULE__
+ @___attr___ unquote(attr)
+ @___struct___ __MODULE__
 
  unquote(block)
  end
  end
 
  defmacro transitions(struct, do: block) do
  quote do
- @__struct__ unquote(struct)
+ @___struct___ unquote(struct)
 
  unquote(block)
  end
@@ -109,8 +109,8 @@ defmodule Machinist do
  """
  defmacro transitions(struct, [attr: attr], do: block) do
  quote do
- @__attr__ unquote(attr)
- @__struct__ unquote(struct)
+ @___attr___ unquote(attr)
+ @___struct___ unquote(struct)
 
  unquote(block)
  end
@@ -226,10 +226,10 @@ defmodule Machinist do
  defp define_transition(state, to: new_state, event: event) do
  quote do
  @impl true
- def transit(%@__struct__{@__attr__ => unquote(state)} = resource, event: unquote(event)) do
+ def transit(%@___struct___{@___attr___ => unquote(state)} = resource, event: unquote(event)) do
  case __set_new_state__(resource, unquote(new_state)) do
  {:error, _msg} = term -> term
- new_state -> {:ok, Map.put(resource, @__attr__, new_state)}
+ new_state -> {:ok, Map.put(resource, @___attr___, new_state)}
  end
  end
  end

test/machinist_test.exs

@@ -356,4 +356,29 @@ defmodule MachinistTest do
  end
  end
  end
+
+ describe "an example w/ typespec" do
+ defmodule Example11 do
+ defstruct state: 1
+
+ @type t :: %__MODULE__{
+ state: integer()
+ }
+
+ use Machinist
+
+ transitions do
+ from 1, to: 2, event: "next"
+ from 2, to: 3, event: "next"
+ end
+ end
+
+ test "all transitions" do
+ assert {:ok, %Example11{state: 2} = step_2} = Example11.transit(%Example11{}, event: "next")
+ assert {:ok, %Example11{state: 3} = step_3} = Example11.transit(step_2, event: "next")
+
+ assert {:error, :not_allowed} = Example11.transit(step_3, event: "next")
+ end
+ end
+
 end