ganeti-qa.py 16.6 KB
Newer Older
Iustin Pop's avatar
Iustin Pop committed
1
#!/usr/bin/python -u
Iustin Pop's avatar
Iustin Pop committed
2
3
#

4
# Copyright (C) 2007, 2008, 2009, 2010, 2011, 2012 Google Inc.
Iustin Pop's avatar
Iustin Pop committed
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
# 02110-1301, USA.


22
"""Script for doing QA on Ganeti.
23
24

"""
Iustin Pop's avatar
Iustin Pop committed
25

26
# pylint: disable=C0103
Iustin Pop's avatar
Iustin Pop committed
27
28
# due to invalid name

Iustin Pop's avatar
Iustin Pop committed
29
import sys
30
31
import datetime
import optparse
Iustin Pop's avatar
Iustin Pop committed
32

33
34
35
36
import qa_cluster
import qa_config
import qa_daemon
import qa_env
37
import qa_group
38
39
import qa_instance
import qa_node
40
import qa_os
41
import qa_job
Oleksiy Mishchenko's avatar
Oleksiy Mishchenko committed
42
import qa_rapi
Michael Hanselmann's avatar
Michael Hanselmann committed
43
import qa_tags
44
import qa_utils
Iustin Pop's avatar
Iustin Pop committed
45

46
from ganeti import utils
47
from ganeti import rapi
48
from ganeti import constants
49

50
import ganeti.rapi.client # pylint: disable=W0611
51

Iustin Pop's avatar
Iustin Pop committed
52

Iustin Pop's avatar
Iustin Pop committed
53
def _FormatHeader(line, end=72):
Iustin Pop's avatar
Iustin Pop committed
54
55
56
57
  """Fill a line up to the end column.

  """
  line = "---- " + line + " "
Andrea Spadaccini's avatar
Andrea Spadaccini committed
58
  line += "-" * (end - len(line))
Iustin Pop's avatar
Iustin Pop committed
59
60
61
62
  line = line.rstrip()
  return line


Iustin Pop's avatar
Iustin Pop committed
63
64
def _DescriptionOf(fn):
  """Computes the description of an item.
Iustin Pop's avatar
Iustin Pop committed
65
66

  """
67
68
  if fn.__doc__:
    desc = fn.__doc__.splitlines()[0].strip()
Iustin Pop's avatar
Iustin Pop committed
69
  else:
Iustin Pop's avatar
Iustin Pop committed
70
    desc = "%r" % fn
Iustin Pop's avatar
Iustin Pop committed
71

Iustin Pop's avatar
Iustin Pop committed
72
73
  return desc.rstrip(".")

74

Iustin Pop's avatar
Iustin Pop committed
75
76
77
78
def RunTest(fn, *args):
  """Runs a test after printing a header.

  """
Iustin Pop's avatar
Iustin Pop committed
79

Iustin Pop's avatar
Iustin Pop committed
80
  tstart = datetime.datetime.now()
Iustin Pop's avatar
Iustin Pop committed
81

Iustin Pop's avatar
Iustin Pop committed
82
83
  desc = _DescriptionOf(fn)

Iustin Pop's avatar
Iustin Pop committed
84
85
86
87
88
89
90
91
92
93
  print
  print _FormatHeader("%s start %s" % (tstart, desc))

  try:
    retval = fn(*args)
    return retval
  finally:
    tstop = datetime.datetime.now()
    tdelta = tstop - tstart
    print _FormatHeader("%s time=%s %s" % (tstop, tdelta, desc))
Iustin Pop's avatar
Iustin Pop committed
94
95


Iustin Pop's avatar
Iustin Pop committed
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
def RunTestIf(testnames, fn, *args):
  """Runs a test conditionally.

  @param testnames: either a single test name in the configuration
      file, or a list of testnames (which will be AND-ed together)

  """
  if qa_config.TestEnabled(testnames):
    RunTest(fn, *args)
  else:
    tstart = datetime.datetime.now()
    desc = _DescriptionOf(fn)
    print _FormatHeader("%s skipping %s, test(s) %s disabled" %
                        (tstart, desc, testnames))


Michael Hanselmann's avatar
Michael Hanselmann committed
112
113
def RunEnvTests():
  """Run several environment tests.
Iustin Pop's avatar
Iustin Pop committed
114
115

  """
