Add response code verification into integration DSL

cardano-wallet.cabal

@@ -174,6 +174,7 @@ test-suite integration
     , generic-lens
     , hspec
     , hspec-core
+    , hspec-expectations-lifted
     , http-client
     , http-api-data
     , http-types

test/integration/Main.hs

@@ -7,10 +7,13 @@ import Control.Concurrent.MVar
 
 import Data.Aeson
     ( Value )
+import Data.ByteString.Lazy
+    ( ByteString )
 import Data.Text
     ( Text )
 import Network.HTTP.Client
-    ( Manager, defaultManagerSettings, newManager )
+    ( Manager, Request, Response, defaultManagerSettings, newManager )
+import Network.HTTP.Types.Status
 import Prelude
 import System.Process
     ( proc, withCreateProcess )
@@ -20,14 +23,6 @@ import Test.Hspec
 import qualified Data.Text as T
 
 import Test.Integration.Framework.DSL
-    ( Context (..)
-    , Scenarios
-    , expectError
-    , request
-    , request_
-    , scenario
-    , verify
-    )
 import Test.Integration.Framework.Request
     ( RequestException (..) )
 
@@ -41,6 +36,9 @@ main = do
         beforeAll (withWallet (newMVar . Context ())) $ do
             describe "Integration test framework" dummySpec
 
+        beforeAll (dummySetup (newMVar . Context ())) $ do
+            describe "Test response codes" respCodesSpec
+
 -- Runs the wallet server only. The API is not implemented yet, so this is
 -- basically a placeholder until then.
 withWallet :: ((Text, Manager) -> IO a) -> IO a
@@ -63,3 +61,37 @@ dummySpec = do
 
     scenario "request_ function is always successful" $ do
         request_ ("GET", "api/xyzzy") Nothing
+
+-- Temporary test setup for testing response codes
+dummySetup :: ((Text, Manager) -> IO a) -> IO a
+dummySetup action = do
+        let baseURL = T.pack ("http://httpbin.org")
+        manager <- newManager defaultManagerSettings
+        action (baseURL, manager)
+
+-- Exercise response codes
+respCodesSpec :: Scenarios Context
+respCodesSpec = do
+    scenario "GET; Response code 200" $ do
+        response <- request' ("GET", "/get") Nothing
+        verify (response :: Either RequestException (Request, Response ByteString))
+            [ expectResponseCode status200
+            ]
+
+    scenario "GET; Response code 404" $ do
+        response <- request' ("GET", "/get/nothing") Nothing
+        verify (response :: Either RequestException (Request, Response ByteString))
+            [ expectResponseCode status404
+            ]
+
+    scenario "POST; Response code 200" $ do
+        response <- request' ("POST", "/post") Nothing
+        verify (response :: Either RequestException (Request, Response ByteString))
+            [ expectResponseCode status200
+            ]
+
+    scenario "POST; Response code 405" $ do
+        response <- request' ("POST", "/get") Nothing
+        verify (response :: Either RequestException (Request, Response ByteString))
+            [ expectResponseCode status405
+            ]

test/integration/Test/Integration/Framework/DSL.hs

@@ -12,13 +12,15 @@ module Test.Integration.Framework.DSL
 
     -- * Steps
     , request
+    , request'
     , request_
     , successfulRequest
     , verify
 
     -- * Expectations
     , expectSuccess
     , expectError
+    , expectResponseCode
     , RequestException(..)
 
     -- * Helpers
@@ -39,6 +41,8 @@ import Control.Monad.IO.Class
     ( MonadIO, liftIO )
 import Data.Aeson.QQ
     ( aesonQQ )
+import Data.ByteString.Lazy
+    ( ByteString )
 import Data.Function
     ( (&) )
 import Data.List
@@ -50,14 +54,21 @@ import GHC.Generics
 import Language.Haskell.TH.Quote
     ( QuasiQuoter )
 import Network.HTTP.Client
-    ( Manager )
+    ( Manager
+    , Request
+    , Response
+    , responseStatus
+    )
+import Network.HTTP.Types.Status
+
 import Test.Hspec.Core.Spec
     ( SpecM, it, xit )
+import Test.Hspec.Expectations.Lifted
 
 import qualified Test.Hspec.Core.Spec as H
 
 import Test.Integration.Framework.Request
-    ( RequestException (..), request, request_, successfulRequest, ($-) )
+    ( RequestException (..), request, request', request_, successfulRequest, ($-) )
 import Test.Integration.Framework.Scenario
     ( Scenario )
 import Web.HttpApiData
@@ -120,6 +131,14 @@ expectError = \case
     Left _  -> return ()
     Right a -> wantedErrorButSuccess a
 
+expectResponseCode
+    :: (MonadIO m, MonadFail m)
+    => Status
+    -> Either RequestException (Request, Response ByteString)
+    -> m ()
+expectResponseCode c = \case
+    Left e   -> wantedSuccessButError e
+    Right a -> (responseStatus (snd a)) `shouldBe` c
 
 -- | Expect a successful response, without any further assumptions
 expectSuccess

test/integration/Test/Integration/Framework/Request.hs

@@ -7,6 +7,7 @@
 
 module Test.Integration.Framework.Request
     ( request
+    , request'
     , request_
     , successfulRequest
     , RequestException(..)