http-conduit 2.1.4 → 2.3.9.1
raw patch · 8 files changed
Files
- ChangeLog.md +155/−0
- Network/HTTP/Client/Conduit.hs +40/−36
- Network/HTTP/Conduit.hs +102/−83
- Network/HTTP/Simple.hs +521/−0
- README.md +10/−0
- http-conduit.cabal +44/−22
- test/CookieTest.hs +40/−19
- test/main.hs +251/−190
+ ChangeLog.md view
@@ -0,0 +1,155 @@+# ChangeLog for http-conduit++## 2.3.9.1++* data-default-class -> data-default [#546](https://github.com/snoyberg/http-client/pull/546/files)++## 2.3.9++* Fix space leaks when closing responses [#539](https://github.com/snoyberg/http-client/pull/539)++## 2.3.8.3++* aeson 2.2 support [#512](https://github.com/snoyberg/http-client/pull/512)++## 2.3.8.2++* Add missing `crypton-connection` dependency++## 2.3.8.1++* Drop `connection` dependency++## 2.3.8++* Adds `setRequestBearerAuth` convenience function. Note that this is only available for `http-client` versions 0.7.6 or greater. [#457](https://github.com/snoyberg/http-client/pull/457/files)+* Adds a convenience function to set a request's response timeout [#456](https://github.com/snoyberg/http-client/pull/456)++## 2.3.7.4++* Introduces the `aeson` cabal file [#448](https://github.com/snoyberg/http-client/issues/448)++## 2.3.7.3++* Relax test suite version bounds++## 2.3.7.2++* Add the `network3` flag++## 2.3.7.1++* Properly skip whitespace after JSON body [#401](https://github.com/snoyberg/http-client/issues/401)++## 2.3.7++* Ensure entire JSON response body is consumed [#395](https://github.com/snoyberg/http-client/issues/395)++## 2.3.6.1++* Add back compatibility with older http-client version [#393](https://github.com/snoyberg/http-client/pull/393)++## 2.3.6++* Add `httpSource` to `Network.HTTP.Client.Conduit` [#390](https://github.com/snoyberg/http-client/pull/390).++## 2.3.5++* Adds `addToRequestQueryString` helper function++## 2.3.4++* Reexport RequestHeaders from Network.HTTP.Types (what was intended in last version)+* Fix mistake in ChangeLog++## 2.3.3++* Reexport Header, QueryItem and ResponseHeaders from Network.HTTP.Types+* Rewrite a type signature of setRequestHeaders with RequestHeaders++## 2.3.2++* Adds `parseRequestThrow`, `parseRequestThrow_`, and+ `setRequestCheckStatus` to `Network.HTTP.Simple`.+ See [#304](https://github.com/snoyberg/http-client/issues/304)++## 2.3.1++* Reexport Query from Network.HTTP.Types+* Rewrite a type signatures of getRequestQueryString and setRequestQueryString with Query++## 2.3.0++* conduit 1.3 support+ * NOTE: Even for older versions of conduit, this includes dropping+ support for finalizers+* `http` returns a `Source` instead of a `ResumableSource` (due to lack of+ finalizers)+* Drop monad-control for unliftio+* Removed some deprecated functions: `withManager`, `withManagerSettings`,+ `conduitManagerSettings`++## 2.2.4++* Add `httpBS` to `Network.HTTP.Simple`++## 2.2.3.2++* Add proper headers for `httpJSON` and `httpJSONEither` [#284](https://github.com/snoyberg/http-client/issues/284)++## 2.2.3.1++* Minor README improvement++## 2.2.3++* Add `withResponse` to `Network.HTTP.Simple`++## 2.2.2.1++* setRequestBodyJSON works with aeson's toEncoding function (>= 0.11)+ [#230](https://github.com/snoyberg/http-client/pull/230)++## 2.2.2++* Add `httpNoBody` to `Network.HTTP.Simple`++## 2.2.1++* Add `httpSource` to `Network.HTTP.Simple`++## 2.2.0.1++* Doc fixes++## 2.2.0++* Upgrade to http-client 0.5++## 2.1.11++* Switch to non-throwing behavior in `Network.HTTP.Simple` [#193](https://github.com/snoyberg/http-client/issues/193)++## 2.1.10.1++* Fix mistaken `@since` comments++## 2.1.10++* Add the `Network.HTTP.Simple` module++## 2.1.9++* cabal file cleanup++## 2.1.8++* Move HasHttpManager from http-conduit to http-client [#147](https://github.com/snoyberg/http-client/pull/147)++## 2.1.7++* Deprecate `conduitManagerSettings`, re-export `tlsManagerSettings` [#136](https://github.com/snoyberg/http-client/issues/136) [#137](https://github.com/snoyberg/http-client/issues/137)++## 2.1.6++* Deprecate `withManager` and `withManagerSettings`
Network/HTTP/Client/Conduit.hs view
@@ -1,20 +1,25 @@-{-# LANGUAGE FlexibleContexts #-}-{-# LANGUAGE MultiParamTypeClasses #-}-{-# LANGUAGE RankNTypes #-}+{-# LANGUAGE FlexibleContexts #-}+{-# LANGUAGE RankNTypes #-} -- | A new, experimental API to replace "Network.HTTP.Conduit".+--+-- For most users, "Network.HTTP.Simple" is probably a better choice. For more+-- information, see:+--+-- <https://haskell-lang.org/library/http-client>+--+-- For more information on using this module, please be sure to read the+-- documentation in the "Network.HTTP.Client" module. module Network.HTTP.Client.Conduit ( -- * Conduit-specific interface withResponse , responseOpen , responseClose , acquireResponse+ , httpSource -- * Manager helpers , defaultManagerSettings , newManager- , withManager- , withManagerSettings , newManagerSettings- , HasHttpManager (..) -- * General HTTP client interface , module Network.HTTP.Client , httpLbs@@ -26,21 +31,21 @@ ) where import Control.Monad (unless)-import Control.Monad.IO.Class (MonadIO, liftIO)-import Control.Monad.Reader (MonadReader (..), ReaderT (..))-import Control.Monad.Trans.Control (MonadBaseControl)+import Control.Monad.IO.Unlift (MonadIO, liftIO, MonadUnliftIO, withRunInIO)+import Control.Monad.Reader (MonadReader (..), runReaderT)+import Control.Monad.Trans.Resource (MonadResource) import Data.Acquire (Acquire, mkAcquire, with) import Data.ByteString (ByteString) import qualified Data.ByteString as S import qualified Data.ByteString.Lazy as L-import Data.Conduit (ConduitM, Producer, Source,- await, yield, ($$+), ($$++))+import Data.Conduit (ConduitM, ($$+), ($$++),+ await, yield, bracketP) import Data.Int (Int64) import Data.IORef (newIORef, readIORef, writeIORef) import Network.HTTP.Client hiding (closeManager, defaultManagerSettings, httpLbs, newManager, responseClose,- responseOpen, withManager,+ responseOpen, withResponse, BodyReader, brRead, brConsume, httpNoBody) import qualified Network.HTTP.Client as H import Network.HTTP.Client.TLS (tlsManagerSettings)@@ -49,18 +54,18 @@ -- -- * Response body is represented as a @Producer@. ----- * Generalized to any instance of @MonadBaseControl@, not just @IO@.+-- * Generalized to any instance of @MonadUnliftIO@, not just @IO@. -- -- * The @Manager@ is contained by a @MonadReader@ context. -- -- Since 2.1.0-withResponse :: (MonadBaseControl IO m, MonadIO n, MonadReader env m, HasHttpManager env)+withResponse :: (MonadUnliftIO m, MonadIO n, MonadReader env m, HasHttpManager env) => Request -> (Response (ConduitM i ByteString n ()) -> m a) -> m a withResponse req f = do env <- ask- with (acquireResponse req env) f+ withRunInIO $ \run -> with (acquireResponse req env) (run . f) -- | An @Acquire@ for getting a @Response@. --@@ -93,18 +98,6 @@ newManagerSettings :: MonadIO m => ManagerSettings -> m Manager newManagerSettings = liftIO . H.newManager --- | Get a new manager with 'defaultManagerSettings' and construct a @ReaderT@ containing it.------ Since 2.1.0-withManager :: MonadIO m => (ReaderT Manager m a) -> m a-withManager = withManagerSettings defaultManagerSettings---- | Get a new manager with the given settings and construct a @ReaderT@ containing it.------ Since 2.1.0-withManagerSettings :: MonadIO m => ManagerSettings -> (ReaderT Manager m a) -> m a-withManagerSettings settings (ReaderT inner) = newManagerSettings settings >>= inner- -- | Conduit-powered version of 'H.responseOpen'. -- -- See 'withResponse' for the differences with 'H.responseOpen'.@@ -123,14 +116,9 @@ responseClose :: MonadIO m => Response body -> m () responseClose = liftIO . H.responseClose -class HasHttpManager a where- getHttpManager :: a -> Manager-instance HasHttpManager Manager where- getHttpManager = id- bodyReaderSource :: MonadIO m => H.BodyReader- -> Producer m ByteString+ -> ConduitM i ByteString m () bodyReaderSource br = loop where@@ -140,13 +128,13 @@ yield bs loop -requestBodySource :: Int64 -> Source IO ByteString -> RequestBody+requestBodySource :: Int64 -> ConduitM () ByteString IO () -> RequestBody requestBodySource size = RequestBodyStream size . srcToPopperIO -requestBodySourceChunked :: Source IO ByteString -> RequestBody+requestBodySourceChunked :: ConduitM () ByteString IO () -> RequestBody requestBodySourceChunked = RequestBodyStreamChunked . srcToPopperIO -srcToPopperIO :: Source IO ByteString -> GivesPopper ()+srcToPopperIO :: ConduitM () ByteString IO () -> GivesPopper () srcToPopperIO src f = do (rsrc0, ()) <- src $$+ return () irsrc <- newIORef rsrc0@@ -186,3 +174,19 @@ env <- ask let man = getHttpManager env liftIO $ H.httpNoBody req man++-- | Same as 'Network.HTTP.Simple.httpSource', but uses 'Manager'+-- from Reader environment instead of the global one.+--+-- Since 2.3.6+httpSource+ :: (MonadResource m, MonadIO n, MonadReader env m, HasHttpManager env)+ => Request+ -> (Response (ConduitM () ByteString n ()) -> ConduitM () r m ())+ -> ConduitM () r m ()+httpSource request withRes = do+ env <- ask+ bracketP+ (runReaderT (responseOpen request) env)+ responseClose+ withRes
Network/HTTP/Conduit.hs view
@@ -1,7 +1,20 @@-{-# LANGUAGE FlexibleContexts #-}-{-# LANGUAGE MultiParamTypeClasses #-}-{-# LANGUAGE OverloadedStrings #-}--- | This module contains everything you need to initiate HTTP connections. If+{-# LANGUAGE CPP #-}+{-# LANGUAGE FlexibleContexts #-}+{-# LANGUAGE OverloadedStrings #-}+-- |+--+-- = Simpler API+--+-- The API below is rather low-level. The "Network.HTTP.Simple" module provides+-- a higher-level API with built-in support for things like JSON request and+-- response bodies. For most users, this will be an easier place to start. You+-- can read the tutorial at:+--+-- <https://haskell-lang.org/library/http-client>+--+-- = Lower-level API+--+-- This module contains everything you need to initiate HTTP connections. If -- you want a simple interface based on URLs, you can use 'simpleHttp'. If you -- want raw power, 'http' is the underlying workhorse of this package. Some -- examples:@@ -15,16 +28,18 @@ -- This example uses interleaved IO to write the response body to a file in -- constant memory space. ----- > import Data.Conduit.Binary (sinkFile)+-- > import Data.Conduit.Binary (sinkFile) -- Exported from the package conduit-extra -- > import Network.HTTP.Conduit--- > import qualified Data.Conduit as C+-- > import Conduit (runConduit, (.|))+-- > import Control.Monad.Trans.Resource (runResourceT) -- > -- > main :: IO () -- > main = do--- > request <- parseUrl "http://google.com/"--- > withManager $ \manager -> do+-- > request <- parseRequest "http://google.com/"+-- > manager <- newManager tlsManagerSettings+-- > runResourceT $ do -- > response <- http request manager--- > responseBody response C.$$+- sinkFile "google.html"+-- > runConduit $ responseBody response .| sinkFile "google.html" -- -- The following headers are automatically set by this module, and should not -- be added to 'requestHeaders':@@ -49,6 +64,7 @@ -- > import Data.Time.Clock -- > import Data.Time.Calendar -- > import qualified Control.Exception as E+-- > import Network.HTTP.Types.Status (statusCode) -- > -- > past :: UTCTime -- > past = UTCTime (ModifiedJulianDay 56200) (secondsToDiffTime 0)@@ -70,25 +86,17 @@ -- > , cookie_http_only = False -- > } -- >--- > main = withSocketsDo $ do--- > request' <- parseUrl "http://example.com/secret-page"+-- > main = do+-- > request' <- parseRequest "http://example.com/secret-page"+-- > manager <- newManager tlsManagerSettings -- > let request = request' { cookieJar = Just $ createCookieJar [cookie] }--- > E.catch (withManager $ httpLbs request)--- > (\(StatusCodeException s _ _) ->--- > if statusCode==403 then putStrLn "login failed" else return ())------ Any network code on Windows requires some initialization, and the network--- library provides withSocketsDo to perform it. Therefore, proper usage of--- this library will always involve calling that function at some point. The--- best approach is to simply call them at the beginning of your main function,--- such as:------ > import Network.HTTP.Conduit--- > import qualified Data.ByteString.Lazy as L--- > import Network (withSocketsDo)--- >--- > main = withSocketsDo--- > $ simpleHttp "http://www.haskell.org/" >>= L.putStr+-- > fmap Just (httpLbs request manager) `E.catch`+-- > (\ex -> case ex of+-- > HttpExceptionRequest _ (StatusCodeException res _) ->+-- > if statusCode (responseStatus res) == 403+-- > then (putStrLn "login failed" >> return Nothing)+-- > else return Nothing+-- > _ -> E.throw ex) -- -- Cookies are implemented according to RFC 6265. --@@ -102,10 +110,11 @@ -- > import Network -- > -- > main :: IO ()--- > main = withSocketsDo $ do--- > request' <- parseUrl "http://www.yesodweb.com/does-not-exist"+-- > main = do+-- > request' <- parseRequest "http://www.yesodweb.com/does-not-exist" -- > let request = request' { checkStatus = \_ _ _ -> Nothing }--- > res <- withManager $ httpLbs request+-- > manager <- newManager tlsManagerSettings+-- > res <- httpLbs request manager -- > print res -- -- By default, when connecting to websites using HTTPS, functions in this@@ -118,10 +127,14 @@ -- > -- > main :: IO () -- > main = do--- > request <- parseUrl "https://github.com/"+-- > request <- parseRequest "https://github.com/" -- > let settings = mkManagerSettings (TLSSettingsSimple True False False) Nothing--- > res <- withManagerSettings settings $ httpLbs request+-- > manager <- newManager settings+-- > res <- httpLbs request manager -- > print res+--+-- For more information, please be sure to read the documentation in the+-- "Network.HTTP.Client" module. module Network.HTTP.Conduit ( -- * Perform a request@@ -146,10 +159,13 @@ , rawBody , decompress , redirectCount- , checkStatus+#if MIN_VERSION_http_client(0,6,2)+ , shouldStripHeaderOnRedirect+#endif+ , checkResponse , responseTimeout , cookieJar- , getConnectionWrapper+ , requestVersion , HCC.setQueryString -- *** Request body , requestBodySource@@ -163,19 +179,23 @@ , responseHeaders , responseBody , responseCookieJar+ , responseEarlyHints -- * Manager , Manager , newManager , closeManager- , withManager- , withManagerSettings -- ** Settings , ManagerSettings- , conduitManagerSettings+ , tlsManagerSettings , mkManagerSettings , managerConnCount , managerResponseTimeout , managerTlsConnection+ -- ** Response timeout+ , HC.ResponseTimeout+ , HC.responseTimeoutMicro+ , HC.responseTimeoutNone+ , HC.responseTimeoutDefault -- * Cookies , Cookie(..) , CookieJar@@ -183,6 +203,10 @@ , destroyCookieJar -- * Utility functions , parseUrl+ , parseUrlThrow+ , parseRequest+ , parseRequest_+ , defaultRequest , applyBasicAuth , addProxy , lbsResponse@@ -191,26 +215,26 @@ , alwaysDecompress , browserDecompress -- * Request bodies- -- | "Network.HTTP.Conduit.MultipartFormData" provides an API for building+ -- | "Network.HTTP.Client.MultipartFormData" provides an API for building -- form-data request bodies. , urlEncodedBody -- * Exceptions , HttpException (..)+ , HCC.HttpExceptionContent (..) ) where import qualified Data.ByteString as S import qualified Data.ByteString.Lazy as L-import Data.Conduit (ResumableSource, ($$+-), await, ($$++), ($$+), Source)-import qualified Data.Conduit.Internal as CI+import Data.Conduit import qualified Data.Conduit.List as CL import Data.IORef (readIORef, writeIORef, newIORef) import Data.Int (Int64)-import Control.Applicative ((<$>))-import Control.Exception.Lifted (bracket)-import Control.Monad.IO.Class (MonadIO (liftIO))+import Control.Applicative as A ((<$>))+import Control.Monad.IO.Unlift (MonadIO (liftIO)) import Control.Monad.Trans.Resource -import qualified Network.HTTP.Client as Client (httpLbs, responseOpen, responseClose)+import qualified Network.HTTP.Client as Client (httpLbs, responseOpen)+import qualified Network.HTTP.Client as HC import qualified Network.HTTP.Client.Conduit as HCC import Network.HTTP.Client.Internal (createCookieJar, destroyCookieJar)@@ -218,10 +242,10 @@ closeManager, managerConnCount, managerResponseTimeout, managerTlsConnection, newManager)-import Network.HTTP.Client (parseUrl, urlEncodedBody, applyBasicAuth)-import Network.HTTP.Client.Internal (addProxy, alwaysDecompress,- browserDecompress)-import Network.HTTP.Client.Internal (getRedirectedRequest)+import Network.HTTP.Client (parseUrl, parseUrlThrow, urlEncodedBody, applyBasicAuth,+ defaultRequest, parseRequest, parseRequest_)+import Network.HTTP.Client.Internal (ResponseClose (..), addProxy, alwaysDecompress,+ browserDecompress, getRedirectedRequest) import Network.HTTP.Client.TLS (mkManagerSettings, tlsManagerSettings) import Network.HTTP.Client.Internal (Cookie (..), CookieJar (..),@@ -244,6 +268,11 @@ -- usage, you'll need to use @conduit@ packages's -- 'C.Source' returned by 'http'. --+-- This function will 'throwIO' an 'HttpException' for any+-- response with a non-2xx status code (besides 3xx redirects up+-- to a limit of 10 redirects). This behavior can be modified by+-- changing the 'checkStatus' field of your request.+-- -- Note: Unlike previous versions, this function will perform redirects, as -- specified by the 'redirectCount' setting. httpLbs :: MonadIO m => Request -> Manager -> m (Response L.ByteString)@@ -254,7 +283,7 @@ -- -- This function will 'throwIO' an 'HttpException' for any -- response with a non-2xx status code (besides 3xx redirects up--- to a limit of 10 redirects). It uses 'parseUrl' to parse the+-- to a limit of 10 redirects). It uses 'parseUrlThrow' to parse the -- input. This function essentially wraps 'httpLbs'. -- -- Note: Even though this function returns a lazy bytestring, it@@ -265,35 +294,19 @@ -- Note: This function creates a new 'Manager'. It should be avoided -- in production code. simpleHttp :: MonadIO m => String -> m L.ByteString-simpleHttp url = liftIO $ withManager $ \man -> do- req <- liftIO $ parseUrl url- responseBody <$> httpLbs (setConnectionClose req) man--conduitManagerSettings :: ManagerSettings-conduitManagerSettings = tlsManagerSettings--withManager :: (MonadIO m, MonadBaseControl IO m)- => (Manager -> ResourceT m a)- -> m a-withManager = withManagerSettings conduitManagerSettings--withManagerSettings :: (MonadIO m, MonadBaseControl IO m)- => ManagerSettings- -> (Manager -> ResourceT m a)- -> m a-withManagerSettings set f = bracket- (liftIO $ newManager set)- (liftIO . closeManager)- (runResourceT . f)+simpleHttp url = liftIO $ do+ man <- newManager tlsManagerSettings+ req <- liftIO $ parseUrlThrow url+ responseBody A.<$> httpLbs (setConnectionClose req) man setConnectionClose :: Request -> Request setConnectionClose req = req{requestHeaders = ("Connection", "close") : requestHeaders req} lbsResponse :: Monad m- => Response (ResumableSource m S.ByteString)+ => Response (ConduitM () S.ByteString m ()) -> m (Response L.ByteString) lbsResponse res = do- bss <- responseBody res $$+- CL.consume+ bss <- runConduit $ responseBody res .| CL.consume return res { responseBody = L.fromChunks bss }@@ -301,21 +314,27 @@ http :: MonadResource m => Request -> Manager- -> m (Response (ResumableSource m S.ByteString))-http req man = do- (key, res) <- allocate (Client.responseOpen req man) Client.responseClose- let rsrc = CI.ResumableSource- (HCC.bodyReaderSource $ responseBody res)- (release key)- return res { responseBody = rsrc }+ -> m (Response (ConduitM i S.ByteString m ()))+http req man = resourceMask $ \_ -> do+ res <- liftIO $ Client.responseOpen req man+ -- Move the cleanup action for the response into `ResourceT` so+ -- that we can release it from the `ReleaseMap` as soon as the+ -- response is closed or the body is consumed.+ let ResponseClose cleanup = responseClose' res+ key <- register cleanup+ pure res { responseClose' = ResponseClose $ release key+ , responseBody = do+ HCC.bodyReaderSource $ responseBody res+ release key+ } -requestBodySource :: Int64 -> Source (ResourceT IO) S.ByteString -> RequestBody+requestBodySource :: Int64 -> ConduitM () S.ByteString (ResourceT IO) () -> RequestBody requestBodySource size = RequestBodyStream size . srcToPopper -requestBodySourceChunked :: Source (ResourceT IO) S.ByteString -> RequestBody+requestBodySourceChunked :: ConduitM () S.ByteString (ResourceT IO) () -> RequestBody requestBodySourceChunked = RequestBodyStreamChunked . srcToPopper -srcToPopper :: Source (ResourceT IO) S.ByteString -> HCC.GivesPopper ()+srcToPopper :: ConduitM () S.ByteString (ResourceT IO) () -> HCC.GivesPopper () srcToPopper src f = runResourceT $ do (rsrc0, ()) <- src $$+ return () irsrc <- liftIO $ newIORef rsrc0@@ -332,8 +351,8 @@ | otherwise -> return bs liftIO $ f popper -requestBodySourceIO :: Int64 -> Source IO S.ByteString -> RequestBody+requestBodySourceIO :: Int64 -> ConduitM () S.ByteString IO () -> RequestBody requestBodySourceIO = HCC.requestBodySource -requestBodySourceChunkedIO :: Source IO S.ByteString -> RequestBody+requestBodySourceChunkedIO :: ConduitM () S.ByteString IO () -> RequestBody requestBodySourceChunkedIO = HCC.requestBodySourceChunked
+ Network/HTTP/Simple.hs view
@@ -0,0 +1,521 @@+{-# LANGUAGE CPP #-}+{-# LANGUAGE DeriveDataTypeable #-}+{-# LANGUAGE RankNTypes #-}+{-# LANGUAGE TupleSections #-}+{-# LANGUAGE OverloadedStrings #-}+-- | Simplified interface for common HTTP client interactions. Tutorial+-- available at+-- <https://haskell-lang.org/library/http-client>+--+-- Important note: 'H.Request' is an instance of 'Data.String.IsString', and+-- therefore recommended usage is to turn on @OverloadedStrings@, e.g.+--+-- > {-# LANGUAGE OverloadedStrings #-}+-- > import Network.HTTP.Simple+-- > import qualified Data.ByteString.Char8 as B8+-- >+-- > main :: IO ()+-- > main = httpBS "http://example.com" >>= B8.putStrLn . getResponseBody+--+-- The `Data.String.IsString` instance uses `H.parseRequest` behind the scenes and inherits its behavior.+module Network.HTTP.Simple+ ( -- * Perform requests+ httpBS+ , httpLBS+ , httpNoBody+#ifdef VERSION_aeson+ , httpJSON+ , httpJSONEither+#endif+ , httpSink+ , httpSource+ , withResponse+ -- * Types+ , H.Header+ , H.Query+ , H.QueryItem+ , H.Request+ , H.RequestHeaders+ , H.Response+ , H.ResponseHeaders+#ifdef VERSION_aeson+ , JSONException (..)+#endif+ , H.HttpException (..)+ , H.Proxy (..)+ -- * Request constructions+ , H.defaultRequest+ , H.parseRequest+ , H.parseRequest_+ , parseRequestThrow+ , parseRequestThrow_+ -- * Request lenses+ -- ** Basics+ , setRequestMethod+ , setRequestSecure+ , setRequestHost+ , setRequestPort+ , setRequestPath+ , addRequestHeader+ , getRequestHeader+ , setRequestHeader+ , setRequestHeaders+ , setRequestQueryString+ , getRequestQueryString+ , addToRequestQueryString+ -- ** Request body+ , setRequestBody+#ifdef VERSION_aeson+ , setRequestBodyJSON+#endif+ , setRequestBodyLBS+ , setRequestBodySource+ , setRequestBodyFile+ , setRequestBodyURLEncoded+ -- ** Special fields+ , H.setRequestIgnoreStatus+ , H.setRequestCheckStatus+ , setRequestBasicAuth+#if MIN_VERSION_http_client(0,7,6)+ , setRequestBearerAuth+#endif+ , setRequestManager+ , setRequestProxy+ , setRequestResponseTimeout+ -- * Response lenses+ , getResponseStatus+ , getResponseStatusCode+ , getResponseHeader+ , getResponseHeaders+ , getResponseBody+ -- * Alternate spellings+ , httpLbs+ ) where++import qualified Data.ByteString as S+import qualified Data.ByteString.Lazy as L+import qualified Network.HTTP.Client as H+import qualified Network.HTTP.Client.Internal as HI+import qualified Network.HTTP.Client.TLS as H+import Network.HTTP.Client.Conduit (bodyReaderSource)+import qualified Network.HTTP.Client.Conduit as HC+import Control.Monad.IO.Unlift (MonadIO, liftIO, MonadUnliftIO, withRunInIO)++#ifdef VERSION_aeson+import Data.Aeson (FromJSON (..), Value)+import Data.Aeson.Parser (json')+import qualified Data.Aeson.Types as A+import qualified Data.Aeson as A+#endif++import qualified Data.Traversable as T+import Control.Exception (throw, throwIO, Exception)+import Data.Monoid+import Data.Typeable (Typeable)+import qualified Data.Conduit as C+import Data.Conduit (runConduit, (.|), ConduitM)+import qualified Data.Conduit.Attoparsec as C+import qualified Network.HTTP.Types as H+import Data.Int (Int64)+import Control.Monad.Trans.Resource (MonadResource, MonadThrow)+import qualified Control.Exception as E (bracket)+import Data.Void (Void)+import qualified Data.Attoparsec.ByteString as Atto+import qualified Data.Attoparsec.ByteString.Char8 as Atto8++-- | Perform an HTTP request and return the body as a @ByteString@.+--+-- @since 2.2.4+httpBS :: MonadIO m => H.Request -> m (H.Response S.ByteString)+httpBS req = liftIO $ do+ man <- H.getGlobalManager+ fmap L.toStrict `fmap` H.httpLbs req man++-- | Perform an HTTP request and return the body as a lazy+-- @ByteString@. Note that the entire value will be read into memory+-- at once (no lazy I\/O will be performed). The advantage of a lazy+-- @ByteString@ here (versus using 'httpBS') is--if needed--a better+-- in-memory representation.+--+-- @since 2.1.10+httpLBS :: MonadIO m => H.Request -> m (H.Response L.ByteString)+httpLBS req = liftIO $ do+ man <- H.getGlobalManager+ H.httpLbs req man++-- | Perform an HTTP request and ignore the response body.+--+-- @since 2.2.2+httpNoBody :: MonadIO m => H.Request -> m (H.Response ())+httpNoBody req = liftIO $ do+ man <- H.getGlobalManager+ H.httpNoBody req man++#ifdef VERSION_aeson+-- | Perform an HTTP request and parse the body as JSON. In the event of an+-- JSON parse errors, a 'JSONException' runtime exception will be thrown.+--+-- NOTE: Depends on the @aeson@ cabal flag being enabled+--+-- @since 2.1.10+httpJSON :: (MonadIO m, FromJSON a) => H.Request -> m (H.Response a)+httpJSON req = liftIO $ httpJSONEither req >>= T.mapM (either throwIO return)++-- | Perform an HTTP request and parse the body as JSON. In the event of an+-- JSON parse errors, a @Left@ value will be returned.+--+-- NOTE: Depends on the @aeson@ cabal flag being enabled+--+-- @since 2.1.10+httpJSONEither :: (MonadIO m, FromJSON a)+ => H.Request+ -> m (H.Response (Either JSONException a))+httpJSONEither req = liftIO $ httpSink req' sink+ where+ req' = addRequestHeader H.hAccept "application/json" req+ sink orig = fmap (\x -> fmap (const x) orig) $ do+ eres1 <- C.sinkParserEither (json' <* (Atto8.skipSpace *> Atto.endOfInput))++ case eres1 of+ Left e -> return $ Left $ JSONParseException req' orig e+ Right value ->+ case A.fromJSON value of+ A.Error e -> return $ Left $ JSONConversionException+ req' (fmap (const value) orig) e+ A.Success x -> return $ Right x++-- | An exception that can occur when parsing JSON+--+-- NOTE: Depends on the @aeson@ cabal flag being enabled+--+-- @since 2.1.10+data JSONException+ = JSONParseException H.Request (H.Response ()) C.ParseError+ | JSONConversionException H.Request (H.Response Value) String+ deriving (Show, Typeable)+instance Exception JSONException+#endif++-- | Perform an HTTP request and consume the body with the given 'C.Sink'+--+-- @since 2.1.10+httpSink :: MonadUnliftIO m+ => H.Request+ -> (H.Response () -> ConduitM S.ByteString Void m a)+ -> m a+httpSink req sink = withRunInIO $ \run -> do+ man <- H.getGlobalManager+ E.bracket+ (H.responseOpen req man)+ H.responseClose+ $ \res -> run+ $ runConduit+ $ bodyReaderSource (getResponseBody res)+ .| sink (fmap (const ()) res)++-- | Perform an HTTP request, and get the response body as a Source.+--+-- The second argument to this function tells us how to make the+-- Source from the Response itself. This allows you to perform actions+-- with the status or headers, for example, in addition to the raw+-- bytes themselves. If you just care about the response body, you can+-- use 'getResponseBody' as the second argument here.+--+-- @+-- \{\-# LANGUAGE OverloadedStrings \#\-}+-- import Control.Monad.IO.Class (liftIO)+-- import Control.Monad.Trans.Resource (runResourceT)+-- import Data.Conduit (($$))+-- import qualified Data.Conduit.Binary as CB+-- import qualified Data.Conduit.List as CL+-- import Network.HTTP.Simple+-- import System.IO (stdout)+--+-- main :: IO ()+-- main =+-- runResourceT+-- $ httpSource "http://httpbin.org/robots.txt" getSrc+-- $$ CB.sinkHandle stdout+-- where+-- getSrc res = do+-- liftIO $ print (getResponseStatus res, getResponseHeaders res)+-- getResponseBody res+-- @+--+-- @since 2.2.1+httpSource :: (MonadResource m, MonadIO n)+ => H.Request+ -> (H.Response (C.ConduitM i S.ByteString n ())+ -> C.ConduitM i o m r)+ -> C.ConduitM i o m r+httpSource req withRes = do+ man <- liftIO H.getGlobalManager+ C.bracketP (H.responseOpen req man) H.responseClose+ (withRes . fmap bodyReaderSource)++-- | Perform an action with the given request. This employes the+-- bracket pattern.+--+-- This is similar to 'httpSource', but does not require+-- 'MonadResource' and allows the result to not contain a 'C.ConduitM'+-- value.+--+-- @since 2.2.3+withResponse :: (MonadUnliftIO m, MonadIO n)+ => H.Request+ -> (H.Response (C.ConduitM i S.ByteString n ()) -> m a)+ -> m a+withResponse req withRes = withRunInIO $ \run -> do+ man <- H.getGlobalManager+ E.bracket+ (H.responseOpen req man)+ H.responseClose+ (run . withRes . fmap bodyReaderSource)++-- | Same as 'parseRequest', except will throw an 'HttpException' in the+-- event of a non-2XX response. This uses 'throwErrorStatusCodes' to+-- implement 'checkResponse'.+--+-- Exactly the same as 'parseUrlThrow', but has a name that is more+-- consistent with the other parseRequest functions.+--+-- @since 2.3.2+parseRequestThrow :: MonadThrow m => String -> m HC.Request+parseRequestThrow = HC.parseUrlThrow++-- | Same as 'parseRequestThrow', but parse errors cause an impure+-- exception. Mostly useful for static strings which are known to be+-- correctly formatted.+--+-- @since 2.3.2+parseRequestThrow_ :: String -> HC.Request+parseRequestThrow_ = either throw id . HC.parseUrlThrow++-- | Alternate spelling of 'httpLBS'+--+-- @since 2.1.10+httpLbs :: MonadIO m => H.Request -> m (H.Response L.ByteString)+httpLbs = httpLBS++-- | Set the request method+--+-- @since 2.1.10+setRequestMethod :: S.ByteString -> H.Request -> H.Request+setRequestMethod x req = req { H.method = x }++-- | Set whether this is a secure/HTTPS (@True@) or insecure/HTTP+-- (@False@) request+--+-- @since 2.1.10+setRequestSecure :: Bool -> H.Request -> H.Request+setRequestSecure x req = req { H.secure = x }++-- | Set the destination host of the request+--+-- @since 2.1.10+setRequestHost :: S.ByteString -> H.Request -> H.Request+setRequestHost x r = r { H.host = x }++-- | Set the destination port of the request+--+-- @since 2.1.10+setRequestPort :: Int -> H.Request -> H.Request+setRequestPort x r = r { H.port = x }++-- | Lens for the requested path info of the request+--+-- @since 2.1.10+setRequestPath :: S.ByteString -> H.Request -> H.Request+setRequestPath x r = r { H.path = x }++-- | Add a request header name/value combination+--+-- @since 2.1.10+addRequestHeader :: H.HeaderName -> S.ByteString -> H.Request -> H.Request+addRequestHeader name val req =+ req { H.requestHeaders = (name, val) : H.requestHeaders req }++-- | Get all request header values for the given name+--+-- @since 2.1.10+getRequestHeader :: H.HeaderName -> H.Request -> [S.ByteString]+getRequestHeader name =+ map snd . filter (\(x, _) -> x == name) . H.requestHeaders++-- | Set the given request header to the given list of values. Removes any+-- previously set header values with the same name.+--+-- @since 2.1.10+setRequestHeader :: H.HeaderName -> [S.ByteString] -> H.Request -> H.Request+setRequestHeader name vals req =+ req { H.requestHeaders =+ filter (\(x, _) -> x /= name) (H.requestHeaders req)+ ++ (map (name, ) vals)+ }++-- | Set the request headers, wiping out __all__ previously set headers. This+-- means if you use 'setRequestHeaders' to set some headers and also use one of+-- the other setters that modifies the @content-type@ header (such as+-- 'setRequestBodyJSON'), be sure that 'setRequestHeaders' is evaluated+-- __first__.+--+-- @since 2.1.10+setRequestHeaders :: H.RequestHeaders -> H.Request -> H.Request+setRequestHeaders x req = req { H.requestHeaders = x }++-- | Get the query string parameters+--+-- @since 2.1.10+getRequestQueryString :: H.Request -> H.Query+getRequestQueryString = H.parseQuery . H.queryString++-- | Set the query string parameters+--+-- @since 2.1.10+setRequestQueryString :: H.Query -> H.Request -> H.Request+setRequestQueryString = H.setQueryString++-- | Add to the existing query string parameters.+--+-- @since 2.3.5+addToRequestQueryString :: H.Query -> H.Request -> H.Request+addToRequestQueryString additions req = setRequestQueryString q req+ where q = additions <> getRequestQueryString req++-- | Set the request body to the given 'H.RequestBody'. You may want to+-- consider using one of the convenience functions in the modules, e.g.+-- 'requestBodyJSON'.+--+-- /Note/: This will not modify the request method. For that, please use+-- 'requestMethod'. You likely don't want the default of @GET@.+--+-- @since 2.1.10+setRequestBody :: H.RequestBody -> H.Request -> H.Request+setRequestBody x req = req { H.requestBody = x }++#ifdef VERSION_aeson+-- | Set the request body as a JSON value+--+-- /Note/: This will not modify the request method. For that, please use+-- 'requestMethod'. You likely don't want the default of @GET@.+--+-- This also sets the @Content-Type@ to @application/json; charset=utf-8@+--+-- NOTE: Depends on the @aeson@ cabal flag being enabled+--+-- @since 2.1.10+setRequestBodyJSON :: A.ToJSON a => a -> H.Request -> H.Request+setRequestBodyJSON x req =+ req { H.requestHeaders+ = (H.hContentType, "application/json; charset=utf-8")+ : filter (\(y, _) -> y /= H.hContentType) (H.requestHeaders req)+ , H.requestBody = H.RequestBodyLBS $ A.encode x+ }+#endif++-- | Set the request body as a lazy @ByteString@+--+-- /Note/: This will not modify the request method. For that, please use+-- 'requestMethod'. You likely don't want the default of @GET@.+--+-- @since 2.1.10+setRequestBodyLBS :: L.ByteString -> H.Request -> H.Request+setRequestBodyLBS = setRequestBody . H.RequestBodyLBS++-- | Set the request body as a 'C.Source'+--+-- /Note/: This will not modify the request method. For that, please use+-- 'requestMethod'. You likely don't want the default of @GET@.+--+-- @since 2.1.10+setRequestBodySource :: Int64 -- ^ length of source+ -> ConduitM () S.ByteString IO ()+ -> H.Request+ -> H.Request+setRequestBodySource len src req = req { H.requestBody = HC.requestBodySource len src }++-- | Set the request body as a file+--+-- /Note/: This will not modify the request method. For that, please use+-- 'requestMethod'. You likely don't want the default of @GET@.+--+-- @since 2.1.10+setRequestBodyFile :: FilePath -> H.Request -> H.Request+setRequestBodyFile = setRequestBody . HI.RequestBodyIO . H.streamFile++-- | Set the request body as URL encoded data+--+-- /Note/: This will change the request method to @POST@ and set the @content-type@+-- to @application/x-www-form-urlencoded@+--+-- @since 2.1.10+setRequestBodyURLEncoded :: [(S.ByteString, S.ByteString)] -> H.Request -> H.Request+setRequestBodyURLEncoded = H.urlEncodedBody++-- | Set basic auth with the given username and password+--+-- @since 2.1.10+setRequestBasicAuth :: S.ByteString -- ^ username+ -> S.ByteString -- ^ password+ -> H.Request+ -> H.Request+setRequestBasicAuth = H.applyBasicAuth++#if MIN_VERSION_http_client(0,7,6)+-- | Set bearer auth with the given token+--+-- @since 2.3.8+setRequestBearerAuth :: S.ByteString -- ^ token+ -> H.Request+ -> H.Request+setRequestBearerAuth = H.applyBearerAuth+#endif++-- | Instead of using the default global 'H.Manager', use the supplied+-- @Manager@.+--+-- @since 2.1.10+setRequestManager :: H.Manager -> H.Request -> H.Request+setRequestManager x req = req { HI.requestManagerOverride = Just x }++-- | Override the default proxy server settings+--+-- @since 2.1.10+setRequestProxy :: Maybe H.Proxy -> H.Request -> H.Request+setRequestProxy x req = req { H.proxy = x }++-- | Set the maximum time to wait for a response+--+-- @since 2.3.8+setRequestResponseTimeout :: H.ResponseTimeout -> H.Request -> H.Request+setRequestResponseTimeout x req = req { H.responseTimeout = x }++-- | Get the status of the response+--+-- @since 2.1.10+getResponseStatus :: H.Response a -> H.Status+getResponseStatus = H.responseStatus++-- | Get the integral status code of the response+--+-- @since 2.1.10+getResponseStatusCode :: H.Response a -> Int+getResponseStatusCode = H.statusCode . getResponseStatus++-- | Get all response header values with the given name+--+-- @since 2.1.10+getResponseHeader :: H.HeaderName -> H.Response a -> [S.ByteString]+getResponseHeader name = map snd . filter (\(x, _) -> x == name) . H.responseHeaders++-- | Get all response headers+--+-- @since 2.1.10+getResponseHeaders :: H.Response a -> [(H.HeaderName, S.ByteString)]+getResponseHeaders = H.responseHeaders++-- | Get the response body+--+-- @since 2.1.10+getResponseBody :: H.Response a -> a+getResponseBody = H.responseBody
+ README.md view
@@ -0,0 +1,10 @@+http-conduit+============++Provides for making efficient HTTP/HTTPS requests, providing either a simple or+streaming interface.++Full tutorial docs are available at:+https://github.com/snoyberg/http-client/blob/master/TUTORIAL.md++The `Network.HTTP.Conduit.Browser` module has been moved to <http://hackage.haskell.org/package/http-conduit-browser/>
http-conduit.cabal view
@@ -1,44 +1,60 @@+cabal-version: >= 1.10 name: http-conduit-version: 2.1.4+version: 2.3.9.1 license: BSD3 license-file: LICENSE author: Michael Snoyman <michael@snoyman.com> maintainer: Michael Snoyman <michael@snoyman.com> synopsis: HTTP client package with conduit interface and HTTPS support.-description:- This package uses conduit for parsing the actual contents of the HTTP connection. It also provides higher-level functions which allow you to avoid directly dealing with streaming data. See <http://www.yesodweb.com/book/http-conduit> for more information.- .- The @Network.HTTP.Conduit.Browser@ module has been moved to <http://hackage.haskell.org/package/http-conduit-browser/>+description: Hackage documentation generation is not reliable. For up to date documentation, please see: <http://www.stackage.org/package/http-conduit>. category: Web, Conduit stability: Stable-cabal-version: >= 1.8 build-type: Simple-homepage: http://www.yesodweb.com/book/http-conduit+homepage: https://github.com/snoyberg/http-client extra-source-files: test/main.hs , test/CookieTest.hs , multipart-example.bin , nyan.gif , certificate.pem , key.pem+ , README.md+ , ChangeLog.md +flag aeson+ manual: True+ description: Enable the dependency on aeson+ default: True+ library- build-depends: base >= 4 && < 5+ default-language: Haskell2010+ build-depends: base >= 4.10 && < 5+ , attoparsec , bytestring >= 0.9.1.4 , transformers >= 0.2- , resourcet >= 1.1 && < 1.2- , conduit >= 0.5.5 && < 1.2+ , resourcet >= 1.1+ , conduit >= 1.2+ , conduit-extra >= 1.1 , http-types >= 0.7- , lifted-base >= 0.1- , http-client >= 0.3.6 && < 0.4- , http-client-tls >= 0.2.2- , monad-control+ , http-client >= 0.5.13 && < 0.8+ , http-client-tls >= 0.3 && < 0.4 , mtl+ , unliftio-core++ if flag(aeson)+ build-depends: aeson >= 0.8+ , attoparsec-aeson >= 2.1++ if !impl(ghc>=7.9)+ build-depends: void >= 0.5.5 exposed-modules: Network.HTTP.Conduit Network.HTTP.Client.Conduit+ Network.HTTP.Simple ghc-options: -Wall test-suite test+ default-language: Haskell2010 main-is: main.hs+ other-modules: CookieTest type: exitcode-stdio-1.0 hs-source-dirs: test @@ -47,29 +63,35 @@ build-depends: base >= 4 && < 5 , HUnit , hspec >= 1.3- , data-default-class- , connection >= 0.2+ , data-default+ , crypton-connection , warp-tls+ , tls < 1.5 || >= 1.5.2 , time , blaze-builder , bytestring , text , transformers- , conduit+ , conduit >= 1.1 , utf8-string , case-insensitive- , lifted-base- , network- , wai >= 3.0 && < 3.1- , warp >= 3.0.0.2 && < 3.1+ , unliftio+ , wai >= 3.0 && < 3.3+ , warp >= 3.0.0.2 && < 3.4 , wai-conduit , http-types , cookie- , network-conduit >= 0.6 , http-client , http-conduit , conduit-extra , streaming-commons+ , temporary+ , resourcet+ , network++ if flag(aeson)+ build-depends: aeson+ , attoparsec-aeson >= 2.1 source-repository head type: git
test/CookieTest.hs view
@@ -1,3 +1,5 @@+{-# OPTIONS_GHC -Wno-orphans #-}+ module CookieTest (cookieTest) where import Prelude hiding (exp)@@ -8,14 +10,32 @@ import qualified Network.HTTP.Conduit as HC import Data.ByteString.UTF8 import Data.Monoid-import Data.Maybe import Data.Time.Clock import Data.Time.Calendar import qualified Data.CaseInsensitive as CI import Web.Cookie +-- We use these Eq instances here because they make sense and may be added to the library in+-- the future. We do not add them now because they would silently break the old Eq behavior,+-- which was `equivCookie`.+instance Eq Cookie where+ (==) = equalCookie++instance Eq CookieJar where+ (==) = equalCookieJar++instance Eq body => Eq (Response body) where+ resp == resp' = and+ [ responseStatus resp == responseStatus resp'+ , responseVersion resp == responseVersion resp'+ , responseHeaders resp == responseHeaders resp'+ , responseBody resp == responseBody resp'+ , responseCookieJar resp `equivCookieJar` responseCookieJar resp' -- !+ -- , responseClose -- !+ ]+ default_request :: HC.Request-default_request = fromJust $ HC.parseUrl "http://www.google.com/"+default_request = HC.parseRequest_ "http://www.google.com/" default_cookie :: Cookie default_cookie = Cookie { cookie_name = fromString "name"@@ -128,8 +148,8 @@ where cookie = default_cookie testCookieEqualityResiliance :: IO ()-testCookieEqualityResiliance = assertEqual "Cookies should still be equal if extra options are changed"- (default_cookie {cookie_persistent = True}) (default_cookie {cookie_host_only = True})+testCookieEqualityResiliance = assertBool "Cookies should still be equal if extra options are changed" $+ (default_cookie {cookie_persistent = True}) `equivCookie` (default_cookie {cookie_host_only = True}) testDomainChangesEquality :: IO () testDomainChangesEquality = assertBool "Changing the domain should make cookies not equal" $@@ -140,7 +160,7 @@ (Just default_cookie, createCookieJar []) (removeExistingCookieFromCookieJar default_cookie $ createCookieJar [default_cookie]) testRemoveNonexistantCookie :: IO ()-testRemoveNonexistantCookie = assertEqual "Removing a nonexistant cookie doesn't work"+testRemoveNonexistantCookie = assertEqual "Removing a nonexistent cookie doesn't work" (Nothing, createCookieJar [default_cookie]) (removeExistingCookieFromCookieJar (default_cookie {cookie_name = fromString "key2"}) $ createCookieJar [default_cookie]) testRemoveCorrectCookie :: IO ()@@ -229,8 +249,9 @@ where cookie_jar = createCookieJar [default_cookie {cookie_http_only = True}] testComputeCookieStringSort :: IO ()-testComputeCookieStringSort = assertEqual "Sorting works correctly"- (fromString "c1=v1;c3=v3;c4=v4;c2=v2", cookie_jar_out) format_output+testComputeCookieStringSort = do+ assertEqual "Sorting works correctly (computed string)" (fst format_output) (fromString "c1=v1;c3=v3;c4=v4;c2=v2")+ assertBool "Sorting works correctly (remaining jar)" $ (snd format_output) `equivCookieJar` cookie_jar_out where now = UTCTime (ModifiedJulianDay 10) (secondsToDiffTime 11) cookie_jar = createCookieJar [ default_cookie { cookie_name = fromString "c1" , cookie_value = fromString "v1"@@ -290,8 +311,8 @@ fromString "otherkey=otherval")]} testReceiveSetCookie :: IO ()-testReceiveSetCookie = assertEqual "Receiving a Set-Cookie"- (createCookieJar [default_cookie]) (receiveSetCookie default_set_cookie default_request default_time True $ createCookieJar [])+testReceiveSetCookie = assertBool "Receiving a Set-Cookie" $+ (createCookieJar [default_cookie]) `equivCookieJar` (receiveSetCookie default_set_cookie default_request default_time True $ createCookieJar []) testReceiveSetCookieTrailingDot :: IO () testReceiveSetCookieTrailingDot = assertEqual "Receiving a Set-Cookie with a trailing domain dot"@@ -299,18 +320,18 @@ where set_cookie = default_set_cookie {setCookieDomain = Just $ fromString "www.google.com."} testReceiveSetCookieLeadingDot :: IO ()-testReceiveSetCookieLeadingDot = assertEqual "Receiving a Set-Cookie with a leading domain dot"- (createCookieJar [default_cookie]) (receiveSetCookie set_cookie default_request default_time True $ createCookieJar [])+testReceiveSetCookieLeadingDot = assertBool "Receiving a Set-Cookie with a leading domain dot" $+ (createCookieJar [default_cookie]) `equivCookieJar` (receiveSetCookie set_cookie default_request default_time True $ createCookieJar []) where set_cookie = default_set_cookie {setCookieDomain = Just $ fromString ".www.google.com"} testReceiveSetCookieNoDomain :: IO ()-testReceiveSetCookieNoDomain = assertEqual "Receiving cookie without domain"- (createCookieJar [default_cookie]) (receiveSetCookie set_cookie default_request default_time True $ createCookieJar [])+testReceiveSetCookieNoDomain = assertBool "Receiving cookie without domain" $+ (createCookieJar [default_cookie]) `equivCookieJar` (receiveSetCookie set_cookie default_request default_time True $ createCookieJar []) where set_cookie = default_set_cookie {setCookieDomain = Nothing} testReceiveSetCookieEmptyDomain :: IO ()-testReceiveSetCookieEmptyDomain = assertEqual "Receiving cookie with empty domain"- (createCookieJar [default_cookie]) (receiveSetCookie set_cookie default_request default_time True $ createCookieJar [])+testReceiveSetCookieEmptyDomain = assertBool "Receiving cookie with empty domain" $+ (createCookieJar [default_cookie]) `equivCookieJar` (receiveSetCookie set_cookie default_request default_time True $ createCookieJar []) where set_cookie = default_set_cookie {setCookieDomain = Just BS.empty} -- Can't test public suffixes until that module is written@@ -380,7 +401,7 @@ where set_cookie = default_set_cookie {setCookiePath = Just $ fromString "/a/path"} testReceiveSetCookieNoPath :: IO ()-testReceiveSetCookieNoPath = assertEqual "Path gets set correctly when nonexistant"+testReceiveSetCookieNoPath = assertEqual "Path gets set correctly when nonexistent" (fromString "/a/path/to") (cookie_path $ head $ destroyCookieJar $ receiveSetCookie set_cookie request default_time True $ createCookieJar []) where set_cookie = default_set_cookie {setCookiePath = Nothing} request = default_request {HC.path = fromString "/a/path/to/nowhere"}@@ -444,7 +465,7 @@ testMonoidPreferRecent :: IO () testMonoidPreferRecent = assertEqual "Monoid prefers more recent cookies"- (cct $ createCookieJar [c2]) (cct $ createCookieJar [c1] `mappend` createCookieJar [c2])+ (cct $ createCookieJar [c2]) (cct $ createCookieJar [c1] `Data.Monoid.mappend` createCookieJar [c2]) where c1 = default_cookie {cookie_creation_time = UTCTime (ModifiedJulianDay 0) (secondsToDiffTime 1)} c2 = default_cookie {cookie_creation_time = UTCTime (ModifiedJulianDay 0) (secondsToDiffTime 2)} cct cj = cookie_creation_time $ head $ destroyCookieJar cj@@ -488,7 +509,7 @@ removeTests :: Spec removeTests = do it "Removing a cookie works" testRemoveCookie- it "Removing a nonexistant cookie doesn't work" testRemoveNonexistantCookie+ it "Removing a nonexistent cookie doesn't work" testRemoveNonexistantCookie it "Removing the correct cookie" testRemoveCorrectCookie evictionTests :: Spec@@ -529,7 +550,7 @@ it "Expiry gets set based on max age if no expiry is given" testReceiveSetCookieNoExpiry it "Expiry gets set based on given value if no max age is given" testReceiveSetCookieNoMaxAge it "Expiry gets set to a future date if no expiry and no max age are given" testReceiveSetCookieNoExpiryNoMaxAge- it "Path gets set correctly when nonexistant" testReceiveSetCookieNoPath+ it "Path gets set correctly when nonexistent" testReceiveSetCookieNoPath it "Path gets set correctly" testReceiveSetCookiePath it "Creation time gets set correctly" testReceiveSetCookieCreationTime it "Last access time gets set correctly" testReceiveSetCookieAccessTime
test/main.hs view
@@ -8,22 +8,22 @@ import Test.HUnit import Network.Wai hiding (requestBody) import Network.Wai.Conduit (responseSource, sourceRequestBody)+import Network.HTTP.Client (streamFile)+import System.IO.Temp (withSystemTempFile) import qualified Network.Wai as Wai-import Network.Wai.Handler.Warp (runSettings, defaultSettings, settingsPort, settingsBeforeMainLoop, Settings, setTimeout)+import Network.Wai.Handler.Warp (runSettings, defaultSettings, setPort, setBeforeMainLoop, Settings, setTimeout) import Network.HTTP.Conduit hiding (port) import qualified Network.HTTP.Conduit as NHC import Network.HTTP.Client.MultipartFormData import Control.Concurrent (forkIO, killThread, putMVar, takeMVar, newEmptyMVar, threadDelay) import Network.HTTP.Types-import Control.Exception.Lifted (try, SomeException, bracket, onException, IOException)+import UnliftIO.Exception (try, SomeException, bracket, onException, IOException) import qualified Data.IORef as I import qualified Control.Exception as E (catch)-import Network (withSocketsDo)-import Network.Socket (sClose)-import qualified Network.BSD+import qualified Network.Socket as NS import CookieTest (cookieTest) #if MIN_VERSION_conduit(1,1,0)-import Data.Conduit.Network (runTCPServer, serverSettings, HostPreference (..), appSink, appSource, ServerSettings)+import Data.Conduit.Network (runTCPServer, serverSettings, appSink, appSource, ServerSettings) import Data.Streaming.Network (bindPortTCP, setAfterBind) #define bindPort bindPortTCP #else@@ -31,7 +31,7 @@ #endif import qualified Data.Conduit.Network import System.IO.Unsafe (unsafePerformIO)-import Data.Conduit (($$), ($$+-), yield, Flush (Chunk, Flush), await)+import Data.Conduit ((.|), yield, Flush (Chunk, Flush), await, runConduit) import Control.Monad (void, forever) import Control.Monad.IO.Class (liftIO) import Data.ByteString.UTF8 (fromString)@@ -48,7 +48,13 @@ import Data.Time.Calendar import qualified Network.Wai.Handler.WarpTLS as WT import Network.Connection (settingDisableCertificateValidation)-import Data.Default.Class (def)+import Data.Default (def)+#ifdef VERSION_aeson+import qualified Data.Aeson as A+#endif+import qualified Network.HTTP.Simple as Simple+import Data.Monoid (mempty)+import Control.Monad.Trans.Resource (runResourceT) past :: UTCTime past = UTCTime (ModifiedJulianDay 56200) (secondsToDiffTime 0)@@ -110,7 +116,7 @@ case esocket of Left (_ :: IOException) -> getPort Right socket -> do- sClose socket+ NS.close socket return port withApp :: (Wai.Request -> IO Wai.Response) -> (Int -> IO ()) -> IO ()@@ -127,10 +133,10 @@ port <- getPort baton <- newEmptyMVar bracket- (forkIO $ runSettings (modSettings defaultSettings- { settingsPort = port- , settingsBeforeMainLoop = putMVar baton ()- }) (app'' port) `onException` putMVar baton ())+ (forkIO $ runSettings (modSettings $+ setPort port+ $ setBeforeMainLoop (putMVar baton ())+ defaultSettings) (app'' port) `onException` putMVar baton ()) killThread (const $ takeMVar baton >> f port) where@@ -146,10 +152,11 @@ port <- getPort baton <- newEmptyMVar bracket- (forkIO $ WT.runTLS WT.defaultTlsSettings defaultSettings- { settingsPort = port- , settingsBeforeMainLoop = putMVar baton ()- } (app'' port) `onException` putMVar baton ())+ (forkIO $ WT.runTLS WT.defaultTlsSettings (+ setPort port+ $ setBeforeMainLoop (putMVar baton ())+ defaultSettings)+ (app'' port) `onException` putMVar baton ()) killThread (const $ takeMVar baton >> f port) where@@ -158,7 +165,7 @@ sendResponse res main :: IO ()-main = withSocketsDo $ do+main = do mapM_ (`hSetBuffering` LineBuffering) [stdout, stderr] hspec $ do cookieTest@@ -169,62 +176,64 @@ it "throws exception on 404" $ withApp app $ \port -> do elbs <- try $ simpleHttp $ concat ["http://127.0.0.1:", show port, "/404"] case elbs of- Left (StatusCodeException _ _ _) -> return ()+ Left (HttpExceptionRequest _ StatusCodeException {}) -> return () _ -> error "Expected an exception" describe "httpLbs" $ do it "preserves 'set-cookie' headers" $ withApp app $ \port -> do- request <- parseUrl $ concat ["http://127.0.0.1:", show port, "/cookies"]- withManager $ \manager -> do- response <- httpLbs request manager- let setCookie = mk (fromString "Set-Cookie")- (setCookieHeaders, _) = partition ((== setCookie) . fst) (NHC.responseHeaders response)- liftIO $ assertBool "response contains a 'set-cookie' header" $ length setCookieHeaders > 0+ request <- parseUrlThrow $ concat ["http://127.0.0.1:", show port, "/cookies"]+ manager <- newManager tlsManagerSettings+ response <- httpLbs request manager+ let setCookie = mk (fromString "Set-Cookie")+ (setCookieHeaders, _) = partition ((== setCookie) . fst) (NHC.responseHeaders response)+ assertBool "response contains a 'set-cookie' header" $ length setCookieHeaders > 0 it "redirects set cookies" $ withApp app $ \port -> do- request <- parseUrl $ concat ["http://127.0.0.1:", show port, "/cookie_redir1"]- withManager $ \manager -> do- response <- httpLbs request manager- liftIO $ (responseBody response) @?= "nom-nom-nom"+ request <- parseUrlThrow $ concat ["http://127.0.0.1:", show port, "/cookie_redir1"]+ manager <- newManager tlsManagerSettings+ response <- httpLbs request manager+ (responseBody response) @?= "nom-nom-nom" it "user-defined cookie jar works" $ withApp app $ \port -> do- request <- parseUrl $ concat ["http://127.0.0.1:", show port, "/dump_cookies"]- withManager $ \manager -> do- response <- httpLbs (request {redirectCount = 1, cookieJar = Just cookie_jar}) manager- liftIO $ (responseBody response) @?= "key=value"+ request <- parseUrlThrow $ concat ["http://127.0.0.1:", show port, "/dump_cookies"]+ manager <- newManager tlsManagerSettings+ response <- httpLbs (request {redirectCount = 1, cookieJar = Just cookie_jar}) manager+ (responseBody response) @?= "key=value" it "user-defined cookie jar is not ignored when redirection is disabled" $ withApp app $ \port -> do- request <- parseUrl $ concat ["http://127.0.0.1:", show port, "/dump_cookies"]- withManager $ \manager -> do- response <- httpLbs (request {redirectCount = 0, cookieJar = Just cookie_jar}) manager- liftIO $ (responseBody response) @?= "key=value"+ request <- parseUrlThrow $ concat ["http://127.0.0.1:", show port, "/dump_cookies"]+ manager <- newManager tlsManagerSettings+ response <- httpLbs (request {redirectCount = 0, cookieJar = Just cookie_jar}) manager+ (responseBody response) @?= "key=value" it "cookie jar is available in response" $ withApp app $ \port -> do- request <- parseUrl $ concat ["http://127.0.0.1:", show port, "/cookies"]- withManager $ \manager -> do- response <- httpLbs (request {cookieJar = Just def}) manager- liftIO $ (length $ destroyCookieJar $ responseCookieJar response) @?= 1+ request <- parseUrlThrow $ concat ["http://127.0.0.1:", show port, "/cookies"]+ manager <- newManager tlsManagerSettings+ response <- httpLbs (request {cookieJar = Just Data.Monoid.mempty}) manager+ (length $ destroyCookieJar $ responseCookieJar response) @?= 1 it "Cookie header isn't touched when no cookie jar supplied" $ withApp app $ \port -> do- request <- parseUrl $ concat ["http://127.0.0.1:", show port, "/dump_cookies"]- withManager $ \manager -> do- let request_headers = (mk "Cookie", "key2=value2") : filter ((/= mk "Cookie") . fst) (NHC.requestHeaders request)- response <- httpLbs (request {NHC.requestHeaders = request_headers, cookieJar = Nothing}) manager- liftIO $ (responseBody response) @?= "key2=value2"+ request <- parseUrlThrow $ concat ["http://127.0.0.1:", show port, "/dump_cookies"]+ manager <- newManager tlsManagerSettings+ let request_headers = (mk "Cookie", "key2=value2") : filter ((/= mk "Cookie") . fst) (NHC.requestHeaders request)+ response <- httpLbs (request {NHC.requestHeaders = request_headers, cookieJar = Nothing}) manager+ (responseBody response) @?= "key2=value2" it "Response cookie jar is nothing when request cookie jar is nothing" $ withApp app $ \port -> do- request <- parseUrl $ concat ["http://127.0.0.1:", show port, "/cookies"]- withManager $ \manager -> do- response <- httpLbs (request {cookieJar = Nothing}) manager- liftIO $ (responseCookieJar response) @?= def+ request <- parseUrlThrow $ concat ["http://127.0.0.1:", show port, "/cookies"]+ manager <- newManager tlsManagerSettings+ response <- httpLbs (request {cookieJar = Nothing}) manager+ (responseCookieJar response) @?= mempty it "TLS" $ withAppTls app $ \port -> do- request <- parseUrl $ "https://127.0.0.1:" ++ show port+ request <- parseUrlThrow $ "https://127.0.0.1:" ++ show port let set = mkManagerSettings def { settingDisableCertificateValidation = True } Nothing- response <- withManagerSettings set $ httpLbs request+ manager <- newManager set+ response <- httpLbs request manager responseBody response @?= "homepage" describe "manager" $ do it "closes all connections" $ withApp app $ \port1 -> withApp app $ \port2 -> do --FIXME clearSocketsList- withManager $ \manager -> do- let Just req1 = parseUrl $ "http://127.0.0.1:" ++ show port1- let Just req2 = parseUrl $ "http://127.0.0.1:" ++ show port2+ manager <- newManager tlsManagerSettings+ let Just req1 = parseUrlThrow $ "http://127.0.0.1:" ++ show port1+ let Just req2 = parseUrlThrow $ "http://127.0.0.1:" ++ show port2+ runResourceT $ do _res1a <- http req1 manager _res1b <- http req1 manager _res2 <- http req2 manager@@ -232,121 +241,122 @@ --FIXME requireAllSocketsClosed describe "http" $ do it "response body" $ withApp app $ \port -> do- withManager $ \manager -> do- req <- liftIO $ parseUrl $ "http://127.0.0.1:" ++ show port+ manager <- newManager tlsManagerSettings+ req <- parseUrlThrow $ "http://127.0.0.1:" ++ show port+ runResourceT $ do res1 <- http req manager- bss <- responseBody res1 $$+- CL.consume+ bss <- runConduit $ responseBody res1 .| CL.consume res2 <- httpLbs req manager liftIO $ L.fromChunks bss `shouldBe` responseBody res2 describe "DOS protection" $ do it "overlong headers" $ overLongHeaders $ \port -> do- withManager $ \manager -> do- let Just req1 = parseUrl $ "http://127.0.0.1:" ++ show port- res1 <- try $ http req1 manager- case res1 of- Left e -> liftIO $ show (e :: SomeException) @?= show OverlongHeaders- _ -> error "Shouldn't have worked"+ manager <- newManager tlsManagerSettings+ let Just req1 = parseUrlThrow $ "http://127.0.0.1:" ++ show port+ res1 <- try $ runResourceT $ http req1 manager+ case res1 of+ Left e -> show (e :: SomeException) @?= show (HttpExceptionRequest req1 OverlongHeaders)+ _ -> error "Shouldn't have worked" it "not overlong headers" $ notOverLongHeaders $ \port -> do- withManager $ \manager -> do- let Just req1 = parseUrl $ "http://127.0.0.1:" ++ show port- _ <- httpLbs req1 manager- return ()+ manager <- newManager tlsManagerSettings+ let Just req1 = parseUrlThrow $ "http://127.0.0.1:" ++ show port+ _ <- httpLbs req1 manager+ return () describe "redirects" $ do it "doesn't double escape" $ redir $ \port -> do- withManager $ \manager -> do- let go (encoded, final) = do- let Just req1 = parseUrl $ concat ["http://127.0.0.1:", show port, "/redir/", encoded]- res <- httpLbs req1 manager- liftIO $ Network.HTTP.Conduit.responseStatus res @?= status200- liftIO $ responseBody res @?= L.fromChunks [TE.encodeUtf8 final]- mapM_ go- [ ("hello world%2F", "hello world/")- , ("%D7%A9%D7%9C%D7%95%D7%9D", "שלום")- , ("simple", "simple")- , ("hello%20world", "hello world")- , ("hello%20world%3f%23", "hello world?#")- ]+ manager <- newManager tlsManagerSettings+ let go (encoded, final) = do+ let Just req1 = parseUrlThrow $ concat ["http://127.0.0.1:", show port, "/redir/", encoded]+ res <- httpLbs req1 manager+ liftIO $ Network.HTTP.Conduit.responseStatus res @?= status200+ liftIO $ responseBody res @?= L.fromChunks [TE.encodeUtf8 final]+ mapM_ go+ [ ("hello world%2F", "hello world/")+ , ("%D7%A9%D7%9C%D7%95%D7%9D", "שלום")+ , ("simple", "simple")+ , ("hello%20world", "hello world")+ , ("hello%20world%3f%23", "hello world?#")+ ] it "TooManyRedirects: redirect request body is preserved" $ withApp app $ \port -> do- let Just req = parseUrl $ concat ["http://127.0.0.1:", show port, "/infredir/0"]+ let Just req = parseUrlThrow $ concat ["http://127.0.0.1:", show port, "/infredir/0"] let go (res, i) = liftIO $ responseBody res @?= (L8.pack $ show i)- E.catch (withManager $ \manager -> do- void $ http req{redirectCount=5} manager) $ \e ->+ manager <- newManager tlsManagerSettings+ E.catch (void $ runResourceT $ http req{redirectCount=5} manager)+ $ \e -> case e of- TooManyRedirects redirs ->+ HttpExceptionRequest _ (TooManyRedirects redirs) -> mapM_ go (zip redirs [5,4..0 :: Int]) _ -> error $ show e describe "chunked request body" $ do it "works" $ echo $ \port -> do- withManager $ \manager -> do- let go bss = do- let Just req1 = parseUrl $ "http://127.0.0.1:" ++ show port- src = sourceList bss- lbs = L.fromChunks bss- res <- httpLbs req1- { method = "POST"- , requestBody = requestBodySourceChunked src- } manager- liftIO $ Network.HTTP.Conduit.responseStatus res @?= status200- let ts = S.concat . L.toChunks- liftIO $ ts (responseBody res) @?= ts lbs- mapM_ go- [ ["hello", "world"]- , replicate 500 "foo\003\n\r"- ]+ manager <- newManager tlsManagerSettings+ let go bss = do+ let Just req1 = parseUrlThrow $ "POST http://127.0.0.1:" ++ show port+ src = sourceList bss+ lbs = L.fromChunks bss+ res <- httpLbs req1+ { requestBody = requestBodySourceChunked src+ } manager+ liftIO $ Network.HTTP.Conduit.responseStatus res @?= status200+ let ts = S.concat . L.toChunks+ liftIO $ ts (responseBody res) @?= ts lbs+ mapM_ go+ [ ["hello", "world"]+ , replicate 500 "foo\003\n\r"+ ] describe "no status message" $ do it "works" $ noStatusMessage $ \port -> do- req <- parseUrl $ "http://127.0.0.1:" ++ show port- withManager $ \manager -> do- res <- httpLbs req manager- liftIO $ do- Network.HTTP.Conduit.responseStatus res `shouldBe` status200- responseBody res `shouldBe` "foo"+ req <- parseUrlThrow $ "http://127.0.0.1:" ++ show port+ manager <- newManager tlsManagerSettings+ res <- httpLbs req manager+ liftIO $ do+ Network.HTTP.Conduit.responseStatus res `shouldBe` status200+ responseBody res `shouldBe` "foo" describe "response body too short" $ do it "throws an exception" $ wrongLength $ \port -> do- req <- parseUrl $ "http://127.0.0.1:" ++ show port- withManager $ \manager -> do- eres <- try $ httpLbs req manager- liftIO $ either (Left . (show :: HttpException -> String)) (Right . id) eres- `shouldBe` Left (show $ ResponseBodyTooShort 50 18)+ req <- parseUrlThrow $ "http://127.0.0.1:" ++ show port+ manager <- newManager tlsManagerSettings+ eres <- try $ httpLbs req manager+ liftIO $ either (Left . (show :: HttpException -> String)) (Right . id) eres+ `shouldBe` Left (show $ HttpExceptionRequest req $ ResponseBodyTooShort 50 18) describe "chunked response body" $ do it "no chunk terminator" $ wrongLengthChunk1 $ \port -> do- req <- parseUrl $ "http://127.0.0.1:" ++ show port- withManager $ \manager -> do- eres <- try $ httpLbs req manager- liftIO $ either (Left . (show :: HttpException -> String)) (Right . id) eres- `shouldBe` Left (show IncompleteHeaders)+ req <- parseUrlThrow $ "http://127.0.0.1:" ++ show port+ manager <- newManager tlsManagerSettings+ eres <- try $ httpLbs req manager+ liftIO $ either (Left . (show :: HttpException -> String)) (Right . id) eres+ `shouldBe` Left (show (HttpExceptionRequest req IncompleteHeaders)) it "incomplete chunk" $ wrongLengthChunk2 $ \port -> do- req <- parseUrl $ "http://127.0.0.1:" ++ show port- withManager $ \manager -> do- eres <- try $ httpLbs req manager- liftIO $ either (Left . (show :: HttpException -> String)) (Right . id) eres- `shouldBe` Left (show InvalidChunkHeaders)+ req <- parseUrlThrow $ "http://127.0.0.1:" ++ show port+ manager <- newManager tlsManagerSettings+ eres <- try $ httpLbs req manager+ liftIO $ either (Left . (show :: HttpException -> String)) (Right . id) eres+ `shouldBe` Left (show (HttpExceptionRequest req InvalidChunkHeaders)) it "invalid chunk" $ invalidChunk $ \port -> do- req <- parseUrl $ "http://127.0.0.1:" ++ show port- withManager $ \manager -> do- eres <- try $ httpLbs req manager- liftIO $ either (Left . (show :: HttpException -> String)) (Right . id) eres- `shouldBe` Left (show InvalidChunkHeaders)+ req <- parseUrlThrow $ "http://127.0.0.1:" ++ show port+ manager <- newManager tlsManagerSettings+ eres <- try $ httpLbs req manager+ liftIO $ either (Left . (show :: HttpException -> String)) (Right . id) eres+ `shouldBe` Left (show (HttpExceptionRequest req InvalidChunkHeaders)) it "missing header" $ rawApp "HTTP/1.1 200 OK\r\nTransfer-Encoding: chunked\r\n\r\n4\r\nabcd\r\n\r\n\r\n" $ \port -> do- req <- parseUrl $ "http://127.0.0.1:" ++ show port- withManager $ \manager -> do- eres <- try $ httpLbs req manager- liftIO $ either (Left . (show :: HttpException -> String)) (Right . id) eres- `shouldBe` Left (show InvalidChunkHeaders)+ req <- parseUrlThrow $ "http://127.0.0.1:" ++ show port+ manager <- newManager tlsManagerSettings+ eres <- try $ httpLbs req manager+ liftIO $ either (Left . (show :: HttpException -> String)) (Right . id) eres+ `shouldBe` Left (show (HttpExceptionRequest req InvalidChunkHeaders)) it "junk header" $ rawApp "HTTP/1.1 200 OK\r\nTransfer-Encoding: chunked\r\n\r\n4\r\nabcd\r\njunk\r\n\r\n" $ \port -> do- req <- parseUrl $ "http://127.0.0.1:" ++ show port- withManager $ \manager -> do- eres <- try $ httpLbs req manager- liftIO $ either (Left . (show :: HttpException -> String)) (Right . id) eres- `shouldBe` Left (show InvalidChunkHeaders)+ req <- parseUrlThrow $ "http://127.0.0.1:" ++ show port+ manager <- newManager tlsManagerSettings+ eres <- try $ httpLbs req manager+ liftIO $ either (Left . (show :: HttpException -> String)) (Right . id) eres+ `shouldBe` Left (show (HttpExceptionRequest req InvalidChunkHeaders)) describe "redirect" $ do it "ignores large response bodies" $ do@@ -354,8 +364,9 @@ case pathInfo req of ["foo"] -> return $ responseLBS status200 [] "Hello World!" _ -> return $ responseSource status301 [("location", S8.pack $ "http://127.0.0.1:" ++ show port ++ "/foo")] $ forever $ yield $ Chunk $ fromByteString "hello\n"- withApp' app' $ \port -> withManager $ \manager -> do- req <- liftIO $ parseUrl $ "http://127.0.0.1:" ++ show port+ manager <- newManager tlsManagerSettings+ withApp' app' $ \port -> do+ req <- liftIO $ parseUrlThrow $ "http://127.0.0.1:" ++ show port res <- httpLbs req manager liftIO $ do Network.HTTP.Conduit.responseStatus res `shouldBe` status200@@ -385,54 +396,55 @@ describe "HTTP/1.0" $ do it "BaseHTTP" $ do let baseHTTP app' = do- _ <- appSource app' $$ await- yield "HTTP/1.0 200 OK\r\n\r\nThis is it!" $$ appSink app'- withCApp baseHTTP $ \port -> withManager $ \manager -> do- req <- liftIO $ parseUrl $ "http://127.0.0.1:" ++ show port+ _ <- runConduit $ appSource app' .| await+ runConduit $ yield "HTTP/1.0 200 OK\r\n\r\nThis is it!" .| appSink app'+ manager <- newManager tlsManagerSettings+ withCApp baseHTTP $ \port -> do+ req <- liftIO $ parseUrlThrow $ "http://127.0.0.1:" ++ show port res1 <- httpLbs req manager res2 <- httpLbs req manager liftIO $ res1 @?= res2 describe "hostAddress" $ do it "overrides host" $ withApp app $ \port -> do- entry <- Network.BSD.getHostByName "127.0.0.1"- req' <- parseUrl $ "http://example.com:" ++ show port- let req = req' { hostAddress = Just $ Network.BSD.hostAddress entry }- res <- withManager $ httpLbs req+ req' <- parseUrlThrow $ "http://example.com:" ++ show port+ let req = req' { hostAddress = Just $ NS.tupleToHostAddress (127, 0, 0, 1) }+ manager <- newManager tlsManagerSettings+ res <- httpLbs req manager responseBody res @?= "homepage for example.com" describe "managerResponseTimeout" $ do it "works" $ withApp app $ \port -> do- req1 <- parseUrl $ "http://localhost:" ++ show port- let req2 = req1 { responseTimeout = Just 5000000 }- withManagerSettings conduitManagerSettings { managerResponseTimeout = Just 1 } $ \man -> do- eres1 <- try $ httpLbs req1 { NHC.path = "/delayed" } man- case eres1 of- Left (FailedConnectionException _ _) -> return ()- _ -> error "Did not time out"- _ <- httpLbs req2 man- return ()+ req1 <- parseUrlThrow $ "http://localhost:" ++ show port+ let req2 = req1 { responseTimeout = responseTimeoutMicro 5000000 }+ man <- newManager tlsManagerSettings { managerResponseTimeout = responseTimeoutMicro 1 }+ eres1 <- try $ httpLbs req1 { NHC.path = "/delayed" } man+ case eres1 of+ Left (HttpExceptionRequest _ ConnectionTimeout{}) -> return ()+ _ -> error "Did not time out"+ _ <- httpLbs req2 man+ return () describe "delayed body" $ do it "works" $ withApp app $ \port -> do- req <- parseUrl $ "http://localhost:" ++ show port ++ "/delayed"- withManager $ \man -> do- _ <- http req man- return ()+ req <- parseUrlThrow $ "http://localhost:" ++ show port ++ "/delayed"+ man <- newManager tlsManagerSettings+ _ <- runResourceT $ http req man+ return () it "reuse/connection close tries again" $ do withAppSettings (setTimeout 1) (const app) $ \port -> do- req <- parseUrl $ "http://localhost:" ++ show port- withManager $ \man -> do- res1 <- httpLbs req man- liftIO $ threadDelay 3000000- res2 <- httpLbs req man- let f res = res- { NHC.responseHeaders = filter (not . isDate) (NHC.responseHeaders res)- }- isDate ("date", _) = True- isDate _ = False- liftIO $ f res2 `shouldBe` f res1+ req <- parseUrlThrow $ "http://localhost:" ++ show port+ man <- newManager tlsManagerSettings+ res1 <- httpLbs req man+ threadDelay 3000000+ res2 <- httpLbs req man+ let f res = res+ { NHC.responseHeaders = filter (not . isDate) (NHC.responseHeaders res)+ }+ isDate ("date", _) = True+ isDate _ = False+ f res2 `shouldBe` f res1 it "setQueryString" $ do ref <- I.newIORef undefined@@ -445,13 +457,44 @@ , (TE.encodeUtf8 "שלום", Just "hola") , ("noval", Nothing) ]- withManager $ \man -> do- req <- parseUrl $ "http://localhost:" ++ show port- _ <- httpLbs (setQueryString qs req) man- return ()+ man <- newManager tlsManagerSettings+ req <- parseUrlThrow $ "http://localhost:" ++ show port+ _ <- httpLbs (setQueryString qs req) man res <- I.readIORef ref res `shouldBe` qs +#ifdef VERSION_aeson+ describe "Simple.JSON" $ do+ it "normal" $ jsonApp $ \port -> do+ req <- parseUrlThrow $ "http://localhost:" ++ show port+ value <- Simple.httpJSON req+ responseBody value `shouldBe` jsonValue+ it "trailing whitespace" $ jsonApp $ \port -> do+ req <- parseUrlThrow $ "http://localhost:" ++ show port ++ "/trailing"+ value <- Simple.httpJSON req+ responseBody value `shouldBe` jsonValue+#endif++ it "RequestBodyIO" $ echo $ \port -> do+ manager <- newManager tlsManagerSettings+ let go bss = withSystemTempFile "request-body-io" $ \tmpfp tmph -> do+ liftIO $ do+ mapM_ (S.hPutStr tmph) bss+ hClose tmph++ let Just req1 = parseUrlThrow $ "POST http://127.0.0.1:" ++ show port+ lbs = L.fromChunks bss+ res <- httpLbs req1+ { requestBody = RequestBodyIO (streamFile tmpfp)+ } manager+ liftIO $ Network.HTTP.Conduit.responseStatus res @?= status200+ let ts = S.concat . L.toChunks+ liftIO $ ts (responseBody res) @?= ts lbs+ mapM_ go+ [ ["hello", "world"]+ , replicate 500 "foo\003\n\r"+ ]+ withCApp :: (Data.Conduit.Network.AppData -> IO ()) -> (Int -> IO ()) -> IO () withCApp app' f = do port <- getPort@@ -471,14 +514,14 @@ overLongHeaders :: (Int -> IO ()) -> IO () overLongHeaders =- withCApp $ \app' -> src $$ appSink app'+ withCApp $ \app' -> runConduit $ src .| appSink app' where src = sourceList $ "HTTP/1.0 200 OK\r\nfoo: " : repeat "bar" notOverLongHeaders :: (Int -> IO ()) -> IO () notOverLongHeaders = withCApp $ \app' -> do- appSource app' $$ CL.drop 1- src $$ appSink app'+ runConduit $ appSource app' .| CL.drop 1+ runConduit $ src .| appSink app' where src = sourceList $ [S.concat $ "HTTP/1.0 200 OK\r\nContent-Type: text/plain\r\nContent-Length: 16384\r\n\r\n" : ( take 16384 $ repeat "x")] @@ -518,20 +561,20 @@ echo :: (Int -> IO ()) -> IO () echo = withApp $ \req -> do- bss <- sourceRequestBody req $$ CL.consume+ bss <- runConduit $ sourceRequestBody req .| CL.consume return $ responseLBS status200 [] $ L.fromChunks bss noStatusMessage :: (Int -> IO ()) -> IO () noStatusMessage =- withCApp $ \app' -> src $$ appSink app'+ withCApp $ \app' -> runConduit $ src .| appSink app' where src = yield "HTTP/1.0 200\r\nContent-Length: 3\r\n\r\nfoo: barbazbin" wrongLength :: (Int -> IO ()) -> IO () wrongLength = withCApp $ \app' -> do- _ <- appSource app' $$ await- src $$ appSink app'+ _ <- runConduit $ appSource app' .| await+ runConduit $ src .| appSink app' where src = do yield "HTTP/1.0 200 OK\r\nContent-Length: 50\r\n\r\n"@@ -540,31 +583,49 @@ wrongLengthChunk1 :: (Int -> IO ()) -> IO () wrongLengthChunk1 = withCApp $ \app' -> do- _ <- appSource app' $$ await- src $$ appSink app'+ _ <- runConduit $ appSource app' .| await+ runConduit $ src .| appSink app' where src = yield "HTTP/1.1 200 OK\r\nTransfer-Encoding: chunked\r\n\r\n4\r\nWiki\r\n" wrongLengthChunk2 :: (Int -> IO ()) -> IO () wrongLengthChunk2 = withCApp $ \app' -> do- _ <- appSource app' $$ await- src $$ appSink app'+ _ <- runConduit $ appSource app' .| await+ runConduit $ src .| appSink app' where src = yield "HTTP/1.1 200 OK\r\nTransfer-Encoding: chunked\r\n\r\n4\r\nWiki\r\n5\r\npedia\r\nE\r\nin\r\n\r\nch\r\n" invalidChunk :: (Int -> IO ()) -> IO () invalidChunk = withCApp $ \app' -> do- _ <- appSource app' $$ await- src $$ appSink app'+ _ <- runConduit $ appSource app' .| await+ runConduit $ src .| appSink app' where src = yield "HTTP/1.1 200 OK\r\nTransfer-Encoding: chunked\r\n\r\n4\r\nabcd\r\ngarbage\r\nef\r\n0\r\n\r\n" rawApp :: S8.ByteString -> (Int -> IO ()) -> IO () rawApp bs = withCApp $ \app' -> do- _ <- appSource app' $$ await- src $$ appSink app'+ _ <- runConduit $ appSource app' .| await+ runConduit $ src .| appSink app' where src = yield bs++#ifdef VERSION_aeson+jsonApp :: (Int -> IO ()) -> IO ()+jsonApp = withApp $ \req -> return $ responseLBS+ status200+ [ ("Content-Type", "application/json")+ ] $+ case pathInfo req of+ [] -> A.encode jsonValue+ ["trailing"] -> L.append (A.encode jsonValue) " \n\r\n\t "+ x -> error $ "unsupported: " ++ show x++jsonValue :: A.Value+jsonValue = A.object+ [ "name" A..= ("Alice" :: String)+ , "age" A..= (35 :: Int)+ ]+#endif