/**
* @param string $name
* @param string|null $class
*
* @return Inheritance
*/
public function map($name, $class = null)
{
if (is_array($name)) {
foreach ($name as $name => $class) {
$this->map($name, $class);
}
return $this;
}
$this->builder->addDiscriminatorMapClass($name, $class);
return $this;
}
public function testAddDiscriminatorMapClass()
{
$this->assertIsFluent($this->builder->addDiscriminatorMapClass('test', 'Doctrine\\Tests\\Models\\CMS\\CmsUser'));
$this->assertIsFluent($this->builder->addDiscriminatorMapClass('test2', 'Doctrine\\Tests\\Models\\CMS\\CmsGroup'));
$this->assertEquals(array('test' => 'Doctrine\\Tests\\Models\\CMS\\CmsUser', 'test2' => 'Doctrine\\Tests\\Models\\CMS\\CmsGroup'), $this->cm->discriminatorMap);
$this->assertEquals('test', $this->cm->discriminatorValue);
}