Iustin Pop's avatar
Iustin Pop committed
116
117
118
  RunTestIf("env", qa_env.TestSshConnection)
  RunTestIf("env", qa_env.TestIcmpPing)
  RunTestIf("env", qa_env.TestGanetiCommands)
Iustin Pop's avatar
Iustin Pop committed
119

120

121
def SetupCluster(rapi_user, rapi_secret):
Michael Hanselmann's avatar
Michael Hanselmann committed
122
  """Initializes the cluster.
Iustin Pop's avatar
Iustin Pop committed
123

124
125
126
  @param rapi_user: Login user for RAPI
  @param rapi_secret: Login secret for RAPI

Michael Hanselmann's avatar
Michael Hanselmann committed
127
  """
Iustin Pop's avatar
Iustin Pop committed
128
129
  RunTestIf("create-cluster", qa_cluster.TestClusterInit,
            rapi_user, rapi_secret)
130
131
132
133

  # Test on empty cluster
  RunTestIf("node-list", qa_node.TestNodeList)
  RunTestIf("instance-list", qa_instance.TestInstanceList)
134
  RunTestIf("job-list", qa_job.TestJobList)
135

Iustin Pop's avatar
Iustin Pop committed
136
137
  RunTestIf("create-cluster", qa_node.TestNodeAddAll)
  if not qa_config.TestEnabled("create-cluster"):
138
139
    # consider the nodes are already there
    qa_node.MarkNodeAddedAll()
140

Iustin Pop's avatar
Iustin Pop committed
141
  RunTestIf("test-jobqueue", qa_cluster.TestJobqueue)
142

143
144
145
  # enable the watcher (unconditionally)
  RunTest(qa_daemon.TestResumeWatcher)

146
147
  RunTestIf("node-list", qa_node.TestNodeList)

148
149
150
  # Test listing fields
  RunTestIf("node-list", qa_node.TestNodeListFields)
  RunTestIf("instance-list", qa_instance.TestInstanceListFields)
151
  RunTestIf("job-list", qa_job.TestJobListFields)
152

Iustin Pop's avatar
Iustin Pop committed
153
  RunTestIf("node-info", qa_node.TestNodeInfo)
Michael Hanselmann's avatar
Michael Hanselmann committed
154
155
156
157


def RunClusterTests():
  """Runs tests related to gnt-cluster.
158

Michael Hanselmann's avatar
Michael Hanselmann committed
159
  """
Iustin Pop's avatar
Iustin Pop committed
160
  for test, fn in [
161
    ("create-cluster", qa_cluster.TestClusterInitDisk),
Iustin Pop's avatar
Iustin Pop committed
162
163
164
    ("cluster-renew-crypto", qa_cluster.TestClusterRenewCrypto),
    ("cluster-verify", qa_cluster.TestClusterVerify),
    ("cluster-reserved-lvs", qa_cluster.TestClusterReservedLvs),
165
    # TODO: add more cluster modify tests
166
    ("cluster-modify", qa_cluster.TestClusterModifyEmpty),
Iustin Pop's avatar
Iustin Pop committed
167
    ("cluster-modify", qa_cluster.TestClusterModifyBe),
168
    ("cluster-modify", qa_cluster.TestClusterModifyDisk),
Iustin Pop's avatar
Iustin Pop committed
169
170
171
172
    ("cluster-rename", qa_cluster.TestClusterRename),
    ("cluster-info", qa_cluster.TestClusterVersion),
    ("cluster-info", qa_cluster.TestClusterInfo),
    ("cluster-info", qa_cluster.TestClusterGetmaster),
Iustin Pop's avatar
Iustin Pop committed
173
    ("cluster-redist-conf", qa_cluster.TestClusterRedistConf),
Iustin Pop's avatar
Iustin Pop committed
174
175
176
177
    ("cluster-copyfile", qa_cluster.TestClusterCopyfile),
    ("cluster-command", qa_cluster.TestClusterCommand),
    ("cluster-burnin", qa_cluster.TestClusterBurnin),
    ("cluster-master-failover", qa_cluster.TestClusterMasterFailover),
178
179
    ("cluster-master-failover",
     qa_cluster.TestClusterMasterFailoverWithDrainedQueue),
Iustin Pop's avatar
Iustin Pop committed
180
    ("cluster-oob", qa_cluster.TestClusterOob),
Iustin Pop's avatar
Iustin Pop committed
181
182
    ("rapi", qa_rapi.TestVersion),
    ("rapi", qa_rapi.TestEmptyCluster),
183
    ("rapi", qa_rapi.TestRapiQuery),
Iustin Pop's avatar
Iustin Pop committed
184
185
    ]:
    RunTestIf(test, fn)
