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 ('SetNodeFcdistro', True),
13 ('DeleteNode' , True),
14 ('UpdateNode' , True),
15 ('AddInterface' , True),
16 ('AddIpAddress' , True),
18 ('GetInterfaces' , True),
20 ('AddTagType' , True),
21 ('AddInterfaceTag' , True),
22 ('GetBootMedium' , "some non-empty-string"),
23 ('GetNodeGroups' , True),
24 ('AddNodeGroup' , True),
25 ('DeleteNodeGroup', True),
26 ('GetNodeTags', True),
28 ('AddInitScript' , True),
29 ('DeleteInitScript', True),
30 ('GetInitScripts', True),
32 ('AddSiteAddress' , True),
33 ('DeleteSite' , True),
34 ('DeleteSlice' , True),
36 ('AddPersonToSlice' , True),
37 ('AddSliceToNodes' , True),
38 ('DeleteSliceFromNodes' , True),
39 ('AddSliceTag' , True),
41 ('UpdatePerson' , True),
42 ('AddRoleToPerson' , True),
43 ('AddPersonToSite' , True),
44 ('DeletePerson' , True),
45 ('AddPersonKey' , True),
46 ('GetPlcRelease', {'build': {'target-arch':'i386'}}),
48 ('GetSlices', [{'name':'dry_run_slice','node_ids':['dry_run']}]),
49 ('GetLeaseGranularity', 180),
52 ('DeleteLeases',True),
54 ('AddConfFile','True'),
56 ('GetNodeFlavour','dry-run-nodeflavour'),
57 ('system.listMethods',[]),
62 def __init__(self, server, dry_run, method, defaults):
64 self.dry_run = dry_run
66 self.defaults = defaults
67 def __call__ (self, *args):
69 print("dry_run:",self.method, end=' ')
70 if len(args) > 0 and type(args[0]) == type({}) and 'AuthMethod' in args[0]:
71 print('<auth>', end=' ')
76 actual_method = getattr(self.server, self.method)
77 return actual_method(*args)
79 def __init__(self, url, dry_run=False):
80 self.apiserver = xmlrpc.client.ServerProxy(url, allow_none=True,
81 use_builtin_types=True)
82 self.dry_run = dry_run
83 for method, defaults in server_methods:
84 setattr(self, method, TestApiserver.Callable(self.apiserver, dry_run, method, defaults))
86 def set_dry_run (self, dry_run):
87 self.dry_run = dry_run
88 for (method, defaults) in server_methods:
89 getattr(self, method).dry_run = dry_run
91 def has_method (self, methodname):
92 return methodname in self.apiserver.system.listMethods()