Synnefo

# Copyright 2012 GRNET S.A. All rights reserved.

#

# Redistribution and use in source and binary forms, with or without

# modification, are permitted provided that the following conditions

# are met:

#

#   1. Redistributions of source code must retain the above copyright

#      notice, this list of conditions and the following disclaimer.

#

#  2. Redistributions in binary form must reproduce the above copyright

#     notice, this list of conditions and the following disclaimer in the

#     documentation and/or other materials provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND

# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE

# ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE

# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL

# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS

# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)

# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY

# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF

# SUCH DAMAGE.

#

# The views and conclusions contained in the software and documentation are

# those of the authors and should not be interpreted as representing official

# policies, either expressed or implied, of GRNET S.A.

# Unit Tests for db

#

# Provides automated tests for db module

from django.test import TestCase

# Import pool tests

from synnefo.db.pools.tests import *

from synnefo.db.models import *

from synnefo.db import models_factory as mfact

from synnefo.db.pools import IPPool

from django.db import IntegrityError

from django.core.exceptions import MultipleObjectsReturned

from mock import patch

class FlavorTest(TestCase):

    def test_flavor_name(self):

        """Test a flavor object name method."""

        flavor = mfact.FlavorFactory(cpu=1, ram=1024, disk=40,

                                     disk_template="temp")

        self.assertEqual(flavor.name, "C1R1024D40temp", "flavor.name is not"

                " generated correctly. Name is %s instead of C1R1024D40temp" %

                flavor.name)

class BackendTest(TestCase):

    def setUp(self):

        self.backend = mfact.BackendFactory()

    @patch("synnefo.db.models.get_rapi_client")

    def test_get_client(self, client):

        id_ = self.backend.id

        hash_ = self.backend.hash

        name = self.backend.clustername

        passwd = self.backend.password

        user = self.backend.username

        port = self.backend.port

        self.backend.get_client()

        client.assert_called_once_with(id_, hash_, name, port, user, passwd)

    def test_save_hash(self):

        """Test that backend hash is generated on credential change"""

        old_hash = self.backend.hash

        for field in ['clustername', 'username', 'password', 'port']:

            value = 5181 if field == 'port' else 'foo'

            self.backend.__setattr__(field, value)

            self.backend.save()

            self.assertNotEqual(old_hash, self.backend.hash)

            old_hash = self.backend.hash

    def test_unique_index(self):

        """Test that each backend gets a unique index"""

        backends = [self.backend]

        for i in xrange(0, 14):

            backends.append(mfact.BackendFactory())

        indexes = map(lambda x: x.index, backends)

        self.assertEqual(len(indexes), len(set(indexes)))

    def test_backend_number(self):

        """Test that no more than 16 backends are created"""

        for i in xrange(0, 14):

            mfact.BackendFactory()

        self.assertRaises(Exception, mfact.BackendFactory, ())

    def test_delete_backend(self):

        vm = mfact.VirtualMachineFactory(backend=self.backend, deleted=True)

        bnet = mfact.BackendNetworkFactory(backend=self.backend)

        self.backend.delete()

        self.assertRaises(Backend.DoesNotExist, Backend.objects.get,

                          id=self.backend.id)

        # Test that VM is not deleted

        vm2 = VirtualMachine.objects.get(id=vm.id)

        self.assertEqual(vm2.backend, None)

        # Test tha backend networks are deleted, but not the network

        self.assertRaises(BackendNetwork.DoesNotExist,

                          BackendNetwork.objects.get, id=bnet.id)

        Network.objects.get(id=bnet.network.id)

    def test_delete_active_backend(self):

        """Test that a backend with non-deleted VMS is not deleted"""

        mfact.VirtualMachineFactory(backend=self.backend)

        self.assertRaises(IntegrityError, self.backend.delete, ())

    def test_password_encryption(self):

        password_hash = self.backend.password

        self.backend.password = '123'

        self.assertNotEqual(self.backend.password_hash, '123')

        self.assertNotEqual(self.backend.password_hash, password_hash)

        self.assertEqual(self.backend.password, '123')

