operator invoke

Author: mtumilowicz

src/main/kotlin/dsl/FsmSpec.kt

@@ -17,4 +17,9 @@ class FsmSpec(
             state = State(initialState)
         )
     }
+
+    operator fun invoke(block: FsmSpec.() -> Unit): Fsm {
+        block()
+        return build()
+    }
 }

src/test/kotlin/dsl/FsmTest.kt

@@ -18,4 +18,19 @@ class FsmTest : BehaviorSpec({
             }
         }
     }
+
+    given("2222") {
+        `when`("I sit on it") {
+            val fsmSpec = FsmSpec(initialState = "locked")
+            val fsm = fsmSpec {
+                add { this on "coin" from "locked" into "unlocked" }
+                add { this on "pass" from "unlocked" into "locked" }
+            }
+
+            then("I should be able to fly") {
+                fsm.initial shouldBe State("locked")
+                fsm.transitions shouldHave haveSize(2)
+            }
+        }
+    }
 })