add tests

ip_pool/tests.py

@@ -1,3 +1,139 @@
+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)
+        netw.save()
+        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))
 
-# Create your tests here.
+    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