From 66d6136a929b13882711dcee8c69e616185552e5 Mon Sep 17 00:00:00 2001
From: zYne <zYne@625475ce-881a-0410-a577-b389adb331d8>
Date: Sat, 1 Sep 2007 22:21:17 +0000
Subject: [PATCH] some tests for MAP keyword

---
 tests/Query/JoinTestCase.php | 43 +++++++++++++++++++++++++++++++-----
 1 file changed, 38 insertions(+), 5 deletions(-)

diff --git a/tests/Query/JoinTestCase.php b/tests/Query/JoinTestCase.php
index 6461b3bd3..56d963082 100644
--- a/tests/Query/JoinTestCase.php
+++ b/tests/Query/JoinTestCase.php
@@ -34,9 +34,9 @@ class Doctrine_Query_Join_TestCase extends Doctrine_UnitTestCase
 {
     public function prepareTables()
     {
-        $this->tables = array('Record_Country', 'Record_City', 'Record_District', 'Entity', 
+        $this->tables = array('Record_Country', 'Record_City', 'Record_District', 'Entity',
                               'User', 'Group', 'Email', 'Phonenumber', 'Groupuser', 'Account');
-        
+
         parent::prepareTables();
     }
     public function prepareData()
@@ -61,7 +61,7 @@ class Doctrine_Query_Join_TestCase extends Doctrine_UnitTestCase
 
         $this->connection->clear();
     }
-    
+
     public function testQuerySupportsCustomJoins()
     {
     	$q = new Doctrine_Query();
@@ -147,7 +147,7 @@ class Doctrine_Query_Join_TestCase extends Doctrine_UnitTestCase
         $q = new Doctrine_Query();
         $q->select('u.id, g.id, e.id')->from('User u')
           ->leftJoin('u.Group g')->leftJoin('g.Email e');
-        
+
         $this->assertEqual($q->getQuery(), 'SELECT e.id AS e__id, e2.id AS e2__id, e3.id AS e3__id FROM entity e LEFT JOIN groupuser g ON e.id = g.user_id LEFT JOIN entity e2 ON e2.id = g.group_id LEFT JOIN email e3 ON e2.email_id = e3.id WHERE (e.type = 0 AND (e2.type = 1 OR e2.type IS NULL))');
         try {
             $q->execute();
@@ -162,7 +162,7 @@ class Doctrine_Query_Join_TestCase extends Doctrine_UnitTestCase
         $q = new Doctrine_Query();
         $q->select('u.id, g.id, e.id')->from('Group g')
           ->leftJoin('g.User u')->leftJoin('u.Account a');
-        
+
         $this->assertEqual($q->getQuery(), 'SELECT e.id AS e__id, e2.id AS e2__id FROM entity e LEFT JOIN groupuser g ON e.id = g.group_id LEFT JOIN entity e2 ON e2.id = g.user_id LEFT JOIN account a ON e2.id = a.entity_id WHERE (e.type = 1 AND (e2.type = 0 OR e2.type IS NULL))');
         try {
             $q->execute();
@@ -171,4 +171,37 @@ class Doctrine_Query_Join_TestCase extends Doctrine_UnitTestCase
             $this->fail();
         }
     }
+
+    public function testMapKeywordForQueryWithOneComponent()
+    {
+        $q = new Doctrine_Query();
+        $coll = $q->from('Record_City c MAP c.name')->fetchArray();
+        
+        $this->assertTrue(isset($coll['City 1']));
+        $this->assertTrue(isset($coll['City 2']));
+        $this->assertTrue(isset($coll['City 3']));
+    }
+
+    public function testMapKeywordSupportsJoins()
+    {
+        $q = new Doctrine_Query();
+        $country = $q->from('Record_Country c LEFT JOIN c.City c2 MAP c2.name')->fetchOne();
+        $coll = $country->City;
+
+        $this->assertTrue(isset($coll['City 1']));
+        $this->assertTrue(isset($coll['City 2']));
+        $this->assertTrue(isset($coll['City 3']));
+    }
+    
+    public function testMapKeywordThrowsExceptionOnNonExistentColumn()
+    {
+    	try {
+            $q = new Doctrine_Query();
+            $country = $q->from('Record_Country c LEFT JOIN c.City c2 MAP c2.unknown')->fetchOne();
+        
+            $this->fail();
+        } catch (Doctrine_Query_Exception $e) {
+            $this->pass();
+        }
+    }
 }