tests/MockXenAPI.py

   1 # Copyright (c) 2011 Nicira Networks
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at:
   6 #
   7 #     http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 # See the License for the specific language governing permissions and
  13 # limitations under the License.
  14
  15 def xapi_local():
  16     return Session()
  17
  18
  19 class Session(object):
  20     def __init__(self):
  21         self.xenapi = XenAPI()
  22
  23
  24 class Failure(Exception):
  25     pass
  26
  27
  28 class XenAPI(object):
  29     def __init__(self):
  30         self.network = Network()
  31         self.pool = Pool()
  32         self.VIF = VIF()
  33
  34     def login_with_password(self, unused_username, unused_password):
  35         pass
  36
  37
  38 class RecordRef(object):
  39     def __init__(self, attrs):
  40         self.attrs = attrs
  41
  42
  43 class Table(object):
  44     def __init__(self, records):
  45         self.records = records
  46
  47     def get_all(self):
  48         return [RecordRef(rec) for rec in self.records]
  49
  50     def get_by_uuid(self, uuid):
  51         recs = [rec for rec in self.records if rec["uuid"] == uuid]
  52         if len(recs) != 1:
  53             raise Failure("No record with UUID %s" % uuid)
  54         return RecordRef(recs[0])
  55
  56     def get_record(self, record_ref):
  57         return record_ref.attrs
  58
  59
  60 class Network(Table):
  61     __records = ({"uuid": "9b66c68b-a74e-4d34-89a5-20a8ab352d1e",
  62                   "bridge": "xenbr0",
  63                   "other_config":
  64                       {"vswitch-controller-fail-mode": "secure",
  65                        "nicira-bridge-id": "custom bridge ID"}},
  66                  {"uuid": "e1c9019d-375b-45ac-a441-0255dd2247de",
  67                   "bridge": "xenbr1",
  68                   "other_config":
  69                       {"vswitch-disable-in-band": "true"}})
  70
  71     def __init__(self):
  72         Table.__init__(self, Network.__records)
  73
  74
  75 class Pool(Table):
  76     __records = ({"uuid": "7a793edf-e5f4-4994-a0f9-cee784c0cda3",
  77                   "other_config":
  78                       {"vswitch-controller-fail-mode": "secure"}},)
  79
  80     def __init__(self):
  81         Table.__init__(self, Pool.__records)
  82
  83 class VIF(Table):
  84     __records = ({"uuid": "6ab1b260-398e-49ba-827b-c7696108964c",
  85                   "other_config":
  86                       {"nicira-iface-id": "custom iface ID"}},)
  87
  88     def __init__(self):
  89         Table.__init__(self, VIF.__records)