ganeti/tests/views/virtual_machine/forms.py

   1 from itertools import chain
   2 import unittest
   3
   4 from django.contrib.auth.models import User, Group
   5 from django.test import TestCase
   6
   7 from ganeti import models
   8 from ganeti import constants
   9 from ganeti.forms.virtual_machine import NewVirtualMachineForm, \
  10     HvmModifyVirtualMachineForm, KvmModifyVirtualMachineForm, \
  11     PvmModifyVirtualMachineForm
  12 from ganeti.tests.rapi_proxy import RapiProxy, XenRapiProxy, INFO, XEN_INFO
  13 from ganeti.tests.views.virtual_machine.base import VirtualMachineTestCaseMixin
  14
  15 __all__ = ['TestNewVirtualMachineForm',
  16     'TestKvmModifyVirtualMachineForm',
  17     'TestHvmModifyVirtualMachineForm',
  18     'TestPvmModifyVirtualMachineForm']
  19
  20 VirtualMachine = models.VirtualMachine
  21 Cluster = models.Cluster
  22
  23 global user, user1, group
  24 global cluster0, cluster1, cluster2, cluster3
  25 global kvm_cluster, hvm_cluster, pvm_cluster
  26 global kvm_vm, hvm_vm, pvm_vm
  27
  28 class TestNewVirtualMachineForm(TestCase, VirtualMachineTestCaseMixin):
  29
  30     def setUp(self):
  31         global user, user1, group, cluster0, cluster1, cluster2, cluster3
  32
  33         models.client.GanetiRapiClient = RapiProxy
  34         cluster0 = Cluster.objects.create(hostname='test0', slug='test0')
  35         cluster1 = Cluster.objects.create(hostname='test1', slug='test1')
  36         cluster2 = Cluster.objects.create(hostname='test2', slug='test2')
  37         cluster3 = Cluster.objects.create(hostname='test3', slug='test3')
  38         cluster0.sync_nodes()
  39         cluster0.info = INFO
  40
  41         user = User(id=67, username='tester0')
  42         user.set_password('secret')
  43         user.save()
  44         user1 = User(id=70, username='tester1')
  45         user1.set_password('secret')
  46         user1.save()
  47         group = Group(id=45, name='testing_group')
  48         group.save()
  49
  50     def tearDown(self):
  51         User.objects.all().delete()
  52         Group.objects.all().delete()
  53         VirtualMachine.objects.all().delete()
  54         Cluster.objects.all().delete()
  55
  56     def test_default_choices(self):
  57         """
  58         Test that ChoiceFields have the correct default options
  59         """
  60         form = NewVirtualMachineForm(user)
  61         self.assertEqual([(u'', u'---------')],
  62             form.fields['nic_type'].choices)
  63         self.assertEqual([(u'', u'---------'),
  64             (u'routed', u'routed'),
  65             (u'bridged', u'bridged')],
  66             form.fields['nic_mode'].choices)
  67         self.assertEqual([(u'', u'---------')],
  68             form.fields['boot_order'].choices)
  69         self.assertEqual([(u'', u'---------'),
  70             (u'plain', u'plain'),
  71             (u'drbd', u'drbd'),
  72             (u'file', u'file'),
  73             (u'diskless', u'diskless')],
  74             form.fields['disk_template'].choices)
  75
  76     def test_cluster_init(self):
  77         """
  78         Tests initializing a form with a Cluster
  79
  80         Verifies:
  81             * cluster choices are set correctly
  82             * node choices are set correctly
  83         """
  84
  85         # no cluster
  86         form = NewVirtualMachineForm(user)
  87         self.assertEqual([(u'', u'---------')], form.fields['pnode'].choices)
  88         self.assertEqual([(u'', u'---------')], form.fields['snode'].choices)
  89         self.assertEqual([(u'', u'---------')], form.fields['os'].choices)
  90
  91         # cluster from initial data
  92         form = NewVirtualMachineForm(user, {'cluster':cluster0.id})
  93         self.assertEqual(set([(u'', u'---------'), (u'gtest1.osuosl.bak', u'gtest1.osuosl.bak'), (u'gtest2.osuosl.bak', u'gtest2.osuosl.bak'), (u'gtest3.osuosl.bak', u'gtest3.osuosl.bak')]), set(form.fields['pnode'].choices))
  94         self.assertEqual(set([(u'', u'---------'), (u'gtest1.osuosl.bak', u'gtest1.osuosl.bak'), (u'gtest2.osuosl.bak', u'gtest2.osuosl.bak'), (u'gtest3.osuosl.bak', u'gtest3.osuosl.bak')]), set(form.fields['snode'].choices))
  95         self.assertEqual(form.fields['os'].choices,
  96             [
  97                 (u'', u'---------'),
  98                 ('Image',
  99                     [('image+debian-osgeo', 'Debian Osgeo'),
 100                     ('image+ubuntu-lucid', 'Ubuntu Lucid')]
 101                 )
 102             ]
 103         )
 104
 105         # cluster from initial data
 106         form = NewVirtualMachineForm(user, {'cluster':cluster0.id})
 107         self.assertEqual(set([(u'', u'---------'), (u'gtest1.osuosl.bak', u'gtest1.osuosl.bak'), (u'gtest2.osuosl.bak', u'gtest2.osuosl.bak'), (u'gtest3.osuosl.bak', u'gtest3.osuosl.bak')]), set(form.fields['pnode'].choices))
 108         self.assertEqual(set([(u'', u'---------'), (u'gtest1.osuosl.bak', u'gtest1.osuosl.bak'), (u'gtest2.osuosl.bak', u'gtest2.osuosl.bak'), (u'gtest3.osuosl.bak', u'gtest3.osuosl.bak')]), set(form.fields['snode'].choices))
 109         self.assertEqual(form.fields['os'].choices,
 110             [
 111                 (u'', u'---------'),
 112                 ('Image',
 113                     [('image+debian-osgeo', 'Debian Osgeo'),
 114                     ('image+ubuntu-lucid', 'Ubuntu Lucid')]
 115                 )
 116             ]
 117         )
 118
 119     def test_cluster_choices_init(self):
 120         """
 121         Tests that cluster choices are based on User permissions
 122
 123         Verifies:
 124             * superusers have all Clusters as choices
 125             * if owner is set, only display clusters the owner has permissions
 126               directly on.  This includes both users and groups
 127             * if no owner is set, choices include clusters that the user has
 128               permission directly on, or through a group
 129         """
 130         global user
 131
 132         # no owner, no permissions
 133         form = NewVirtualMachineForm(user)
 134         self.assertEqual(set([(u'', u'---------')]), set(form.fields['cluster'].choices))
 135
 136         # no owner, group and direct permissions
 137         user.grant('admin', cluster0)
 138         user.grant('create_vm', cluster1)
 139         group.grant('admin', cluster2)
 140         group.user_set.add(user)
 141         self.assertEqual(set([(u'', u'---------'), (cluster0.pk, u'test0'), (cluster1.pk, u'test1'), (cluster2.pk, u'test2')]), set(form.fields['cluster'].choices))
 142         user.revoke_all(cluster0)
 143         user.revoke_all(cluster1)
 144         group.revoke_all(cluster2)
 145
 146         # owner, user with no choices
 147         form = NewVirtualMachineForm(user, initial={'owner':user.get_profile().id})
 148         self.assertEqual(set([(u'', u'---------')]), set(form.fields['cluster'].choices))
 149
 150         # owner, user with choices
 151         user.grant('admin', cluster0)
 152         user.grant('create_vm', cluster1)
 153         form = NewVirtualMachineForm(user, initial={'owner':user.get_profile().id})
 154         self.assertEqual(set([(u'', u'---------'), (cluster0.pk, u'test0'), (cluster1.pk, u'test1')]), set(form.fields['cluster'].choices))
 155
 156         # owner, group with no choices
 157         form = NewVirtualMachineForm(user, initial={'owner':group.organization.id})
 158         self.assertEqual(set([(u'', u'---------')]), set(form.fields['cluster'].choices))
 159
 160         # owner, group with choices
 161         group.grant('admin', cluster2)
 162         group.grant('create_vm', cluster3)
 163         form = NewVirtualMachineForm(user, initial={'owner':group.organization.id})
 164         self.assertEqual(set([(u'', u'---------'), (cluster2.pk, u'test2'), (cluster3.pk, u'test3')]), set(form.fields['cluster'].choices))
 165
 166         # user - superuser
 167         user.is_superuser = True
 168         user.save()
 169         form = NewVirtualMachineForm(user, initial={'owner':user.get_profile().id})
 170         self.assertEqual(set([(u'', u'---------'), (cluster0.pk, u'test0'), (cluster1.pk, u'test1'), (cluster2.pk, u'test2'), (cluster3.pk, u'test3')]), set(form.fields['cluster'].choices))
 171
 172         # group - superuser
 173         form = NewVirtualMachineForm(user, initial={'owner':group.organization.id})
 174         self.assertEqual(set([(u'', u'---------'), (cluster0.pk, u'test0'), (cluster1.pk, u'test1'), (cluster2.pk, u'test2'), (cluster3.pk, u'test3')]), set(form.fields['cluster'].choices))
 175
 176     def test_owner_choices_init(self):
 177         """
 178         Tests that owner choices are set based on User permissions
 179
 180         Verifies:
 181             * superusers have all clusterusers as choices
 182             * user receives themselves as a choice if they have perms
 183             * user receives all groups they are a member of
 184         """
 185
 186         # user with no choices
 187         form = NewVirtualMachineForm(user)
 188         self.assertEqual([(u'', u'---------')], form.fields['owner'].choices)
 189
 190         # user with perms on self, no groups
 191         user.grant('admin', cluster0)
 192         form = NewVirtualMachineForm(user)
 193         self.assertEqual(
 194             [
 195                 (u'', u'---------'),
 196                 (user.profile.id, u'tester0'),
 197             ], form.fields['owner'].choices)
 198         user.set_perms(['create_vm'], cluster0)
 199         form = NewVirtualMachineForm(user)
 200         self.assertEqual(
 201             [
 202                 (u'', u'---------'),
 203                 (user.profile.id, u'tester0'),
 204             ], form.fields['owner'].choices)
 205
 206         # user with perms on self and groups
 207         group.user_set.add(user)
 208         group.grant('admin', cluster0)
 209         form = NewVirtualMachineForm(user)
 210         self.assertEqual(
 211             [
 212                 (u'', u'---------'),
 213                 (group.organization.id, u'testing_group'),
 214                 (user.profile.id, u'tester0'),
 215             ], form.fields['owner'].choices)
 216         user.revoke_all(cluster0)
 217
 218         # user with no perms on self, but groups
 219         form = NewVirtualMachineForm(user)
 220         self.assertEqual(
 221             [
 222                 (u'', u'---------'),
 223                 (group.organization.id, u'testing_group'),
 224             ], form.fields['owner'].choices)
 225         group.set_perms(['create_vm'], cluster0)
 226         form = NewVirtualMachineForm(user)
 227         self.assertEqual(
 228             [
 229                 (u'', u'---------'),
 230                 (group.organization.id, u'testing_group'),
 231             ], form.fields['owner'].choices)
 232         group.revoke_all(cluster0)
 233
 234         # superuser
 235         user.is_superuser = True
 236         user.save()
 237         form = NewVirtualMachineForm(user)
 238         self.assertEqual(
 239             [
 240                 (u'', u'---------'),
 241                 (user.profile.id, u'tester0'),
 242                 (user1.profile.id, u'tester1'),
 243                 (group.organization.id, u'testing_group'),
 244             ], list(form.fields['owner'].choices))
 245
 246
 247 class TestKvmModifyVirtualMachineForm(TestCase, VirtualMachineTestCaseMixin):
 248     def setUp(self):
 249         global kvm_cluster, kvm_vm
 250
 251         models.client.GanetiRapiClient = RapiProxy
 252         kvm_cluster = Cluster.objects.create(hostname='test0.kvm_cluster',
 253             slug='test0')
 254         kvm_vm = VirtualMachine.objects.create(hostname='kvm.osuosl',
 255             cluster=kvm_cluster)
 256
 257     def tearDown(self):
 258         VirtualMachine.objects.all().delete()
 259         Cluster.objects.all().delete()
 260
 261     def test_meta_rapiproxy_set(self):
 262         self.assertEqual(models.client.GanetiRapiClient, RapiProxy)
 263
 264     def test_form_defaults(self):
 265         """
 266         Test the default choices for ChoiceFields on the form.
 267         """
 268         choices = constants.KVM_CHOICES
 269         disk_type = choices['disk_type']
 270         nic_type = choices['nic_type']
 271         boot_order = choices['boot_order']
 272         disk_caches = constants.HV_DISK_CACHES
 273         kvm_flags = constants.KVM_FLAGS
 274         security_models = constants.HV_SECURITY_MODELS
 275         usb_mice = constants.HV_USB_MICE
 276
 277         form = KvmModifyVirtualMachineForm(kvm_vm)
 278         fields = form.fields
 279         self.assertEqual(set(disk_type),
 280             set(fields['disk_type'].choices))
 281         self.assertEqual(set(nic_type),
 282             set(fields['nic_type'].choices))
 283         self.assertEqual(set(boot_order),
 284             set(fields['boot_order'].choices))
 285         self.assertEqual(set(disk_caches),
 286             set(fields['disk_cache'].choices))
 287         self.assertEqual(set(kvm_flags),
 288             set(fields['kvm_flag'].choices))
 289         self.assertEqual(set(security_models),
 290             set(fields['security_model'].choices))
 291         self.assertEqual(set(usb_mice),
 292             set(fields['usb_mouse'].choices))
 293
 294     def test_initial_form_fields(self):
 295         """
 296         Test that the form does not contain any extra fields.
 297         """
 298         modify_fields = ('vcpus', 'memory', 'nic_link', 'nic_mac', 'os_name')
 299         hv_fields = KvmModifyVirtualMachineForm.hvparam_fields
 300         form = KvmModifyVirtualMachineForm(kvm_vm)
 301
 302         for field in chain(modify_fields, hv_fields):
 303             self.assertTrue(field in form.fields)
 304
 305         for field in form.Meta.exclude:
 306             self.assertFalse(field in form.fields)
 307
 308
 309 class TestHvmModifyVirtualMachineForm(TestCase, VirtualMachineTestCaseMixin):
 310     def setUp(self):
 311         from ganeti.tests.rapi_proxy import XEN_HVM_INSTANCE
 312         global hvm_cluster, hvm_vm
 313
 314         models.client.GanetiRapiClient = XenRapiProxy
 315         hvm_cluster = Cluster.objects.create(hostname='test2.hvm_cluster',
 316             slug='test2')
 317         hvm_cluster.info = XEN_INFO.copy()
 318         hvm_cluster.info['default_hypervisor'] = 'xen-hvm'
 319         hvm_vm = VirtualMachine.objects.create(hostname='hvm.osuosl',
 320             cluster=hvm_cluster)
 321         hvm_vm.info = XEN_HVM_INSTANCE
 322
 323     def tearDown(self):
 324         User.objects.all().delete()
 325         Group.objects.all().delete()
 326         VirtualMachine.objects.all().delete()
 327         Cluster.objects.all().delete()
 328
 329     def test_meta_xenrapiproxy_set(self):
 330         self.assertEqual(models.client.GanetiRapiClient, XenRapiProxy)
 331
 332     def test_meta_default_hypervisor(self):
 333         self.assertEqual(hvm_cluster.info['default_hypervisor'], 'xen-hvm')
 334
 335     def test_form_defaults(self):
 336         choices = constants.HVM_CHOICES
 337         disk_type = choices['disk_type']
 338         nic_type = choices['nic_type']
 339         boot_order = choices['boot_order']
 340
 341         form = HvmModifyVirtualMachineForm(hvm_vm)
 342         self.assertEqual(set(disk_type),
 343             set(form.fields['disk_type'].choices))
 344         self.assertEqual(set(nic_type),
 345             set(form.fields['nic_type'].choices))
 346         self.assertEqual(set(boot_order),
 347             set(form.fields['boot_order'].choices))
 348
 349     def test_initial_form_fields(self):
 350         """
 351         Test that the form does not contain any extra fields.
 352         """
 353         modify_fields = ('vcpus', 'memory', 'nic_link', 'nic_mac', 'os_name')
 354         hv_fields = HvmModifyVirtualMachineForm.hvparam_fields
 355         form = HvmModifyVirtualMachineForm(hvm_vm)
 356
 357         for field in chain(modify_fields, hv_fields):
 358             self.assertTrue(field in form.fields)
 359
 360         for field in form.Meta.exclude:
 361             self.assertFalse(field in form.fields)
 362         """
 363         # Uncomment to see if modify contains all available hvparam fields
 364         hvparam_fields = HvmModifyVirtualMachineForm.hvparam_fields
 365         hvparams = hvm_cluster.info['hvparams']['xen-hvm']
 366         self.assertEqual(set(hvparams), set(hvparam_fields))
 367         """
 368
 369     def test_field_initial_values(self):
 370         """
 371         Test that fields contain the correct initial values taken from a vm.
 372         """
 373         hvparam_fields = HvmModifyVirtualMachineForm.hvparam_fields
 374         hvparams = hvm_vm.info['hvparams']
 375         form = HvmModifyVirtualMachineForm(hvm_vm)
 376
 377         for field in hvparam_fields:
 378             self.assertEqual(form.fields[field].initial, hvparams[field])
 379
 380 class TestPvmModifyVirtualMachineForm(TestCase, VirtualMachineTestCaseMixin):
 381     def setUp(self):
 382         from ganeti.tests.rapi_proxy import XEN_PVM_INSTANCE
 383         global pvm_cluster, pvm_vm
 384
 385         models.client.GanetiRapiClient = XenRapiProxy
 386         pvm_cluster = Cluster.objects.create(hostname='test1.pvm_cluster',
 387             slug='test1')
 388         pvm_cluster.info = XEN_INFO
 389         pvm_vm = VirtualMachine.objects.create(hostname='pvm.osuosl',
 390             cluster=pvm_cluster)
 391         pvm_vm.info = XEN_PVM_INSTANCE
 392
 393     def tearDown(self):
 394         User.objects.all().delete()
 395         Group.objects.all().delete()
 396         VirtualMachine.objects.all().delete()
 397         Cluster.objects.all().delete()
 398
 399     def test_meta_default_hypervisor(self):
 400         self.assertEqual(pvm_cluster.info['default_hypervisor'], 'xen-pvm')
 401
 402     def test_meta_xenrapiproxy_set(self):
 403         self.assertEqual(models.client.GanetiRapiClient, XenRapiProxy)
 404
 405     def test_form_defaults(self):
 406         form = PvmModifyVirtualMachineForm(pvm_vm)
 407
 408     def test_initial_form_fields(self):
 409         """
 410         Test that the form does not contain any extra fields.
 411         """
 412         modify_fields = ('vcpus', 'memory', 'nic_link', 'nic_mac', 'os_name')
 413         hv_fields = PvmModifyVirtualMachineForm.hvparam_fields
 414         form = PvmModifyVirtualMachineForm(pvm_vm)
 415
 416         for field in chain(modify_fields, hv_fields):
 417             self.assertTrue(field in form.fields)
 418
 419         for field in form.Meta.exclude:
 420             self.assertFalse(field in form.fields)
 421
 422     def test_bound_form(self):
 423         data = dict(
 424             os_name = 'image+default',
 425             vcpus = 2,
 426             memory = 200,
 427         )
 428         form = PvmModifyVirtualMachineForm(pvm_vm, data)
 429         self.assertTrue(form.is_bound)
 430         self.assertTrue(form.is_valid(), msg=form.errors)
 431
 432     def test_field_initial_values(self):
 433         """
 434         Test that fields contain the correct initial values taken from a vm.
 435         """
 436         hvparam_fields = PvmModifyVirtualMachineForm.hvparam_fields
 437         hvparams = pvm_vm.info['hvparams']
 438         form = PvmModifyVirtualMachineForm(pvm_vm)
 439
 440         for field in hvparam_fields:
 441             self.assertEqual(form.fields[field].initial, hvparams[field])