1 files changed, 57 insertions, 2 deletions
diff --git a/test/Spec.hs b/test/Spec.hs
index 298eb04..24a17e2 100644
--- a/test/Spec.hs
+++ b/test/Spec.hs
@@ -1,7 +1,19 @@
+{-# LANGUAGE MultiWayIf #-}
 {-# LANGUAGE ScopedTypeVariables #-}
-import Test.QuickCheck
+{-# OPTIONS_GHC -fno-warn-orphans #-}
+import Blessings.Internal as B
+import Blessings.String
+import Control.Exception
+import System.IO.Unsafe
+import System.Timeout
 import Test.Hspec
-import Blessings
+import Test.QuickCheck
+
+unsafeTimeout :: Int -> a -> a
+unsafeTimeout n f =
+    case unsafePerformIO $ timeout n (evaluate f) of
+      Nothing -> error "timeout"
+      Just y -> y
 
 instance Arbitrary a => Arbitrary (Blessings a) where
   arbitrary =
@@ -23,3 +35,46 @@ main =
       it "obeys the Monoid laws" $
         property $ \(x :: Blessings String) ->
           x <> mempty == x && x == mempty <> x
+
+      it "pp (normalize x) == pp x" $
+        property $ \(x :: Blessings String) ->
+          pp (stripSGR (normalize x)) == pp (stripSGR x)
+
+      it "take 1 x <> drop 1 x == x" $
+        property $ \(x :: Blessings String) ->
+          normalize (B.take 1 x <> B.drop 1 x) == normalize x
+
+      it "uncurry (<>) (splitAt i x) == x" $
+        property $ \(i :: Int, x :: Blessings String) ->
+          unsafeTimeout 100000 $
+            normalize (uncurry (<>) (B.splitAt i x)) == normalize x
+
+      it "splitAt produces pairs with elements of proper length" $
+        property $ \(i :: Int, x :: Blessings String) ->
+          unsafeTimeout 100000 $
+            let
+              (l, r) = B.splitAt i x
+              n = B.length x
+            in
+            if | i <= 0    -> B.length l == 0 && B.length r == n
+               | i <= n    -> B.length l == i && B.length r == n - i
+               | otherwise -> B.length l == n && B.length r == 0
+
+      let infx = mconcat (repeat (Plain "x" :: Blessings String))
+
+      it "can take from infinite structure" $
+        property $ \(n :: NonNegative Int) ->
+          unsafeTimeout 100000 $
+            let i = getNonNegative n in
+            B.length (B.take i infx) == i
+
+      it "can drop from infinite structure" $
+        property $ \(n :: NonNegative Int) ->
+          unsafeTimeout 100000 $
+            let i = getNonNegative n in
+            B.length (B.take i (B.drop i infx)) == i
+
+      it "can take concat of infinite structures" $
+        property $ \(x :: Blessings String) ->
+          unsafeTimeout 100000 $
+            B.length (B.take 1 $ infx <> x) <= 1