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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
|
{-# LANGUAGE CPP #-}
{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE NoImplicitPrelude #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE RecordWildCards #-}
{-# OPTIONS_GHC -fno-warn-orphans #-}
module Instances () where
import Prelude.Compat
import Control.Applicative (empty)
import Control.Monad
import Data.Aeson.Types
import Data.Function (on)
import Data.Time (ZonedTime(..), TimeZone(..))
import Data.Time.Clock (UTCTime(..))
import Functions
import Test.QuickCheck (Arbitrary(..), elements, oneof)
import Types
import qualified Data.DList as DList
import qualified Data.HashMap.Strict as HM
import Data.Orphans ()
import Test.QuickCheck.Instances ()
-- "System" types.
instance Arbitrary DotNetTime where
arbitrary = DotNetTime `fmap` arbitrary
shrink = map DotNetTime . shrink . fromDotNetTime
-- | Compare timezone part only on 'timeZoneMinutes'
instance Eq ZonedTime where
ZonedTime a (TimeZone a' _ _) == ZonedTime b (TimeZone b' _ _) =
a == b && a' == b'
-- Compare equality to within a millisecond, allowing for rounding
-- error (ECMA 262 requires milliseconds to rounded to zero, not
-- rounded to nearest).
instance ApproxEq UTCTime where
a =~ b = ((==) `on` utctDay) a b &&
(approxEqWith 1 1 `on` ((* 1e3) . utctDayTime)) a b
instance ApproxEq DotNetTime where
(=~) = (=~) `on` fromDotNetTime
instance ApproxEq Float where
a =~ b
| isNaN a && isNaN b = True
| otherwise = approxEq a b
instance ApproxEq Double where
a =~ b
| isNaN a && isNaN b = True
| otherwise = approxEq a b
instance (ApproxEq k, Eq v) => ApproxEq (HM.HashMap k v) where
a =~ b = and $ zipWith eq (HM.toList a) (HM.toList b)
where
eq (x,y) (u,v) = x =~ u && y == v
-- Test-related types.
instance Arbitrary Foo where
arbitrary = liftM4 Foo arbitrary arbitrary arbitrary arbitrary
instance Eq Foo where
a == b = fooInt a == fooInt b &&
fooDouble a `approxEq` fooDouble b &&
fooTuple a == fooTuple b
instance ToJSON Foo where
toJSON Foo{..} = object [ "fooInt" .= fooInt
, "fooDouble" .= fooDouble
, "fooTuple" .= fooTuple
, "fooMap" .= fooMap
]
instance FromJSON Foo where
parseJSON (Object v) = Foo <$>
v .: "fooInt" <*>
v .: "fooDouble" <*>
v .: "fooTuple" <*>
v .: "fooMap"
parseJSON _ = empty
instance Arbitrary UFoo where
arbitrary = UFoo <$> arbitrary <*> arbitrary
where _ = uFooInt
instance Arbitrary OneConstructor where
arbitrary = return OneConstructor
instance FromJSON OneConstructor
instance ToJSON OneConstructor
instance (Arbitrary a, Arbitrary b) => Arbitrary (Product2 a b) where
arbitrary = liftM2 Product2 arbitrary arbitrary
instance (FromJSON a, FromJSON b) => FromJSON (Product2 a b)
instance (ToJSON a, ToJSON b) => ToJSON (Product2 a b)
instance (Arbitrary a, Arbitrary b, Arbitrary c, Arbitrary d, Arbitrary e,
Arbitrary f) => Arbitrary (Product6 a b c d e f) where
arbitrary = Product6 <$> arbitrary <*> arbitrary <*> arbitrary <*>
arbitrary <*> arbitrary <*> arbitrary
instance (FromJSON a, FromJSON b, FromJSON c, FromJSON d, FromJSON e,
FromJSON f) => FromJSON (Product6 a b c d e f)
instance (ToJSON a, ToJSON b, ToJSON c, ToJSON d, ToJSON e,
ToJSON f) => ToJSON (Product6 a b c d e f)
instance (Arbitrary a, Arbitrary b, Arbitrary c, Arbitrary d)
=> Arbitrary (Sum4 a b c d) where
arbitrary = oneof [Alt1 <$> arbitrary, Alt2 <$> arbitrary,
Alt3 <$> arbitrary, Alt4 <$> arbitrary]
instance (FromJSON a, FromJSON b, FromJSON c, FromJSON d)
=> FromJSON (Sum4 a b c d)
instance (ToJSON a, ToJSON b, ToJSON c, ToJSON d) => ToJSON (Sum4 a b c d)
instance (Arbitrary a) => Arbitrary (Approx a) where
arbitrary = Approx <$> arbitrary
instance (FromJSON a) => FromJSON (Approx a) where
parseJSON a = Approx <$> parseJSON a
instance (ToJSON a) => ToJSON (Approx a) where
toJSON = toJSON . fromApprox
instance Arbitrary Nullary where
arbitrary = elements [C1, C2, C3]
instance Arbitrary a => Arbitrary (SomeType a) where
arbitrary = oneof [ pure Nullary
, Unary <$> arbitrary
, Product <$> arbitrary <*> arbitrary <*> arbitrary
, Record <$> arbitrary <*> arbitrary <*> arbitrary
, List <$> arbitrary
]
instance Arbitrary EitherTextInt where
arbitrary = oneof
[ LeftBool <$> arbitrary
, RightInt <$> arbitrary
, BothTextInt <$> arbitrary <*> arbitrary
, pure NoneNullary
]
instance Arbitrary (GADT String) where
arbitrary = GADT <$> arbitrary
#if !MIN_VERSION_base(4,16,0)
instance Arbitrary OptionField where
arbitrary = OptionField <$> arbitrary
#endif
instance ApproxEq Char where
(=~) = (==)
instance (ApproxEq a) => ApproxEq [a] where
a =~ b = length a == length b && all (uncurry (=~)) (zip a b)
instance Arbitrary a => Arbitrary (DList.DList a) where
arbitrary = DList.fromList <$> arbitrary
|
