keep `safe` and `{-# SOURCE #-}` import annotations

1 files changed, 17 insertions, 0 deletions

diff --git a/tests/Language/Haskell/Stylish/Step/Imports/Tests.hs b/tests/Language/Haskell/Stylish/Step/Imports/Tests.hs
index 5ca60de..86a4141 100644
--- a/tests/Language/Haskell/Stylish/Step/Imports/Tests.hs
+++ b/tests/Language/Haskell/Stylish/Step/Imports/Tests.hs
@@ -46,6 +46,7 @@ tests = testGroup "Language.Haskell.Stylish.Step.Imports.Tests"
     , testCase "case 19b" case19b
     , testCase "case 19d" case19c
     , testCase "case 19d" case19d
+    , testCase "case 20" case20
     ]
 
 
@@ -512,3 +513,19 @@ case19input = unlines
         , ""
         , "import Data.List (foldl', intercalate, intersperse)"
         ]
+
+--------------------------------------------------------------------------------
+case20 :: Assertion
+case20 = expected
+    @=? testStep (step 80 defaultOptions) input'
+  where
+    expected = unlines
+        [ "import qualified Data.Map  as Map"
+        , "import           Data.Set  (empty)"
+        , "import {-# SOURCE #-} qualified Data.Text as T"
+        ]
+    input' = unlines
+        [ "import {-# SOURCE #-} qualified Data.Text as T"
+        , "import qualified   Data.Map as Map"
+        , "import Data.Set (empty)"
+        ]