module B9.QCUtil where
import Control.Monad
import Test.QuickCheck
arbitraryEnv :: Arbitrary a => Gen [(String, a)]
arbitraryEnv = listOf ((,) <$> listOf1 (choose ('a', 'z')) <*> arbitrary)
halfSize :: Gen a -> Gen a
halfSize g = sized (flip resize g . flip div 2)
smaller :: Gen a -> Gen a
smaller g = sized (flip resize g . max 0 . flip (-) 1)
arbitraryFilePath :: Gen FilePath
arbitraryFilePath = do
path <- join <$> listOf
(elements
[ "/"
, "../"
, "./"
, "etc/"
, "opt/"
, "user/"
, "var/"
, "tmp/"
, "doc/"
, "share/"
, "conf.d/"
]
)
prefix <- elements ["foo_", "", "alt_", "ssh-", ""]
body <- elements ["www", "passwd", "cert", "opnsfe", "runtime"]
extension <- elements [".txt", ".png", ".ps", ".erl", ""]
return (path ++ prefix ++ body ++ extension)
arbitraryLetter :: Gen Char
arbitraryLetter = oneof [arbitraryLetterUpper, arbitraryLetterLower]
arbitraryLetterUpper :: Gen Char
arbitraryLetterUpper = elements ['A' .. 'Z']
arbitraryLetterLower :: Gen Char
arbitraryLetterLower = elements ['a' .. 'z']
arbitraryDigit :: Gen Char
arbitraryDigit = elements ['0' .. '9']