Synnefo

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.neutron.models_factory as mf

NEUTRON_URL = get_service_path(cyclades_services, "neutron", "v2.0")

NETWORKS_URL = join_urls(NEUTRON_URL, "networks/")

SUBNETS_URL = join_urls(NEUTRON_URL, "subnets/")

class NetworkTest(BaseAPITest):

    def test_list_networks(self):

        response = self.get(NETWORKS_URL)

        self.assertSuccess(response)

        networks = json.loads(response.content)

        self.assertEqual(networks, {'networks': []})

    def test_create_network(self):

        request = {"network":

                    {"name": "sample_network"}

                }

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

        code = response.status_code

        self.assertEqual(code, 201)

    def test_get_unfound_network(self):

        url = join_urls(NETWORKS_URL,"123")

        response = self.get(url)

        self.assertItemNotFound(response)

    def test_get_network(self):

        test_net = mf.NetworkFactory.create()

        url = join_urls(NETWORKS_URL,str(test_net.id))

        response = self.get(url, user=test_net.userid)

        self.assertEqual(response.status_code, 200)

    def test_delete_unfound_network(self):

        url = join_urls(NETWORKS_URL,"123")

        response = self.delete(url)

        self.assertItemNotFound(response)

    def test_delete_network(self):

        test_net = mf.NetworkFactory.create()

        url = join_urls(NETWORKS_URL,str(test_net.id))

        response = self.delete(url, user=test_net.userid)

        self.assertEqual(response.status_code, 204)

    def test_put_unfound_network(self):

        url = join_urls(NETWORKS_URL,"123")

        response = self.delete(url)

        self.assertItemNotFound(response)

    def test_put_network(self):

        test_net = mf.NetworkFactory.create()

        url = join_urls(NETWORKS_URL,str(test_net.id))

        request = {"network":

            {"name": "new_name"}

        }

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

        self.assertEqual(response.status_code, 200)

    def test_put_network_wrong(self):

        test_net = mf.NetworkFactory.create()

        url = join_urls(NETWORKS_URL,str(test_net.id))

        request = {"network":

            {"wronng_field": "new_name"}

        }

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

        self.assertEqual(response.status_code, 400)

class SubnetTest(BaseAPITest):

    def test_list_subnets(self):

        '''Test Subnet list'''

        response = self.get(SUBNETS_URL)

        self.assertSuccess(response)

        subnets = json.loads(response.content)

        self.assertEqual(subnets, {'subnets': []})

    def test_get_subnet(self):

        '''Test get info of a subnet'''

        url = join_urls(SUBNETS_URL, '42')

        response = self.get(url)

        self.assertSuccess(response)

        subnet = json.loads(response.content)

        self.assertEqual(subnet, {'subnet': []})

    def test_get_subnet_404(self):

        '''Test get info of a subnet that doesn't exist'''

        url = join_urls(SUBNETS_URL, '52')

        response = self.get(url)

        self.assertItemNotFound(response)

    def test_subnet_delete_not_in_use(self):

        '''Test delete a subnet that's not in use'''

        url = join_urls(SUBNETS_URL, '42')

        response = self.get(url)

        self.assertItemNotFound(response)