Record alignment takes max columns into account

1 files changed, 46 insertions, 2 deletions

diff --git a/tests/Language/Haskell/Stylish/Step/Records/Tests.hs b/tests/Language/Haskell/Stylish/Step/Records/Tests.hs
index 312c6fa..5b24c2f 100644
--- a/tests/Language/Haskell/Stylish/Step/Records/Tests.hs
+++ b/tests/Language/Haskell/Stylish/Step/Records/Tests.hs
@@ -20,12 +20,14 @@ tests :: Test
 tests = testGroup "Language.Haskell.Stylish.Step.Records.Tests"
     [ testCase "case 01" case01
     , testCase "case 02" case02
+    , testCase "case 03" case03
+    , testCase "case 04" case04
     ]
 
 
 --------------------------------------------------------------------------------
 case01 :: Assertion
-case01 = expected @=? testStep step input
+case01 = expected @=? testStep (step 80) input
   where
     input = unlines
         [ "data Foo = Foo"
@@ -44,7 +46,7 @@ case01 = expected @=? testStep step input
 
 --------------------------------------------------------------------------------
 case02 :: Assertion
-case02 = input @=? testStep step input
+case02 = input @=? testStep (step 80) input
   where
     -- Don't attempt to align this since a field spans multiple lines
     input = unlines
@@ -54,3 +56,45 @@ case02 = input @=? testStep step input
         , "         :: String"
         , "    } deriving (Show)"
         ]
+
+
+--------------------------------------------------------------------------------
+case03 :: Assertion
+case03 =
+    -- 22 max columns is /just/ enough to align this stuff.
+    expected @=? testStep (step 22) input
+  where
+    input = unlines
+        [ "data Foo = Foo"
+        , "    { foo :: String"
+        , "    , barqux :: Int"
+        , "    }"
+        ]
+
+    expected = unlines
+        [ "data Foo = Foo"
+        , "    { foo    :: String"
+        , "    , barqux :: Int"
+        , "    }"
+        ]
+
+
+--------------------------------------------------------------------------------
+case04 :: Assertion
+case04 =
+    -- 21 max columns is /just NOT/ enough to align this stuff.
+    expected @=? testStep (step 21) input
+  where
+    input = unlines
+        [ "data Foo = Foo"
+        , "    { foo :: String"
+        , "    , barqux :: Int"
+        , "    }"
+        ]
+
+    expected = unlines
+        [ "data Foo = Foo"
+        , "    { foo :: String"
+        , "    , barqux :: Int"
+        , "    }"
+        ]