186

187

188
189
190
191
192
193
194
def RunRepairDiskSizes():
  """Run the repair disk-sizes test.

  """
  RunTestIf("cluster-repair-disk-sizes", qa_cluster.TestClusterRepairDiskSizes)


Michael Hanselmann's avatar
Michael Hanselmann committed
195
196
def RunOsTests():
  """Runs all tests related to gnt-os.
Michael Hanselmann's avatar
Michael Hanselmann committed
197

Michael Hanselmann's avatar
Michael Hanselmann committed
198
  """
199
200
201
202
203
  if qa_config.TestEnabled("rapi"):
    rapi_getos = qa_rapi.GetOperatingSystems
  else:
    rapi_getos = None

Iustin Pop's avatar
Iustin Pop committed
204
205
206
  for fn in [
    qa_os.TestOsList,
    qa_os.TestOsDiagnose,
207
208
209
210
    ]:
    RunTestIf("os", fn)

  for fn in [
Iustin Pop's avatar
Iustin Pop committed
211
212
213
    qa_os.TestOsValid,
    qa_os.TestOsInvalid,
    qa_os.TestOsPartiallyValid,
214
215
216
217
    ]:
    RunTestIf("os", fn, rapi_getos)

  for fn in [
Iustin Pop's avatar
Iustin Pop committed
218
219
    qa_os.TestOsModifyValid,
    qa_os.TestOsModifyInvalid,
220
    qa_os.TestOsStatesNonExisting,
Iustin Pop's avatar
Iustin Pop committed
221
222
    ]:
    RunTestIf("os", fn)
Michael Hanselmann's avatar
Michael Hanselmann committed
223
224
225
226
227
228


def RunCommonInstanceTests(instance):
  """Runs a few tests that are common to all disk types.

  """
Iustin Pop's avatar
Iustin Pop committed
229
  RunTestIf("instance-shutdown", qa_instance.TestInstanceShutdown, instance)
230
231
  RunTestIf(["instance-shutdown", "instance-console", "rapi"],
            qa_rapi.TestRapiStoppedInstanceConsole, instance)
232
233
  RunTestIf(["instance-shutdown", "instance-modify"],
            qa_instance.TestInstanceStoppedModify, instance)
Iustin Pop's avatar
Iustin Pop committed
234
  RunTestIf("instance-shutdown", qa_instance.TestInstanceStartup, instance)
Iustin Pop's avatar
Iustin Pop committed
235

236
237
238
239
240
241
  # Test shutdown/start via RAPI
  RunTestIf(["instance-shutdown", "rapi"],
            qa_rapi.TestRapiInstanceShutdown, instance)
  RunTestIf(["instance-shutdown", "rapi"],
            qa_rapi.TestRapiInstanceStartup, instance)

Iustin Pop's avatar
Iustin Pop committed
242
  RunTestIf("instance-list", qa_instance.TestInstanceList)
Michael Hanselmann's avatar
Michael Hanselmann committed
243

Iustin Pop's avatar
Iustin Pop committed
244
  RunTestIf("instance-info", qa_instance.TestInstanceInfo, instance)
245

Iustin Pop's avatar
Iustin Pop committed
246
247
248
  RunTestIf("instance-modify", qa_instance.TestInstanceModify, instance)
  RunTestIf(["instance-modify", "rapi"],
            qa_rapi.TestRapiInstanceModify, instance)
249

Iustin Pop's avatar
Iustin Pop committed
250
  RunTestIf("instance-console", qa_instance.TestInstanceConsole, instance)
251
252
  RunTestIf(["instance-console", "rapi"],
            qa_rapi.TestRapiInstanceConsole, instance)
253