class VirtualMachineTest(TestCase):

    def setUp(self):

        self.vm = mfact.VirtualMachineFactory()

    @patch("synnefo.db.models.get_rapi_client")

    def test_get_client(self, client):

        backend = self.vm.backend

        id_ = backend.id

        hash_ = backend.hash

        name = backend.clustername

        passwd = backend.password

        user = backend.username

        port = backend.port

        self.vm.get_client()

        client.assert_called_once_with(id_, hash_, name, port, user, passwd)

    def test_create(self):

        vm = mfact.VirtualMachineFactory()

        self.assertEqual(vm.action, None)

        self.assertEqual(vm.backendjobid, None)

        self.assertEqual(vm.backendjobstatus, None)

        self.assertEqual(vm.backendopcode, None)

        self.assertEqual(vm.backendlogmsg, None)

        self.assertEqual(vm.operstate, 'BUILD')

class NetworkTest(TestCase):

    def setUp(self):

        self.net = mfact.NetworkFactory()

    def test_tags(self):

        net1 = mfact.NetworkFactory(flavor='IP_LESS_ROUTED')

        self.assertEqual(net1.backend_tag, ['ip-less-routed'])

        net1 = mfact.NetworkFactory(flavor='CUSTOM')

        self.assertEqual(net1.backend_tag, [])

    def test_create_backend_network(self):

        len_backends = len(Backend.objects.all())

        back = mfact.BackendFactory()

        self.net.create_backend_network(backend=back)

        BackendNetwork.objects.get(network=self.net, backend=back)

        back1 = mfact.BackendFactory()

        back2 = mfact.BackendFactory()

        self.net.create_backend_network()

        BackendNetwork.objects.get(network=self.net, backend=back1)

        BackendNetwork.objects.get(network=self.net, backend=back2)

        self.assertEqual(len(BackendNetwork.objects.filter(network=self.net)),

                         len_backends + 3)

    def test_pool(self):

        pool = self.net.get_pool()

        pool.network = self.net

        self.assertTrue(isinstance(pool, IPPool))

    def test_reserve_ip(self):

        net1 = mfact.NetworkFactory(subnet='192.168.2.0/24')

        net1.reserve_address('192.168.2.12')

        pool = net1.get_pool()

        self.assertFalse(pool.is_available('192.168.2.12'))

        net1.release_address('192.168.2.12')

        pool = net1.get_pool()

        self.assertTrue(pool.is_available('192.168.2.12'))

class BackendNetworkTest(TestCase):

    def test_mac_prefix(self):

        network = mfact.NetworkFactory(mac_prefix='aa:bb:c')

        backend = mfact.BackendFactory()

        bnet = mfact.BackendNetworkFactory(network=network, backend=backend)

        self.assertTrue(backend.index < 10)

        self.assertEqual(bnet.mac_prefix, 'aa:bb:c%s' % backend.index)

    def test_invalid_mac(self):

        network = mfact.NetworkFactory(mac_prefix='zz:bb:c')

        backend = mfact.BackendFactory()

        self.assertRaises(utils.InvalidMacAddress,

                mfact.BackendNetworkFactory, network=network, backend=backend)

class BridgePoolTest(TestCase):

    def test_no_pool(self):

        self.assertRaises(BridgePoolTable.DoesNotExist,

                          BridgePoolTable.get_pool)

    def test_two_pools(self):

        mfact.BridgePoolTableFactory()

        mfact.BridgePoolTableFactory()

        self.assertRaises(MultipleObjectsReturned, BridgePoolTable.get_pool)

class AESTest(TestCase):

    def test_encrypt_decrtypt(self):

        from synnefo.db import aes_encrypt as aes

        old = 'bar'

        new = aes.decrypt_db_charfield(aes.encrypt_db_charfield(old))

        self.assertEqual(old, new)

    def test_password_change(self):

        from synnefo.db import aes_encrypt as aes

        old_pass = aes.SECRET_ENCRYPTION_KEY

        old = 'bar'

        encrypted = aes.encrypt_db_charfield(old)

        aes.SECRET_ENCRYPTION_KEY = 'foo2'

        self.assertRaises(aes.CorruptedPassword, aes.decrypt_db_charfield,

                          encrypted)

        aes.SECRET_ENCRYPTION_KEY = old_pass

        new = aes.decrypt_db_charfield(encrypted)

        self.assertEqual(old, new)

    def test_big_secret(self):

        from synnefo.db import aes_encrypt as aes

        old = aes.SECRET_ENCRYPTION_KEY

        aes.SECRET_ENCRYPTION_KEY = \

            '91490231234814234812348913289481294812398421893489'

        self.assertRaises(ValueError, aes.encrypt_db_charfield, 'la')

        aes.SECRET_ENCRYPTION_KEY = old