system/TestApiserver.py

   1 # $Id$
   2 # wrapper to xmlrpc server, that support dry-run commands
   3 # we dont want to have to depend on PLCAPI, so:
   4 import xmlrpclib
   5
   6 server_methods = [ ('GetNodes' ,  []),
   7                    ('AddNode' , True),
   8                    ('UpdateNode' , True),
   9                    ('AddNodeNetwork' , True),
  10                    ('GetNodeNetworks' , True),
  11                    ('GetNodeNetworkSettingTypes' , True),
  12                    ('AddNodeNetworkSettingType' , True),
  13                    ('AddNodeNetworkSetting' , True),
  14                    ('GetBootMedium' , True),
  15                    ('GetNodeGroups' , True),
  16                    ('AddNodeGroup' , True),
  17                    ('AddNodeToNodeGroup' , True),
  18                    ('AddInitScript' , True),
  19                    ('AddSite' , True),
  20                    ('AddSiteAddress' , True),
  21                    ('DeleteSite' , True),
  22                    ('DeleteSlice' , True),
  23                    ('AddSlice' , True),
  24                    ('AddPersonToSlice' , True),
  25                    ('AddSliceToNodes' , True),
  26                    ('AddSliceAttribute' , True),
  27                    ('AddPerson' , True),
  28                    ('UpdatePerson' , True),
  29                    ('AddRoleToPerson' , True),
  30                    ('AddPersonToSite' , True),
  31                    ('DeletePerson' , True),
  32                    ('AddPersonKey' , True),
  33                    ('GetPlcRelease', {'build': {'target-arch':'i386'}}),
  34                    ]
  35
  36 class TestApiserver:
  37     class Callable:
  38         def __init__(self,server,dry_run,method,defaults):
  39             self.server=server
  40             self.dry_run=dry_run
  41             self.method=method
  42             self.defaults=defaults
  43         def __call__ (self, *args):
  44             if self.dry_run:
  45                 print "dry_run:",self.method,
  46                 if len(args)>0 and type(args[0])==type({}) and args[0].has_key('AuthMethod'):
  47                     print '<auth>',
  48                     args=args[1:]
  49                 print '(',args,')'
  50                 return self.defaults
  51             else:
  52                 actual_method=getattr(self.server,self.method)
  53                 return actual_method(*args)
  54
  55     def __init__(self,url,dry_run=False):
  56         self.apiserver = xmlrpclib.Server(url,allow_none=True)
  57         self.dry_run=dry_run
  58         for (method,defaults) in server_methods:
  59             setattr(self,method,TestApiserver.Callable(self.apiserver,dry_run,method,defaults))