Add spec for lenient decoding

2 files changed, 39 insertions, 1 deletions

diff --git a/tests/Data/ByteString/Base32/HexSpec.hs b/tests/Data/ByteString/Base32/HexSpec.hs
index 4e41d09..3a04b09 100644
--- a/tests/Data/ByteString/Base32/HexSpec.hs
+++ b/tests/Data/ByteString/Base32/HexSpec.hs
@@ -45,3 +45,23 @@ spec = do
     it "fail gracefully if encoded data contains non alphabet chars" $ do
       evaluate (decode "#=======")         `shouldThrow` anyErrorCall
       evaluate (decode "AAAAAAAA#=======") `shouldThrow` anyErrorCall
+
+  describe "decodeLenient" $ do
+    it "conform RFC examples" $ do
+      decodeLenient ""                 `shouldBe` ""
+      decodeLenient "MY======"         `shouldBe` "f"
+      decodeLenient "MZXQ===="         `shouldBe` "fo"
+      decodeLenient  "MZXW6==="        `shouldBe` "foo"
+      decodeLenient "MZXW6YQ="         `shouldBe` "foob"
+      decodeLenient "MZXW6YTB"         `shouldBe` "fooba"
+      decodeLenient "MZXW6YTBOI======" `shouldBe` "foobar"
+
+    it "inverse for encode" $ property $ \bs ->
+      decodeLenient (encode bs) == bs
+
+    it "case insensitive" $ property $ \bs ->
+      decodeLenient (BC.map toLower (encode bs)) == bs
+
+    it "skip non alphabet chars" $ do
+      decodeLenient "|"   `shouldBe` ""
+      decodeLenient "M|Y" `shouldBe` "f"
diff --git a/tests/Data/ByteString/Base32Spec.hs b/tests/Data/ByteString/Base32Spec.hs
index 66df53a..148539a 100644
--- a/tests/Data/ByteString/Base32Spec.hs
+++ b/tests/Data/ByteString/Base32Spec.hs
@@ -52,4 +52,22 @@ spec = do
       evaluate (decode "0=======")         `shouldThrow` anyErrorCall
       evaluate (decode "AAAAAAAA0=======") `shouldThrow` anyErrorCall
 
---  describe "decodeLenient" $ do
+  describe "decodeLenient" $ do
+    it "conform RFC examples" $ do
+      decodeLenient ""                 `shouldBe` ""
+      decodeLenient "MY======"         `shouldBe` "f"
+      decodeLenient "MZXQ===="         `shouldBe` "fo"
+      decodeLenient  "MZXW6==="        `shouldBe` "foo"
+      decodeLenient "MZXW6YQ="         `shouldBe` "foob"
+      decodeLenient "MZXW6YTB"         `shouldBe` "fooba"
+      decodeLenient "MZXW6YTBOI======" `shouldBe` "foobar"
+
+    it "inverse for encode" $ property $ \bs ->
+      decodeLenient (encode bs) == bs
+
+    it "case insensitive" $ property $ \bs ->
+      decodeLenient (BC.map toLower (encode bs)) == bs
+
+    it "skip non alphabet chars" $ do
+      decodeLenient "|"   `shouldBe` ""
+      decodeLenient "M|Y" `shouldBe` "f"