From f11e95497c02ad5229f305a8c99d5a3238710828 Mon Sep 17 00:00:00 2001
From: "Jonathan.Wage" <Jonathan.Wage@625475ce-881a-0410-a577-b389adb331d8>
Date: Thu, 15 Nov 2007 19:46:17 +0000
Subject: [PATCH] Fixed issue with data dumping. Added exception throwing when
 you use data key that does not exist.

---
 lib/Doctrine/Data/Export.php |  2 +-
 lib/Doctrine/Data/Import.php | 17 +++++++++++++----
 2 files changed, 14 insertions(+), 5 deletions(-)

diff --git a/lib/Doctrine/Data/Export.php b/lib/Doctrine/Data/Export.php
index 7cb6b9185..5079666e6 100644
--- a/lib/Doctrine/Data/Export.php
+++ b/lib/Doctrine/Data/Export.php
@@ -174,7 +174,7 @@ class Doctrine_Data_Export extends Doctrine_Data
                         
                         $relationValue = $relationClassName . '_' . $value;
                         
-                        $preparedData[$className][$recordKey][$relationClassName] = $relationValue;
+                        $preparedData[$className][$recordKey][$relationAlias] = $relationValue;
                     } else {                        
                         $preparedData[$className][$recordKey][$key] = $value;
                     }
diff --git a/lib/Doctrine/Data/Import.php b/lib/Doctrine/Data/Import.php
index 6a3d24682..08c98dfce 100644
--- a/lib/Doctrine/Data/Import.php
+++ b/lib/Doctrine/Data/Import.php
@@ -114,6 +114,15 @@ class Doctrine_Data_Import extends Doctrine_Data
         }
     }
     
+    protected function _getImportedObject($rowKey)
+    {
+        if (isset($this->_importedObjects[$rowKey])) {
+            return $this->_importedObjects[$rowKey];
+        } else {
+            throw new Doctrine_Data_Exception('Invalid row key specified: ' . $rowKey);
+        }
+    }
+    
     protected function _processRow($rowKey, $row)
     {
         $obj = $this->_importedObjects[$rowKey];
@@ -130,18 +139,18 @@ class Doctrine_Data_Import extends Doctrine_Data
                         foreach ($value as $link) {
                             
                             if ($obj->getTable()->getRelation($key)->getType() === Doctrine_Relation::ONE) {
-                                $obj->set($key, $this->_importedObjects[$link]);
+                                $obj->set($key, $this->_getImportedObject($link));
                             } else if ($obj->getTable()->getRelation($key)->getType() === Doctrine_Relation::MANY) {
                                 $relation = $obj->$key;
                                 
-                                $relation[] = $this->_importedObjects[$link];
+                                $relation[] = $this->_getImportedObject($link);
                             }
                         }
                     } else {
                         $obj->$key->fromArray($value);
                     }
-                } else if (isset($this->_importedObjects[$value])) {
-                    $obj->set($key, $this->_importedObjects[$value]);
+                } else {
+                    $obj->set($key, $this->_getImportedObject($value));
                 }
             }
         }