Tests: if disk is full, it's not our problem, skip the test

Author: Bodigrim

tests/Tests/Regressions.hs

@@ -10,12 +10,12 @@ module Tests.Regressions
       tests
     ) where
 
-import Control.Exception (ErrorCall, SomeException, handle, evaluate)
+import Control.Exception (ErrorCall, SomeException, handle, evaluate, displayException, try)
 import Data.Char (isLetter, chr)
 import GHC.Exts (Int(..), sizeofByteArray#)
 import System.IO
 import System.IO.Temp (withSystemTempFile)
-import Test.Tasty.HUnit (assertBool, assertEqual, assertFailure)
+import Test.Tasty.HUnit (assertBool, assertEqual, assertFailure, (@?=))
 import qualified Data.ByteString as B
 import Data.ByteString.Char8 ()
 import qualified Data.ByteString.Lazy as LB
@@ -34,9 +34,8 @@ import qualified Data.Text.Lazy.Encoding as LE
 import qualified Data.Text.Unsafe as T
 import qualified Test.Tasty as F
 import qualified Test.Tasty.HUnit as F
-import Test.Tasty.HUnit ((@?=))
-
 import Tests.Utils (withTempFile)
+import System.IO.Error (isFullError)
 
 -- Reported by Michael Snoyman: UTF-8 encoding a large lazy bytestring
 -- caused either a segfault or attempt to allocate a negative number
@@ -50,12 +49,19 @@ lazy_encode_crash = withTempFile $ \ _ h ->
 -- exception).
 hGetContents_crash :: IO ()
 hGetContents_crash = withSystemTempFile "crashy.txt" $ \path h -> do
-  B.hPut h (B.pack [0x78, 0xc4 ,0x0a]) >> hClose h
-  h' <- openFile path ReadMode
-  hSetEncoding h' utf8
-  handle (\(_::SomeException) -> return ()) $
-    T.hGetContents h' >> assertFailure "T.hGetContents should crash"
-  hClose h'
+  putRes <- try $ B.hPut h (B.pack [0x78, 0xc4 ,0x0a])
+  case putRes of
+    Left e
+      -- If disk is full (as it happens on some of our CI runners), it's not our issue, skip it
+      | isFullError e -> pure ()
+      | otherwise -> assertFailure $ "B.hPut crashed because of " ++ displayException e
+    Right () -> do
+      hClose h
+      h' <- openFile path ReadMode
+      hSetEncoding h' utf8
+      handle (\(_::SomeException) -> pure ()) $
+        T.hGetContents h' >> assertFailure "T.hGetContents should crash"
+      hClose h'
 
 -- Reported by Ian Lynagh: attempting to allocate a sufficiently large
 -- string (via either Array.new or Text.replicate) could result in an