Iustin Pop's avatar
Iustin Pop committed
254
255
256
257
258
  DOWN_TESTS = ["instance-reinstall", "instance-rename"]
  # shutdown instance for any 'down' tests
  RunTestIf(DOWN_TESTS, qa_instance.TestInstanceShutdown, instance)

  # now run the 'down' state tests
Iustin Pop's avatar
Iustin Pop committed
259
  RunTestIf("instance-reinstall", qa_instance.TestInstanceReinstall, instance)
260
261
  RunTestIf(["instance-reinstall", "rapi"],
            qa_rapi.TestRapiInstanceReinstall, instance)
Iustin Pop's avatar
Iustin Pop committed
262
263
264
265
  # RAPI reinstall will leave the instance up by default, so we have
  # to stop it again
  RunTestIf(["instance-reinstall", "rapi"],
            qa_rapi.TestRapiInstanceShutdown, instance)
266

Iustin Pop's avatar
Iustin Pop committed
267
  if qa_config.TestEnabled("instance-rename"):
268
    rename_source = instance["name"]
269
    rename_target = qa_config.get("rename", None)
270
    # perform instance rename to the same name
Iustin Pop's avatar
Iustin Pop committed
271
272
273
    RunTest(qa_instance.TestInstanceRenameAndBack,
            rename_source, rename_source)
    RunTestIf("rapi", qa_rapi.TestRapiInstanceRenameAndBack,
274
              rename_source, rename_source)
275
276
    if rename_target is not None:
      # perform instance rename to a different name, if we have one configured
Iustin Pop's avatar
Iustin Pop committed
277
278
279
      RunTest(qa_instance.TestInstanceRenameAndBack,
              rename_source, rename_target)
      RunTestIf("rapi", qa_rapi.TestRapiInstanceRenameAndBack,
280
                rename_source, rename_target)
Iustin Pop's avatar
Iustin Pop committed
281
282
283
284
285

  # and now start the instance again
  RunTestIf(DOWN_TESTS, qa_instance.TestInstanceStartup, instance)

  RunTestIf("instance-reboot", qa_instance.TestInstanceReboot, instance)
286

Iustin Pop's avatar
Iustin Pop committed
287
  RunTestIf("tags", qa_tags.TestInstanceTags, instance)
Michael Hanselmann's avatar
Michael Hanselmann committed
288

Michael Hanselmann's avatar
Michael Hanselmann committed
289
  RunTestIf("cluster-verify", qa_cluster.TestClusterVerify)
Michael Hanselmann's avatar
Michael Hanselmann committed
290

Iustin Pop's avatar
Iustin Pop committed
291
  RunTestIf("rapi", qa_rapi.TestInstance, instance)
292

293
294
295
  # Lists instances, too
  RunTestIf("node-list", qa_node.TestNodeList)

296
297
298
  # Some jobs have been run, let's test listing them
  RunTestIf("job-list", qa_job.TestJobList)

299
300
301
302
303

def RunCommonNodeTests():
  """Run a few common node tests.

  """
Iustin Pop's avatar
Iustin Pop committed
304
305
  RunTestIf("node-volumes", qa_node.TestNodeVolumes)
  RunTestIf("node-storage", qa_node.TestNodeStorage)
306
  RunTestIf("node-oob", qa_node.TestOutOfBand)
307

308

309
310
311
312
def RunGroupListTests():
  """Run tests for listing node groups.

  """
313
314
  RunTestIf("group-list", qa_group.TestGroupList)
  RunTestIf("group-list", qa_group.TestGroupListFields)
315
316


317
318
319
320
321
def RunGroupRwTests():
  """Run tests for adding/removing/renaming groups.

  """
  RunTestIf("group-rwops", qa_group.TestGroupAddRemoveRename)
322
323
  RunTestIf("group-rwops", qa_group.TestGroupAddWithOptions)
  RunTestIf("group-rwops", qa_group.TestGroupModify)
324
  RunTestIf(["group-rwops", "rapi"], qa_rapi.TestRapiNodeGroups)
325
326
  RunTestIf(["group-rwops", "tags"], qa_tags.TestGroupTags,
            qa_group.GetDefaultGroup())
327

328

329
def RunExportImportTests(instance, pnode, snode):
Michael Hanselmann's avatar
Michael Hanselmann committed
330
  """Tries to export and import the instance.
Iustin Pop's avatar
Iustin Pop committed
331

332
333
334
335
  @param pnode: current primary node of the instance
  @param snode: current secondary node of the instance, if any,
      otherwise None

Michael Hanselmann's avatar
Michael Hanselmann committed
336
  """
