Add hook to print memory info before/after tests.

Author: cgrindel

rules_haskell_tests/tests/RunTests.hs

@@ -23,7 +23,7 @@ import IntegrationTesting
 
 main :: IO ()
 main = hspec $ do
-  afterAll_ shutdownBazel $ do
+  afterAll_ shutdownBazel $ after_ printMemoryHook $ do
     it "bazel test" $ do
       assertSuccess (bazel ["test", "//..."])
 
@@ -179,6 +179,14 @@ shutdownBazel = do
   printMemory "AFTER"
   pure ()
 
+-- | Print memory information before and after each test
+printMemoryHook :: IO () -> IO ()
+printMemoryHook action = bracket
+  (printMemory "BEFORE")
+  (printMemory "AFTER")
+  (const action)
+
+
 -- | Print information about the current memory state to debug intermittent failures
 -- Related to https://github.com/tweag/rules_haskell/issues/2089
 printMemory :: String -> IO ()