1 # Thierry Parmentelat <thierry.parmentelat@inria.fr>
2 # Copyright (C) 2010 INRIA
4 # wrapper to xmlrpc server, that support dry-run commands
5 # we dont want to have to depend on PLCAPI, so:
8 # the default value is for the dry run mode
9 server_methods = [ ('GetNodes' , []),
11 ('SetNodePlainBootstrapfs', True),
12 ('DeleteNode' , True),
13 ('UpdateNode' , True),
14 ('AddInterface' , True),
15 ('AddIpAddress' , True),
17 ('GetInterfaces' , True),
19 ('AddTagType' , True),
20 ('AddInterfaceTag' , True),
21 ('GetBootMedium' , True),
22 ('GetNodeGroups' , True),
23 ('AddNodeGroup' , True),
24 ('DeleteNodeGroup', True),
25 ('GetNodeTags', True),
27 ('AddInitScript' , True),
28 ('DeleteInitScript', True),
29 ('GetInitScripts', True),
31 ('AddSiteAddress' , True),
32 ('DeleteSite' , True),
33 ('DeleteSlice' , True),
35 ('AddPersonToSlice' , True),
36 ('AddSliceToNodes' , True),
37 ('DeleteSliceFromNodes' , True),
38 ('AddSliceTag' , True),
40 ('UpdatePerson' , True),
41 ('AddRoleToPerson' , True),
42 ('AddPersonToSite' , True),
43 ('DeletePerson' , True),
44 ('AddPersonKey' , True),
45 ('GetPlcRelease', {'build': {'target-arch':'i386'}}),
47 ('GetSlices', [{'name':'dry_run_slice','node_ids':['dry_run']}]),
48 ('GetLeaseGranularity', 180),
51 ('DeleteLeases',True),
53 ('AddConfFile','True'),
55 ('system.listMethods',[]),
60 def __init__(self, server, dry_run, method, defaults):
62 self.dry_run = dry_run
64 self.defaults = defaults
65 def __call__ (self, *args):
67 print("dry_run:",self.method, end=' ')
68 if len(args) > 0 and type(args[0]) == type({}) and 'AuthMethod' in args[0]:
69 print('<auth>', end=' ')
74 actual_method = getattr(self.server, self.method)
75 return actual_method(*args)
77 def __init__(self, url, dry_run=False):
78 self.apiserver = xmlrpc.client.ServerProxy(url, allow_none=True,
79 use_builtin_types=True)
80 self.dry_run = dry_run
81 for method, defaults in server_methods:
82 setattr(self, method, TestApiserver.Callable(self.apiserver, dry_run, method, defaults))
84 def set_dry_run (self, dry_run):
85 self.dry_run = dry_run
86 for (method, defaults) in server_methods:
87 getattr(self, method).dry_run = dry_run
89 def has_method (self, methodname):
90 return methodname in self.apiserver.system.listMethods()