- Add TestCase::setLocale().

sebastianbergmann · Jun 21, 2007 · 530ec6f · 530ec6f
1 parent a637caf
commit 530ec6f
Showing 1 changed file with 36 additions and 0 deletions.
diff --git a/3.1/PHPUnit/Framework/TestCase.php b/3.1/PHPUnit/Framework/TestCase.php
@@ -153,6 +153,12 @@ abstract class PHPUnit_Framework_TestCase extends PHPUnit_Framework_Assert imple
      */
     private $iniSettings = array();
 
+    /**
+     * @var    Array
+     * @access private
+     */
+    private $locale = array();
+
     /**
      * @var    Array
      * @access private
@@ -320,6 +326,11 @@ public function runBare()
             ini_set($varName, $oldValue);
         }
 
+        // Clean up locale settings.
+        foreach ($this->locale as $category => $locale) {
+            setlocale($category, $locale);
+        }
+
         $this->iniSettings = array();
 
         // Workaround for missing "finally".
@@ -404,6 +415,31 @@ protected function iniSet($varName, $newValue)
         }
     }
 
+    /**
+     * This method is a wrapper for the setlocate() function that automatically
+     * resets the locale to its original value after the test is run.
+     *
+     * @param  integer $category
+     * @param  string  $locale
+     * @throws InvalidArgumentException
+     * @throws RuntimeException
+     * @access protected
+     * @since  Method available since Release 3.1.0
+     */
+    protected function setLocale($category, $locale)
+    {
+        if (!in_array($category, array(LC_ALL, LC_COLLATE, LC_CTYPE, LC_MONETARY, LC_NUMERIC, LC_TIME, LC_MESSAGES))) {
+            throw new InvalidArgumentException;
+        }
+
+        if (!is_array($locale) && !is_string($locale)) {
+            throw new InvalidArgumentException;
+        }
+
+        $this->locale[$category] = setlocale($category, NULL);
+        setlocale($category, $locale);
+    }
+
     /**
      * Returns a mock object for the specified class.
      *