From aebb08bdac0ff5ccc147dac5699b9490ba3b590d Mon Sep 17 00:00:00 2001
From: adamgsoc2013 <adamgsoc2013@gmail.com>
Date: Sat, 20 Jul 2013 20:40:36 +0800
Subject: [PATCH] add test case for PMA_relation

---
 test/libraries/PMA_relation_test.php | 84 +++++++++++++++++++++++++++-
 1 file changed, 81 insertions(+), 3 deletions(-)

diff --git a/test/libraries/PMA_relation_test.php b/test/libraries/PMA_relation_test.php
index 0e6dc31cc8..d1272cd975 100644
--- a/test/libraries/PMA_relation_test.php
+++ b/test/libraries/PMA_relation_test.php
@@ -25,9 +25,11 @@ class PMA_Relation_Test extends PHPUnit_Framework_TestCase
         $GLOBALS['cfg']['Server']['pmadb'] = 'phpmyadmin';
         $_SESSION['relation'][$GLOBALS['server']] = "PMA_relation";
         $_SESSION['PMA_Theme'] = new PMA_Theme();
+        $_SESSION['relation'] = array();
 
         $GLOBALS['pmaThemePath'] = $_SESSION['PMA_Theme']->getPath();
         $GLOBALS['pmaThemeImage'] = 'theme/';
+        $GLOBALS['cfg']['ServerDefault'] = 0;
 
         include_once 'libraries/relation.lib.php';
     }
@@ -71,9 +73,6 @@ class PMA_Relation_Test extends PHPUnit_Framework_TestCase
      */
     public function testPMA_getRelationsParam()
     {
-        $GLOBALS['cfg']['ServerDefault'] = 0;
-        $_SESSION['relation'] = array();
-
         $relationsPara = PMA_getRelationsParam();
         $this->assertEquals(
             false,
@@ -137,5 +136,84 @@ class PMA_Relation_Test extends PHPUnit_Framework_TestCase
             $retval
         );
     }
+
+    /**
+     * Test for PMA_getDisplayField
+     *
+     * @return void
+     */
+    public function testPMA_getDisplayField()
+    {
+        
+        $db = 'information_schema';
+        $table = 'CHARACTER_SETS';
+        $this->assertEquals(
+            'DESCRIPTION',
+            PMA_getDisplayField($db, $table)
+        );  
+              
+        $db = 'information_schema';
+        $table = 'TABLES';
+        $this->assertEquals(
+            'TABLE_COMMENT',
+            PMA_getDisplayField($db, $table)
+        );
+              
+        $db = 'information_schema';
+        $table = 'PMA';
+        $this->assertEquals(
+            false,
+            PMA_getDisplayField($db, $table)
+        );
+        
+    }
+
+    /**
+     * Test for PMA_getComments
+     *
+     * @return void
+     */
+    public function testPMA_getComments()
+    {
+        $GLOBALS['cfg']['ServerDefault'] = 0;
+        $_SESSION['relation'] = array();
+        
+        $dbi = $this->getMockBuilder('PMA_DatabaseInterface')
+            ->disableOriginalConstructor()
+            ->getMock();
+        
+        $getColumnsResult = array(
+                array(
+                        'Field' => 'field1',
+                        'Type' => 'int(11)',
+                        'Comment' => 'Comment1'
+                ),
+                array(
+                        'Field' => 'field2',
+                        'Type' => 'text',
+                        'Comment' => 'Comment1'
+                )
+        );
+        $dbi->expects($this->any())->method('getColumns')
+            ->will($this->returnValue($getColumnsResult));
+        
+        $GLOBALS['dbi'] = $dbi;
+        
+        $db = 'information_schema';
+        $this->assertEquals(
+            array(''),
+            PMA_getComments($db)
+        );  
+              
+        $db = 'information_schema';
+        $table = 'TABLES';
+        $this->assertEquals(
+            array(
+                'field1' => 'Comment1',
+                'field2' => 'Comment1'
+            ),
+            PMA_getComments($db, $table)
+        );    
+    }
 }