Ability to start or stop recording data (#5)

fmizzell · Mar 12, 2020 · d35dee1 · d35dee1
1 parent 8c67f18
commit d35dee1
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/src/StateMachine/Execution.php b/src/StateMachine/Execution.php
@@ -7,8 +7,13 @@ trait Execution
 {
     public $execution;
 
+    private $recordsData = TRUE;
+
     private function recordStateExecution($next_states)
     {
+        if (!$this->recordsData) {
+            return;
+        }
         if (json_encode($this->currentStates) != json_encode($next_states)) {
             array_push($this->execution, $next_states);
             array_push($this->execution, "");
@@ -17,8 +22,20 @@ private function recordStateExecution($next_states)
 
     private function recordInputExecution($input)
     {
+        if (!$this->recordsData) {
+            return;
+        }
         $inputs = array_pop($this->execution);
         $inputs .= $input;
         array_push($this->execution, $inputs);
     }
+
+    public function startRecording() {
+        $this->recordsData = TRUE;
+    }
+
+    public function stopRecording() {
+        $this->recordsData = FALSE;
+    }
+
 }
diff --git a/test/NonDeterministicStateMachineTest.php b/test/NonDeterministicStateMachineTest.php
@@ -54,4 +54,13 @@ public function testBadPath()
         $machine = $this->stateMachine;
         \Maquina\Feeder::feed("acdd", $machine);
     }
+
+    public function testStoppingAndStartingExecutionRecording()
+    {
+        $machine = $this->stateMachine;
+        $machine->stopRecording();
+        \Maquina\Feeder::feed("abd", $machine);
+        $machine->startRecording();
+        $this->assertTrue($machine->isCurrentlyAtAnEndState());
+    }
 }