public function testGetWorkflowStep() { $entity = new Task(); $workflowStep = $this->getMock('Oro\\Bundle\\WorkflowBundle\\Entity\\WorkflowStep'); $this->assertNull($entity->getWorkflowStep()); $entity->setWorkflowStep($workflowStep); $this->assertEquals($workflowStep, $entity->getWorkflowStep()); }