Iustin Pop's avatar
Iustin Pop committed
337
  if qa_config.TestEnabled("instance-export"):
338
339
    RunTest(qa_instance.TestInstanceExportNoTarget, instance)

Michael Hanselmann's avatar
Michael Hanselmann committed
340
341
342
343
344
345
    expnode = qa_config.AcquireNode(exclude=pnode)
    try:
      name = RunTest(qa_instance.TestInstanceExport, instance, expnode)

      RunTest(qa_instance.TestBackupList, expnode)

Iustin Pop's avatar
Iustin Pop committed
346
      if qa_config.TestEnabled("instance-import"):
Michael Hanselmann's avatar
Michael Hanselmann committed
347
        newinst = qa_config.AcquireInstance()
Michael Hanselmann's avatar
Michael Hanselmann committed
348
        try:
Michael Hanselmann's avatar
Michael Hanselmann committed
349
350
351
          RunTest(qa_instance.TestInstanceImport, pnode, newinst,
                  expnode, name)
          RunTest(qa_instance.TestInstanceRemove, newinst)
Michael Hanselmann's avatar
Michael Hanselmann committed
352
        finally:
Michael Hanselmann's avatar
Michael Hanselmann committed
353
354
355
          qa_config.ReleaseInstance(newinst)
    finally:
      qa_config.ReleaseNode(expnode)
Michael Hanselmann's avatar
Michael Hanselmann committed
356

Iustin Pop's avatar
Iustin Pop committed
357
  if qa_config.TestEnabled(["rapi", "inter-cluster-instance-move"]):
358
359
    newinst = qa_config.AcquireInstance()
    try:
360
361
362
363
364
      if snode is None:
        excl = [pnode]
      else:
        excl = [pnode, snode]
      tnode = qa_config.AcquireNode(exclude=excl)
365
366
      try:
        RunTest(qa_rapi.TestInterClusterInstanceMove, instance, newinst,
367
                pnode, snode, tnode)
368
      finally:
369
        qa_config.ReleaseNode(tnode)
370
371
372
    finally:
      qa_config.ReleaseInstance(newinst)

Michael Hanselmann's avatar
Michael Hanselmann committed
373

374
def RunDaemonTests(instance):
Michael Hanselmann's avatar
Michael Hanselmann committed
375
  """Test the ganeti-watcher script.
376

Michael Hanselmann's avatar
Michael Hanselmann committed
377
  """
378
  RunTest(qa_daemon.TestPauseWatcher)
379

Iustin Pop's avatar
Iustin Pop committed
380
  RunTestIf("instance-automatic-restart",
381
            qa_daemon.TestInstanceAutomaticRestart, instance)
Iustin Pop's avatar
Iustin Pop committed
382
  RunTestIf("instance-consecutive-failures",
383
            qa_daemon.TestInstanceConsecutiveFailures, instance)
384

385
386
  RunTest(qa_daemon.TestResumeWatcher)

387

Michael Hanselmann's avatar
Michael Hanselmann committed
388
389
def RunHardwareFailureTests(instance, pnode, snode):
  """Test cluster internal hardware failure recovery.
Iustin Pop's avatar
Iustin Pop committed
390

Michael Hanselmann's avatar
Michael Hanselmann committed
391
  """
Iustin Pop's avatar
Iustin Pop committed
392
  RunTestIf("instance-failover", qa_instance.TestInstanceFailover, instance)
393
394
  RunTestIf(["instance-failover", "rapi"],
            qa_rapi.TestRapiInstanceFailover, instance)
Michael Hanselmann's avatar
Michael Hanselmann committed
395

Iustin Pop's avatar
Iustin Pop committed
396
397
398
  RunTestIf("instance-migrate", qa_instance.TestInstanceMigrate, instance)
  RunTestIf(["instance-migrate", "rapi"],
            qa_rapi.TestRapiInstanceMigrate, instance)
399

Iustin Pop's avatar
Iustin Pop committed
400
  if qa_config.TestEnabled("instance-replace-disks"):
401
    othernode = qa_config.AcquireNode(exclude=[pnode, snode])
