mirror of
https://github.com/peeringdb/peeringdb.git
synced 2024-05-11 05:55:09 +00:00
81 lines
2.8 KiB
Python
81 lines
2.8 KiB
Python
|
import pytest
|
||
|
|
import json
|
||
|
|
import uuid
|
||
|
|
import re
|
||
|
|
|
||
|
|
from django.test import Client, TestCase, RequestFactory
|
||
|
|
from django.contrib.auth.models import Group, AnonymousUser
|
||
|
|
from django.contrib.auth import get_user
|
||
|
|
import django_namespace_perms as nsp
|
||
|
|
|
||
|
|
import peeringdb_server.views as views
|
||
|
|
import peeringdb_server.models as models
|
||
|
|
|
||
|
|
|
||
|
|
class ViewTestCase(TestCase):
|
||
|
|
|
||
|
|
entities = ["ix", "net", "fac"]
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def setUpTestData(cls):
|
||
|
|
# create user and guest group
|
||
|
|
|
||
|
|
guest_group = Group.objects.create(name="guest")
|
||
|
|
user_group = Group.objects.create(name="user")
|
||
|
|
|
||
|
|
cls.guest_user = models.User.objects.create_user(
|
||
|
|
"guest", "guest@localhost", "guest")
|
||
|
|
cls.guest_user.set_password("guest")
|
||
|
|
guest_group.user_set.add(cls.guest_user)
|
||
|
|
|
||
|
|
# create organizations
|
||
|
|
cls.organizations = dict((k,
|
||
|
|
models.Organization.objects.create(
|
||
|
|
name="Partner Org %s" % k, status="ok"))
|
||
|
|
for k in ["a", "b", "c", "d"])
|
||
|
|
|
||
|
|
# create partnerships
|
||
|
|
cls.partnerships = {
|
||
|
|
"a": models.Partnership.objects.create(
|
||
|
|
org=cls.organizations.get("a"), logo="fake.png",
|
||
|
|
url="org-a.com", level=1),
|
||
|
|
"b": models.Partnership.objects.create(
|
||
|
|
org=cls.organizations.get("b"), logo="fake.png", level=1),
|
||
|
|
"c": models.Partnership.objects.create(
|
||
|
|
org=cls.organizations.get("c"), logo="fake.png", level=2),
|
||
|
|
"d": models.Partnership.objects.create(
|
||
|
|
org=cls.organizations.get("d"), level=1)
|
||
|
|
}
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self.factory = RequestFactory()
|
||
|
|
|
||
|
|
def test_view(self):
|
||
|
|
c = Client()
|
||
|
|
resp = c.get("/partners", follow=True)
|
||
|
|
self.assertEqual(resp.status_code, 200)
|
||
|
|
|
||
|
|
#make sure org a,b and c exist in the partners page
|
||
|
|
self.assertGreater(resp.content.find(self.organizations["a"].name), -1)
|
||
|
|
self.assertGreater(resp.content.find(self.organizations["b"].name), -1)
|
||
|
|
self.assertGreater(resp.content.find(self.organizations["c"].name), -1)
|
||
|
|
|
||
|
|
#make sure org d does not exist in the partners page
|
||
|
|
self.assertEqual(resp.content.find(self.organizations["d"].name), -1)
|
||
|
|
|
||
|
|
#make sure partnership a url exists in the partners page
|
||
|
|
self.assertGreater(resp.content.find(self.partnerships["a"].url), -1)
|
||
|
|
|
||
|
|
#makre sure order is randomized with each view
|
||
|
|
i = 0
|
||
|
|
rgx = re.compile("fake.png\" alt=\"([^\"]+)\"")
|
||
|
|
a = re.findall(rgx, resp.content)
|
||
|
|
while i < 100:
|
||
|
|
resp = c.get("/partners", follow=True)
|
||
|
|
b = re.findall(rgx, resp.content)
|
||
|
|
self.assertEqual(len(a), len(b))
|
||
|
|
if b != a:
|
||
|
|
break
|
||
|
|
i += 1
|
||
|
|
self.assertLess(i, 99)
|