1 files changed, 5 insertions, 0 deletions

diff --git a/server/test/Thermoprint/ServerSpec.hs b/server/test/Thermoprint/ServerSpec.hs
index 8af210d..334f785 100644
--- a/server/test/Thermoprint/ServerSpec.hs
+++ b/server/test/Thermoprint/ServerSpec.hs
@@ -42,6 +42,7 @@ import           System.IO.Temp
 import qualified Data.Text as T
 
 import           Network.Wai.Handler.Warp (defaultSettings, setBeforeMainLoop)
+import           Network.HTTP.Types.Status (Status(..))
 
 import qualified Data.Map as Map
 
@@ -153,7 +154,11 @@ spec = withSetup $ do
     (atomically . takeTMVar $ outputChan printer) `shouldReturn` p
     draftDelete dId
     drafts `shouldReturn` []
+    draftReplace dId Nothing p `shouldThrow` is404
   where
     Client{..} = mkClient' $ BaseUrl Http "localhost" 3000 ""
+    is404 :: ServantError -> Bool
+    is404 e@(FailureResponse {}) = statusCode (responseStatus e) == 404
+    is404 _ = False