From b9ec48d68e21c8b9041de77573022e3840fa859e Mon Sep 17 00:00:00 2001
From: romanb <romanb@625475ce-881a-0410-a577-b389adb331d8>
Date: Sat, 3 Oct 2009 22:00:01 +0000
Subject: [PATCH] [2.0][DDC-31] Fixed.

---
 lib/Doctrine/ORM/Mapping/ClassMetadata.php | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/lib/Doctrine/ORM/Mapping/ClassMetadata.php b/lib/Doctrine/ORM/Mapping/ClassMetadata.php
index f25c55b82..4fc92b84f 100644
--- a/lib/Doctrine/ORM/Mapping/ClassMetadata.php
+++ b/lib/Doctrine/ORM/Mapping/ClassMetadata.php
@@ -1566,14 +1566,14 @@ final class ClassMetadata
      */
     public function setDiscriminatorColumn($columnDef)
     {
-        $this->discriminatorColumn = $columnDef;
-        
-        if ( ! isset($columnDef['fieldName'])) {
-            $this->discriminatorColumn['fieldName'] = $columnDef['name'];
-        }
-        
-        if ( ! isset($columnDef['name'])) {
-            $this->discriminatorColumn['name'] = $this->discriminatorColumn['fieldName'];
+        if ($columnDef !== null) {
+            if ( ! isset($columnDef['name'])) {
+                throw new MappingException("'name' attribute is mandatory for discriminator columns.");
+            }
+            if ( ! isset($columnDef['fieldName'])) {
+                $columnDef['fieldName'] = $columnDef['name'];
+            }
+            $this->discriminatorColumn = $columnDef;
         }
     }
 
@@ -1883,6 +1883,7 @@ final class ClassMetadata
             'customRepositoryClassName',
             'discriminatorColumn',
             'discriminatorValue',
+            'discriminatorMap',
             'fieldMappings',
             'fieldNames',
             'generatorType',
@@ -1900,6 +1901,7 @@ final class ClassMetadata
             'namespace',
             'parentClasses',
             'primaryTable',
+            'resultColumnNames',
             'rootEntityName',
             'sequenceGeneratorDefinition',
             'subClasses',