/** * Run tests to ensure the output is correct for add/remove/getShift(s) * * @return void */ public function testShift() { $shift = $this->getMock('\\OpenSkedge\\AppBundle\\Entity\\Shift'); $schedule = new Schedule(); $schedule->addShift($shift); $shifts = $schedule->getShifts(); $this->assertInstanceOf('\\OpenSkedge\\AppBundle\\Entity\\Shift', $shifts[0]); $schedule->removeShift($shift); $this->assertTrue($schedule->getShifts()->isEmpty()); }