{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE TemplateHaskell #-}

-- Copyright (C) 2010 John Millikin <jmillikin@gmail.com>
--
-- See license.txt for details
module EnumeratorTests.Binary.Drop
	( test_Drop
	, test_DropWhile
	, test_Filter
	, test_FilterM
	) where

import           Test.Chell

import           Data.Enumerator ((=$))
import qualified Data.Enumerator as E
import qualified Data.Enumerator.Binary as EB
import qualified Data.Enumerator.List as EL

test_Drop :: Suite
test_Drop = assertions "drop" $ do
	$expect $ equal
		["ABCDE"]
		(E.runLists_ [["ABCDE"]] $ do
			EB.drop 0
			EL.consume)
	$expect $ equal
		["CDE"]
		(E.runLists_ [["ABCDE"]] $ do
			EB.drop 2
			EL.consume)
	$expect $ equal
		["CDE"]
		(E.runLists_ [["A"], ["BCDE"]] $ do
			EB.drop 2
			EL.consume)

test_DropWhile :: Suite
test_DropWhile = assertions "dropWhile" $ do
	$expect $ equal
		["CDE"]
		(E.runLists_ [["ABCDE"]] $ do
			EB.dropWhile (< 0x43)
			EL.consume)
	$expect $ equal
		[]
		(E.runLists_ [["ABCDE"]] $ do
			EB.dropWhile (\_ -> True)
			EL.consume)

test_Filter :: Suite
test_Filter = assertions "filter" $ do
	$expect $ equal
		["A", "B", "", "D", "E"]
		(E.runLists_ [["ABCDE"]] $ do
			EB.filter (/= 0x43) =$ EL.consume)

test_FilterM :: Suite
test_FilterM = assertions "filterM" $ do
	$expect $ equal
		["A", "B", "", "D", "E"]
		(E.runLists_ [["ABCDE"]] $ do
			EB.filterM (\x -> return (x /= 0x43)) =$ EL.consume)