def setUp(self):
self.tearDown()
models.client.GanetiRapiClient = RapiProxy
cluster = Cluster(hostname='test.cluster',
slug='test',
username='******',
password='******')
cluster.id = 23 # XXX MySQL DB does not reset auto-increment IDs when an object is removed
cluster.save()
cluster.sync_nodes()
template = VirtualMachineTemplate(template_name="Template1",
cluster=cluster)
template.disks = [{'size': 500}]
template.nics = [{'mode': 'bridged', 'link': ''}]
template.save()
instance = VirtualMachine(hostname='new.vm.hostname', cluster=cluster)
instance.info = INSTANCE
instance.disks = []
instance.nics = []
instance.save()
# Users
self.create_users([
('superuser', {
'is_superuser': True
}),
'cluster_admin',
'create_vm',
'unauthorized',
])
self.cluster_admin.grant('admin', cluster)
self.create_vm.grant('create_vm', cluster)
self.create_template_data = dict(
cluster=cluster.pk,
template_name='foo_bar',
memory=512,
disk_template='plain',
disk_count=0,
nic_count=0,
)
self.cluster = cluster
self.template = template
self.instance = instance
self.c = Client()
def setUp(self):
self.tearDown()
models.client.GanetiRapiClient = RapiProxy
cluster = Cluster(hostname='test.cluster', slug='test',
username='******', password='******')
cluster.id = 23 # XXX MySQL DB does not reset auto-increment
# IDs when an object is removed
cluster.save()
cluster.sync_nodes()
template = VirtualMachineTemplate(template_name="Template1",
cluster=cluster)
template.disks = [{'size': 500}]
template.nics = [{'mode': 'bridged', 'link': ''}]
template.save()
instance = VirtualMachine(hostname='new.vm.hostname', cluster=cluster)
instance.info = INSTANCE
instance.disks = []
instance.nics = []
instance.save()
# Users
self.create_users([
('superuser', {'is_superuser': True}),
'cluster_admin',
'create_vm',
'unauthorized',
])
self.cluster_admin.grant('admin', cluster)
self.create_vm.grant('create_vm', cluster)
self.create_template_data = dict(
cluster=cluster.pk,
template_name='foo_bar',
memory=512,
disk_template='plain',
disk_count=0,
nic_count=0,
)
self.cluster = cluster
self.template = template
self.instance = instance
self.c = Client()
def _update_vm(self, cluster, info, data, updated, callback):
"""
updates an individual VirtualMachine, this is the actual work function
@param cluster - cluster this node is on
@param info - info from ganeti
@param data - data from database
@param updated - counter object
@param callback - callback fired when method is complete.
"""
name = info['name']
if name in data:
id, mtime, status = data[name]
if not mtime or mtime < info['mtime'] \
or status != info['status']:
print ' Virtual Machine (updated) : %s' % name
#print ' %s :: %s' % (mtime, datetime.fromtimestamp(info['mtime']))
# only update the whole object if it is new or modified.
#
# XXX status changes will not always be reflected in mtime
# explicitly check status to see if it has changed. failing
# to check this would result in state changes being lost
parsed = VirtualMachine.parse_persistent_info(info)
if 'delete' in parsed:
VirtualMachine.objects.filter(pk=id).delete()
else:
VirtualMachine.objects.filter(pk=id) \
.update(serialized_info=cPickle.dumps(info), **parsed)
updated += 1
else:
# new vm
vm = VirtualMachine(cluster=cluster, hostname=info['name'])
vm.info = info
vm.save()
id = vm.id
updated += 1
callback(id)
