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 __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)