1
0
mirror of https://github.com/peeringdb/peeringdb.git synced 2024-05-11 05:55:09 +00:00
Files
peeringdb-peeringdb/tests/test_data_views.py
Matt Griswold 636f37eb12 Support 202108 (#1051)
* add OPERATIONS_EMAIL setting

* fixes #1019: redundant saves to deleted netixlans during ix-f import

* private pocs are no longer valid (#944)

* poetry relock (handleref update)

* fixes #1032: __id api filter not working

* Additional self-selection fields for Facilities #800

* advanced search fields for available voltage, property and diverse serving substations (#1016)

* When network sets netixlan speed to 1200000 only 1T is shown instead of 1.2T ... sometimes #500

* add search-data

* comment out mount points for api-cache, search-data, django-peeringdb

* poetry relock (django-peeringdb 2.9.0)

* linting

Co-authored-by: Stefan Pratter <stefan@20c.com>
2021-09-14 08:59:09 -05:00

116 lines
2.9 KiB
Python

import pytest
from django.core.management import call_command
from django.test import Client
from peeringdb_server.models import (
Organization,
Sponsorship,
SponsorshipOrganization,
User,
)
@pytest.mark.django_db
@pytest.mark.parametrize(
"name",
[
"RATIOS",
"RATIOS_TRUNC",
"RATIOS_ADVS",
"TRAFFIC",
"SCOPES",
"SCOPES_TRUNC",
"SCOPES_ADVS",
"NET_TYPES",
"NET_TYPES_TRUNC",
"NET_TYPES_ADVS",
"POLICY_GENERAL",
"POLICY_LOCATIONS",
"POLICY_CONTRACTS",
"REGIONS",
"POC_ROLES",
"POC_VISIBILITY",
"MEDIA",
"PROTOCOLS",
"ORG_GROUPS",
"BOOL_CHOICE_STR",
"VISIBILITY",
],
)
def test_enum(name):
client = Client()
response = client.get(f"/data/enum/{name}")
assert response.status_code == 200
assert len(response.json()[f"enum/{name}"]) > 0
@pytest.mark.django_db
@pytest.mark.parametrize(
"name,user,count,status",
[
("countries_b", "user", 1, 200),
("countries", "user", 1, 200),
("sponsors", "user", 1, 200),
("facilities", "user", 4, 200),
("organizations", "user", 0, 403),
("organizations", "su", 1, 200),
],
)
def test_generic(name, user, count, status):
call_command("pdb_generate_test_data", limit=3, commit=True)
org = Organization.objects.first()
sponsorship = Sponsorship.objects.create(level=1)
SponsorshipOrganization.objects.create(
org=org, sponsorship=sponsorship, logo="fake.png"
)
_user = User.objects.create_user(
username="user_a", password="user_a", email="user_a@localhost"
)
if user == "su":
_user.is_superuser = True
_user.is_staff = True
_user.save()
org.usergroup.user_set.add(_user)
client = Client()
client.login(username="user_a", password="user_a")
response = client.get(f"/data/{name}")
assert response.status_code == status
if status == 200:
assert len(response.json()[name]) >= count
@pytest.mark.django_db
def test_my_organizations():
call_command("pdb_generate_test_data", limit=3, commit=True)
org = Organization.objects.first()
_user = User.objects.create_user(
username="user_a", password="user_a", email="user_a@localhost"
)
org.usergroup.user_set.add(_user)
client = Client()
client.login(username="user_a", password="user_a")
response = client.get("/data/my_organizations")
assert response.status_code == 200
assert len(response.json()["my_organizations"]) == 1
@pytest.mark.django_db
def test_my_organizations_anon():
call_command("pdb_generate_test_data", limit=3, commit=True)
client = Client()
response = client.get("/data/my_organizations")
assert response.status_code == 200
assert len(response.json()["my_organizations"]) == 0