djing/ip_pool/tests.py

from abc import ABCMeta

from django.conf import settings
from django.shortcuts import resolve_url
from django.test import TestCase
from django.utils.translation import gettext_lazy as _

from accounts_app.models import UserProfile
from group_app.models import Group
from ip_pool.models import NetworkModel


class MyBaseTestCase(metaclass=ABCMeta):
    def _client_get_check_login(self, url):
        """
        Checks if url is protected from unauthorized access
        :param url:
        :return: authorized response
        """
        r = self.client.get(url)
        self.assertRedirects(r, "%s?next=%s" % (getattr(settings, 'LOGIN_URL'), url))
        self.client.force_login(self.adminuser)
        r = self.client.get(url)
        self.assertEqual(r.status_code, 200)
        return r

    def setUp(self):
        grp = Group.objects.create(title='Grp1')
        my_admin = UserProfile.objects.create_superuser('+79781234567', 'local_superuser', 'ps')
        self.adminuser = my_admin
        self.group = grp


class NetworksTestCase(MyBaseTestCase, TestCase):
    def setUp(self):
        super(NetworksTestCase, self).setUp()
        netw = NetworkModel.objects.create(
            network='192.168.23.0/24',
            kind='inet',
            description='SomeDescr',
            ip_start='192.168.23.2',
            ip_end='192.168.23.254'
        )
        netw.groups.add(self.group.pk)
        self.network = netw

    def test_add_network(self):
        print('test_add_network')
        url = resolve_url('ip_pool:net_add')
        self._client_get_check_login(url)

        # Ip outside range
        r = self.client.post(url, data={
            'network': '192.168.23.0/24',
            'kind': 'inet',
            'description': 'SomeDescr',
            'ip_start': '192.168.23.2',
            'ip_end': '192.168.24.254'
        })
        self.assertFormError(r, form='form', field='ip_end', errors=_('End ip must be in subnet of specified network'))

        # Invalid ip
        r = self.client.post(url, data={
            'network': '192.168.23.0/24',
            'kind': 'inet',
            'description': 'SomeDescr',
            'groups': ('1',),
            'ip_start': '192.168.23.2',
            'ip_end': '192.168.23.g'
        })
        self.assertFormError(r, form='form', field='ip_end', errors=_('Enter a valid IPv4 or IPv6 address.'))

        # Not existed group
        r = self.client.post(url, data={
            'network': '192.168.23.0/24',
            'kind': 'inet',
            'description': 'SomeDescr',
            'groups': ('2',),
            'ip_start': '192.168.23.2',
            'ip_end': '192.168.23.6'
        })
        self.assertFormError(
            r, form='form', field='groups',
            errors=_('Select a valid choice. %(value)s is not one of the available choices.') % {
                'value': 2
            }
        )

        # Successfully add
        r = self.client.post(url, data={
            'network': '192.168.12.0/24',
            'kind': 'inet',
            'description': 'SomeDescr',
            'groups': ('1',),
            'ip_start': '192.168.12.2',
            'ip_end': '192.168.12.254'
        })
        self.assertRedirects(r, resolve_url('ip_pool:net_edit', 2))
        created_network = NetworkModel.objects.get(pk=2)
        self.assertEqual('192.168.12.0/24', str(created_network.network))
        self.assertEqual('inet', created_network.kind)
        self.assertEqual('SomeDescr', created_network.description)
        self.assertEqual(1, created_network.groups.all().first().pk)
        self.assertEqual('192.168.12.2', str(created_network.ip_start))
        self.assertEqual('192.168.12.254', str(created_network.ip_end))

    def test_edit_network(self):
        print('test_edit_network')
        url = resolve_url('ip_pool:net_edit', net_id=self.network.pk)
        self._client_get_check_login(url)
        r = self.client.post(url, data={
            'network': '192.168.0.0/24',
            'kind': 'guest',
            'description': 'Описание',
            'groups': ('1',),
            'ip_start': '192.168.0.2',
            'ip_end': '192.168.0.254'
        })
        self.assertRedirects(r, resolve_url('ip_pool:net_edit', 1))
        updated_network = NetworkModel.objects.get(pk=1)
        self.assertEqual('192.168.0.0/24', str(updated_network.network))
        self.assertEqual('guest', updated_network.kind)
        self.assertEqual('Описание', updated_network.description)
        self.assertEqual(1, updated_network.groups.all().first().pk)
        self.assertEqual('192.168.0.2', str(updated_network.ip_start))
        self.assertEqual('192.168.0.254', str(updated_network.ip_end))

    def test_remove_network(self):
        print('test_remove_network')
        url = resolve_url('ip_pool:net_delete', net_id=self.network.pk)
        self._client_get_check_login(url)
        r = self.client.post(url)
        self.assertRedirects(r, resolve_url('ip_pool:networks'))
        try:
            NetworkModel.objects.get(pk=self.network.pk)
            raise self.failureException('Network must will be deleted')
        except NetworkModel.DoesNotExist:
            pass