@@ -24,40 +24,57 @@
24{-# OPTIONS_GHC -fno-warn-name-shadowing #-}
24{-# OPTIONS_GHC -fno-warn-name-shadowing #-}
26#include "overlapping-compat.h"
26#include "overlapping-compat.h"
27−module Servant.ClientSpec where
27+module Servant.ClientSpec (spec) where
29−import Control.Arrow (left)
32+import Control.Arrow (left)
30−import Control.Concurrent (forkIO, killThread, ThreadId)
33+import Control.Concurrent (ThreadId, forkIO,
31−import Control.Exception (bracket)
35+import Control.Exception (bracket)
32−import Control.Monad.Error.Class (throwError )
36+import Control.Monad.Error.Class (throwError)
33import Data.Aeson
37import Data.Aeson
34−import qualified Data.ByteString.Lazy as BS
35−import Data.Char (chr, isPrint)
38+import Data.Char (chr, isPrint)
36−import Data.Foldable (forM_)
39+import Data.Foldable (forM_)
37−import Data.Monoid hiding (getLast)
40+import Data.Monoid hiding (getLast)
38import Data.Proxy
41import Data.Proxy
39−import qualified Generics.SOP as SOP
42+import qualified Generics.SOP as SOP
40−import GHC.Generics (Generic)
43+import GHC.Generics (Generic)
41−import qualified Network.HTTP.Client as C
44+import qualified Network.HTTP.Client as C
42−import Network.HTTP.Media
43−import qualified Network.HTTP.Types as HTTP
45+import qualified Network.HTTP.Types as HTTP
44import Network.Socket
46import Network.Socket
45−import Network.Wai (Request, requestHeaders, responseLBS)
47+import qualified Network.Wai as Wai
46import Network.Wai.Handler.Warp
48import Network.Wai.Handler.Warp
49−import System.IO.Unsafe (unsafePerformIO)
49+import System.IO.Unsafe (unsafePerformIO)
51import Test.Hspec
50import Test.Hspec
52import Test.Hspec.QuickCheck
51import Test.Hspec.QuickCheck
53import Test.QuickCheck
53import Test.QuickCheck
54−import Web.FormUrlEncoded (FromForm, ToForm)
54+import Web.FormUrlEncoded (FromForm, ToForm)
56−import Servant.API
56+import Servant.API ((:<|>) ((:<|>)),
67+ NoContent (NoContent),
57import Servant.API.Internal.Test.ComprehensiveAPI
74import Servant.API.Internal.Test.ComprehensiveAPI
58import Servant.Client
75import Servant.Client
59−import Servant.Client.Generic
76+import qualified Servant.Client.Core.Internal.Request as Req
60−import qualified Servant.Common.Req as SCR
77+import qualified Servant.Client.Core.Internal.Auth as Auth
61import Servant.Server
78import Servant.Server
62import Servant.Server.Experimental.Auth
79import Servant.Server.Experimental.Auth
@@ -75,17 +92,16 @@ spec = describe "Servant.Client" $ do
76-- * test data types
93-- * test data types
78−data Person = Person {
95+data Person = Person
79− name :: String,
96+ { _name :: String
80− age :: Integer
97+ , _age :: Integer
82− deriving (Eq, Show, Generic)
98+ } deriving (Eq, Show, Generic)
84instance ToJSON Person
100instance ToJSON Person
85instance FromJSON Person
101instance FromJSON Person
87−instance ToForm Person where
103+instance ToForm Person
88−instance FromForm Person where
104+instance FromForm Person
90alice :: Person
106alice :: Person
91alice = Person "Alice" 42
107alice = Person "Alice" 42
@@ -116,22 +132,20 @@ type Api =
116api :: Proxy Api
132api :: Proxy Api
117api = Proxy
133api = Proxy
119−getGet :: SCR.ClientM Person
135+getGet :: ClientM Person
120−getDeleteEmpty :: SCR.ClientM NoContent
136+getDeleteEmpty :: ClientM NoContent
121−getCapture :: String -> SCR.ClientM Person
137+getCapture :: String -> ClientM Person
122−getCaptureAll :: [String] -> SCR.ClientM [Person]
138+getCaptureAll :: [String] -> ClientM [Person]
123−getBody :: Person -> SCR.ClientM Person
139+getBody :: Person -> ClientM Person
124−getQueryParam :: Maybe String -> SCR.ClientM Person
140+getQueryParam :: Maybe String -> ClientM Person
125−getQueryParams :: [String] -> SCR.ClientM [Person]
141+getQueryParams :: [String] -> ClientM [Person]
126−getQueryFlag :: Bool -> SCR.ClientM Bool
142+getQueryFlag :: Bool -> ClientM Bool
127−getRawSuccess :: HTTP.Method
143+getRawSuccess :: HTTP.Method -> ClientM Response
128− -> SCR.ClientM (Int, BS.ByteString, MediaType, [HTTP.Header], C.Response BS.ByteString)
129−getRawFailure :: HTTP.Method
144+getRawFailure :: HTTP.Method -> ClientM Response
130− -> SCR.ClientM (Int, BS.ByteString, MediaType, [HTTP.Header], C.Response BS.ByteString)
131getMultiple :: String -> Maybe Int -> Bool -> [(String, [Rational])]
145getMultiple :: String -> Maybe Int -> Bool -> [(String, [Rational])]
132− -> SCR.ClientM (String, Maybe Int, Bool, [(String, [Rational])])
146+ -> ClientM (String, Maybe Int, Bool, [(String, [Rational])])
133−getRespHeaders :: SCR.ClientM (Headers TestHeaders Bool)
147+getRespHeaders :: ClientM (Headers TestHeaders Bool)
134−getDeleteContentType :: SCR.ClientM NoContent
148+getDeleteContentType :: ClientM NoContent
137 :<|> getDeleteEmpty
151 :<|> getDeleteEmpty
@@ -161,8 +175,8 @@ server = serve api (
161 Nothing -> throwError $ ServantErr 400 "missing parameter" "" [])
175 Nothing -> throwError $ ServantErr 400 "missing parameter" "" [])
162 :<|> (\ names -> return (zipWith Person names [0..]))
176 :<|> (\ names -> return (zipWith Person names [0..]))
163 :<|> return
177 :<|> return
164− :<|> (Tagged $ \ _request respond -> respond $ responseLBS HTTP.ok200 [] "rawSuccess")
178+ :<|> (Tagged $ \ _request respond -> respond $ Wai.responseLBS HTTP.ok200 [] "rawSuccess")
165− :<|> (Tagged $ \ _request respond -> respond $ responseLBS HTTP.badRequest400 [] "rawFailure")
179+ :<|> (Tagged $ \ _request respond -> respond $ Wai.responseLBS HTTP.badRequest400 [] "rawFailure")
166 :<|> (\ a b c d -> return (a, b, c, d))
180 :<|> (\ a b c d -> return (a, b, c, d))
167 :<|> (return $ addHeader 1729 $ addHeader "eg2" True)
181 :<|> (return $ addHeader 1729 $ addHeader "eg2" True)
168 :<|> return NoContent
182 :<|> return NoContent
@@ -178,9 +192,9 @@ failApi = Proxy
179failServer :: Application
193failServer :: Application
180failServer = serve failApi (
194failServer = serve failApi (
181− (Tagged $ \ _request respond -> respond $ responseLBS HTTP.ok200 [] "")
195+ (Tagged $ \ _request respond -> respond $ Wai.responseLBS HTTP.ok200 [] "")
182− :<|> (\ _capture -> Tagged $ \_request respond -> respond $ responseLBS HTTP.ok200 [("content-type", "application/json")] "")
196+ :<|> (\ _capture -> Tagged $ \_request respond -> respond $ Wai.responseLBS HTTP.ok200 [("content-type", "application/json")] "")
183− :<|> (Tagged $ \_request respond -> respond $ responseLBS HTTP.ok200 [("content-type", "fooooo")] "")
197+ :<|> (Tagged $ \_request respond -> respond $ Wai.responseLBS HTTP.ok200 [("content-type", "fooooo")] "")
186-- * basic auth stuff
200-- * basic auth stuff
@@ -214,16 +228,16 @@ genAuthAPI :: Proxy GenAuthAPI
214genAuthAPI = Proxy
228genAuthAPI = Proxy
216type instance AuthServerData (AuthProtect "auth-tag") = ()
230type instance AuthServerData (AuthProtect "auth-tag") = ()
217−type instance AuthClientData (AuthProtect "auth-tag") = ()
231+type instance Auth.AuthClientData (AuthProtect "auth-tag") = ()
219−genAuthHandler :: AuthHandler Request ()
233+genAuthHandler :: AuthHandler Wai.Request ()
220genAuthHandler =
234genAuthHandler =
221− let handler req = case lookup "AuthHeader" (requestHeaders req) of
235+ let handler req = case lookup "AuthHeader" (Wai.requestHeaders req) of
222 Nothing -> throwError (err401 { errBody = "Missing auth header" })
236 Nothing -> throwError (err401 { errBody = "Missing auth header" })
223 Just _ -> return ()
237 Just _ -> return ()
224 in mkAuthHandler handler
238 in mkAuthHandler handler
226−genAuthServerContext :: Context '[ AuthHandler Request () ]
240+genAuthServerContext :: Context '[ AuthHandler Wai.Request () ]
227genAuthServerContext = genAuthHandler :. EmptyContext
241genAuthServerContext = genAuthHandler :. EmptyContext
229genAuthServer :: Application
243genAuthServer :: Application
@@ -236,11 +250,11 @@ type GenericClientAPI
236 :<|> Capture "foo" String :> NestedAPI1
250 :<|> Capture "foo" String :> NestedAPI1
238data GenericClient = GenericClient
252data GenericClient = GenericClient
239− { getSqr :: Maybe Int -> SCR.ClientM Int
253+ { getSqr :: Maybe Int -> ClientM Int
240 , mkNestedClient1 :: String -> NestedClient1
254 , mkNestedClient1 :: String -> NestedClient1
241 } deriving Generic
255 } deriving Generic
242instance SOP.Generic GenericClient
256instance SOP.Generic GenericClient
243−instance (Client GenericClientAPI ~ client) => ClientLike client GenericClient
257+instance (Client ClientM GenericClientAPI ~ client) => ClientLike client GenericClient
245type NestedAPI1
259type NestedAPI1
246 = QueryParam "int" Int :> NestedAPI2
260 = QueryParam "int" Int :> NestedAPI2
@@ -248,21 +262,21 @@ type NestedAPI1
249data NestedClient1 = NestedClient1
263data NestedClient1 = NestedClient1
250 { mkNestedClient2 :: Maybe Int -> NestedClient2
264 { mkNestedClient2 :: Maybe Int -> NestedClient2
251− , idChar :: Maybe Char -> SCR.ClientM Char
265+ , idChar :: Maybe Char -> ClientM Char
252 } deriving Generic
266 } deriving Generic
253instance SOP.Generic NestedClient1
267instance SOP.Generic NestedClient1
254−instance (Client NestedAPI1 ~ client) => ClientLike client NestedClient1
268+instance (Client ClientM NestedAPI1 ~ client) => ClientLike client NestedClient1
256type NestedAPI2
270type NestedAPI2
257 = "sum" :> Capture "first" Int :> Capture "second" Int :> Get '[JSON] Int
271 = "sum" :> Capture "first" Int :> Capture "second" Int :> Get '[JSON] Int
258 :<|> "void" :> Post '[JSON] ()
272 :<|> "void" :> Post '[JSON] ()
260data NestedClient2 = NestedClient2
274data NestedClient2 = NestedClient2
261− { getSum :: Int -> Int -> SCR.ClientM Int
275+ { getSum :: Int -> Int -> ClientM Int
262− , doNothing :: SCR.ClientM ()
276+ , doNothing :: ClientM ()
263 } deriving Generic
277 } deriving Generic
264instance SOP.Generic NestedClient2
278instance SOP.Generic NestedClient2
265−instance (Client NestedAPI2 ~ client) => ClientLike client NestedClient2
279+instance (Client ClientM NestedAPI2 ~ client) => ClientLike client NestedClient2
267genericClientServer :: Application
281genericClientServer :: Application
268genericClientServer = serve (Proxy :: Proxy GenericClientAPI) (
282genericClientServer = serve (Proxy :: Proxy GenericClientAPI) (
@@ -276,67 +290,70 @@ genericClientServer = serve (Proxy :: Proxy GenericClientAPI) (
276 nestedServer1 _str = nestedServer2 :<|> (maybe (throwError $ ServantErr 400 "missing parameter" "" []) return)
290 nestedServer1 _str = nestedServer2 :<|> (maybe (throwError $ ServantErr 400 "missing parameter" "" []) return)
277 nestedServer2 _int = (\ x y -> return (x + y)) :<|> return ()
291 nestedServer2 _int = (\ x y -> return (x + y)) :<|> return ()
279−{-# NOINLINE manager #-}
293+{-# NOINLINE manager' #-}
280−manager :: C.Manager
294+manager' :: C.Manager
281−manager = unsafePerformIO $ C.newManager C.defaultManagerSettings
295+manager' = unsafePerformIO $ C.newManager C.defaultManagerSettings
297+runClient :: ClientM a -> BaseUrl -> IO (Either ServantError a)
298+runClient x baseUrl' = runClientM x (ClientEnv manager' baseUrl')
283sucessSpec :: Spec
300sucessSpec :: Spec
284sucessSpec = beforeAll (startWaiApp server) $ afterAll endWaiApp $ do
301sucessSpec = beforeAll (startWaiApp server) $ afterAll endWaiApp $ do
286 it "Servant.API.Get" $ \(_, baseUrl) -> do
303 it "Servant.API.Get" $ \(_, baseUrl) -> do
287− (left show <$> (runClientM getGet (ClientEnv manager baseUrl))) `shouldReturn` Right alice
304+ left show <$> runClient getGet baseUrl `shouldReturn` Right alice
289 describe "Servant.API.Delete" $ do
306 describe "Servant.API.Delete" $ do
290 it "allows empty content type" $ \(_, baseUrl) -> do
307 it "allows empty content type" $ \(_, baseUrl) -> do
291− (left show <$> (runClientM getDeleteEmpty (ClientEnv manager baseUrl))) `shouldReturn` Right NoContent
308+ left show <$> runClient getDeleteEmpty baseUrl `shouldReturn` Right NoContent
293 it "allows content type" $ \(_, baseUrl) -> do
310 it "allows content type" $ \(_, baseUrl) -> do
294− (left show <$> (runClientM getDeleteContentType (ClientEnv manager baseUrl))) `shouldReturn` Right NoContent
311+ left show <$> runClient getDeleteContentType baseUrl `shouldReturn` Right NoContent
296 it "Servant.API.Capture" $ \(_, baseUrl) -> do
313 it "Servant.API.Capture" $ \(_, baseUrl) -> do
297− (left show <$> (runClientM (getCapture "Paula") (ClientEnv manager baseUrl))) `shouldReturn` Right (Person "Paula" 0)
314+ left show <$> runClient (getCapture "Paula") baseUrl `shouldReturn` Right (Person "Paula" 0)
299 it "Servant.API.CaptureAll" $ \(_, baseUrl) -> do
316 it "Servant.API.CaptureAll" $ \(_, baseUrl) -> do
300 let expected = [(Person "Paula" 0), (Person "Peta" 1)]
317 let expected = [(Person "Paula" 0), (Person "Peta" 1)]
301− (left show <$> (runClientM (getCaptureAll ["Paula", "Peta"]) (ClientEnv manager baseUrl))) `shouldReturn` Right expected
318+ left show <$> runClient (getCaptureAll ["Paula", "Peta"]) baseUrl `shouldReturn` Right expected
303 it "Servant.API.ReqBody" $ \(_, baseUrl) -> do
320 it "Servant.API.ReqBody" $ \(_, baseUrl) -> do
304 let p = Person "Clara" 42
321 let p = Person "Clara" 42
305− (left show <$> runClientM (getBody p) (ClientEnv manager baseUrl)) `shouldReturn` Right p
322+ left show <$> runClient (getBody p) baseUrl `shouldReturn` Right p
307 it "Servant.API.QueryParam" $ \(_, baseUrl) -> do
324 it "Servant.API.QueryParam" $ \(_, baseUrl) -> do
308− left show <$> runClientM (getQueryParam (Just "alice")) (ClientEnv manager baseUrl) `shouldReturn` Right alice
325+ left show <$> runClient (getQueryParam (Just "alice")) baseUrl `shouldReturn` Right alice
309− Left FailureResponse{..} <- runClientM (getQueryParam (Just "bob")) (ClientEnv manager baseUrl)
326+ Left (FailureResponse r) <- runClient (getQueryParam (Just "bob")) baseUrl
310− responseStatus `shouldBe` HTTP.Status 400 "bob not found"
327+ responseStatusCode r `shouldBe` HTTP.Status 400 "bob not found"
312 it "Servant.API.QueryParam.QueryParams" $ \(_, baseUrl) -> do
329 it "Servant.API.QueryParam.QueryParams" $ \(_, baseUrl) -> do
313− (left show <$> runClientM (getQueryParams []) (ClientEnv manager baseUrl)) `shouldReturn` Right []
330+ left show <$> runClient (getQueryParams []) baseUrl `shouldReturn` Right []
314− (left show <$> runClientM (getQueryParams ["alice", "bob"]) (ClientEnv manager baseUrl))
331+ left show <$> runClient (getQueryParams ["alice", "bob"]) baseUrl
315 `shouldReturn` Right [Person "alice" 0, Person "bob" 1]
332 `shouldReturn` Right [Person "alice" 0, Person "bob" 1]
317 context "Servant.API.QueryParam.QueryFlag" $
334 context "Servant.API.QueryParam.QueryFlag" $
318 forM_ [False, True] $ \ flag -> it (show flag) $ \(_, baseUrl) -> do
335 forM_ [False, True] $ \ flag -> it (show flag) $ \(_, baseUrl) -> do
319− (left show <$> runClientM (getQueryFlag flag) (ClientEnv manager baseUrl)) `shouldReturn` Right flag
336+ left show <$> runClient (getQueryFlag flag) baseUrl `shouldReturn` Right flag
321 it "Servant.API.Raw on success" $ \(_, baseUrl) -> do
338 it "Servant.API.Raw on success" $ \(_, baseUrl) -> do
322− res <- runClientM (getRawSuccess HTTP.methodGet) (ClientEnv manager baseUrl)
339+ res <- runClient (getRawSuccess HTTP.methodGet) baseUrl
323 case res of
340 case res of
324 Left e -> assertFailure $ show e
341 Left e -> assertFailure $ show e
325− Right (code, body, ct, _, response) -> do
342+ Right r -> do
326− (code, body, ct) `shouldBe` (200, "rawSuccess", "application"//"octet-stream")
343+ responseStatusCode r `shouldBe` HTTP.status200
327− C.responseBody response `shouldBe` body
344+ responseBody r `shouldBe` "rawSuccess"
328− C.responseStatus response `shouldBe` HTTP.ok200
330 it "Servant.API.Raw should return a Left in case of failure" $ \(_, baseUrl) -> do
346 it "Servant.API.Raw should return a Left in case of failure" $ \(_, baseUrl) -> do
331− res <- runClientM (getRawFailure HTTP.methodGet) (ClientEnv manager baseUrl)
347+ res <- runClient (getRawFailure HTTP.methodGet) baseUrl
332 case res of
348 case res of
333 Right _ -> assertFailure "expected Left, but got Right"
349 Right _ -> assertFailure "expected Left, but got Right"
334− Left e -> do
350+ Left (FailureResponse r) -> do
335− Servant.Client.responseStatus e `shouldBe` HTTP.status400
351+ responseStatusCode r `shouldBe` HTTP.status400
336− Servant.Client.responseBody e `shouldBe` "rawFailure"
352+ responseBody r `shouldBe` "rawFailure"
353+ Left e -> assertFailure $ "expected FailureResponse, but got " ++ show e
338 it "Returns headers appropriately" $ \(_, baseUrl) -> do
355 it "Returns headers appropriately" $ \(_, baseUrl) -> do
339− res <- runClientM getRespHeaders (ClientEnv manager baseUrl)
356+ res <- runClient getRespHeaders baseUrl
340 case res of
357 case res of
341 Left e -> assertFailure $ show e
358 Left e -> assertFailure $ show e
342 Right val -> getHeaders val `shouldBe` [("X-Example1", "1729"), ("X-Example2", "eg2")]
359 Right val -> getHeaders val `shouldBe` [("X-Example1", "1729"), ("X-Example2", "eg2")]
@@ -345,7 +362,7 @@ sucessSpec = beforeAll (startWaiApp server) $ afterAll endWaiApp $ do
345 it "works for a combination of Capture, QueryParam, QueryFlag and ReqBody" $ \(_, baseUrl) ->
362 it "works for a combination of Capture, QueryParam, QueryFlag and ReqBody" $ \(_, baseUrl) ->
346 property $ forAllShrink pathGen shrink $ \(NonEmpty cap) num flag body ->
363 property $ forAllShrink pathGen shrink $ \(NonEmpty cap) num flag body ->
347 ioProperty $ do
364 ioProperty $ do
348− result <- left show <$> runClientM (getMultiple cap num flag body) (ClientEnv manager baseUrl)
365+ result <- left show <$> runClient (getMultiple cap num flag body) baseUrl
350 result === Right (cap, num, flag, body)
367 result === Right (cap, num, flag, body)
@@ -357,10 +374,10 @@ wrappedApiSpec = describe "error status codes" $ do
357 let test :: (WrappedApi, String) -> Spec
374 let test :: (WrappedApi, String) -> Spec
358 test (WrappedApi api, desc) =
375 test (WrappedApi api, desc) =
359 it desc $ bracket (startWaiApp $ serveW api) endWaiApp $ \(_, baseUrl) -> do
376 it desc $ bracket (startWaiApp $ serveW api) endWaiApp $ \(_, baseUrl) -> do
360− let getResponse :: SCR.ClientM ()
377+ let getResponse :: ClientM ()
361 getResponse = client api
378 getResponse = client api
362− Left FailureResponse{..} <- runClientM getResponse (ClientEnv manager baseUrl)
379+ Left (FailureResponse r) <- runClient getResponse baseUrl
363− responseStatus `shouldBe` (HTTP.Status 500 "error message")
380+ responseStatusCode r `shouldBe` (HTTP.Status 500 "error message")
364 in mapM_ test $
381 in mapM_ test $
365 (WrappedApi (Proxy :: Proxy (Delete '[JSON] ())), "Delete") :
382 (WrappedApi (Proxy :: Proxy (Delete '[JSON] ())), "Delete") :
366 (WrappedApi (Proxy :: Proxy (Get '[JSON] ())), "Get") :
383 (WrappedApi (Proxy :: Proxy (Get '[JSON] ())), "Get") :
@@ -374,42 +391,42 @@ failSpec = beforeAll (startWaiApp failServer) $ afterAll endWaiApp $ do
374 context "client returns errors appropriately" $ do
391 context "client returns errors appropriately" $ do
375 it "reports FailureResponse" $ \(_, baseUrl) -> do
392 it "reports FailureResponse" $ \(_, baseUrl) -> do
376 let (_ :<|> getDeleteEmpty :<|> _) = client api
393 let (_ :<|> getDeleteEmpty :<|> _) = client api
377− Left res <- runClientM getDeleteEmpty (ClientEnv manager baseUrl)
394+ Left res <- runClient getDeleteEmpty baseUrl
378 case res of
395 case res of
379− FailureResponse _ (HTTP.Status 404 "Not Found") _ _ -> return ()
396+ FailureResponse r | responseStatusCode r == HTTP.status404 -> return ()
380 _ -> fail $ "expected 404 response, but got " <> show res
397 _ -> fail $ "expected 404 response, but got " <> show res
382 it "reports DecodeFailure" $ \(_, baseUrl) -> do
399 it "reports DecodeFailure" $ \(_, baseUrl) -> do
383 let (_ :<|> _ :<|> getCapture :<|> _) = client api
400 let (_ :<|> _ :<|> getCapture :<|> _) = client api
384− Left res <- runClientM (getCapture "foo") (ClientEnv manager baseUrl)
401+ Left res <- runClient (getCapture "foo") baseUrl
385 case res of
402 case res of
386− DecodeFailure _ ("application/json") _ -> return ()
403+ DecodeFailure _ _ -> return ()
387 _ -> fail $ "expected DecodeFailure, but got " <> show res
404 _ -> fail $ "expected DecodeFailure, but got " <> show res
389 it "reports ConnectionError" $ \_ -> do
406 it "reports ConnectionError" $ \_ -> do
390 let (getGetWrongHost :<|> _) = client api
407 let (getGetWrongHost :<|> _) = client api
391− Left res <- runClientM getGetWrongHost (ClientEnv manager (BaseUrl Http "127.0.0.1" 19872 ""))
408+ Left res <- runClient getGetWrongHost (BaseUrl Http "127.0.0.1" 19872 "")
392 case res of
409 case res of
393 ConnectionError _ -> return ()
410 ConnectionError _ -> return ()
394 _ -> fail $ "expected ConnectionError, but got " <> show res
411 _ -> fail $ "expected ConnectionError, but got " <> show res
396 it "reports UnsupportedContentType" $ \(_, baseUrl) -> do
413 it "reports UnsupportedContentType" $ \(_, baseUrl) -> do
397 let (getGet :<|> _ ) = client api
414 let (getGet :<|> _ ) = client api
398− Left res <- runClientM getGet (ClientEnv manager baseUrl)
415+ Left res <- runClient getGet baseUrl
399 case res of
416 case res of
400 UnsupportedContentType ("application/octet-stream") _ -> return ()
417 UnsupportedContentType ("application/octet-stream") _ -> return ()
401 _ -> fail $ "expected UnsupportedContentType, but got " <> show res
418 _ -> fail $ "expected UnsupportedContentType, but got " <> show res
403 it "reports InvalidContentTypeHeader" $ \(_, baseUrl) -> do
420 it "reports InvalidContentTypeHeader" $ \(_, baseUrl) -> do
404 let (_ :<|> _ :<|> _ :<|> _ :<|> getBody :<|> _) = client api
421 let (_ :<|> _ :<|> _ :<|> _ :<|> getBody :<|> _) = client api
405− Left res <- runClientM (getBody alice) (ClientEnv manager baseUrl)
422+ Left res <- runClient (getBody alice) baseUrl
406 case res of
423 case res of
407− InvalidContentTypeHeader "fooooo" _ -> return ()
424+ InvalidContentTypeHeader _ -> return ()
408 _ -> fail $ "expected InvalidContentTypeHeader, but got " <> show res
425 _ -> fail $ "expected InvalidContentTypeHeader, but got " <> show res
410data WrappedApi where
427data WrappedApi where
411 WrappedApi :: (HasServer (api :: *) '[], Server api ~ Handler a,
428 WrappedApi :: (HasServer (api :: *) '[], Server api ~ Handler a,
412− HasClient api, Client api ~ SCR.ClientM ()) =>
429+ HasClient ClientM api, Client ClientM api ~ ClientM ()) =>
413 Proxy api -> WrappedApi
430 Proxy api -> WrappedApi
415basicAuthSpec :: Spec
432basicAuthSpec :: Spec
@@ -419,15 +436,15 @@ basicAuthSpec = beforeAll (startWaiApp basicAuthServer) $ afterAll endWaiApp $ d
419 it "Authenticates a BasicAuth protected server appropriately" $ \(_,baseUrl) -> do
436 it "Authenticates a BasicAuth protected server appropriately" $ \(_,baseUrl) -> do
420 let getBasic = client basicAuthAPI
437 let getBasic = client basicAuthAPI
421 let basicAuthData = BasicAuthData "servant" "server"
438 let basicAuthData = BasicAuthData "servant" "server"
422− (left show <$> runClientM (getBasic basicAuthData) (ClientEnv manager baseUrl)) `shouldReturn` Right alice
439+ left show <$> runClient (getBasic basicAuthData) baseUrl `shouldReturn` Right alice
424 context "Authentication is rejected when requests are not authenticated properly" $ do
441 context "Authentication is rejected when requests are not authenticated properly" $ do
426 it "Authenticates a BasicAuth protected server appropriately" $ \(_,baseUrl) -> do
443 it "Authenticates a BasicAuth protected server appropriately" $ \(_,baseUrl) -> do
427 let getBasic = client basicAuthAPI
444 let getBasic = client basicAuthAPI
428 let basicAuthData = BasicAuthData "not" "password"
445 let basicAuthData = BasicAuthData "not" "password"
429− Left FailureResponse{..} <- runClientM (getBasic basicAuthData) (ClientEnv manager baseUrl)
446+ Left (FailureResponse r) <- runClient (getBasic basicAuthData) baseUrl
430− responseStatus `shouldBe` HTTP.Status 403 "Forbidden"
447+ responseStatusCode r `shouldBe` HTTP.Status 403 "Forbidden"
432genAuthSpec :: Spec
449genAuthSpec :: Spec
433genAuthSpec = beforeAll (startWaiApp genAuthServer) $ afterAll endWaiApp $ do
450genAuthSpec = beforeAll (startWaiApp genAuthServer) $ afterAll endWaiApp $ do
@@ -435,16 +452,16 @@ genAuthSpec = beforeAll (startWaiApp genAuthServer) $ afterAll endWaiApp $ do
436 it "Authenticates a AuthProtect protected server appropriately" $ \(_, baseUrl) -> do
453 it "Authenticates a AuthProtect protected server appropriately" $ \(_, baseUrl) -> do
437 let getProtected = client genAuthAPI
454 let getProtected = client genAuthAPI
438− let authRequest = mkAuthenticateReq () (\_ req -> SCR.addHeader "AuthHeader" ("cool" :: String) req)
455+ let authRequest = Auth.mkAuthenticatedRequest () (\_ req -> Req.addHeader "AuthHeader" ("cool" :: String) req)
439− (left show <$> runClientM (getProtected authRequest) (ClientEnv manager baseUrl)) `shouldReturn` Right alice
456+ left show <$> runClient (getProtected authRequest) baseUrl `shouldReturn` Right alice
441 context "Authentication is rejected when requests are not authenticated properly" $ do
458 context "Authentication is rejected when requests are not authenticated properly" $ do
443 it "Authenticates a AuthProtect protected server appropriately" $ \(_, baseUrl) -> do
460 it "Authenticates a AuthProtect protected server appropriately" $ \(_, baseUrl) -> do
444 let getProtected = client genAuthAPI
461 let getProtected = client genAuthAPI
445− let authRequest = mkAuthenticateReq () (\_ req -> SCR.addHeader "Wrong" ("header" :: String) req)
462+ let authRequest = Auth.mkAuthenticatedRequest () (\_ req -> Req.addHeader "Wrong" ("header" :: String) req)
446− Left FailureResponse{..} <- runClientM (getProtected authRequest) (ClientEnv manager baseUrl)
463+ Left (FailureResponse r) <- runClient (getProtected authRequest) baseUrl
447− responseStatus `shouldBe` (HTTP.Status 401 "Unauthorized")
464+ responseStatusCode r `shouldBe` (HTTP.Status 401 "Unauthorized")
449genericClientSpec :: Spec
466genericClientSpec :: Spec
450genericClientSpec = beforeAll (startWaiApp genericClientServer) $ afterAll endWaiApp $ do
467genericClientSpec = beforeAll (startWaiApp genericClientServer) $ afterAll endWaiApp $ do
@@ -454,13 +471,13 @@ genericClientSpec = beforeAll (startWaiApp genericClientServer) $ afterAll endWa
454 NestedClient1{..} = mkNestedClient1 "example"
471 NestedClient1{..} = mkNestedClient1 "example"
455 NestedClient2{..} = mkNestedClient2 (Just 42)
472 NestedClient2{..} = mkNestedClient2 (Just 42)
457− it "works for top-level client function" $ \(_, baseUrl) -> do
474+ it "works for top-level client inClientM function" $ \(_, baseUrl) -> do
458− (left show <$> (runClientM (getSqr (Just 5)) (ClientEnv manager baseUrl))) `shouldReturn` Right 25
475+ left show <$> runClient (getSqr (Just 5)) baseUrl `shouldReturn` Right 25
460 it "works for nested clients" $ \(_, baseUrl) -> do
477 it "works for nested clients" $ \(_, baseUrl) -> do
461− (left show <$> (runClientM (idChar (Just 'c')) (ClientEnv manager baseUrl))) `shouldReturn` Right 'c'
478+ left show <$> runClient (idChar (Just 'c')) baseUrl `shouldReturn` Right 'c'
462− (left show <$> (runClientM (getSum 3 4) (ClientEnv manager baseUrl))) `shouldReturn` Right 7
479+ left show <$> runClient (getSum 3 4) baseUrl `shouldReturn` Right 7
463− (left show <$> (runClientM doNothing (ClientEnv manager baseUrl))) `shouldReturn` Right ()
480+ left show <$> runClient doNothing baseUrl `shouldReturn` Right ()
465-- * utils
482-- * utils