Synnefo

# Copyright 2011-2013 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 GRNET S.A. ``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 GRNET S.A 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.i

from snf_django.utils.testing import (BaseAPITest, override_settings)

from snf_django.utils.testing import BaseAPITest

from django.utils import simplejson as json

from synnefo.cyclades_settings import cyclades_services

from synnefo.lib.services import get_service_path

from synnefo.lib import join_urls

import synnefo.db.models_factory as dbmf

COMPUTE_URL = get_service_path(cyclades_services, 'compute',

                               version='v2.0')

PORTS_URL = join_urls(COMPUTE_URL, "ports/")

class PortTest(BaseAPITest):

    def test_get_ports(self):

        url = join_urls(PORTS_URL)

        response = self.get(url)

        self.assertEqual(response.status_code, 200)

        ports = json.loads(response.content)

        self.assertEqual(ports, {"ports": []})

    def test_get_port_unfound(self):

        url = join_urls(PORTS_URL, "123")

        response = self.get(url)

        self.assertEqual(response.status_code, 404)

    def test_get_port(self):

        nic = dbmf.NetworkInterfaceFactory.create()

        url = join_urls(PORTS_URL, str(nic.id))

        response = self.get(url, user=nic.network.userid)

        self.assertEqual(response.status_code, 200)

    def test_delete_port(self):

        nic = dbmf.NetworkInterfaceFactory.create(device_owner='vm')

        url = join_urls(PORTS_URL, str(nic.id))

        response = self.delete(url, user=nic.network.userid)

        self.assertEqual(response.status_code, 204)

    def test_update_port_name(self):

        nic = dbmf.NetworkInterfaceFactory.create(device_owner='vm')

        url = join_urls(PORTS_URL, str(nic.id))

        request = {'port': {"name": "test-name"}}

        response = self.put(url, params=json.dumps(request),

                            user=nic.network.userid)

        self.assertEqual(response.status_code, 200)

        res = json.loads(response.content)

        self.assertEqual(res['port']['name'], "test-name")

    def test_update_port_sg_unfound(self):

        sg1 = dbmf.SecurityGroupFactory.create()

        nic = dbmf.NetworkInterfaceFactory.create(device_owner='vm')

        nic.security_groups.add(sg1)

        nic.save()

        url = join_urls(PORTS_URL, str(nic.id))

        request = {'port': {"security_groups": ["123"]}}

        response = self.put(url, params=json.dumps(request),

                            user=nic.network.userid)

        self.assertEqual(response.status_code, 404)

    def test_update_port_sg(self):

        sg1 = dbmf.SecurityGroupFactory.create()

        sg2 = dbmf.SecurityGroupFactory.create()

        sg3 = dbmf.SecurityGroupFactory.create()

        nic = dbmf.NetworkInterfaceFactory.create(device_owner='vm')

        nic.security_groups.add(sg1)

        nic.save()

        url = join_urls(PORTS_URL, str(nic.id))

        request = {'port': {"security_groups": [str(sg2.id), str(sg3.id)]}}

        response = self.put(url, params=json.dumps(request),

                            user=nic.network.userid)

        res = json.loads(response.content)

        self.assertEqual(res['port']['security_groups'],

                         [str(sg2.id), str(sg3.id)])

    def test_create_port_no_network(self):

        request = {

            "port": {

                "device_id": "123",

                "name": "port1",

                "network_id": "123"

            }

        }

        response = self.post(PORTS_URL, params=json.dumps(request))

        self.assertEqual(response.status_code, 404)

    def test_create_port(self):

        net = dbmf.NetworkFactory.create()

        subnet1 = dbmf.IPv4SubnetFactory.create(network=net)

        subnet2 = dbmf.IPv6SubnetFactory.create(network=net)

        sg1 = dbmf.SecurityGroupFactory.create()

        sg2 = dbmf.SecurityGroupFactory.create()

        vm = dbmf.VirtualMachineFactory.create(userid=net.userid)

        request = {

            "port": {

                "name": "port1",

                "network_id": str(net.id),

                "device_id": str(vm.id),

                "security_groups": [str(sg1.id), str(sg2.id)]

            }

        }

        response = self.post(PORTS_URL, params=json.dumps(request),

                             user=net.userid)

        self.assertEqual(response.status_code, 201)