/** * @covers OperationalCountryService::getLocationsMappedToOperationalCountry */ public function testGetLocationsMappedToOperationalCountry_Success() { $locationList = array(); $locationList[] = new Location(); $locationList[] = new Location(); $locationList[] = new Location(); $sriLanka = new Country(); $sriLanka->setName('Sri Lanka'); $operationalCountry = new OperationalCountry(); $operationalCountry->setId(1); $operationalCountry->setCountry($sriLanka); $operationalCountry->setCountryCode('LK'); $operationalCountryDaoMock = $this->getMock('OperationalCountryDao', array('getLocationsMappedToOperationalCountry')); $operationalCountryDaoMock->expects($this->once())->method('getLocationsMappedToOperationalCountry')->will($this->returnValue($locationList)); $result = $this->service->setOperationalCountryDao($operationalCountryDaoMock); $result = $this->service->getLocationsMappedToOperationalCountry($operationalCountry); $this->assertEquals($locationList, $result); }