mirror of
https://github.com/facebook/Haxl.git
synced 2024-12-23 16:53:02 +03:00
dfd8a4655e
Summary: Test that the ApplicativeDo extension batches things correctly in the Haxl monad. Test Plan: beholdunittests Reviewed By: bnitka@fb.com Subscribers: ldbrandy, memo, watashi, smarlow, akr, bnitka, jcoens FB internal diff: D2039149 Tasks: 5504687 Signature: t1:2039149:1430501733:98fd1cf0f69663d6db3b07c3aed6e261ae9884d6
50 lines
953 B
Haskell
50 lines
953 B
Haskell
{-# LANGUAGE RebindableSyntax, OverloadedStrings, ApplicativeDo #-}
|
|
module AdoTests (tests) where
|
|
|
|
import TestUtils
|
|
import MockTAO
|
|
|
|
import Control.Applicative
|
|
import Test.HUnit
|
|
|
|
import Prelude()
|
|
import Haxl.Prelude
|
|
|
|
-- -----------------------------------------------------------------------------
|
|
|
|
--
|
|
-- Test ApplicativeDo batching
|
|
--
|
|
ado1 = expectRounds 1 12 ado1_
|
|
|
|
ado1_ = do
|
|
a <- friendsOf =<< id1
|
|
b <- friendsOf =<< id2
|
|
return (length (a ++ b))
|
|
|
|
ado2 = expectRounds 1 12 ado2_
|
|
|
|
ado2_ = do
|
|
x <- id1
|
|
a <- friendsOf x
|
|
y <- id2
|
|
b <- friendsOf y
|
|
return (length (a ++ b))
|
|
|
|
ado3 = expectRounds 2 11 ado3_
|
|
|
|
ado3_ = do
|
|
x <- id1
|
|
a <- friendsOf x
|
|
a' <- friendsOf =<< if null a then id3 else id4
|
|
y <- id2
|
|
b <- friendsOf y
|
|
b' <- friendsOf =<< if null b then id4 else id3
|
|
return (length (a' ++ b'))
|
|
|
|
tests = TestList
|
|
[ TestLabel "ado1" $ TestCase ado1
|
|
, TestLabel "ado2" $ TestCase ado2
|
|
, TestLabel "ado3" $ TestCase ado3
|
|
]
|