conntrackt Changeset - c966cf604d1c

Changeset - c966cf604d1c

Parent rev.

Child rev.

[Not reviewed]

default

0 1 0

Branko Majic (branko) - 8 years ago 2017-12-19 14:52:39
branko@majic.rs

CONNT-28: Updated ordering of elements in one of the tests for expected results to avoid test failure.

1 file changed with 2 insertions and 2 deletions:

conntrackt/tests/test_views.py

0 comments (0 inline, 0 general)

conntrackt/tests/test_views.py

➞

Show inline comments

@@ @@ -26,386 +26,386 @@ from zipfile import ZipFile, ZIP_DEFLATE @@
 # Python third-party library imports.
 import mock
 # Django imports.
 from django.core.exceptions import ValidationError
 from django.core.urlresolvers import reverse
 from django.http import Http404
 from django.test import RequestFactory
 from django.test import TestCase
 from django.utils.http import urlquote
 # Application imports
 from conntrackt.models import Project, Location, Entity, Interface, Communication
 from conntrackt.views import IndexView, SearchView, APISearchView
 from conntrackt.views import entity_iptables, project_iptables, project_diagram
 from conntrackt.views import ProjectView, ProjectCreateView, ProjectUpdateView, ProjectDeleteView
 from conntrackt.views import LocationCreateView, LocationUpdateView, LocationDeleteView
 from conntrackt.views import EntityView, EntityCreateView, EntityUpdateView, EntityDeleteView
 from conntrackt.views import InterfaceCreateView, InterfaceUpdateView, InterfaceDeleteView
 from conntrackt.views import CommunicationCreateView, CommunicationUpdateView, CommunicationDeleteView
 # Test imports.
 from .forms import FormWithWidgetCSSClassFormMixin, FormWithPlaceholderFormMixin
 from .helpers import PermissionTestMixin, create_get_request, generate_get_response, FakeMessages
 from .views import RedirectToNextMixinView
 from .factories import setup_test_data
 class IndexViewTest(PermissionTestMixin, TestCase):
     sufficient_permissions = ("view",)
     view_class = IndexView
     def setUp(self):
         """
         Set-up some test data.
         """
         setup_test_data()
     def test_context_no_projects(self):
         """
         Verifies that the context is properly set-up when the view is called and
         no projects are available.
         """
         Project.objects.all().delete()
         # Get the view.
         view = IndexView.as_view()
         # Get the response.
         response = generate_get_response(view)
         # Validate the response.
         self.assertQuerysetEqual(response.context_data["projects"], [])
     def test_context_no_locations(self):
         """
         Verifies that the context is properly set-up when the view is called and
         no locations are available.
         """
         Location.objects.all().delete()
         # Get the view.
         view = IndexView.as_view()
         # Get the response.
         response = generate_get_response(view)
         # Validate the response.
         self.assertQuerysetEqual(response.context_data["locations"], [])
     def test_context_projects(self):
         """
         Verifies that the context is properly set-up when the view is called and
         there's multiple projects available.
         """
         # Get the view.
         view = IndexView.as_view()
         # Get the response.
         response = generate_get_response(view)
         self.assertQuerysetEqual(response.context_data["projects"], ["<Project: Test Project 1>", "<Project: Test Project 2>"])
     def test_locations_available(self):
         """
         Verifies that the context is properly set-up when the view is called and
         there's multiple locationsg available.
         """
         # Get the view.
         view = IndexView.as_view()
         # Get the response.
         response = generate_get_response(view)
         # Validate the response.
         self.assertQuerysetEqual(response.context_data["locations"], ["<Location: Test Location 1>", "<Location: Test Location 2>"])
 class ProjectViewTest(PermissionTestMixin, TestCase):
     sufficient_permissions = ("view",)
     permission_test_view_kwargs = {"pk": "1"}
     view_class = ProjectView
     def setUp(self):
         """
         Set-up some test data.
         """
         setup_test_data()
     def test_context(self):
         """
         Verifies that the context is properly set-up when the view is called.
         """
         # Get the view.
         view = ProjectView.as_view()
         # Get the response.
         response = generate_get_response(view, pk=1)
         # Fetch context data from response.
         location, entities = response.context_data["location_entities"][0]
         # Set-up expected context data values.
         expected_entities = ["<Entity: Test Entity 1 (Test Project 1 - Test Location 1)>",
                              "<Entity: Test Entity 2 (Test Project 1 - Test Location 1)>"]
         # Validate context data.
         self.assertEqual(location.name, "Test Location 1")
         self.assertQuerysetEqual(entities, expected_entities)
         # Fetch context data from response.
         location, entities = response.context_data["location_entities"][1]
         # Set-up expected context data values.
         expected_entities = ["<Entity: Test Entity 3 (Test Project 1 - Test Location 2)>",
                              "<Entity: Test Subnet 4 (Test Project 1 - Test Location 2)>"]
         # Validate context data.
         self.assertEqual(location.name, "Test Location 2")
         self.assertQuerysetEqual(entities, expected_entities)
         # Validate context data.
         self.assertEqual(str(response.context_data["project"]), "Test Project 1")
         # Validate context data is present.
         self.assertIn("communications", response.context_data.keys())
 class EntityViewTest(PermissionTestMixin, TestCase):
     view_class = EntityView
     sufficient_permissions = ("view",)
     permission_test_view_kwargs = {"pk": "1"}
     def setUp(self):
         """
         Set-up some test data.
         """
         setup_test_data()
     def test_context(self):
         """
         Tests if the form comes pre-populated with proper content.
         """
         # Get the view.
         view = EntityView.as_view()
         # Get the response.
         response = generate_get_response(view, pk=1)
         # Set-up expected context data.
         expected_entity = Entity.objects.get(pk=1)
         expected_incoming_communications = ["<Communication: Test Entity 2 -> Test Entity 1 (TCP:22)>",
                                             "<Communication: Test Entity 2 -> Test Entity 1 (ICMP:8)>",
                                             "<Communication: Test Entity 3 -> Test Entity 1 (TCP:3306)>",
                                             "<Communication: Test Subnet 4 -> Test Entity 1 (TCP:22)>"]
         expected_outgoing_communications = ["<Communication: Test Entity 1 -> Test Entity 2 (UDP:123)>",
                                             "<Communication: Test Entity 1 -> Test Entity 3 (UDP:53)>"]
         expected_outgoing_communications = ["<Communication: Test Entity 1 -> Test Entity 3 (UDP:53)>",
                                             "<Communication: Test Entity 1 -> Test Entity 2 (UDP:123)>"]
         expected_interfaces = ["<Interface: Test Entity 1 (192.168.1.1)>"]
         # Validate the response.
         self.assertQuerysetEqual(response.context_data["interfaces"], expected_interfaces)
         self.assertQuerysetEqual(response.context_data["incoming_communications"], expected_incoming_communications)
         self.assertQuerysetEqual(response.context_data["outgoing_communications"], expected_outgoing_communications)
         self.assertEqual(response.context_data["entity"], expected_entity)
         self.assertTrue("entity_iptables" in response.context_data)
 class EntityIptablesTest(PermissionTestMixin, TestCase):
     view_function = staticmethod(entity_iptables)
     sufficient_permissions = ("view",)
     permission_test_view_kwargs = {"pk": "1"}
     def setUp(self):
         """
         Set-up some test data.
         """
         setup_test_data()
     def test_invalid_entity(self):
         """
         Tests if a 404 is returned if no entity was found (invalid ID).
         """
         # Set-up a request.
         request = create_get_request()
         # Get the view.
         view = entity_iptables
         # Validate the response.
         self.assertRaises(Http404, view, request, pk=200)
     def test_content_type(self):
         """
         Test if correct content type is being returned by the response.
         """
         # Get the view.
         view = entity_iptables
         # Get the response.
         response = generate_get_response(view, pk=1)
         self.assertEqual(response['Content-Type'], "text/plain")
     def test_content_disposition(self):
         """
         Test if the correct content disposition has been set.
         """
         # Get the view.
         view = entity_iptables
         # Get the response.
         response = generate_get_response(view, pk=1)
         self.assertEqual(response['Content-Disposition'], "attachment; filename=test_entity_1-iptables.conf")
     def test_content(self):
         """
         Tests content produced by the view.
         """
         # Get the view.
         view = entity_iptables
         # Get the response.
         response = generate_get_response(view, pk=1)
         self.assertContains(response, ":INPUT")
         self.assertContains(response, ":OUTPUT")
         self.assertContains(response, ":FORWARD")
 class ProjectIptablesTest(PermissionTestMixin, TestCase):
     view_function = staticmethod(project_iptables)
     sufficient_permissions = ("view",)
     permission_test_view_kwargs = {"project_id": 1}
     def setUp(self):
         """
         Set-up some test data.
         """
         setup_test_data()
     def test_invalid_project(self):
         """
         Tests if a 404 is returned if no project was found (invalid ID).
         """
         # Set-up a request.
         request = create_get_request()
         # Get the view.
         view = project_iptables
         # Request iptables for whole project.
         self.assertRaises(Http404, view, request, 200)
         # Request iptables for project location
         self.assertRaises(Http404, view, request, 200, 1)
     def test_invalid_location(self):
         """
         Tests if a 404 is returned if no location was found (invalid ID).
         """
         # Set-up a request.
         request = create_get_request()
         # Get the view.
         view = project_iptables
         # Request iptables for project location
         self.assertRaises(Http404, view, request, 1, 200)
     def test_content_type(self):
         """
         Test if correct content type is being returned by the response.
         """
         # Get the view.
         view = project_iptables
         # Get the response.
         response = generate_get_response(view, None, 1)
         # Validate the response.
         self.assertEqual(response['Content-Type'], "application/zip")
     def test_content_disposition(self):
         """
         Test if the correct content disposition has been set.
         """
         # Get the view.
         view = project_iptables
         # Get the response.
         response = generate_get_response(view, None, 1)
         self.assertEqual(response['Content-Disposition'], 'attachment; filename="test_project_1-iptables.zip"')
         response = generate_get_response(view, None, 1, 1)
         self.assertEqual(response['Content-Disposition'], 'attachment; filename="test_project_1-test_location_1-iptables.zip"')
     def test_content_project(self):
         """
         Verifies that the content is properly generated when the view is called
         for an entire project.
         """
         # Get the view.
         view = project_iptables
         # Get the response.
         response = generate_get_response(project_iptables, None, 1)
         buff = StringIO(response.content)
         zipped_iptables = ZipFile(buff, "r", ZIP_DEFLATED)
         expected_zip_files = ["test_entity_1-iptables.conf",
                               "test_entity_2-iptables.conf",
                               "test_entity_3-iptables.conf",
                               "test_subnet_4-iptables.conf"]
         self.assertEqual(len(zipped_iptables.namelist()), 4)
         self.assertEqual(zipped_iptables.namelist(), expected_zip_files)
         for filename in expected_zip_files:
             iptables_file = zipped_iptables.read(filename)
             self.assertIn(":INPUT", iptables_file)
             self.assertIn(":OUTPUT", iptables_file)
             self.assertIn(":FORWARD", iptables_file)
         zipped_iptables.close()
     def test_content_location(self):
         """
         Verifies that the content is properly generated when the view is called
         for an entire project.
         """
         # Get the view.
         view = project_iptables

0 comments (0 inline, 0 general)