def test_get_quota(self): """ Tests cluster.get_quota() method Verifies: * if no user is passed, return default quota values * if user has quota, return values from Quota * if user doesn't have quota, return default cluster values """ default_quota = {'default': 1, 'ram': 1, 'virtual_cpus': None, 'disk': 3} user_quota = {'default': 0, 'ram': 4, 'virtual_cpus': 5, 'disk': None} cluster = Cluster(hostname='foo.fake.hostname') cluster.__dict__.update(default_quota) cluster.save() user = User(username='******') user.save() # default quota self.assertEqual(default_quota, cluster.get_quota()) # user without quota, defaults to default self.assertEqual(default_quota, cluster.get_quota(user.get_profile())) # user with custom quota quota = Quota(cluster=cluster, user=user.get_profile()) quota.__dict__.update(user_quota) quota.save() self.assertEqual(user_quota, cluster.get_quota(user.get_profile())) quota.delete() cluster.delete() user.delete()
def create_virtual_machine(self, cluster=None, hostname="vm1.example.bak"): if cluster is None: cluster = Cluster(hostname="test.example.bak", slug="OSL_TEST", username="******", password="******") cluster.save() cluster.sync_nodes() vm = VirtualMachine(cluster=cluster, hostname=hostname) vm.save() return vm, cluster
def setUp(self): # Cluster cluster = Cluster(hostname='test.cluster.gwm', slug='test', username='******', password='******') # cluster.info = INFO cluster.save() # Template template_data = dict( template_name='new.vm.template', description='A new template.', cluster=cluster.id, start=True, name_check=True, disk_template='plain', disk_count=0, memory=256, vcpus=2, root_path='/', kernel_path='', cdrom_image_path='', serial_console=False, nic_type='paravirtual', disk_type='paravirtual', nic_count=0, boot_order='disk', os='image+ubuntu-lucid', ) data = template_data.copy() data['cluster'] = cluster del data['disk_count'] del data['nic_count'] template = VirtualMachineTemplate(**data) template.save() # Template Fields fields = vars(template).keys() # Users self.create_users([ ('superuser', { 'is_superuser': True }), 'cluster_admin', ]) self.cluster_admin.grant('admin', cluster) self.users = [self.superuser, self.cluster_admin] self.template = template self.cluster = cluster self.template_data = template_data self.template_fields = fields
def create_virtual_machine(self, cluster=None, hostname='vm1.example.bak'): if cluster is None: cluster = Cluster(hostname='test.example.bak', slug='OSL_TEST', username='******', password='******') cluster.save() cluster.sync_nodes() vm = VirtualMachine(cluster=cluster, hostname=hostname) vm.save() return vm, cluster
def test_view_list(self): """ Tests displaying the list of clusters """ url = '/clusters/' # create extra user and tests cluster1 = Cluster(hostname='cluster1', slug='cluster1') cluster1.save() # anonymous user response = self.c.get(url, follow=True) self.assertEqual(200, response.status_code) self.assertTemplateUsed(response, 'registration/login.html') # unauthorized user self.assertTrue( self.c.login(username=self.unauthorized.username, password='******')) response = self.c.get(url) self.assertEqual(200, response.status_code) self.assertEquals('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/cluster/list.html') clusters = response.context['cluster_list'] self.assertFalse(clusters) # authorized permissions self.assertTrue( self.c.login(username=self.cluster_admin.username, password='******')) response = self.c.get(url) self.assertEqual(200, response.status_code) self.assertEquals('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/cluster/list.html') clusters = response.context['cluster_list'] self.assertTrue(self.cluster in clusters) self.assertTrue(cluster1 not in clusters) self.assertEqual(1, len(clusters)) # authorized (superuser) self.assertTrue( self.c.login(username=self.superuser.username, password='******')) response = self.c.get(url) self.assertEqual(200, response.status_code) self.assertEquals('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/cluster/list.html') clusters = response.context['cluster_list'] self.assertTrue(self.cluster in clusters) self.assertTrue(cluster1 in clusters) self.assertEqual(2, len(clusters)) cluster1.delete()
def test_view_list(self): """ Tests displaying the list of clusters """ url = '/clusters/' # create extra user and tests cluster1 = Cluster(hostname='cluster1', slug='cluster1') cluster1.save() # anonymous user response = self.c.get(url, follow=True) self.assertEqual(200, response.status_code) self.assertTemplateUsed(response, 'registration/login.html') # unauthorized user self.assertTrue(self.c.login(username=self.unauthorized.username, password='******')) response = self.c.get(url) self.assertEqual(200, response.status_code) self.assertEquals('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/cluster/list.html') clusters = response.context['cluster_list'] self.assertFalse(clusters) # authorized permissions self.assertTrue(self.c.login(username=self.cluster_admin.username, password='******')) response = self.c.get(url) self.assertEqual(200, response.status_code) self.assertEquals('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/cluster/list.html') clusters = response.context['cluster_list'] self.assertTrue(self.cluster in clusters) self.assertTrue(cluster1 not in clusters) self.assertEqual(1, len(clusters)) # authorized (superuser) self.assertTrue(self.c.login(username=self.superuser.username, password='******')) response = self.c.get(url) self.assertEqual(200, response.status_code) self.assertEquals('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/cluster/list.html') clusters = response.context['cluster_list'] self.assertTrue(self.cluster in clusters) self.assertTrue(cluster1 in clusters) self.assertEqual(2, len(clusters)) cluster1.delete()
def setUp(self): # Cluster cluster = Cluster(hostname='test.cluster.gwm', slug='test', username='******', password='******') # cluster.info = INFO cluster.save() # Template template_data = dict( template_name='new.vm.template', description='A new template.', cluster=cluster.id, start=True, name_check=True, disk_template='plain', disk_count=0, memory=256, vcpus=2, root_path='/', kernel_path='', cdrom_image_path='', serial_console=False, nic_type='paravirtual', disk_type='paravirtual', nic_count=0, boot_order='disk', os='image+ubuntu-lucid', ) data = template_data.copy() data['cluster'] = cluster del data['disk_count'] del data['nic_count'] template = VirtualMachineTemplate(**data) template.save() # Template Fields fields = vars(template).keys() # Users self.create_users([ ('superuser', {'is_superuser': True}), 'cluster_admin', ]) self.cluster_admin.grant('admin', cluster) self.users = [self.superuser, self.cluster_admin] self.template = template self.cluster = cluster self.template_data = template_data self.template_fields = fields
class TestVMWizardAdvancedForm(TestCase): def setUp(self): # XXX #8895 means we need a cluster here self.cluster = Cluster() self.cluster.hostname = "cluster.example.com" self.cluster.save() self.pnode = Node() self.pnode.cluster = self.cluster self.pnode.hostname = "pnode.example.com" self.pnode.save() self.snode = Node() self.snode.cluster = self.cluster self.snode.hostname = "snode.example.com" self.snode.save() self.valid_data = { "pnode": self.pnode.id, "snode": self.snode.id, } def tearDown(self): self.pnode.delete() self.snode.delete() self.cluster.delete() def test_trivial(self): pass def test_validate_valid(self): form = VMWizardAdvancedForm(self.valid_data) self.assertTrue(form.is_valid(), form.errors) def test_validate_ip_check_without_name_check(self): data = self.valid_data.copy() data["ip_check"] = True form = VMWizardAdvancedForm(data) self.assertFalse(form.is_valid(), "IP check shouldn't be allowed without name check") def test_validate_pnode_equals_snode(self): invalid_data = self.valid_data.copy() invalid_data["snode"] = invalid_data["pnode"] form = VMWizardAdvancedForm(invalid_data) self.assertFalse(form.is_valid(), "The secondary node cannot be the primary node.")
def test_view_delete_superuser(self): """ Superusers can delete clusters. """ cluster = Cluster(hostname='test.cluster.bak', slug='cluster1') cluster.save() url = '/cluster/%s/edit/' % cluster.slug self.assertTrue( self.c.login(username=self.superuser.username, password='******')) response = self.c.delete(url, follow=True) self.assertEqual(200, response.status_code) self.assertEquals('application/json', response['content-type']) self.assertEquals('1', response.content) self.assertFalse(Cluster.objects.all().filter(id=cluster.id).exists())
def test_view_delete_superuser(self): """ Superusers can delete clusters. """ cluster = Cluster(hostname='test.cluster.bak', slug='cluster1') cluster.save() url = '/cluster/%s/edit/' % cluster.slug self.assertTrue(self.c.login(username=self.superuser.username, password='******')) response = self.c.delete(url, follow=True) self.assertEqual(200, response.status_code) self.assertEquals('application/json', response['content-type']) self.assertEquals('1', response.content) self.assertFalse(Cluster.objects.all().filter(id=cluster.id).exists())
def test_view_delete_authorized(self): """ Users with admin on the cluster should be able to delete the cluster. """ cluster = Cluster(hostname='test.cluster.bak', slug='cluster1') cluster.save() url = '/cluster/%s/edit/' % cluster.slug self.user.grant('admin', cluster) self.assertTrue(self.c.login(username=self.user.username, password='******')) response = self.c.delete(url, follow=True) self.assertEqual(200, response.status_code) self.assertEquals('application/json', response['content-type']) self.assertEquals('1', response.content) self.assertFalse(Cluster.objects.all().filter(id=cluster.id).exists())
def test_view_delete_authorized(self): """ Users with admin on the cluster should be able to delete the cluster. """ cluster = Cluster(hostname='test.cluster.bak', slug='cluster1') cluster.save() url = '/cluster/%s/edit/' % cluster.slug self.user.grant('admin', cluster) self.assertTrue( self.c.login(username=self.user.username, password='******')) response = self.c.delete(url, follow=True) self.assertEqual(200, response.status_code) self.assertEquals('application/json', response['content-type']) self.assertEquals('1', response.content) self.assertFalse(Cluster.objects.all().filter(id=cluster.id).exists())
def test_set_quota(self): """ Tests cluster.set_quota() Verifies: * passing values with no quota, creates a new quota object * passing values with an existing quota, updates it. * passing a None with an existing quota deletes it * passing a None with no quota, does nothing """ default_quota = { 'default': 1, 'ram': 1, 'virtual_cpus': None, 'disk': 3 } user_quota = {'default': 0, 'ram': 4, 'virtual_cpus': 5, 'disk': None} user_quota2 = {'default': 0, 'ram': 7, 'virtual_cpus': 8, 'disk': 9} cluster = Cluster(hostname='foo.fake.hostname') cluster.__dict__.update(default_quota) cluster.save() user = User(username='******') user.save() # create new quota cluster.set_quota(user.get_profile(), user_quota) query = Quota.objects.filter(cluster=cluster, user=user.get_profile()) self.assertTrue(query.exists()) self.assertEqual(user_quota, cluster.get_quota(user.get_profile())) # update quota with new values cluster.set_quota(user.get_profile(), user_quota2) query = Quota.objects.filter(cluster=cluster, user=user.get_profile()) self.assertEqual(1, query.count()) self.assertEqual(user_quota2, cluster.get_quota(user.get_profile())) # delete quota cluster.set_quota(user.get_profile(), None) query = Quota.objects.filter(cluster=cluster, user=user.get_profile()) self.assertFalse(query.exists()) self.assertEqual(default_quota, cluster.get_quota(user.get_profile())) cluster.delete() user.delete()
def test_parse_info(self): """ Test parsing values from cached info Verifies: * mtime and ctime are parsed * ram, virtual_cpus, and disksize are parsed """ cluster = Cluster(hostname='foo.fake.hostname') cluster.save() cluster.info = INFO self.assertEqual(cluster.ctime, datetime.fromtimestamp(1270685309.818239)) self.assertEqual(cluster.mtime, datetime.fromtimestamp(1283552454.2998919)) cluster.delete()
def test_missing_in_ganeti(self): """ Tests missing_in_ganeti property """ cluster = Cluster(hostname='ganeti.example.test') cluster.save() vm_current = VirtualMachine(cluster=cluster, hostname='gimager2.example.bak') vm_removed = VirtualMachine(cluster=cluster, hostname='does.not.exist.org') vm_current.save() vm_removed.save() self.assertEqual([u'does.not.exist.org'], cluster.missing_in_ganeti) vm_current.delete() vm_removed.delete() cluster.delete()
def test_sync_virtual_machines(self): """ Tests synchronizing cached virtuals machines (stored in db) with info the ganeti cluster is storing Verifies: * VMs no longer in ganeti are deleted * VMs missing from the database are added """ cluster = Cluster(hostname='ganeti.example.test') cluster.save() vm_missing = 'gimager.example.bak' vm_current = VirtualMachine(cluster=cluster, hostname='gimager2.example.bak') vm_removed = VirtualMachine(cluster=cluster, hostname='does.not.exist.org') vm_current.save() vm_removed.save() cluster.sync_virtual_machines() self.assertTrue(VirtualMachine.objects.get(cluster=cluster, hostname=vm_missing), 'missing vm was not created') self.assertTrue( VirtualMachine.objects.get( cluster=cluster, hostname=vm_current.hostname), 'previously existing vm was not created') self.assertTrue( VirtualMachine.objects.filter( cluster=cluster, hostname=vm_removed.hostname), "vm not in ganeti was not removed from db") cluster.sync_virtual_machines(True) self.assertFalse( VirtualMachine.objects.filter( cluster=cluster, hostname=vm_removed.hostname), 'vm not present in ganeti was not removed from db') vm_removed.delete() vm_current.delete() cluster.delete()
def setUp(self): user = User(id=2, username='******') user.set_password('secret') user.save() group = Group(name='testing_group') group.save() cluster = Cluster(hostname='test.example.test', slug='OSL_TEST') cluster.save() self.create_standard_users() self.create_users(['cluster_admin']) self.cluster_admin.grant('admin', cluster) self.user = user self.group = group self.cluster = cluster self.c = Client()
def test_save(self): """ test saving a cluster object Verifies: * object is saved and queryable * hash is updated """ # XXX any reason why these are in a single test? cluster = Cluster() cluster.save() self.assertTrue(cluster.hash) cluster.delete() cluster = Cluster(hostname='foo.fake.hostname', slug='different') cluster.save() self.assertTrue(cluster.hash) cluster.delete()
def test_set_quota(self): """ Tests cluster.set_quota() Verifies: * passing values with no quota, creates a new quota object * passing values with an existing quota, updates it. * passing a None with an existing quota deletes it * passing a None with no quota, does nothing """ default_quota = {'default': 1, 'ram': 1, 'virtual_cpus': None, 'disk': 3} user_quota = {'default': 0, 'ram': 4, 'virtual_cpus': 5, 'disk': None} user_quota2 = {'default': 0, 'ram': 7, 'virtual_cpus': 8, 'disk': 9} cluster = Cluster(hostname='foo.fake.hostname') cluster.__dict__.update(default_quota) cluster.save() user = User(username='******') user.save() # create new quota cluster.set_quota(user.get_profile(), user_quota) query = Quota.objects.filter(cluster=cluster, user=user.get_profile()) self.assertTrue(query.exists()) self.assertEqual(user_quota, cluster.get_quota(user.get_profile())) # update quota with new values cluster.set_quota(user.get_profile(), user_quota2) query = Quota.objects.filter(cluster=cluster, user=user.get_profile()) self.assertEqual(1, query.count()) self.assertEqual(user_quota2, cluster.get_quota(user.get_profile())) # delete quota cluster.set_quota(user.get_profile(), None) query = Quota.objects.filter(cluster=cluster, user=user.get_profile()) self.assertFalse(query.exists()) self.assertEqual(default_quota, cluster.get_quota(user.get_profile())) cluster.delete() user.delete()
def test_sync_virtual_machines(self): """ Tests synchronizing cached virtuals machines (stored in db) with info the ganeti cluster is storing Verifies: * VMs no longer in ganeti are deleted * VMs missing from the database are added """ cluster = Cluster(hostname='ganeti.example.test') cluster.save() vm_missing = 'gimager.example.bak' vm_current = VirtualMachine(cluster=cluster, hostname='gimager2.example.bak') vm_removed = VirtualMachine(cluster=cluster, hostname='does.not.exist.org') vm_current.save() vm_removed.save() cluster.sync_virtual_machines() self.assertTrue( VirtualMachine.objects.get(cluster=cluster, hostname=vm_missing), 'missing vm was not created') self.assertTrue( VirtualMachine.objects.get(cluster=cluster, hostname=vm_current.hostname), 'previously existing vm was not created') self.assertTrue( VirtualMachine.objects.filter(cluster=cluster, hostname=vm_removed.hostname), "vm not in ganeti was not removed from db") cluster.sync_virtual_machines(True) self.assertFalse( VirtualMachine.objects.filter(cluster=cluster, hostname=vm_removed.hostname), 'vm not present in ganeti was not removed from db') vm_removed.delete() vm_current.delete() cluster.delete()
def test_get_quota(self): """ Tests cluster.get_quota() method Verifies: * if no user is passed, return default quota values * if user has quota, return values from Quota * if user doesn't have quota, return default cluster values """ default_quota = { 'default': 1, 'ram': 1, 'virtual_cpus': None, 'disk': 3 } user_quota = {'default': 0, 'ram': 4, 'virtual_cpus': 5, 'disk': None} cluster = Cluster(hostname='foo.fake.hostname') cluster.__dict__.update(default_quota) cluster.save() user = User(username='******') user.save() # default quota self.assertEqual(default_quota, cluster.get_quota()) # user without quota, defaults to default self.assertEqual(default_quota, cluster.get_quota(user.get_profile())) # user with custom quota quota = Quota(cluster=cluster, user=user.get_profile()) quota.__dict__.update(user_quota) quota.save() self.assertEqual(user_quota, cluster.get_quota(user.get_profile())) quota.delete() cluster.delete() user.delete()
class ImportViews(TestCase): def setUp(self): self.tearDown() self.user = User(id=2, username='******') self.user.set_password('secret') self.user.save() self.group = Group(name='testing_group') self.group.save() self.cluster0 = Cluster(hostname='test0', slug='OSL_TEST0') self.cluster0.save() self.cluster1 = Cluster(hostname='test1', slug='OSL_TEST1') self.cluster1.save() self.vm0 = VirtualMachine(hostname='vm0', cluster=self.cluster0) self.vm1 = VirtualMachine(hostname='vm1', cluster=self.cluster0, owner=self.user.get_profile()) # self.vm2 = VirtualMachine(hostname='vm2', cluster=cluster0) self.vm3 = VirtualMachine(hostname='vm3', cluster=self.cluster1) self.vm4 = VirtualMachine(hostname='vm4', cluster=self.cluster1, owner=self.user.get_profile()) # self.vm5 = VirtualMachine(hostname='vm5', cluster=cluster1) self.vm0.save() self.vm1.save() self.vm3.save() self.vm4.save() self.c = Client() self.owner = self.user.get_profile() def tearDown(self): VirtualMachine.objects.all().delete() Cluster.objects.all().delete() Organization.objects.all().delete() Profile.objects.all().delete() User.objects.all().delete() Group.objects.all().delete() def test_orphans_view(self): """ Test orphans view """ url = '/import/orphans/' # anonymous user response = self.c.get(url, follow=True) self.assertEqual(200, response.status_code) self.assertTemplateUsed(response, 'registration/login.html') # unauthorized user self.assertTrue(self.c.login(username=self.user.username, password='******')) response = self.c.get(url) self.assertEqual(403, response.status_code) # authorized get (cluster admin perm) self.user.grant('admin', self.cluster0) response = self.c.get(url) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/orphans.html') self.assertEqual([(self.vm0.id, 'test0', 'vm0')], response.context['vms']) self.user.revoke_all(self.cluster0) # authorized get (superuser) self.user.is_superuser = True self.user.save() response = self.c.get(url) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/orphans.html') self.assertEqual([(self.vm0.id, 'test0', 'vm0'), (self.vm3.id, 'test1', 'vm3')], response.context['vms']) self.user.is_superuser = False self.user.save() # POST - invalid vm self.user.grant('admin', self.cluster0) data = {'virtual_machines': [-1], 'owner': self.owner.id} response = self.c.post(url, data) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/orphans.html') self.assertTrue(response.context['form'].errors) # POST - invalid owner data = {'virtual_machines': [self.vm0.id], 'owner': -1} response = self.c.post(url, data) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/orphans.html') self.assertTrue(response.context['form'].errors) # POST - user does not have perms for cluster data = {'virtual_machines': [self.vm3.id], 'owner': self.owner.id} response = self.c.post(url, data) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/orphans.html') self.assertTrue(response.context['form'].errors) # POST - success data = {'virtual_machines': [self.vm0.id], 'owner': self.owner.id} response = self.c.post(url, data) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/orphans.html') self.assertFalse(response.context['form'].errors) self.assertEqual([], response.context['vms']) def test_missing_ganeti(self): """ Tests view for Virtual Machines missing from ganeti """ url = '/import/missing/' self.cluster0.rapi.GetInstances.response = ['vm0', 'vm2'] self.cluster1.rapi.GetInstances.response = ['vm3', 'vm5'] # anonymous user response = self.c.get(url, follow=True) self.assertEqual(200, response.status_code) self.assertTemplateUsed(response, 'registration/login.html') # unauthorized user self.assertTrue(self.c.login(username=self.user.username, password='******')) response = self.c.get(url) self.assertEqual(403, response.status_code) # authorized get (cluster admin perm) self.user.grant('admin', self.cluster0) response = self.c.get(url) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/missing.html') self.assertEqual([('vm1', 'test0', 'vm1')], response.context['vms']) self.user.revoke_all(self.cluster0) # authorized get (superuser) self.user.is_superuser = True self.user.save() response = self.c.get(url) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/missing.html') self.assertEqual([('vm1', 'test0', 'vm1'), ('vm4', 'test1', 'vm4')], response.context['vms']) self.user.is_superuser = False self.user.save() # POST - invalid vm self.user.grant('admin', self.cluster0) data = {'virtual_machines': [-1]} response = self.c.post(url, data) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/missing.html') self.assertTrue(response.context['form'].errors) # POST - user does not have perms for cluster data = {'virtual_machines': [self.vm3.hostname]} response = self.c.post(url, data) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/missing.html') self.assertTrue(response.context['form'].errors) # POST - success data = {'virtual_machines': ['vm1']} response = self.c.post(url, data) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/missing.html') self.assertFalse(response.context['form'].errors) self.assertEqual([], response.context['vms']) def test_missing_db(self): """ Tests view for Virtual Machines missing from database """ url = '/import/missing_db/' self.cluster0.rapi.GetInstances.response = ['vm0', 'vm2'] self.cluster1.rapi.GetInstances.response = ['vm3', 'vm5'] # anonymous user response = self.c.get(url, follow=True) self.assertEqual(200, response.status_code) self.assertTemplateUsed(response, 'registration/login.html') # unauthorized user self.assertTrue(self.c.login(username=self.user.username, password='******')) response = self.c.get(url) self.assertEqual(403, response.status_code) # authorized get (cluster admin perm) self.user.grant('admin', self.cluster0) response = self.c.get(url) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/missing_db.html') cid_vm_hostname = Template('$cid:$hostname') c_vm2 = cid_vm_hostname.substitute( cid=self.cluster0.id, hostname='vm2') self.assertEqual([(c_vm2, u'test0', u'vm2')], response.context['vms']) self.user.revoke_all(self.cluster0) # authorized get (superuser) self.user.is_superuser = True self.user.save() response = self.c.get(url) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/missing_db.html') c_vm5 = cid_vm_hostname.substitute( cid=self.cluster1.id, hostname='vm5') self.assertEqual([(c_vm2, u'test0', u'vm2'), (c_vm5, u'test1', u'vm5')], response.context['vms']) self.user.is_superuser = False self.user.save() # POST - invalid vm self.user.grant('admin', self.cluster0) data = {'virtual_machines': [-1]} response = self.c.post(url, data) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/missing_db.html') self.assertTrue(response.context['form'].errors) # POST - invalid owner data = {'virtual_machines': [self.vm0.id], 'owner': -1} response = self.c.post(url, data) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/missing_db.html') self.assertTrue(response.context['form'].errors) # POST - user does not have perms for cluster data = {'virtual_machines': [self.vm3.hostname], 'owner': self.owner.id} response = self.c.post(url, data) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/missing_db.html') self.assertTrue(response.context['form'].errors) # POST - success data = {'virtual_machines': [c_vm2], 'owner': self.owner.id} response = self.c.post(url, data) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/missing_db.html') self.assertFalse(response.context['form'].errors) self.assertEqual([], response.context['vms']) self.assertTrue(VirtualMachine.objects.filter(hostname='vm2').exists())
class ImportViews(TestCase): def setUp(self): self.tearDown() self.user = User(id=2, username='******') self.user.set_password('secret') self.user.save() self.group = Group(name='testing_group') self.group.save() self.cluster0 = Cluster(hostname='test0', slug='OSL_TEST0') self.cluster0.save() self.cluster1 = Cluster(hostname='test1', slug='OSL_TEST1') self.cluster1.save() self.vm0 = VirtualMachine(hostname='vm0', cluster=self.cluster0) self.vm1 = VirtualMachine(hostname='vm1', cluster=self.cluster0, owner=self.user.get_profile()) # self.vm2 = VirtualMachine(hostname='vm2', cluster=cluster0) self.vm3 = VirtualMachine(hostname='vm3', cluster=self.cluster1) self.vm4 = VirtualMachine(hostname='vm4', cluster=self.cluster1, owner=self.user.get_profile()) # self.vm5 = VirtualMachine(hostname='vm5', cluster=cluster1) self.vm0.save() self.vm1.save() self.vm3.save() self.vm4.save() self.c = Client() self.owner = self.user.get_profile() def tearDown(self): VirtualMachine.objects.all().delete() Cluster.objects.all().delete() Organization.objects.all().delete() Profile.objects.all().delete() User.objects.all().delete() Group.objects.all().delete() def test_orphans_view(self): """ Test orphans view """ url = '/import/orphans/' # anonymous user response = self.c.get(url, follow=True) self.assertEqual(200, response.status_code) self.assertTemplateUsed(response, 'registration/login.html') # unauthorized user self.assertTrue( self.c.login(username=self.user.username, password='******')) response = self.c.get(url) self.assertEqual(403, response.status_code) # authorized get (cluster admin perm) self.user.grant('admin', self.cluster0) response = self.c.get(url) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/orphans.html') self.assertEqual([(self.vm0.id, 'test0', 'vm0')], response.context['vms']) self.user.revoke_all(self.cluster0) # authorized get (superuser) self.user.is_superuser = True self.user.save() response = self.c.get(url) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/orphans.html') self.assertEqual([(self.vm0.id, 'test0', 'vm0'), (self.vm3.id, 'test1', 'vm3')], response.context['vms']) self.user.is_superuser = False self.user.save() # POST - invalid vm self.user.grant('admin', self.cluster0) data = {'virtual_machines': [-1], 'owner': self.owner.id} response = self.c.post(url, data) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/orphans.html') self.assertTrue(response.context['form'].errors) # POST - invalid owner data = {'virtual_machines': [self.vm0.id], 'owner': -1} response = self.c.post(url, data) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/orphans.html') self.assertTrue(response.context['form'].errors) # POST - user does not have perms for cluster data = {'virtual_machines': [self.vm3.id], 'owner': self.owner.id} response = self.c.post(url, data) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/orphans.html') self.assertTrue(response.context['form'].errors) # POST - success data = {'virtual_machines': [self.vm0.id], 'owner': self.owner.id} response = self.c.post(url, data) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/orphans.html') self.assertFalse(response.context['form'].errors) self.assertEqual([], response.context['vms']) def test_missing_ganeti(self): """ Tests view for Virtual Machines missing from ganeti """ url = '/import/missing/' self.cluster0.rapi.GetInstances.response = ['vm0', 'vm2'] self.cluster1.rapi.GetInstances.response = ['vm3', 'vm5'] # anonymous user response = self.c.get(url, follow=True) self.assertEqual(200, response.status_code) self.assertTemplateUsed(response, 'registration/login.html') # unauthorized user self.assertTrue( self.c.login(username=self.user.username, password='******')) response = self.c.get(url) self.assertEqual(403, response.status_code) # authorized get (cluster admin perm) self.user.grant('admin', self.cluster0) response = self.c.get(url) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/missing.html') self.assertEqual([('vm1', 'test0', 'vm1')], response.context['vms']) self.user.revoke_all(self.cluster0) # authorized get (superuser) self.user.is_superuser = True self.user.save() response = self.c.get(url) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/missing.html') self.assertEqual([('vm1', 'test0', 'vm1'), ('vm4', 'test1', 'vm4')], response.context['vms']) self.user.is_superuser = False self.user.save() # POST - invalid vm self.user.grant('admin', self.cluster0) data = {'virtual_machines': [-1]} response = self.c.post(url, data) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/missing.html') self.assertTrue(response.context['form'].errors) # POST - user does not have perms for cluster data = {'virtual_machines': [self.vm3.hostname]} response = self.c.post(url, data) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/missing.html') self.assertTrue(response.context['form'].errors) # POST - success data = {'virtual_machines': ['vm1']} response = self.c.post(url, data) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/missing.html') self.assertFalse(response.context['form'].errors) self.assertEqual([], response.context['vms']) def test_missing_db(self): """ Tests view for Virtual Machines missing from database """ url = '/import/missing_db/' self.cluster0.rapi.GetInstances.response = ['vm0', 'vm2'] self.cluster1.rapi.GetInstances.response = ['vm3', 'vm5'] # anonymous user response = self.c.get(url, follow=True) self.assertEqual(200, response.status_code) self.assertTemplateUsed(response, 'registration/login.html') # unauthorized user self.assertTrue( self.c.login(username=self.user.username, password='******')) response = self.c.get(url) self.assertEqual(403, response.status_code) # authorized get (cluster admin perm) self.user.grant('admin', self.cluster0) response = self.c.get(url) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/missing_db.html') cid_vm_hostname = Template('$cid:$hostname') c_vm2 = cid_vm_hostname.substitute(cid=self.cluster0.id, hostname='vm2') self.assertEqual([(c_vm2, u'test0', u'vm2')], response.context['vms']) self.user.revoke_all(self.cluster0) # authorized get (superuser) self.user.is_superuser = True self.user.save() response = self.c.get(url) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/missing_db.html') c_vm5 = cid_vm_hostname.substitute(cid=self.cluster1.id, hostname='vm5') self.assertEqual([(c_vm2, u'test0', u'vm2'), (c_vm5, u'test1', u'vm5')], response.context['vms']) self.user.is_superuser = False self.user.save() # POST - invalid vm self.user.grant('admin', self.cluster0) data = {'virtual_machines': [-1]} response = self.c.post(url, data) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/missing_db.html') self.assertTrue(response.context['form'].errors) # POST - invalid owner data = {'virtual_machines': [self.vm0.id], 'owner': -1} response = self.c.post(url, data) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/missing_db.html') self.assertTrue(response.context['form'].errors) # POST - user does not have perms for cluster data = { 'virtual_machines': [self.vm3.hostname], 'owner': self.owner.id } response = self.c.post(url, data) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/missing_db.html') self.assertTrue(response.context['form'].errors) # POST - success data = {'virtual_machines': [c_vm2], 'owner': self.owner.id} response = self.c.post(url, data) self.assertEqual(200, response.status_code) self.assertEqual('text/html; charset=utf-8', response['content-type']) self.assertTemplateUsed(response, 'ganeti/importing/missing_db.html') self.assertFalse(response.context['form'].errors) self.assertEqual([], response.context['vms']) self.assertTrue(VirtualMachine.objects.filter(hostname='vm2').exists())
def test_view_overview(self): """ Tests overview (status) page """ # TODO: in future, add Ganeti errors checking cluster1 = Cluster(hostname='cluster1', slug='cluster1') cluster1.save() vm1 = VirtualMachine(hostname='vm2.example.bak', cluster=cluster1) vm1.save() job = Job(job_id=233, obj=self.vm, cluster=self.cluster, finished="2011-01-07 21:59", status="error") job.save() job1 = Job(job_id=1234, obj=vm1, cluster=cluster1, finished="2011-01-05 21:59", status="error") job1.save() job.rapi.GetJobStatus.response = JOB_ERROR url = "/" args = [] template = "ganeti/overview.html" mimetype = "text/html; charset=utf-8" status = 200 # anonymous user response = self.c.get(url % args, follow=True) self.assertEqual(200, response.status_code) self.assertTemplateUsed(response, 'registration/login.html') # authorized user (non-admin) self.user.grant("admin", self.vm) self.user.save() self.assertTrue(self.c.login(username=self.user.username, password='******')) response = self.c.get(url % args) self.assertEqual(status, response.status_code) self.assertEqual(mimetype, response['content-type']) self.assertTemplateUsed(response, template) clusters = response.context['cluster_list'] self.assertTrue(self.cluster not in clusters) self.assertEqual(0, len(clusters)) self.assertEqual(0, response.context["orphaned"]) self.assertEqual(0, response.context["missing"]) self.assertEqual(0, response.context["import_ready"]) # authorized user (admin on one cluster) self.assertTrue(self.c.login(username=self.user1.username, password='******')) response = self.c.get(url % args) self.assertEqual(status, response.status_code) self.assertEqual(mimetype, response['content-type']) self.assertTemplateUsed(response, template) clusters = response.context['cluster_list'] self.assertTrue(self.cluster in clusters) self.assertEqual(1, len(clusters)) self.assertEqual(1, response.context["orphaned"]) self.assertEqual(1, response.context["missing"]) self.assertEqual(2, response.context["import_ready"]) # authorized user (superuser) self.assertTrue(self.c.login(username=self.user2.username, password='******')) response = self.c.get(url % args) self.assertEqual(status, response.status_code) self.assertEqual(mimetype, response['content-type']) self.assertTemplateUsed(response, template) clusters = response.context['cluster_list'] self.assertTrue(self.cluster in clusters) self.assertTrue(cluster1 in clusters) self.assertEqual(2, len(clusters)) self.assertEqual(2, response.context["orphaned"]) self.assertEqual(2, response.context["missing"]) self.assertEqual(4, response.context["import_ready"])
class TestGeneralViews(TestCase, ViewTestMixin): def setUp(self): self.tearDown() self.cluster = Cluster(hostname='test.example.test', slug='OSL_TEST') self.cluster.save() self.vm = VirtualMachine(hostname='vm1.example.bak', cluster=self.cluster) self.vm.save() User(id=1, username='******').save() settings.ANONYMOUS_USER_ID = 1 self.user = User(id=2, username='******') self.user.set_password('secret') self.user.save() self.user1 = User(id=3, username='******') self.user1.set_password('secret') self.user1.grant("admin", self.cluster) self.user1.grant("admin", self.vm) self.user1.save() self.user2 = User(id=4, username="******") self.user2.set_password("secret") self.user2.is_superuser = True self.user2.save() self.c = Client() def tearDown(self): SSHKey.objects.all().delete() Cluster.objects.all().delete() User.objects.all().delete() Group.objects.all().delete() Job.objects.all().delete() def test_view_overview(self): """ Tests overview (status) page """ # TODO: in future, add Ganeti errors checking cluster1 = Cluster(hostname='cluster1', slug='cluster1') cluster1.save() vm1 = VirtualMachine(hostname='vm2.example.bak', cluster=cluster1) vm1.save() job = Job(job_id=233, obj=self.vm, cluster=self.cluster, finished="2011-01-07 21:59", status="error") job.save() job1 = Job(job_id=1234, obj=vm1, cluster=cluster1, finished="2011-01-05 21:59", status="error") job1.save() job.rapi.GetJobStatus.response = JOB_ERROR url = "/" args = [] template = "ganeti/overview.html" mimetype = "text/html; charset=utf-8" status = 200 # anonymous user response = self.c.get(url % args, follow=True) self.assertEqual(200, response.status_code) self.assertTemplateUsed(response, 'registration/login.html') # authorized user (non-admin) self.user.grant("admin", self.vm) self.user.save() self.assertTrue(self.c.login(username=self.user.username, password='******')) response = self.c.get(url % args) self.assertEqual(status, response.status_code) self.assertEqual(mimetype, response['content-type']) self.assertTemplateUsed(response, template) clusters = response.context['cluster_list'] self.assertTrue(self.cluster not in clusters) self.assertEqual(0, len(clusters)) self.assertEqual(0, response.context["orphaned"]) self.assertEqual(0, response.context["missing"]) self.assertEqual(0, response.context["import_ready"]) # authorized user (admin on one cluster) self.assertTrue(self.c.login(username=self.user1.username, password='******')) response = self.c.get(url % args) self.assertEqual(status, response.status_code) self.assertEqual(mimetype, response['content-type']) self.assertTemplateUsed(response, template) clusters = response.context['cluster_list'] self.assertTrue(self.cluster in clusters) self.assertEqual(1, len(clusters)) self.assertEqual(1, response.context["orphaned"]) self.assertEqual(1, response.context["missing"]) self.assertEqual(2, response.context["import_ready"]) # authorized user (superuser) self.assertTrue(self.c.login(username=self.user2.username, password='******')) response = self.c.get(url % args) self.assertEqual(status, response.status_code) self.assertEqual(mimetype, response['content-type']) self.assertTemplateUsed(response, template) clusters = response.context['cluster_list'] self.assertTrue(self.cluster in clusters) self.assertTrue(cluster1 in clusters) self.assertEqual(2, len(clusters)) self.assertEqual(2, response.context["orphaned"]) self.assertEqual(2, response.context["missing"]) self.assertEqual(4, response.context["import_ready"]) def test_used_resources(self): """ tests the used_resources view """ group0 = Group.objects.create(name='group0') group1 = Group.objects.create(name='group1') self.user.groups.add(group0) self.user1.groups.add(group1) url = "/used_resources/" args = {} template = "ganeti/overview/used_resources_data.html" mimetype = "text/html; charset=utf-8" # anonymous user response = self.c.get(url, args, follow=True) self.assertEqual(200, response.status_code) self.assertTemplateUsed(response, 'registration/login.html') # 404 - no id self.assertTrue(self.c.login(username=self.user.username, password='******')) response = self.c.get(url, {}) self.assertEqual(404, response.status_code) # 404 - invalid id response = self.c.get(url, {'id': 1234567}) self.assertEqual(404, response.status_code) # unauthorized user (different user) response = self.c.get(url, {'id': self.user2.get_profile().pk}) self.assertEqual(403, response.status_code) # unauthorized user (in different group) self.assertTrue(self.c.login(username=self.user.username, password='******')) response = self.c.get(url, {'id': group1.organization.pk}) self.assertEqual(403, response.status_code) # authorized user (same user) response = self.c.get(url, {'id': self.user.get_profile().pk}) self.assertEqual(200, response.status_code) self.assertEqual(mimetype, response['content-type']) self.assertTemplateUsed(response, template) # authorized user (in group) response = self.c.get(url, {'id': group0.organization.pk}) self.assertEqual(200, response.status_code) self.assertEqual(mimetype, response['content-type']) self.assertTemplateUsed(response, template) # authorized user (superuser) self.assertTrue(self.c.login(username=self.user2.username, password='******')) response = self.c.get(url, {'id': self.user.get_profile().pk}) self.assertEqual(200, response.status_code) self.assertEqual(mimetype, response['content-type']) self.assertTemplateUsed(response, template) # authorized user (superuser) self.assertTrue(self.c.login(username=self.user2.username, password='******')) response = self.c.get(url, {'id': group1.organization.pk}) self.assertEqual(200, response.status_code) self.assertEqual(mimetype, response['content-type']) self.assertTemplateUsed(response, template) def test_view_ssh_keys(self): """ tests retrieving all sshkeys from the gwm instance """ # add some keys SSHKey.objects.create(key="ssh-rsa test test@test", user=self.user) SSHKey.objects.create(key="ssh-dsa test asd@asd", user=self.user) SSHKey.objects.create(key="ssh-dsa test foo@bar", user=self.user1) self.user.revoke_all(self.vm) self.user.revoke_all(self.cluster) self.user1.revoke_all(self.vm) self.user1.revoke_all(self.cluster) self.user2.revoke_all(self.vm) self.user2.revoke_all(self.cluster) # get API key # import settings key = settings.WEB_MGR_API_KEY url = '/keys/%s/' args = (key,) # cluster without users who have admin perms response = self.c.get(url % args) self.assertEqual(200, response.status_code) self.assertEquals("application/json", response["content-type"]) self.assertEqual(len(json.loads(response.content)), 0) self.assertNotContains(response, "test@test") self.assertNotContains(response, "asd@asd") # vm with users who have admin perms # grant admin permission to first user self.user.grant("admin", self.vm) self.user1.grant("admin", self.cluster) response = self.c.get(url % args) self.assertEqual(200, response.status_code) self.assertEquals("application/json", response["content-type"]) self.assertEqual(len(json.loads(response.content)), 3) self.assertContains(response, "test@test", count=1) self.assertContains(response, "asd@asd", count=1) self.assertContains(response, "foo@bar", count=1)