1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
module Data.Conduit.ExtraSpec where
import Data.Conduit
import Test.Hspec
import Test.Hspec.QuickCheck
import Data.Conduit.List (isolate, peek, consume)
import qualified Data.Conduit.List as CL
import qualified Data.Text as T
import qualified Data.Text.Encoding as T
import qualified Data.ByteString as S
import qualified Data.Conduit.Text as CT
spec :: Spec
spec = describe "Data.Conduit.Extra" $ do
it "basic test" $ do
let sink2 :: ConduitT a o IO (Maybe a, Maybe a)
sink2 = do
ma1 <- fuseLeftovers id (isolate 10) peek
ma2 <- peek
return (ma1, ma2)
source = yield 1 >> yield 2
res <- runConduit $ source .| sink2
res `shouldBe` (Just 1, Just (1 :: Int))
it "get leftovers" $ do
let sink2 :: ConduitT a o IO ([a], [a], [a])
sink2 = do
(x, y) <- fuseReturnLeftovers (isolate 2) peek3
z <- CL.consume
return (x, y, z)
peek3 = do
x <- CL.take 3
mapM_ leftover $ reverse x
return x
source = mapM_ yield [1..5 :: Int]
res <- runConduit $ source .| sink2
res `shouldBe` ([1..2], [1..2], [3..5])
it "multiple values" $ do
let sink2 :: ConduitT a o IO ([a], Maybe a)
sink2 = do
ma1 <- fuseLeftovers id (isolate 10) peek3
ma2 <- peek
return (ma1, ma2)
peek3 = do
x <- CL.take 3
mapM_ leftover $ reverse x
return x
source = mapM_ yield [1..5]
res <- runConduit $ source .| sink2
res `shouldBe` ([1..3], Just (1 :: Int))
prop "more complex" $ \ss cnt -> do
let ts = map T.pack ss
src = mapM_ (yield . T.encodeUtf8) ts
conduit = CL.map T.decodeUtf8
sink = CT.take cnt .| consume
undo = return . T.encodeUtf8 . T.concat
res <- runConduit $ src .| do
x <- fuseLeftovers undo conduit sink
y <- consume
return (T.concat x, T.decodeUtf8 $ S.concat y)
res `shouldBe` T.splitAt cnt (T.concat ts)
main :: IO ()
main = hspec spec
|
