def __init__(self):
super( VMStompEngine, self).__init__()
networkInterface = self.config.get('VM', 'network_interface')
networkInterfaceData = netifaces.ifaddresses(networkInterface)
self._id, self._ip = [ networkInterfaceData[af][0]['addr'] for af in (netifaces.AF_LINK, netifaces.AF_INET) ]
self._id = self._id.upper()
self.logger.debug("VM instantiated with id/ip %s/%s" % (self._id, self._ip) )
self._descriptor = EntityDescriptor(self._id, ip=self._ip)
def setUp(self):
self.cmdRegistry = Host.CommandRegistry()
self.names = ("testVM1", "testVM2")
self.ids = ("DE:AD:BE:EF:00:01", "DE:AD:BE:EF:00:02")
self.descriptors = map(lambda i: EntityDescriptor(i), self.ids)
for d in self.descriptors:
self.cmdRegistry.vmRegistry.addVM(d)
self.fakeTransport = proto_helpers.StringTransport()
def setUp(self):
self.vmRegistry = Host.VMRegistry()
self.names = ("testVM1", "testVM2")
self.ids = ("DE:AD:BE:EF:00:01", "DE:AD:BE:EF:00:02")
self.extras = ({
"extraData": "extra extra!"
}, {
"extraData": "extra extra! (bis)"
})
self.descriptors = map(lambda i, d: EntityDescriptor(i, **d), self.ids,
self.extras)
for d in self.descriptors:
self.vmRegistry.addVM(d)
class TestEntityDescriptor(unittest.TestCase):
def setUp(self):
self.descriptor1 = EntityDescriptor(id=123, name='test1', foo=1, bar=2)
self.descriptor2 = EntityDescriptor(456, foo=3)
self.descriptor3 = EntityDescriptor(0)
def test_contains(self):
self.assertTrue('foo' in self.descriptor1)
self.assertTrue('id' in self.descriptor1)
self.assertTrue('name' in self.descriptor1)
self.assertTrue('id' in self.descriptor2)
def test_serialization(self):
serialized = self.descriptor1.serialize()
deserialized = EntityDescriptor.deserialize(serialized)
self.assertEquals(self.descriptor1, deserialized)
def test_setattr(self):
self.descriptor1.foobar = 3
self.assertEquals(3, self.descriptor1.foobar)
#overwriting
self.descriptor1.foo = 11
self.assertEquals(11, self.descriptor1.foo)
def test_getattr(self):
self.assertRaises(AttributeError, lambda: self.descriptor1.nonexistent)
self.assertEquals(123, self.descriptor1.id)
self.assertEquals('test1', self.descriptor1.name)
self.assertEquals(1, self.descriptor1.foo)
self.assertEquals(2, self.descriptor1.bar)
self.assertRaises(AttributeError, lambda: self.descriptor2.bar)
self.assertEquals(3, self.descriptor2.foo)
self.assertEquals(456, self.descriptor2.id)
def test_repr(self):
self.assertEquals(
"""EntityDescriptor<id=123>: {bar: 2, foo: 1, name: test1}""",
repr(self.descriptor1))
class TestEntityDescriptor(unittest.TestCase):
def setUp(self):
self.descriptor1 = EntityDescriptor(id=123, name="test1", foo=1, bar=2)
self.descriptor2 = EntityDescriptor(456, foo=3)
self.descriptor3 = EntityDescriptor(0)
def test_contains(self):
self.assertTrue("foo" in self.descriptor1)
self.assertTrue("id" in self.descriptor1)
self.assertTrue("name" in self.descriptor1)
self.assertTrue("id" in self.descriptor2)
def test_serialization(self):
serialized = self.descriptor1.serialize()
deserialized = EntityDescriptor.deserialize(serialized)
self.assertEquals(self.descriptor1, deserialized)
def test_setattr(self):
self.descriptor1.foobar = 3
self.assertEquals(3, self.descriptor1.foobar)
# overwriting
self.descriptor1.foo = 11
self.assertEquals(11, self.descriptor1.foo)
def test_getattr(self):
self.assertRaises(AttributeError, lambda: self.descriptor1.nonexistent)
self.assertEquals(123, self.descriptor1.id)
self.assertEquals("test1", self.descriptor1.name)
self.assertEquals(1, self.descriptor1.foo)
self.assertEquals(2, self.descriptor1.bar)
self.assertRaises(AttributeError, lambda: self.descriptor2.bar)
self.assertEquals(3, self.descriptor2.foo)
self.assertEquals(456, self.descriptor2.id)
def test_repr(self):
self.assertEquals("""EntityDescriptor<id=123>: {bar: 2, foo: 1, name: test1}""", repr(self.descriptor1))
def setUp(self):
self.descriptor1 = EntityDescriptor(id=123, name='test1', foo=1, bar=2)
self.descriptor2 = EntityDescriptor(456, foo=3)
self.descriptor3 = EntityDescriptor(0)
def test_serialization(self):
serialized = self.descriptor1.serialize()
deserialized = EntityDescriptor.deserialize(serialized)
self.assertEquals(self.descriptor1, deserialized)
def test_addVM(self):
nonExistantVM = EntityDescriptor("00:00:00:00:00:00")
self.assertRaises(Exceptions.NoSuchVirtualMachine,
self.vmRegistry.addVM, nonExistantVM)
def listenAndAct(self, msg): headers = msg['headers'] vmDescriptor = EntityDescriptor.deserialize(headers['descriptor']) self.subject.addVM(vmDescriptor)
def listenAndAct(self, msg):
headers = msg['headers']
vmDescriptor = EntityDescriptor.deserialize(headers['descriptor'])
self.subject.addVM(vmDescriptor)
def __init__(self, anId):
self.descriptor = EntityDescriptor(anId)
self.stuffDone = False
def __init__(self):
self._descriptor = EntityDescriptor('Host-ID')
def setUp(self):
self.descriptor1 = EntityDescriptor(id=123, name="test1", foo=1, bar=2)
self.descriptor2 = EntityDescriptor(456, foo=3)
self.descriptor3 = EntityDescriptor(0)
def test_serialization(self):
serialized = self.descriptor1.serialize()
deserialized = EntityDescriptor.deserialize(serialized)
self.assertEquals(self.descriptor1, deserialized)
