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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
{-# LANGUAGE BangPatterns #-}
{-# LANGUAGE CPP #-}
{-# OPTIONS_GHC -fno-warn-orphans #-}
module QuickCheckUtils (NonEmptyText(..), LatinSpoofableText(..),
NonSpoofableText(..), Utf8Text(..)) where
import Data.Text.ICU (Collator, LocaleName(..))
import Data.Text.ICU.Normalize2
import Data.Text.ICU.Break (available)
import Test.QuickCheck (Arbitrary(..), Gen, elements, listOf1, suchThat, vectorOf)
import Data.Word (Word8)
import qualified Data.ByteString as BS
import qualified Data.Text as T
import qualified Data.Text.ICU as I
#if !MIN_VERSION_base(4,11,0)
import Data.Semigroup ((<>))
#endif
instance Arbitrary T.Text where
arbitrary = T.pack `fmap` arbitrary
shrink = map T.pack . shrink . T.unpack
instance Arbitrary BS.ByteString where
arbitrary = BS.pack <$> arbitrary
shrink xs = BS.pack <$> shrink (BS.unpack xs)
instance Arbitrary LocaleName where
arbitrary = elements (Root:available)
instance Arbitrary NormalizationMode where
arbitrary = elements [NFD .. NFKCCasefold]
instance Arbitrary Collator where
arbitrary = I.collator <$> arbitrary
newtype NonEmptyText = NonEmptyText { nonEmptyText :: T.Text } deriving Show
instance Arbitrary NonEmptyText where
arbitrary = NonEmptyText . T.pack <$> listOf1 arbitrary
newtype LatinSpoofableText = LatinSpoofableText { latinSpoofableText :: T.Text }
deriving Show
instance Arbitrary LatinSpoofableText where
arbitrary = LatinSpoofableText <$> T.pack . (<>) "latin" <$>
listOf1 genCyrillicLatinSpoofableChar
genCyrillicLatinSpoofableChar :: Gen Char
genCyrillicLatinSpoofableChar = elements (
"\x043A\x043E\x0433\x0435\x043A\x043C" ++
['\x043E'..'\x0443'] ++
['\x0445'..'\x0446'] ++
"\x044A" ++
['\x0454'..'\x0456'] ++
"\x0458\x045B\x048D\x0491\x0493\x049B\x049F\x04AB\x04AD\x04AF\x04B1\x04BB" ++
"\x04BD\x04BF" ++
['\x04CE'..'\x04CF'] ++
"\x04D5\x04D9\x04E9\x0501\x0511\x051B\x051D")
newtype NonSpoofableText = NonSpoofableText { nonSpoofableText :: T.Text }
deriving Show
instance Arbitrary NonSpoofableText where
arbitrary = NonSpoofableText <$> T.pack <$> listOf1 genNonSpoofableChar
genNonSpoofableChar :: Gen Char
genNonSpoofableChar = elements "QDFRz"
newtype Utf8Text = Utf8Text { utf8Text :: BS.ByteString }
deriving Show
instance Arbitrary Utf8Text where
arbitrary = Utf8Text . BS.pack <$> vectorOf 300
(suchThat
(arbitrary :: Gen Word8)
(`elem` ([0x41..0x5A] ++ [0x61..0x7A]))
)
|
