Statistics
| Branch: | Tag: | Revision:

root / htest / Test / Ganeti / HTools / Types.hs @ d83903ee

History | View | Annotate | Download (5.4 kB)

1 e1ee7d5a Iustin Pop
{-# LANGUAGE TemplateHaskell #-}
2 e1ee7d5a Iustin Pop
{-# OPTIONS_GHC -fno-warn-orphans #-}
3 e1ee7d5a Iustin Pop
4 e1ee7d5a Iustin Pop
{-| Unittests for ganeti-htools.
5 e1ee7d5a Iustin Pop
6 e1ee7d5a Iustin Pop
-}
7 e1ee7d5a Iustin Pop
8 e1ee7d5a Iustin Pop
{-
9 e1ee7d5a Iustin Pop
10 e1ee7d5a Iustin Pop
Copyright (C) 2009, 2010, 2011, 2012 Google Inc.
11 e1ee7d5a Iustin Pop
12 e1ee7d5a Iustin Pop
This program is free software; you can redistribute it and/or modify
13 e1ee7d5a Iustin Pop
it under the terms of the GNU General Public License as published by
14 e1ee7d5a Iustin Pop
the Free Software Foundation; either version 2 of the License, or
15 e1ee7d5a Iustin Pop
(at your option) any later version.
16 e1ee7d5a Iustin Pop
17 e1ee7d5a Iustin Pop
This program is distributed in the hope that it will be useful, but
18 e1ee7d5a Iustin Pop
WITHOUT ANY WARRANTY; without even the implied warranty of
19 e1ee7d5a Iustin Pop
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
20 e1ee7d5a Iustin Pop
General Public License for more details.
21 e1ee7d5a Iustin Pop
22 e1ee7d5a Iustin Pop
You should have received a copy of the GNU General Public License
23 e1ee7d5a Iustin Pop
along with this program; if not, write to the Free Software
24 e1ee7d5a Iustin Pop
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
25 e1ee7d5a Iustin Pop
02110-1301, USA.
26 e1ee7d5a Iustin Pop
27 e1ee7d5a Iustin Pop
-}
28 e1ee7d5a Iustin Pop
29 e1ee7d5a Iustin Pop
module Test.Ganeti.HTools.Types
30 e09c1fa0 Iustin Pop
  ( testHTools_Types
31 e1ee7d5a Iustin Pop
  , Types.AllocPolicy(..)
32 e1ee7d5a Iustin Pop
  , Types.DiskTemplate(..)
33 e1ee7d5a Iustin Pop
  , Types.FailMode(..)
34 e1ee7d5a Iustin Pop
  , Types.EvacMode(..)
35 e1ee7d5a Iustin Pop
  , Types.OpResult(..)
36 e1ee7d5a Iustin Pop
  , Types.ISpec(..)
37 e1ee7d5a Iustin Pop
  , Types.IPolicy(..)
38 e1ee7d5a Iustin Pop
  , nullIPolicy
39 e1ee7d5a Iustin Pop
  ) where
40 e1ee7d5a Iustin Pop
41 e1ee7d5a Iustin Pop
import Test.QuickCheck
42 e1ee7d5a Iustin Pop
43 e1ee7d5a Iustin Pop
import Control.Applicative
44 e1ee7d5a Iustin Pop
45 e1ee7d5a Iustin Pop
import Test.Ganeti.TestHelper
46 e1ee7d5a Iustin Pop
import Test.Ganeti.TestCommon
47 e1ee7d5a Iustin Pop
import Test.Ganeti.TestHTools
48 e1ee7d5a Iustin Pop
49 e1ee7d5a Iustin Pop
import qualified Ganeti.HTools.Types as Types
50 e1ee7d5a Iustin Pop
51 e1ee7d5a Iustin Pop
-- * Helpers
52 e1ee7d5a Iustin Pop
53 e1ee7d5a Iustin Pop
-- | All disk templates (used later)
54 e1ee7d5a Iustin Pop
allDiskTemplates :: [Types.DiskTemplate]
55 e1ee7d5a Iustin Pop
allDiskTemplates = [minBound..maxBound]
56 e1ee7d5a Iustin Pop
57 e1ee7d5a Iustin Pop
-- * Arbitrary instance
58 e1ee7d5a Iustin Pop
59 7022db83 Iustin Pop
$(genArbitrary ''Types.AllocPolicy)
60 e1ee7d5a Iustin Pop
61 7022db83 Iustin Pop
$(genArbitrary ''Types.DiskTemplate)
62 e1ee7d5a Iustin Pop
63 7022db83 Iustin Pop
$(genArbitrary ''Types.FailMode)
64 e1ee7d5a Iustin Pop
65 7022db83 Iustin Pop
$(genArbitrary ''Types.EvacMode)
66 e1ee7d5a Iustin Pop
67 e1ee7d5a Iustin Pop
instance Arbitrary a => Arbitrary (Types.OpResult a) where
68 e1ee7d5a Iustin Pop
  arbitrary = arbitrary >>= \c ->
69 e1ee7d5a Iustin Pop
              if c
70 e1ee7d5a Iustin Pop
                then Types.OpGood <$> arbitrary
71 e1ee7d5a Iustin Pop
                else Types.OpFail <$> arbitrary
72 e1ee7d5a Iustin Pop
73 e1ee7d5a Iustin Pop
instance Arbitrary Types.ISpec where
74 e1ee7d5a Iustin Pop
  arbitrary = do
75 e1ee7d5a Iustin Pop
    mem_s <- arbitrary::Gen (NonNegative Int)
76 e1ee7d5a Iustin Pop
    dsk_c <- arbitrary::Gen (NonNegative Int)
77 e1ee7d5a Iustin Pop
    dsk_s <- arbitrary::Gen (NonNegative Int)
78 e1ee7d5a Iustin Pop
    cpu_c <- arbitrary::Gen (NonNegative Int)
79 e1ee7d5a Iustin Pop
    nic_c <- arbitrary::Gen (NonNegative Int)
80 e1ee7d5a Iustin Pop
    su    <- arbitrary::Gen (NonNegative Int)
81 e1ee7d5a Iustin Pop
    return Types.ISpec { Types.iSpecMemorySize = fromIntegral mem_s
82 e1ee7d5a Iustin Pop
                       , Types.iSpecCpuCount   = fromIntegral cpu_c
83 e1ee7d5a Iustin Pop
                       , Types.iSpecDiskSize   = fromIntegral dsk_s
84 e1ee7d5a Iustin Pop
                       , Types.iSpecDiskCount  = fromIntegral dsk_c
85 e1ee7d5a Iustin Pop
                       , Types.iSpecNicCount   = fromIntegral nic_c
86 e1ee7d5a Iustin Pop
                       , Types.iSpecSpindleUse = fromIntegral su
87 e1ee7d5a Iustin Pop
                       }
88 e1ee7d5a Iustin Pop
89 e1ee7d5a Iustin Pop
-- | Generates an ispec bigger than the given one.
90 e1ee7d5a Iustin Pop
genBiggerISpec :: Types.ISpec -> Gen Types.ISpec
91 e1ee7d5a Iustin Pop
genBiggerISpec imin = do
92 e1ee7d5a Iustin Pop
  mem_s <- choose (Types.iSpecMemorySize imin, maxBound)
93 e1ee7d5a Iustin Pop
  dsk_c <- choose (Types.iSpecDiskCount imin, maxBound)
94 e1ee7d5a Iustin Pop
  dsk_s <- choose (Types.iSpecDiskSize imin, maxBound)
95 e1ee7d5a Iustin Pop
  cpu_c <- choose (Types.iSpecCpuCount imin, maxBound)
96 e1ee7d5a Iustin Pop
  nic_c <- choose (Types.iSpecNicCount imin, maxBound)
97 e1ee7d5a Iustin Pop
  su    <- choose (Types.iSpecSpindleUse imin, maxBound)
98 e1ee7d5a Iustin Pop
  return Types.ISpec { Types.iSpecMemorySize = fromIntegral mem_s
99 e1ee7d5a Iustin Pop
                     , Types.iSpecCpuCount   = fromIntegral cpu_c
100 e1ee7d5a Iustin Pop
                     , Types.iSpecDiskSize   = fromIntegral dsk_s
101 e1ee7d5a Iustin Pop
                     , Types.iSpecDiskCount  = fromIntegral dsk_c
102 e1ee7d5a Iustin Pop
                     , Types.iSpecNicCount   = fromIntegral nic_c
103 e1ee7d5a Iustin Pop
                     , Types.iSpecSpindleUse = fromIntegral su
104 e1ee7d5a Iustin Pop
                     }
105 e1ee7d5a Iustin Pop
106 e1ee7d5a Iustin Pop
instance Arbitrary Types.IPolicy where
107 e1ee7d5a Iustin Pop
  arbitrary = do
108 e1ee7d5a Iustin Pop
    imin <- arbitrary
109 e1ee7d5a Iustin Pop
    istd <- genBiggerISpec imin
110 e1ee7d5a Iustin Pop
    imax <- genBiggerISpec istd
111 e1ee7d5a Iustin Pop
    num_tmpl <- choose (0, length allDiskTemplates)
112 e1ee7d5a Iustin Pop
    dts  <- genUniquesList num_tmpl
113 e1ee7d5a Iustin Pop
    vcpu_ratio <- choose (1.0, maxVcpuRatio)
114 e1ee7d5a Iustin Pop
    spindle_ratio <- choose (1.0, maxSpindleRatio)
115 e1ee7d5a Iustin Pop
    return Types.IPolicy { Types.iPolicyMinSpec = imin
116 e1ee7d5a Iustin Pop
                         , Types.iPolicyStdSpec = istd
117 e1ee7d5a Iustin Pop
                         , Types.iPolicyMaxSpec = imax
118 e1ee7d5a Iustin Pop
                         , Types.iPolicyDiskTemplates = dts
119 e1ee7d5a Iustin Pop
                         , Types.iPolicyVcpuRatio = vcpu_ratio
120 e1ee7d5a Iustin Pop
                         , Types.iPolicySpindleRatio = spindle_ratio
121 e1ee7d5a Iustin Pop
                         }
122 e1ee7d5a Iustin Pop
123 e1ee7d5a Iustin Pop
-- * Test cases
124 e1ee7d5a Iustin Pop
125 20bc5360 Iustin Pop
prop_AllocPolicy_serialisation :: Types.AllocPolicy -> Property
126 63b068c1 Iustin Pop
prop_AllocPolicy_serialisation = testSerialisation
127 e1ee7d5a Iustin Pop
128 20bc5360 Iustin Pop
prop_DiskTemplate_serialisation :: Types.DiskTemplate -> Property
129 63b068c1 Iustin Pop
prop_DiskTemplate_serialisation = testSerialisation
130 e1ee7d5a Iustin Pop
131 20bc5360 Iustin Pop
prop_ISpec_serialisation :: Types.ISpec -> Property
132 63b068c1 Iustin Pop
prop_ISpec_serialisation = testSerialisation
133 e1ee7d5a Iustin Pop
134 20bc5360 Iustin Pop
prop_IPolicy_serialisation :: Types.IPolicy -> Property
135 63b068c1 Iustin Pop
prop_IPolicy_serialisation = testSerialisation
136 e1ee7d5a Iustin Pop
137 20bc5360 Iustin Pop
prop_EvacMode_serialisation :: Types.EvacMode -> Property
138 63b068c1 Iustin Pop
prop_EvacMode_serialisation = testSerialisation
139 e1ee7d5a Iustin Pop
140 20bc5360 Iustin Pop
prop_opToResult :: Types.OpResult Int -> Bool
141 20bc5360 Iustin Pop
prop_opToResult op =
142 e1ee7d5a Iustin Pop
  case op of
143 e1ee7d5a Iustin Pop
    Types.OpFail _ -> Types.isBad r
144 e1ee7d5a Iustin Pop
    Types.OpGood v -> case r of
145 e1ee7d5a Iustin Pop
                        Types.Bad _ -> False
146 e1ee7d5a Iustin Pop
                        Types.Ok v' -> v == v'
147 e1ee7d5a Iustin Pop
  where r = Types.opToResult op
148 e1ee7d5a Iustin Pop
149 20bc5360 Iustin Pop
prop_eitherToResult :: Either String Int -> Bool
150 20bc5360 Iustin Pop
prop_eitherToResult ei =
151 e1ee7d5a Iustin Pop
  case ei of
152 e1ee7d5a Iustin Pop
    Left _ -> Types.isBad r
153 e1ee7d5a Iustin Pop
    Right v -> case r of
154 e1ee7d5a Iustin Pop
                 Types.Bad _ -> False
155 e1ee7d5a Iustin Pop
                 Types.Ok v' -> v == v'
156 e1ee7d5a Iustin Pop
    where r = Types.eitherToResult ei
157 e1ee7d5a Iustin Pop
158 e09c1fa0 Iustin Pop
testSuite "HTools/Types"
159 20bc5360 Iustin Pop
            [ 'prop_AllocPolicy_serialisation
160 20bc5360 Iustin Pop
            , 'prop_DiskTemplate_serialisation
161 20bc5360 Iustin Pop
            , 'prop_ISpec_serialisation
162 20bc5360 Iustin Pop
            , 'prop_IPolicy_serialisation
163 20bc5360 Iustin Pop
            , 'prop_EvacMode_serialisation
164 20bc5360 Iustin Pop
            , 'prop_opToResult
165 20bc5360 Iustin Pop
            , 'prop_eitherToResult
166 e1ee7d5a Iustin Pop
            ]