/** * Run tests to ensure the output is correct for add/remove/getLateShift(s) * * @return void */ public function testLateShift() { $lateShift = $this->getMock('\\OpenSkedge\\AppBundle\\Entity\\LateShift'); $schedule = new Schedule(); $schedule->addLateShift($lateShift); $lateShifts = $schedule->getLateShifts(); $this->assertInstanceOf('\\OpenSkedge\\AppBundle\\Entity\\LateShift', $lateShifts[0]); $schedule->removeLateShift($lateShift); $this->assertTrue($schedule->getLateShifts()->isEmpty()); }