peeringdb-peeringdb/tests/test_geocode.py

import pytest
import json
import uuid
import re
import io

from django.test import Client, TestCase, RequestFactory
from django.contrib.auth.models import Group, AnonymousUser
from django.contrib.auth import get_user
from django.core.management import call_command

import django_namespace_perms as nsp

import peeringdb_server.models as models


class ViewTestCase(TestCase):
    @classmethod
    def setUpTestData(cls):

        # create organizations
        cls.organizations = {
            k: models.Organization.objects.create(
                name="Geocode Org %s" % k, status="ok"
            )
            for k in ["a", "b", "c", "d"]
        }

        # create facilities
        cls.facilities = {
            k: models.Facility.objects.create(
                name=f"Geocode Fac {k}",
                status="ok",
                org=cls.organizations[k],
                address1="Some street",
                address2=k,
                city="Chicago",
                country="US",
                state="IL",
                zipcode="1234",
                latitude=1.23,
                longitude=-1.23,
                geocode_status=True,
            )
            for k in ["a", "b", "c", "d"]
        }

    def test_base(self):
        self.assertEqual(
            self.facilities["a"].geocode_address, "Some street a, Chicago, IL 1234"
        )
        self.assertEqual(self.facilities["a"].geocode_coordinates, (1.23, -1.23))

    def test_change(self):
        self.assertEqual(self.facilities["b"].geocode_status, True)
        self.facilities["b"].address1 = "Another street b"
        self.facilities["b"].save()
        self.assertEqual(self.facilities["b"].geocode_status, False)
        self.assertEqual(self.facilities["c"].geocode_status, True)
        self.facilities["c"].lat = 4567.0
        self.facilities["c"].save()
        self.assertEqual(self.facilities["c"].geocode_status, True)
        self.assertEqual(self.facilities["d"].geocode_status, True)
        self.facilities["d"].website = "http://www.test.com"
        self.facilities["d"].save()
        self.assertEqual(self.facilities["d"].geocode_status, True)

    def test_command(self):
        self.assertEqual(self.facilities["a"].geocode_status, True)

        # change address to flag facility for geocoding

        self.facilities["a"].address1 = "Another street a"

        # test unicode output from command by adding special characters
        # to the new address

        self.facilities["a"].name = "sdílených služeb"
        self.facilities["a"].save()

        out = io.StringIO()
        call_command("pdb_geosync", "fac", limit=1, stdout=out)
        out = out.getvalue()

        assert "[fac 1/1 ID:1]" in out