/** * @return StateWorkflow */ private function createValidStateWorkflow() { $stateA = new StateA(); $stateB = new StateB(); $stateC = new StateC(); $stateWorkflow = new StateWorkflow('Booking Workflow', 'key'); $stateWorkflow->addAvailableState($stateA); $stateWorkflow->addAvailableState($stateB); $stateWorkflow->addAvailableState($stateC); $stateWorkflow->setStateAsDefault($stateA->getKey()); return $stateWorkflow; }