1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
module TestUtil(
(==>), QFilePath(..), QFilePathValidW(..), QFilePathValidP(..),
module Test.QuickCheck,
module Data.List,
module Data.Maybe
) where
import Test.QuickCheck hiding ((==>))
import Data.List
import Data.Maybe
import Control.Monad
import qualified System.FilePath.Windows as W
import qualified System.FilePath.Posix as P
infixr 0 ==>
a ==> b = not a || b
newtype QFilePathValidW = QFilePathValidW FilePath deriving Show
instance Arbitrary QFilePathValidW where
arbitrary = fmap (QFilePathValidW . W.makeValid) arbitraryFilePath
shrink (QFilePathValidW x) = shrinkValid QFilePathValidW W.makeValid x
newtype QFilePathValidP = QFilePathValidP FilePath deriving Show
instance Arbitrary QFilePathValidP where
arbitrary = fmap (QFilePathValidP . P.makeValid) arbitraryFilePath
shrink (QFilePathValidP x) = shrinkValid QFilePathValidP P.makeValid x
newtype QFilePath = QFilePath FilePath deriving Show
instance Arbitrary QFilePath where
arbitrary = fmap QFilePath arbitraryFilePath
shrink (QFilePath x) = shrinkValid QFilePath id x
-- | Generate an arbitrary FilePath use a few special (interesting) characters.
arbitraryFilePath :: Gen FilePath
arbitraryFilePath = sized $ \n -> do
k <- choose (0,n)
replicateM k $ elements "?./:\\a ;_"
-- | Shrink, but also apply a validity function. Try and make shorter, or use more
-- @a@ (since @a@ is pretty dull), but make sure you terminate even after valid.
shrinkValid :: (FilePath -> a) -> (FilePath -> FilePath) -> FilePath -> [a]
shrinkValid wrap valid o =
[ wrap y
| y <- map valid $ shrinkList (\x -> ['a' | x /= 'a']) o
, length y < length o || (length y == length o && countA y > countA o)]
where countA = length . filter (== 'a')
|
