1 {-| Unittest runner for ganeti-htools.
7 Copyright (C) 2009, 2011, 2012 Google Inc.
9 This program is free software; you can redistribute it and/or modify
10 it under the terms of the GNU General Public License as published by
11 the Free Software Foundation; either version 2 of the License, or
12 (at your option) any later version.
14 This program is distributed in the hope that it will be useful, but
15 WITHOUT ANY WARRANTY; without even the implied warranty of
16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17 General Public License for more details.
19 You should have received a copy of the GNU General Public License
20 along with this program; if not, write to the Free Software
21 Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
26 module Main(main) where
28 import Data.Monoid (mappend)
30 import System.Environment (getArgs)
32 import Test.Ganeti.TestImports ()
33 import Test.Ganeti.BasicTypes
34 import Test.Ganeti.Confd.Utils
35 import Test.Ganeti.HTools.CLI
36 import Test.Ganeti.HTools.Cluster
37 import Test.Ganeti.HTools.Container
38 import Test.Ganeti.HTools.Loader
39 import Test.Ganeti.HTools.Instance
40 import Test.Ganeti.HTools.Node
41 import Test.Ganeti.HTools.PeerMap
42 import Test.Ganeti.HTools.Simu
43 import Test.Ganeti.HTools.Text
44 import Test.Ganeti.HTools.Types
45 import Test.Ganeti.HTools.Utils
46 import Test.Ganeti.Jobs
47 import Test.Ganeti.JSON
48 import Test.Ganeti.Luxi
49 import Test.Ganeti.Objects
50 import Test.Ganeti.OpCodes
51 import Test.Ganeti.Query.Filter
52 import Test.Ganeti.Query.Language
53 import Test.Ganeti.Query.Query
54 import Test.Ganeti.Rpc
55 import Test.Ganeti.Ssconf
57 -- | Our default test options, overring the built-in test-framework
62 , topt_maximum_generated_tests = Just 500
63 , topt_maximum_unsuitable_generated_tests = Just 5000
64 , topt_maximum_test_size = Nothing
65 , topt_maximum_test_depth = Nothing
66 , topt_timeout = Nothing
69 -- | Our slow test options.
72 { topt_maximum_generated_tests = Just 50
73 , topt_maximum_unsuitable_generated_tests = Just 500
76 -- | All our defined tests.
77 allTests :: [(Bool, (String, [Test]))]
79 [ (True, testBasicTypes)
80 , (True, testConfd_Utils)
81 , (True, testHTools_CLI)
82 , (True, testHTools_Container)
83 , (True, testHTools_Instance)
84 , (True, testHTools_Loader)
85 , (True, testHTools_Node)
86 , (True, testHTools_PeerMap)
87 , (True, testHTools_Simu)
88 , (True, testHTools_Text)
89 , (True, testHTools_Types)
90 , (True, testHTools_Utils)
96 , (True, testQuery_Filter)
97 , (True, testQuery_Language)
98 , (True, testQuery_Query)
101 , (False, testHTools_Cluster)
102 , (False, testSlowObjects)
105 -- | Slow a test's max tests, if provided as such.
106 makeSlowOrFast :: Bool -> TestOptions -> TestOptions
107 makeSlowOrFast is_fast opts =
108 let template = if is_fast then fast else slow
109 fn_val v = if is_fast then v else v `div` 10
110 in case topt_maximum_generated_tests opts of
111 -- user didn't override the max_tests, so we'll do it here
112 Nothing -> opts `mappend` template
113 -- user did override, so we ignore the template and just directly
114 -- decrease the max_tests, if needed
115 Just max_tests -> opts { topt_maximum_generated_tests =
116 Just (fn_val max_tests)
119 -- | Main function. Note we don't use defaultMain since we want to
120 -- control explicitly our test sizes (and override the default).
123 ropts <- getArgs >>= interpretArgsOrExit
124 -- note: we do this overriding here since we need some groups to
125 -- have a smaller test count; so in effect we're basically
126 -- overriding t-f's inheritance here, but only for max_tests
127 let (act_fast, act_slow) =
128 case ropt_test_options ropts of
129 Nothing -> (fast, slow)
130 Just topts -> (makeSlowOrFast True topts, makeSlowOrFast False topts)
131 actual_opts is_fast = if is_fast then act_fast else act_slow
132 let tests = map (\(is_fast, (group_name, group_tests)) ->
133 plusTestOptions (actual_opts is_fast) $
134 testGroup group_name group_tests) allTests
135 defaultMainWithOpts tests ropts