Haxl/tests/AdoTests.hs
Simon Marlow c1c789a71f Change license to plain BSD3
Reviewed By: mic47, niteria

Differential Revision: D6519157

fbshipit-source-id: 7fd977837bb7dd8463d697d685107aaf07c95255
2017-12-08 04:33:35 -08:00

56 lines
1.1 KiB
Haskell

-- Copyright (c) 2014-present, Facebook, Inc.
-- All rights reserved.
--
-- This source code is distributed under the terms of a BSD license,
-- found in the LICENSE file.
{-# 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 future = TestList
[ TestLabel "ado1" $ TestCase (ado1 future)
, TestLabel "ado2" $ TestCase (ado2 future)
, TestLabel "ado3" $ TestCase (ado3 future)
]