mirror of
https://github.com/peeringdb/peeringdb.git
synced 2024-05-11 05:55:09 +00:00
5147028bee
* stub in poetry for pipenv * re-add tester image * add pre-commit / formatting * fix ghactions * revert test data whitespace, exclude tests/data * revert ws * decruft, rm tox/pipenv * install dev packages for base image * add lgtm config to force to py3
55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
import io
|
|
import sys
|
|
|
|
from django.core.management import call_command
|
|
|
|
from peeringdb_server.models import REFTAG_MAP
|
|
|
|
from .util import ClientCase
|
|
|
|
|
|
class TestWhois(ClientCase):
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
super().setUpTestData()
|
|
cls.org = REFTAG_MAP["org"].objects.create(name="Test org", status="ok")
|
|
cls.net = REFTAG_MAP["net"].objects.create(
|
|
name="Test net", status="ok", asn=63311, org=cls.org
|
|
)
|
|
cls.pocs = []
|
|
for visibility in ["Private", "Users", "Public"]:
|
|
cls.pocs.append(
|
|
REFTAG_MAP["poc"].objects.create(
|
|
network=cls.net,
|
|
status="ok",
|
|
role="Abuse",
|
|
name=f"POC-{visibility}",
|
|
email=f"{visibility}@localhost",
|
|
visible=visibility,
|
|
)
|
|
)
|
|
|
|
def test_whois_perms(self):
|
|
"""
|
|
test that anonymous user permissions are applied
|
|
to whois output - any pocs other than public ones should
|
|
be excluded
|
|
"""
|
|
|
|
# whois does not go to the command's stdout so we need to
|
|
# capture the output through sys.stdout
|
|
out = io.StringIO()
|
|
oldout = sys.stdout
|
|
sys.stdout = out
|
|
|
|
call_command("pdb_whois", "as63311")
|
|
|
|
# restore sys.stdout
|
|
sys.stdout = oldout
|
|
|
|
out = out.getvalue()
|
|
|
|
assert "POC-Private" not in out
|
|
assert "POC-Users" not in out
|
|
assert "POC-Public" in out
|