88 lines
2.8 KiB
Haskell
88 lines
2.8 KiB
Haskell
module Test.Pipes.CBOR where
|
|
|
|
import Prelude
|
|
|
|
import Control.Monad.Cont (lift)
|
|
import Control.Monad.Gen (chooseInt)
|
|
import Data.Array as Array
|
|
import Data.DateTime (DateTime)
|
|
import Data.List ((:))
|
|
import Data.List as List
|
|
import Data.Maybe (Maybe(..), fromJust)
|
|
import Data.Newtype (wrap)
|
|
import Data.PreciseDateTime (fromRFC3339String, toDateTimeLossy)
|
|
import Data.Tuple.Nested ((/\))
|
|
import Effect (Effect)
|
|
import Effect.Aff (delay)
|
|
import Effect.CBOR as CBOR
|
|
import Effect.Class (liftEffect)
|
|
import Node.Buffer (Buffer)
|
|
import Node.Buffer as Buffer
|
|
import Node.Encoding (Encoding(..))
|
|
import Partial.Unsafe (unsafePartial)
|
|
import Pipes (yield, (>->))
|
|
import Pipes.CBOR as Pipes.CBOR
|
|
import Pipes.Collect as Pipes.Collect
|
|
import Pipes.Node.Stream as Pipes.Stream
|
|
import Pipes.Prelude (toListM) as Pipes
|
|
import Test.QuickCheck.Gen (randomSample')
|
|
import Test.Spec (Spec, before, describe, it)
|
|
import Test.Spec.Assertions (shouldEqual)
|
|
|
|
cborHex :: String
|
|
cborHex = "82b90002646e616d656568656e72796174c1fb41d990ee6d671aa0b90002646e616d65656a756c696f6174c1fbc1d756dad0bbb646"
|
|
|
|
cborBuf :: Effect Buffer
|
|
cborBuf = Buffer.fromString cborHex Hex
|
|
|
|
exp :: Array {name :: String, t :: DateTime}
|
|
exp =
|
|
[{name: "henry", t: toDateTimeLossy $ unsafePartial fromJust $ fromRFC3339String $ wrap "2024-05-14T19:21:25.611Z"}
|
|
,{name: "julio", t: toDateTimeLossy $ unsafePartial fromJust $ fromRFC3339String $ wrap "1920-05-14T20:21:17.067Z"}
|
|
]
|
|
|
|
|
|
dt :: String -> DateTime
|
|
dt = toDateTimeLossy <<< unsafePartial fromJust <<< fromRFC3339String <<< wrap
|
|
|
|
spec :: Spec Unit
|
|
spec =
|
|
describe "Pipes.CBOR" do
|
|
it "encode" do
|
|
bytes
|
|
<- Pipes.Collect.toBuffer
|
|
$ Pipes.Stream.withEOS (yield exp)
|
|
>-> Pipes.CBOR.encode
|
|
>-> Pipes.Stream.unEOS
|
|
act <- liftEffect $ CBOR.decode bytes
|
|
act `shouldEqual` exp
|
|
|
|
describe "parse" do
|
|
it "parses csv" do
|
|
buf <- liftEffect $ cborBuf
|
|
rows <- Pipes.toListM
|
|
$ Pipes.Stream.withEOS (yield buf *> lift (delay $ wrap 10.0))
|
|
>-> Pipes.CBOR.decode
|
|
|
|
rows `shouldEqual` ((Just exp) : Nothing : List.Nil)
|
|
before
|
|
(do
|
|
nums <- liftEffect $ randomSample' 100000 (chooseInt 0 9)
|
|
let
|
|
objs = (\n -> {id: n}) <$> nums
|
|
bytes <-
|
|
Pipes.Collect.toBuffer
|
|
$ Pipes.Stream.withEOS (yield objs)
|
|
>-> Pipes.CBOR.encode
|
|
>-> Pipes.Stream.unEOS
|
|
pure $ nums /\ bytes
|
|
)
|
|
$ it "parses large csv" \(nums /\ bytes) -> do
|
|
rows <-
|
|
Pipes.Collect.toArray
|
|
$ Pipes.Stream.withEOS (yield bytes)
|
|
>-> Pipes.CBOR.decode @(Array {id :: Int})
|
|
>-> Pipes.Stream.unEOS
|
|
|
|
rows `shouldEqual` [(\id -> { id }) <$> nums]
|