public function testDefaultDriverGetAllClassNames()
{
$companyDriver = $this->getMock('Doctrine\\Common\\Persistence\\Mapping\\Driver\\MappingDriver');
$defaultDriver = $this->getMock('Doctrine\\Common\\Persistence\\Mapping\\Driver\\MappingDriver');
$chain = new MappingDriverChain();
$companyDriver->expects($this->once())->method('getAllClassNames')->will($this->returnValue(array('Doctrine\\Tests\\Models\\Company\\Foo')));
$defaultDriver->expects($this->once())->method('getAllClassNames')->will($this->returnValue(array('Other\\Class')));
$chain->setDefaultDriver($defaultDriver);
$chain->addDriver($companyDriver, 'Doctrine\\Tests\\Models\\Company');
$classNames = $chain->getAllClassNames();
$this->assertEquals(array('Doctrine\\Tests\\Models\\Company\\Foo', 'Other\\Class'), $classNames);
}
public function testGatherAllClassNames()
{
$className = 'Doctrine\\Tests\\Common\\Persistence\\Mapping\\DriverChainEntity';
$classMetadata = $this->getMock('Doctrine\\Common\\Peristence\\ClassMetadata');
$chain = new MappingDriverChain();
$driver1 = $this->getMock('Doctrine\\Common\\Persistence\\Mapping\\Driver\\MappingDriver');
$driver1->expects($this->once())->method('getAllClassNames')->will($this->returnValue(array('Doctrine\\Tests\\Models\\Company\\Foo')));
$driver2 = $this->getMock('Doctrine\\Common\\Persistence\\Mapping\\Driver\\MappingDriver');
$driver2->expects($this->once())->method('getAllClassNames')->will($this->returnValue(array('Doctrine\\Tests\\ORM\\Mapping\\Bar', 'Doctrine\\Tests\\ORM\\Mapping\\Baz', 'FooBarBaz')));
$chain->addDriver($driver1, 'Doctrine\\Tests\\Models\\Company');
$chain->addDriver($driver2, 'Doctrine\\Tests\\ORM\\Mapping');
$this->assertEquals(array('Doctrine\\Tests\\Models\\Company\\Foo', 'Doctrine\\Tests\\ORM\\Mapping\\Bar', 'Doctrine\\Tests\\ORM\\Mapping\\Baz'), $chain->getAllClassNames());
}
