From d0f1c60b634644b9cb1d844a123aa067f61b9c4b Mon Sep 17 00:00:00 2001
From: Michael Snoyman <michael@snoyman.com>
Date: Sat, 14 Aug 2010 21:50:56 +0300
Subject: [PATCH] Added UTF8 path tests

---
 Yesod.hs          |  3 ++-
 Yesod/Dispatch.hs |  7 +++++-
 Yesod/Yesod.hs    | 60 ++++++++++++++++++++++++++++++++++++++++++++---
 runtests.hs       |  2 ++
 4 files changed, 67 insertions(+), 5 deletions(-)

diff --git a/Yesod.hs b/Yesod.hs
index 73c3f733..3edf55e7 100644
--- a/Yesod.hs
+++ b/Yesod.hs
@@ -19,15 +19,16 @@ module Yesod
 import Yesod.Content hiding (testSuite)
 import Yesod.Json hiding (testSuite)
 import Yesod.Dispatch hiding (testSuite)
+import Yesod.Yesod hiding (testSuite)
 #else
 import Yesod.Content
 import Yesod.Json
 import Yesod.Dispatch
+import Yesod.Yesod
 #endif
 
 import Yesod.Request
 import Yesod.Form
-import Yesod.Yesod
 import Yesod.Widget
 import Yesod.Handler hiding (runHandler)
 import Network.Wai (Application)
diff --git a/Yesod/Dispatch.hs b/Yesod/Dispatch.hs
index 4dc89406..c6bd9e1a 100644
--- a/Yesod/Dispatch.hs
+++ b/Yesod/Dispatch.hs
@@ -23,8 +23,13 @@ module Yesod.Dispatch
 #endif
     ) where
 
-import Yesod.Handler
+#if TEST
+import Yesod.Yesod hiding (testSuite)
+#else
 import Yesod.Yesod
+#endif
+
+import Yesod.Handler
 import Yesod.Request
 import Yesod.Internal
 
diff --git a/Yesod/Yesod.hs b/Yesod/Yesod.hs
index 2ed95adf..06475539 100644
--- a/Yesod/Yesod.hs
+++ b/Yesod/Yesod.hs
@@ -4,6 +4,7 @@
 {-# LANGUAGE FlexibleContexts #-}
 {-# LANGUAGE MultiParamTypeClasses #-}
 {-# LANGUAGE OverloadedStrings #-}
+{-# LANGUAGE CPP #-}
 -- | The basic typeclass for a Yesod application.
 module Yesod.Yesod
     ( -- * Type classes
@@ -25,28 +26,45 @@ module Yesod.Yesod
     , defaultErrorHandler
       -- * Data types
     , AuthResult (..)
+#if TEST
+    , testSuite
+#endif
     ) where
 
+#if TEST
+import Yesod.Content hiding (testSuite)
+import Yesod.Json hiding (testSuite)
+#else
 import Yesod.Content
+import Yesod.Json
+#endif
+
 import Yesod.Request
 import Yesod.Hamlet
 import Yesod.Handler
 import qualified Network.Wai as W
-import Yesod.Json
 import Yesod.Internal
 import Web.ClientSession (getKey, defaultKeyFile)
 import qualified Web.ClientSession as CS
 import Data.Monoid (mempty)
-import Data.ByteString.UTF8 (toString)
+import qualified Data.ByteString.UTF8 as BSU
 import Database.Persist
 import Web.Routes.Site (Site)
 import Control.Monad.Trans.Class (MonadTrans (..))
 import Control.Monad.Attempt (Failure)
 import qualified Data.ByteString as S
+import qualified Data.ByteString.Char8 as S8
 import qualified Network.Wai.Middleware.CleanPath
 import Web.Routes (encodePathInfo)
 import qualified Data.ByteString.Lazy as L
 
+#if TEST
+import Test.Framework (testGroup, Test)
+import Test.Framework.Providers.HUnit
+import Test.Framework.Providers.QuickCheck2 (testProperty)
+import Test.HUnit hiding (Test)
+#endif
+
 -- | This class is automatically instantiated when you use the template haskell
 -- mkYesod function. You should never need to deal with it directly.
 class Eq (Route y) => YesodSite y where
@@ -250,9 +268,10 @@ applyLayout' s = fmap chooseRep . applyLayout s mempty
 defaultErrorHandler :: Yesod y => ErrorResponse -> GHandler sub y ChooseRep
 defaultErrorHandler NotFound = do
     r <- waiRequest
+    let pi = BSU.toString $ pathInfo r
     applyLayout' "Not Found" $ [$hamlet|
 %h1 Not Found
-%p $toString.pathInfo.r$
+%p $pi$
 |]
   where
     pathInfo = W.pathInfo
@@ -305,3 +324,38 @@ maybeAuthorized :: Yesod a
 maybeAuthorized r isWrite = do
     x <- isAuthorized r isWrite
     return $ if x == Authorized then Just r else Nothing
+
+#if TEST
+testSuite :: Test
+testSuite = testGroup "Yesod.Yesod"
+    [ testProperty "join/split path" propJoinSplitPath
+    , testCase "utf8 split path" caseUtf8SplitPath
+    , testCase "utf8 join path" caseUtf8JoinPath
+    ]
+
+data TmpYesod = TmpYesod
+data TmpRoute = TmpRoute deriving Eq
+type instance Route TmpYesod = TmpRoute
+instance Yesod TmpYesod where approot _ = ""
+
+propJoinSplitPath ss =
+    splitPath TmpYesod (BSU.fromString $ joinPath TmpYesod "" ss' [])
+        == Right ss'
+  where
+    ss' = filter (not . null) ss
+
+caseUtf8SplitPath :: Assertion
+caseUtf8SplitPath = do
+    Right ["שלום"] @=?
+        splitPath TmpYesod (BSU.fromString "/שלום/")
+    Right ["page", "Fooé"] @=?
+        splitPath TmpYesod (BSU.fromString "/page/Fooé/")
+    Right ["\156"] @=?
+        splitPath TmpYesod (BSU.fromString "/\156/")
+    Right ["ð"] @=?
+        splitPath TmpYesod (BSU.fromString "/%C3%B0/")
+
+caseUtf8JoinPath :: Assertion
+caseUtf8JoinPath = do
+    "/%D7%A9%D7%9C%D7%95%D7%9D/" @=? joinPath TmpYesod "" ["שלום"] []
+#endif
diff --git a/runtests.hs b/runtests.hs
index 208e38fe..7e06ab98 100644
--- a/runtests.hs
+++ b/runtests.hs
@@ -4,6 +4,7 @@ import qualified Yesod.Content
 import qualified Yesod.Json
 import qualified Yesod.Dispatch
 import qualified Yesod.Helpers.Static
+import qualified Yesod.Yesod
 
 main :: IO ()
 main = defaultMain
@@ -11,4 +12,5 @@ main = defaultMain
     , Yesod.Json.testSuite
     , Yesod.Dispatch.testSuite
     , Yesod.Helpers.Static.testSuite
+    , Yesod.Yesod.testSuite
     ]