/** * @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); }