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

-- Copyright (C) 2010 John Millikin <jmillikin@gmail.com>
--
-- See license.txt for details
module EnumeratorTests.Text.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.List as EL
import qualified Data.Enumerator.Text as ET

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

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

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

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