# -*- 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