mirror of
https://github.com/peeringdb/peeringdb.git
synced 2024-05-11 05:55:09 +00:00
5eb925e319
* fix next redirect when using U2F 2FA auth (#1191) * Added self identifier to API * fix migrations hierarchy after merging in previous support branch * campus object Co-authored-by: Stefan Pratter <stefan@20c.com> * fix out of bound error message add city / country to campus view * fix tests * relock poetry * linting * linting * fix docs regen * regen docs * linting * refactor self entity view to support carrier and campus object types and also make it easier to support additional object types in the future * remove debug message --------- Co-authored-by: Gajanan Patil <dipaksavaliya.python@gmail.com>
145 lines
4.4 KiB
Python
145 lines
4.4 KiB
Python
import json
|
|
import re
|
|
import uuid
|
|
|
|
import pytest
|
|
from django.contrib.auth import get_user
|
|
from django.contrib.auth.models import AnonymousUser, Group
|
|
from django.test import Client, RequestFactory, TestCase
|
|
|
|
import peeringdb_server.models as models
|
|
import peeringdb_server.views as views
|
|
|
|
|
|
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 = {
|
|
f"{k}": models.Organization.objects.create(
|
|
name="Sponsor Org %s" % k, status="ok"
|
|
)
|
|
for k in range(1, 7)
|
|
}
|
|
|
|
# create sponsorships
|
|
|
|
cls.sponsorships = {
|
|
"1": models.Sponsorship.objects.create(level=1),
|
|
"2": models.Sponsorship.objects.create(level=1),
|
|
"3": models.Sponsorship.objects.create(level=2),
|
|
"4": models.Sponsorship.objects.create(level=1),
|
|
"5_and_6": models.Sponsorship.objects.create(level=3),
|
|
}
|
|
|
|
# org sponsorship with logo and url set lvl1
|
|
|
|
models.SponsorshipOrganization.objects.create(
|
|
sponsorship=cls.sponsorships["1"],
|
|
org=cls.organizations["1"],
|
|
logo="fake.png",
|
|
url="org-1.com",
|
|
)
|
|
|
|
# org sponsorship with logo set lvl1
|
|
|
|
models.SponsorshipOrganization.objects.create(
|
|
sponsorship=cls.sponsorships["2"],
|
|
org=cls.organizations["2"],
|
|
logo="fake.png",
|
|
)
|
|
|
|
# org sponsorship with logo set lvl2
|
|
|
|
models.SponsorshipOrganization.objects.create(
|
|
sponsorship=cls.sponsorships["3"],
|
|
org=cls.organizations["3"],
|
|
logo="fake.png",
|
|
)
|
|
|
|
# org sponsorship without logo or url set lvl1
|
|
|
|
models.SponsorshipOrganization.objects.create(
|
|
sponsorship=cls.sponsorships["4"],
|
|
org=cls.organizations["4"],
|
|
)
|
|
|
|
# two orgs in one sponsorship
|
|
|
|
models.SponsorshipOrganization.objects.create(
|
|
sponsorship=cls.sponsorships["5_and_6"],
|
|
org=cls.organizations["5"],
|
|
logo="fake.png",
|
|
url="org-5.com",
|
|
)
|
|
|
|
models.SponsorshipOrganization.objects.create(
|
|
sponsorship=cls.sponsorships["5_and_6"],
|
|
org=cls.organizations["6"],
|
|
logo="fake.png",
|
|
url="org-6.com",
|
|
)
|
|
|
|
def setUp(self):
|
|
self.factory = RequestFactory()
|
|
|
|
def test_data_view(self):
|
|
c = Client()
|
|
resp = c.get("/data/sponsors", follow=True)
|
|
self.assertEqual(resp.status_code, 200)
|
|
expected_sponsorships = [
|
|
"silver",
|
|
"silver",
|
|
"gold",
|
|
"silver",
|
|
"platinum",
|
|
"platinum",
|
|
]
|
|
print(resp.json())
|
|
output_sponsorships = [
|
|
spon["name"] for spon in resp.json()["sponsors"].values()
|
|
]
|
|
self.assertEqual(output_sponsorships, expected_sponsorships)
|
|
|
|
def test_view(self):
|
|
c = Client()
|
|
resp = c.get("/sponsors", follow=True)
|
|
self.assertEqual(resp.status_code, 200)
|
|
|
|
# make sure orgs 1,2,3,5 and 6 exists in the sponsor page
|
|
assert self.organizations["1"].name in resp.content.decode()
|
|
assert self.organizations["2"].name in resp.content.decode()
|
|
assert self.organizations["3"].name in resp.content.decode()
|
|
assert self.organizations["5"].name in resp.content.decode()
|
|
assert self.organizations["6"].name in resp.content.decode()
|
|
|
|
# make sure org 4 does not exist in the sponsors page
|
|
assert self.organizations["4"].name not in resp.content.decode()
|
|
|
|
# makre sure order is randomized with each view
|
|
i = 0
|
|
rgx = re.compile(r'fake.png" alt="([^"]+)"')
|
|
a = re.findall(rgx, resp.content.decode())
|
|
while i < 100:
|
|
resp = c.get("/sponsors", follow=True)
|
|
b = re.findall(rgx, resp.content.decode())
|
|
self.assertEqual(len(a), len(b))
|
|
if b != a:
|
|
break
|
|
i += 1
|
|
self.assertLess(i, 99)
|