402
    try:
403
      RunTestIf("rapi", qa_rapi.TestRapiInstanceReplaceDisks, instance)
404
405
406
407
408
      RunTest(qa_instance.TestReplaceDisks,
              instance, pnode, snode, othernode)
    finally:
      qa_config.ReleaseNode(othernode)

Iustin Pop's avatar
Iustin Pop committed
409
  RunTestIf("node-evacuate", qa_node.TestNodeEvacuate, pnode, snode)
Michael Hanselmann's avatar
Michael Hanselmann committed
410

Iustin Pop's avatar
Iustin Pop committed
411
  RunTestIf("node-failover", qa_node.TestNodeFailover, pnode, snode)
Michael Hanselmann's avatar
Michael Hanselmann committed
412

Iustin Pop's avatar
Iustin Pop committed
413
  RunTestIf("instance-disk-failure", qa_instance.TestInstanceMasterDiskFailure,
Michael Hanselmann's avatar
Michael Hanselmann committed
414
            instance, pnode, snode)
Iustin Pop's avatar
Iustin Pop committed
415
416
417
  RunTestIf("instance-disk-failure",
            qa_instance.TestInstanceSecondaryDiskFailure, instance,
            pnode, snode)
Michael Hanselmann's avatar
Michael Hanselmann committed
418
419


420
421
def RunQa():
  """Main QA body.
Michael Hanselmann's avatar
Michael Hanselmann committed
422
423

  """
424
425
426
  rapi_user = "ganeti-qa"
  rapi_secret = utils.GenerateSecret()

Michael Hanselmann's avatar
Michael Hanselmann committed
427
  RunEnvTests()
428
  SetupCluster(rapi_user, rapi_secret)
Michael Hanselmann's avatar
Michael Hanselmann committed
429
430

  # Load RAPI certificate
431
  qa_rapi.Setup(rapi_user, rapi_secret)
Michael Hanselmann's avatar
Michael Hanselmann committed
432

Michael Hanselmann's avatar
Michael Hanselmann committed
433
434
  RunClusterTests()
  RunOsTests()
435

Iustin Pop's avatar
Iustin Pop committed
436
  RunTestIf("tags", qa_tags.TestClusterTags)
Michael Hanselmann's avatar
Michael Hanselmann committed
437

438
  RunCommonNodeTests()
439
  RunGroupListTests()
440
  RunGroupRwTests()
441

442
443
  pnode = qa_config.AcquireNode(exclude=qa_config.GetMasterNode())
  try:
Iustin Pop's avatar
Iustin Pop committed
444
445
    RunTestIf("node-readd", qa_node.TestNodeReadd, pnode)
    RunTestIf("node-modify", qa_node.TestNodeModify, pnode)
446
    RunTestIf("delay", qa_cluster.TestDelay, pnode)
447
448
  finally:
    qa_config.ReleaseNode(pnode)
449

Michael Hanselmann's avatar
Michael Hanselmann committed
450
451
  pnode = qa_config.AcquireNode()
  try:
Iustin Pop's avatar
Iustin Pop committed
452
    RunTestIf("tags", qa_tags.TestNodeTags, pnode)
Michael Hanselmann's avatar
Michael Hanselmann committed
453

Oleksiy Mishchenko's avatar
Oleksiy Mishchenko committed
454
455
456
    if qa_rapi.Enabled():
      RunTest(qa_rapi.TestNode, pnode)

457
      if qa_config.TestEnabled("instance-add-plain-disk"):
458
459
460
461
462
463
        for use_client in [True, False]:
          rapi_instance = RunTest(qa_rapi.TestRapiInstanceAdd, pnode,
                                  use_client)
          RunCommonInstanceTests(rapi_instance)
          RunTest(qa_rapi.TestRapiInstanceRemove, rapi_instance, use_client)
          del rapi_instance
464

465
    if qa_config.TestEnabled("instance-add-plain-disk"):
Michael Hanselmann's avatar
Michael Hanselmann committed
466
467
      instance = RunTest(qa_instance.TestInstanceAddWithPlainDisk, pnode)
      RunCommonInstanceTests(instance)
468
      RunGroupListTests()
469
      RunTestIf("cluster-epo", qa_cluster.TestClusterEpo)
