peeringdb-peeringdb/tests/test_geocode.py

# -*- coding: utf-8 -*-
import pytest
import json
import uuid
import re
import StringIO

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 = dict((k,
                                  models.Organization.objects.create(
                                      name="Geocode Org %s" % k, status="ok"))
                                 for k in ["a", "b", "c", "d"])

        # create facilities
        cls.facilities = dict(
            (k,
             models.Facility.objects.create(
                 name=u"Geocode Fac {}".format(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,
                         u"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 = StringIO.StringIO()
        call_command("pdb_geosync", "fac", limit=1, stdout=out)
        out = out.getvalue()

        assert out.find("[fac 1/1 ID:1]") > -1