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 ('SetNodePldistro', True),
14 ('DeleteNode' , True),
15 ('UpdateNode' , True),
16 ('AddInterface' , True),
17 ('AddIpAddress' , True),
19 ('GetInterfaces' , True),
21 ('AddTagType' , True),
22 ('AddInterfaceTag' , True),
23 ('GetBootMedium' , "some non-empty-string"),
24 ('GetNodeGroups' , True),
25 ('AddNodeGroup' , True),
26 ('DeleteNodeGroup', True),
27 ('GetNodeTags', True),
29 ('AddInitScript' , True),
30 ('DeleteInitScript', True),
31 ('GetInitScripts', True),
33 ('AddSiteAddress' , True),
34 ('DeleteSite' , True),
35 ('DeleteSlice' , True),
37 ('AddPersonToSlice' , True),
38 ('AddSliceToNodes' , True),
39 ('DeleteSliceFromNodes' , True),
40 ('AddSliceTag' , True),
42 ('UpdatePerson' , True),
43 ('AddRoleToPerson' , True),
44 ('AddPersonToSite' , True),
45 ('DeletePerson' , True),
46 ('AddPersonKey' , True),
47 ('GetPlcRelease', {'build': {'target-arch':'i386'}}),
49 ('GetSlices', [{'name':'dry_run_slice','node_ids':['dry_run']}]),
50 ('GetLeaseGranularity', 180),
53 ('DeleteLeases',True),
55 ('AddConfFile','True'),
57 ('GetNodeFlavour','dry-run-nodeflavour'),
58 ('system.listMethods',[]),
63 def __init__(self, server, dry_run, method, defaults):
65 self.dry_run = dry_run
67 self.defaults = defaults
68 def __call__ (self, *args):
70 print("dry_run:",self.method, end=' ')
71 if len(args) > 0 and type(args[0]) == type({}) and 'AuthMethod' in args[0]:
72 print('<auth>', end=' ')
77 actual_method = getattr(self.server, self.method)
78 return actual_method(*args)
80 def __init__(self, url, dry_run=False):
81 self.apiserver = xmlrpc.client.ServerProxy(url, allow_none=True,
82 use_builtin_types=True)
83 self.dry_run = dry_run
84 for method, defaults in server_methods:
85 setattr(self, method, TestApiserver.Callable(self.apiserver, dry_run, method, defaults))
87 def set_dry_run (self, dry_run):
88 self.dry_run = dry_run
89 for (method, defaults) in server_methods:
90 getattr(self, method).dry_run = dry_run
92 def has_method (self, methodname):
93 return methodname in self.apiserver.system.listMethods()