initial commit of code

tests/test_exporters.py

@@ -0,0 +1,166 @@
+"""
+Test peeringdb data export views
+"""
+import os
+import pytest
+import json
+import difflib
+
+from django.test import Client
+
+from util import ClientCase
+
+from peeringdb_server.models import (Organization, Network, InternetExchange,
+                                     Facility, NetworkFacility, NetworkIXLan,
+                                     IXLan)
+
+
+class AdvancedSearchExportTest(ClientCase):
+    """
+    Tests advanced search result exports
+    """
+
+    @classmethod
+    def setUpTestData(cls):
+
+        ClientCase.setUpTestData()
+
+        # create organization
+        cls.org = Organization.objects.create(name="Test Org", status="ok")
+
+        entity_count = range(1, 4)
+
+        countries = ["US", "FI", ""]
+
+        # create networks
+        cls.net = [
+            Network.objects.create(
+                name="Network {}".format(i), status="ok",
+                aka="AKA {}".format(i), policy_general="Open",
+                info_traffic="0-20 Mbps", asn=i, org=cls.org)
+            for i in entity_count
+        ]
+
+        # create exchanges
+        cls.ix = [
+            InternetExchange.objects.create(
+                name="Exchange {}".format(i), media="Ethernet",
+                country=countries[i - 1], city="City {}".format(i),
+                status="ok", org=cls.org) for i in entity_count
+        ]
+
+        # create facilities
+        cls.fac = [
+            Facility.objects.create(
+                name="Facility {}".format(i), status="ok",
+                city="City {}".format(i), clli="CLLI{}".format(i),
+                state="State {}".format(i), npanxx="{}-{}".format(
+                    i, i), country=countries[i - 1], zipcode=i, org=cls.org)
+            for i in entity_count
+        ]
+
+        # create network facility relationships
+        cls.netfac = [
+            NetworkFacility.objects.create(
+                network=cls.net[i - 1], facility=cls.fac[i - 1], status="ok")
+            for i in entity_count
+        ]
+
+        # create ixlans
+        cls.ixlan = [
+            IXLan.objects.create(ix=cls.ix[i - 1], status="ok")
+            for i in entity_count
+        ]
+
+        # create netixlans
+        cls.netixlan = [
+            NetworkIXLan.objects.create(ixlan=cls.ixlan[i - 1],
+                                        network=cls.net[i - 1], asn=i, speed=0,
+                                        status="ok") for i in entity_count
+        ]
+
+    def expected_data(self, tag, fmt):
+        path = os.path.join(
+            os.path.dirname(__file__), "data", "export", "advancedsearch",
+            "{}.{}".format(tag, fmt))
+        with open(path, "r") as fh:
+            data = fh.read().rstrip()
+        return data
+
+    def test_export_net_json(self):
+        """ test json export of network search """
+        client = Client()
+        response = client.get(
+            "/export/advanced-search/net/json?name_search=Network")
+        self.assertEqual(
+            json.loads(response.content),
+            json.loads(self.expected_data("net", "json")))
+
+    def test_export_net_json_pretty(self):
+        """ test pretty json export of network search """
+        client = Client()
+        response = client.get(
+            "/export/advanced-search/net/json-pretty?name_search=Network")
+        self.assertEqual(response.content,
+                         self.expected_data("net", "jsonpretty"))
+
+    def test_export_net_csv(self):
+        """ test csv export of network search """
+        client = Client()
+        response = client.get(
+            "/export/advanced-search/net/csv?name_search=Network")
+        self.assertEqual(
+            response.content.replace("\r\n", "\n").rstrip(),
+            self.expected_data("net", "csv"))
+
+    def test_export_fac_json(self):
+        """ test json export of facility search """
+        client = Client()
+        response = client.get(
+            "/export/advanced-search/fac/json?name__contains=Facility")
+        self.assertEqual(
+            json.loads(response.content),
+            json.loads(self.expected_data("fac", "json")))
+
+    def test_export_fac_json_pretty(self):
+        """ test pretty json export of facility search """
+        client = Client()
+        response = client.get(
+            "/export/advanced-search/fac/json-pretty?name__contains=Facility")
+        self.assertEqual(response.content,
+                         self.expected_data("fac", "jsonpretty"))
+
+    def test_export_fac_csv(self):
+        """ test csv export of facility search """
+        client = Client()
+        response = client.get(
+            "/export/advanced-search/fac/csv?name__contains=Facility")
+        self.assertEqual(
+            response.content.replace("\r\n", "\n").rstrip(),
+            self.expected_data("fac", "csv"))
+
+    def test_export_ix_json(self):
+        """ test json export of exchange search """
+        client = Client()
+        response = client.get(
+            "/export/advanced-search/ix/json?name__contains=Exchange")
+        self.assertEqual(
+            json.loads(response.content),
+            json.loads(self.expected_data("ix", "json")))
+
+    def test_export_ix_json_pretty(self):
+        """ test pretty json export of exchange search """
+        client = Client()
+        response = client.get(
+            "/export/advanced-search/ix/json-pretty?name__contains=Exchange")
+        self.assertEqual(response.content,
+                         self.expected_data("ix", "jsonpretty"))
+
+    def test_export_ix_csv(self):
+        """ test csv export of exchange search """
+        client = Client()
+        response = client.get(
+            "/export/advanced-search/ix/csv?name__contains=Exchange")
+        self.assertEqual(
+            response.content.replace("\r\n", "\n").rstrip(),
+            self.expected_data("ix", "csv"))