aboutsummaryrefslogtreecommitdiffstats
path: root/tools/xm-test/lib/XmTestLib/DomainTracking.py
blob: 7e0bb3013a69d896fe25a50d09eb89ea3ef54860 (plain)
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
#!/usr/bin/python
"""
 Copyright (C) International Business Machines Corp., 2005
 Author: Dan Smith <danms@us.ibm.com>

 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; under version 2 of the License.

 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

"""

import atexit
import Test
import xapi

# Tracking of managed domains
_managedDomains = []
_VMuuids = []
registered = 0

def addManagedDomain(name):
    global registered
    _managedDomains.append(name)
    if not registered:
        atexit.register(destroyManagedDomains)
        registered = 1

def delManagedDomain(name):
    if name in _managedDomains:
        del _managedDomains[_managedDomains.index(name)]

def addXAPIDomain(uuid):
    global registered
    _VMuuids.append(uuid)
    if not registered:
        atexit.register(destroyManagedDomains)
        registered = 1

def delXAPIDomain(uuid):
    _VMuuids.remove(uuid)

def destroyManagedDomains():
    if len(_managedDomains) > 0:
        for m in _managedDomains:
            Test.traceCommand("xm destroy %s" % m)
            Test.traceCommand("xm delete %s" % m)
    if len(_VMuuids) > 0:
        for uuid in _VMuuids:
            Test.traceCommand("xm destroy %s" % uuid)
            Test.traceCommand("xm delete %s" % uuid)