{-# LANGUAGE CPP #-}
module Main where

import qualified PaymentChannel.Test as Pay
import Criterion.Main
import Test.QuickCheck  (sample')

import qualified Data.ByteString as B
#if !MIN_VERSION_bytestring(0,10,0)
instance NFData B.ByteString
#endif

payCount = 1000 :: Integer

main :: IO ()
main = do
  (arbPair,_)   <- fmap head $ sample' $ Pay.mkChanPairInitAmount 0
  let mkPayment = Pay.createPayment (Pay.sendChan arbPair)
  (_,payment)   <- either (fail . show) return =<< mkPayment 1
  let multiMkVerify = Pay.runChanPair arbPair (fromIntegral <$> [1..payCount])
  defaultMain
    [ bench "Create payment" $ nfIO ( either (error . show) id <$> mkPayment 2 )
    , bench "Verify payment" $ nfIO
        ( either (error . show) id <$> Pay.acceptPayment (Pay.recvChan arbPair) payment )
    , bench (show payCount ++ " rounds of create+verify payment") $ nfIO multiMkVerify
    ]