mirror of
https://github.com/peeringdb/peeringdb.git
synced 2024-05-11 05:55:09 +00:00
93 lines
2.9 KiB
Python
93 lines
2.9 KiB
Python
# -*- 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
|