| 30 |
30 |
( testTypes
|
| 31 |
31 |
, AllocPolicy(..)
|
| 32 |
32 |
, DiskTemplate(..)
|
|
33 |
, allDiskTemplates
|
| 33 |
34 |
, InstanceStatus(..)
|
| 34 |
35 |
, NonEmpty(..)
|
| 35 |
36 |
, Hypervisor(..)
|
| 36 |
37 |
, JobId(..)
|
| 37 |
38 |
) where
|
| 38 |
39 |
|
| 39 |
|
import Data.List (sort)
|
|
40 |
import Data.List (delete, sort)
|
| 40 |
41 |
import Test.QuickCheck as QuickCheck hiding (Result)
|
| 41 |
42 |
import Test.HUnit
|
| 42 |
43 |
import qualified Text.JSON as J
|
| ... | ... | |
| 78 |
79 |
|
| 79 |
80 |
$(genArbitrary ''AllocPolicy)
|
| 80 |
81 |
|
| 81 |
|
$(genArbitrary ''DiskTemplate)
|
|
82 |
-- | Valid disk templates (depending on configure options).
|
|
83 |
allDiskTemplates :: [DiskTemplate]
|
|
84 |
allDiskTemplates =
|
|
85 |
let all_vals = [minBound..maxBound]::[DiskTemplate]
|
|
86 |
sel1 = if C.enableFileStorage
|
|
87 |
then all_vals
|
|
88 |
else delete DTFile all_vals
|
|
89 |
sel2 = if C.enableSharedFileStorage
|
|
90 |
then sel1
|
|
91 |
else delete DTSharedFile sel1
|
|
92 |
in sel2
|
|
93 |
|
|
94 |
-- | Custom 'Arbitrary' instance for 'DiskTemplate', which needs to
|
|
95 |
-- handle the case of file storage being disabled at configure time.
|
|
96 |
instance Arbitrary DiskTemplate where
|
|
97 |
arbitrary = elements allDiskTemplates
|
| 82 |
98 |
|
| 83 |
99 |
$(genArbitrary ''InstanceStatus)
|
| 84 |
100 |
|
| ... | ... | |
| 96 |
112 |
|
| 97 |
113 |
$(genArbitrary ''OobCommand)
|
| 98 |
114 |
|
| 99 |
|
$(genArbitrary ''StorageType)
|
|
115 |
-- | Valid storage types.
|
|
116 |
allStorageTypes :: [StorageType]
|
|
117 |
allStorageTypes =
|
|
118 |
let all_vals = [minBound..maxBound]::[StorageType]
|
|
119 |
in if C.enableFileStorage
|
|
120 |
then all_vals
|
|
121 |
else delete StorageFile all_vals
|
|
122 |
|
|
123 |
-- | Custom 'Arbitrary' instance for 'StorageType', which needs to
|
|
124 |
-- handle the case of file storage being disabled at configure time.
|
|
125 |
instance Arbitrary StorageType where
|
|
126 |
arbitrary = elements allStorageTypes
|
| 100 |
127 |
|
| 101 |
128 |
$(genArbitrary ''NodeEvacMode)
|
| 102 |
129 |
|