470
      RunExportImportTests(instance, pnode, None)
471
      RunDaemonTests(instance)
472
      RunRepairDiskSizes()
Michael Hanselmann's avatar
Michael Hanselmann committed
473
474
      RunTest(qa_instance.TestInstanceRemove, instance)
      del instance
475

476
    multinode_tests = [
Iustin Pop's avatar
Iustin Pop committed
477
      ("instance-add-drbd-disk",
478
479
480
481
482
483
484
485
486
       qa_instance.TestInstanceAddWithDrbdDisk),
    ]

    for name, func in multinode_tests:
      if qa_config.TestEnabled(name):
        snode = qa_config.AcquireNode(exclude=pnode)
        try:
          instance = RunTest(func, pnode, snode)
          RunCommonInstanceTests(instance)
487
          RunGroupListTests()
488
489
490
          RunTest(qa_group.TestAssignNodesIncludingSplit,
                  constants.INITIAL_NODE_GROUP_NAME,
                  pnode["primary"], snode["primary"])
Iustin Pop's avatar
Iustin Pop committed
491
          if qa_config.TestEnabled("instance-convert-disk"):
492
            RunTest(qa_instance.TestInstanceShutdown, instance)
493
            RunTest(qa_instance.TestInstanceConvertDisk, instance, snode)
494
            RunTest(qa_instance.TestInstanceStartup, instance)
495
          RunExportImportTests(instance, pnode, snode)
496
          RunHardwareFailureTests(instance, pnode, snode)
497
          RunRepairDiskSizes()
498
499
500
501
          RunTest(qa_instance.TestInstanceRemove, instance)
          del instance
        finally:
          qa_config.ReleaseNode(snode)
Iustin Pop's avatar
Iustin Pop committed
502

Iustin Pop's avatar
Iustin Pop committed
503
    if qa_config.TestEnabled(["instance-add-plain-disk", "instance-export"]):
504
505
506
507
508
509
510
511
512
513
514
515
516
      for shutdown in [False, True]:
        instance = RunTest(qa_instance.TestInstanceAddWithPlainDisk, pnode)
        expnode = qa_config.AcquireNode(exclude=pnode)
        try:
          if shutdown:
            # Stop instance before exporting and removing it
            RunTest(qa_instance.TestInstanceShutdown, instance)
          RunTest(qa_instance.TestInstanceExportWithRemove, instance, expnode)
          RunTest(qa_instance.TestBackupList, expnode)
        finally:
          qa_config.ReleaseNode(expnode)
        del expnode
        del instance
517

Iustin Pop's avatar
Iustin Pop committed
518
  finally:
Michael Hanselmann's avatar
Michael Hanselmann committed
519
    qa_config.ReleaseNode(pnode)
Iustin Pop's avatar
Iustin Pop committed
520

Iustin Pop's avatar
Iustin Pop committed
521
  RunTestIf("create-cluster", qa_node.TestNodeRemoveAll)
Iustin Pop's avatar
Iustin Pop committed
522

Iustin Pop's avatar
Iustin Pop committed
523
  RunTestIf("cluster-destroy", qa_cluster.TestClusterDestroy)
Iustin Pop's avatar
Iustin Pop committed
524

525

526
527
528
529
530
531
@rapi.client.UsesRapiClient
def main():
  """Main program.

  """
  parser = optparse.OptionParser(usage="%prog [options] <config-file>")
Iustin Pop's avatar
Iustin Pop committed
532
  parser.add_option("--yes-do-it", dest="yes_do_it",
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
      action="store_true",
      help="Really execute the tests")
  (qa_config.options, args) = parser.parse_args()

  if len(args) == 1:
    (config_file, ) = args
  else:
    parser.error("Wrong number of arguments.")

  if not qa_config.options.yes_do_it:
    print ("Executing this script irreversibly destroys any Ganeti\n"
           "configuration on all nodes involved. If you really want\n"
           "to start testing, supply the --yes-do-it option.")
    sys.exit(1)

  qa_config.Load(config_file)

  qa_utils.StartMultiplexer(qa_config.GetMasterNode()["primary"])
  try:
    RunQa()
  finally:
    qa_utils.CloseMultiplexers()

Iustin Pop's avatar
Iustin Pop committed
556
if __name__ == "__main__":
557
  main()