{-# LANGUAGE CPP #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE TupleSections #-}
{-# OPTIONS_GHC -fno-warn-incomplete-patterns #-}
module Data.Conduit.AttoparsecSpec (spec) where
import Control.Exception (fromException)
import Test.Hspec
import Control.Applicative ((<*), (<|>))
import Control.Monad
import Control.Monad.Trans.Resource (runExceptionT)
import qualified Data.Attoparsec.ByteString.Char8
import qualified Data.Attoparsec.Text
import Data.Conduit
import Data.Conduit.Attoparsec
import qualified Data.Conduit.List as CL
spec :: Spec
spec = describe "Data.Conduit.AttoparsecSpec" $ do
describe "error position" $ do
it "works for text" $ do
let input = ["aaa\na", "aaa\n\n", "aaa", "aab\n\naaaa"]
badLine = 4
badCol = 6
badOff = 15
parser = Data.Attoparsec.Text.endOfInput <|> (Data.Attoparsec.Text.notChar 'b' >> parser)
sink = sinkParser parser
sink' = sinkParserEither parser
ea <- runExceptionT $ CL.sourceList input $$ sink
case ea of
Left e ->
case fromException e of
Just pe -> do
errorPosition pe `shouldBe` Position badLine badCol badOff
ea' <- CL.sourceList input $$ sink'
case ea' of
Left pe ->
errorPosition pe `shouldBe` Position badLine badCol badOff
it "works for bytestring" $ do
let input = ["aaa\na", "aaa\n\n", "aaa", "aab\n\naaaa"]
badLine = 4
badCol = 6
badOff = 15
parser = Data.Attoparsec.ByteString.Char8.endOfInput <|> (Data.Attoparsec.ByteString.Char8.notChar 'b' >> parser)
sink = sinkParser parser
sink' = sinkParserEither parser
ea <- runExceptionT $ CL.sourceList input $$ sink
case ea of
Left e ->
case fromException e of
Just pe -> do
errorPosition pe `shouldBe` Position badLine badCol badOff
ea' <- CL.sourceList input $$ sink'
case ea' of
Left pe ->
errorPosition pe `shouldBe` Position badLine badCol badOff
it "works in last chunk" $ do
let input = ["aaa\na", "aaa\n\n", "aaa", "aab\n\naaaa"]
badLine = 6
badCol = 5
badOff = 22
parser = Data.Attoparsec.Text.char 'c' <|> (Data.Attoparsec.Text.anyChar >> parser)
sink = sinkParser parser
sink' = sinkParserEither parser
ea <- runExceptionT $ CL.sourceList input $$ sink
case ea of
Left e ->
case fromException e of
Just pe -> do
errorPosition pe `shouldBe` Position badLine badCol badOff
ea' <- CL.sourceList input $$ sink'
case ea' of
Left pe ->
errorPosition pe `shouldBe` Position badLine badCol badOff
it "works in last chunk" $ do
let input = ["aaa\na", "aaa\n\n", "aaa", "aa\n\naaaab"]
badLine = 6
badCol = 6
badOff = 22
parser = Data.Attoparsec.Text.string "bc" <|> (Data.Attoparsec.Text.anyChar >> parser)
sink = sinkParser parser
sink' = sinkParserEither parser
ea <- runExceptionT $ CL.sourceList input $$ sink
case ea of
Left e ->
case fromException e of
Just pe -> do
errorPosition pe `shouldBe` Position badLine badCol badOff
ea' <- CL.sourceList input $$ sink'
case ea' of
Left pe ->
errorPosition pe `shouldBe` Position badLine badCol badOff
it "works after new line in text" $ do
let input = ["aaa\n", "aaa\n\n", "aaa", "aa\nb\naaaa"]
badLine = 5
badCol = 1
badOff = 15
parser = Data.Attoparsec.Text.endOfInput <|> (Data.Attoparsec.Text.notChar 'b' >> parser)
sink = sinkParser parser
sink' = sinkParserEither parser
ea <- runExceptionT $ CL.sourceList input $$ sink
case ea of
Left e ->
case fromException e of
Just pe -> do
errorPosition pe `shouldBe` Position badLine badCol badOff
ea' <- CL.sourceList input $$ sink'
case ea' of
Left pe ->
errorPosition pe `shouldBe` Position badLine badCol badOff
it "works after new line in bytestring" $ do
let input = ["aaa\n", "aaa\n\n", "aaa", "aa\nb\naaaa"]
badLine = 5
badCol = 1
badOff = 15
parser = Data.Attoparsec.ByteString.Char8.endOfInput <|> (Data.Attoparsec.ByteString.Char8.notChar 'b' >> parser)
sink = sinkParser parser
sink' = sinkParserEither parser
ea <- runExceptionT $ CL.sourceList input $$ sink
case ea of
Left e ->
case fromException e of
Just pe -> do
errorPosition pe `shouldBe` Position badLine badCol badOff
ea' <- CL.sourceList input $$ sink'
case ea' of
Left pe ->
errorPosition pe `shouldBe` Position badLine badCol badOff
it "works for first line" $ do
let input = ["aab\na", "aaa\n\n", "aaa", "aab\n\naaaa"]
badLine = 1
badCol = 3
badOff = 2
parser = Data.Attoparsec.Text.endOfInput <|> (Data.Attoparsec.Text.notChar 'b' >> parser)
sink = sinkParser parser
sink' = sinkParserEither parser
ea <- runExceptionT $ CL.sourceList input $$ sink
case ea of
Left e ->
case fromException e of
Just pe -> do
errorPosition pe `shouldBe` Position badLine badCol badOff
ea' <- CL.sourceList input $$ sink'
case ea' of
Left pe ->
errorPosition pe `shouldBe` Position badLine badCol badOff
describe "conduitParser" $ do
it "parses a repeated stream" $ do
let input = ["aaa\n", "aaa\naaa\n", "aaa\n"]
parser = Data.Attoparsec.Text.string "aaa" <* Data.Attoparsec.Text.endOfLine
sink = conduitParserEither parser =$= CL.consume
(Right ea) <- runExceptionT $ CL.sourceList input $$ sink
let chk a = case a of
Left{} -> False
Right (_, xs) -> xs == "aaa"
chkp l = PositionRange (Position l 1 ((l - 1) * 4)) (Position (l+1) 1 (l * 4))
forM_ ea $ \ a -> a `shouldSatisfy` chk :: Expectation
forM_ (zip ea [1..]) $ \ (Right (pos, _), l) -> pos `shouldBe` chkp l
length ea `shouldBe` 4
it "positions on first line" $ do
results <- yield "hihihi\nhihi"
$$ conduitParser (Data.Attoparsec.Text.string "\n" <|> Data.Attoparsec.Text.string "hi")
=$ CL.consume
let f (a, b, c, d, e, f, g) = (PositionRange (Position a b c) (Position d e f), g)
results `shouldBe` map f
[ (1, 1, 0, 1, 3, 2, "hi")
, (1, 3, 2, 1, 5, 4, "hi")
, (1, 5, 4, 1, 7, 6, "hi")
, (1, 7, 6, 2, 1, 7, "\n")
, (2, 1, 7, 2, 3, 9, "hi")
, (2, 3, 9, 2, 5, 11, "hi")
]