mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-24 07:52:14 +03:00
22 lines
688 B
Haskell
22 lines
688 B
Haskell
|
module Control.Concurrent.ExtendedSpec
|
||
|
|
( spec,
|
||
|
|
)
|
||
|
|
where
|
||
|
|
|
||
|
|
import Control.Concurrent.Extended qualified as C
|
||
|
|
import Hasura.Prelude
|
||
|
|
import Test.Hspec
|
||
|
|
|
||
|
|
spec :: Spec
|
||
|
|
spec = do
|
||
|
|
describe "forConcurrentlyEIO" $ do
|
||
|
|
it "chunks correctly" $ do
|
||
|
|
for_ [1, 2, 3, 4, 5] $ \chunkSize -> do
|
||
|
|
out <- runExceptT $ C.forConcurrentlyEIO chunkSize [pure 1, pure 2, pure 3, pure 4] id
|
||
|
|
out `shouldBe` (Right [1, 2, 3, 4] :: Either Int [Int])
|
||
|
|
|
||
|
|
it "behaves like ExceptT" $ do
|
||
|
|
for_ [1, 2, 3, 4, 5] $ \chunkSize -> do
|
||
|
|
out <- runExceptT $ C.forConcurrentlyEIO chunkSize [pure 1, throwError 2, throwError 3, pure 4] id
|
||
|
|
out `shouldBe` (Left 2 :: Either Int [Int])
|