Synnefo

from datetime import datetime

from mock import patch

from synnefo.api.util import allocate_resource

                                     update_build_progress)

from snf_django.utils.testing import mocked_quotaholder

    def create_msg(self, **kwargs):

        """Create snf-ganeti-hook message"""

        msg = {'event_time': split_time(time())}

        msg['status'] = 'success'

        msg['jobId'] = 1

        msg['logmsg'] = 'Dummy Log'

        return message

    def test_missing_attribute(self, client):

        client.basic_nack.assert_called_once()

    def test_unhandled_exception(self, client):

        client.basic_reject.assert_called_once()

    def test_wrong_type(self, client):

        msg = self.create_msg(type="WRONG_TYPE")

        client.basic_ack.assert_called_once()

    def test_missing_instance(self, client):

        msg = self.create_msg(operation='OP_INSTANCE_STARTUP',

                              instance='foo')

        client.basic_nack.assert_called_once()

    def test_no_nics(self, client):

        self.assertEqual(len(vm.nics.all()), 3)

        msg = self.create_msg(nics=[],

                              instance=vm.backend_vm_id)

        client.basic_ack.assert_called_once()

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

        self.assertEqual(len(db_vm.nics.all()), 0)

            net = mfactory.NetworkFactory(public=public)

            msg = self.create_msg(nics=[{'network': net.backend_id}],

                                  instance=vm.backend_vm_id)

            client.basic_ack.assert_called_once()

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

            nics = db_vm.nics.all()

                                     'ip': '10.0.0.22',

                                     'mac': 'aa:bb:cc:00:11:22'}],

                              instance=vm.backend_vm_id)

        client.basic_ack.assert_called_once()

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

        nics = db_vm.nics.all()