chore: add `ConduitT` instancesmain

1 files changed, 50 insertions, 3 deletions

diff --git a/test/Main.hs b/test/Main.hs
index 2f20997..f6bfaf6 100644
--- a/test/Main.hs
+++ b/test/Main.hs
@@ -5,6 +5,7 @@
 
 module Main (main) where
 
+import Conduit
 import Data.ByteString.Char8 qualified as B
 import Data.ByteString.Lazy.Char8 qualified as LB
 import Data.Text qualified as T
@@ -14,25 +15,67 @@ import Test.Hspec
 
 main :: IO ()
 main = hspec do
+  describe "input" do
+    it "capture stdin" do
+      (`shouldBe` "stdin") =<< runConduitRes (yield "stdin\n" .| [sh|cat|])
+      (`shouldBe` "stdin") =<< runConduitRes (yield "stdin\n" .| [sh|cat|] .| foldC)
   describe "output" do
     it "capture stdout" do
-      (`shouldBe` "stdout") . fst @String @String =<< [sh|echo stdout|]
+      (`shouldBe` "stdout") . fst @String @() =<< [sh|echo stdout|]
+      (`shouldBe` "stdout") . fst @String @String =<< runConduitRes [sh|echo stdout|]
+      (`shouldBe` "stdout") . snd @String @String =<< runConduitRes ([sh|echo stdout|] .| partitionEithersC)
+    -- TODO conduit
     it "capture stderr" do
-      (`shouldBe` "stderr") . snd @String =<< [sh|>&2 echo stderr|]
+      (`shouldBe` "stderr") . snd @() @String =<< [sh|>&2 echo stderr|]
+      (`shouldBe` "stderr") . snd @() @String =<< runConduitRes [sh|>&2 echo stderr|]
+      (`shouldBe` "stderr") . fst @String @String
+        =<< runConduitRes ([sh|>&2 echo stderr|] .| partitionEithersC)
     it "capture stdout and stderr" do
       (`shouldBe` ("stdout", "stderr"))
         =<< [sh|
         echo stdout
         >&2 echo stderr
         |]
+      (`shouldBe` ("stdout", "stderr"))
+        =<< runConduitRes
+          ( [sh|
+              echo stdout
+              >&2 echo stderr
+              |]
+          )
+      (`shouldBe` ("stderr", "stdout"))
+        =<< runConduitRes
+          ( [sh|
+              echo stdout
+              >&2 echo stderr
+              |]
+              .| partitionEithersC
+          )
     it "capture stdout and stderr interleaved" do
       (`shouldBe` "stdout\nstderr")
         =<< [sh|
         echo stdout
         >&2 echo stderr
         |]
+      (`shouldBe` "stdout\nstderr")
+        =<< runConduitRes
+          ( [sh|
+              echo stdout
+              >&2 echo stderr
+              |]
+          )
+      (`shouldBe` "stdout\nstderr")
+        =<< runConduitRes
+          ( [sh|
+              echo stdout
+              >&2 echo stderr
+              |]
+              .| foldC
+          )
     it "preserve trailing newline" do
-      (`shouldBe` "stdout\n") . fst @String @String =<< [sh|echo stdout \|]
+      (`shouldBe` "stdout\n") =<< [sh|echo stdout \|]
+      (`shouldBe` "stdout\n") =<< runConduitRes [sh|echo stdout \|]
+      (`shouldBe` "stdout\n") =<< runConduitRes ([sh|echo stdout \|] .| foldC)
   describe "arguments" do
     it "passes `Int`" do
       (`shouldBe` "1") =<< let x = 1 :: Int in [sh|echo '#{x}'|]
@@ -64,3 +107,7 @@ main = hspec do
       (`shouldBe` "foobar") =<< [sh|echo 'foobar'|]
     it "parses double cross" do
       (`shouldBe` "0") =<< [sh|echo $#|]
+
+partitionEithersC :: (Monad m, Monoid a, Monoid b) => ConduitT (Either a b) o m (a, b)
+partitionEithersC =
+  foldlC (\(es, rs) x -> either (\e -> (e `mappend` es, rs)) (\r -> (es, r `mappend` rs)) x) (mempty, mempty)