mirror of
https://github.com/peeringdb/peeringdb.git
synced 2024-05-11 05:55:09 +00:00
ba6f9b6432
* use new peeringdb client (1.0.0) for pdb_load_data sync (#599) * drop django-mobi for lack of py3/dj2 support (#492) remove django-forms-bootstrap for lack of py3/dj2 support (#492) * black formatted * django2.2 and py3 upgrade (#492) * drop ixlans (#21) ui and api changes * drop local_asn (#168) * org search (#193) * phone number validation (#50) * implement help text tooltips (#228) * Mark own ASN as transit-free (#394) * py3 fix for `pdb_migrate_ixlans` command when writing migration report * pdb_migrate_ixlans: properly handle py3 Runtime error if ixlan dict changes during iteration * set rest DEFAULT_SCHEMA_CLASS to coreapi to fix swagger apidocs fix migration 0027 missing from facsimile manifest * fix swagger doc strings * fix tests that were broken from api doc fixes * fix UniqueFieldValidator for netixlan ipaddress validation that broke during django/drf upgrade * fix org merge tool layout issues * travis config * update pipfile and lock * black formatting * update travis dist * beta mode banner (#411) * add beta banner template (#411) * automatically scheduled sync may not always be on, add a flag that lets us reflect that state in the beta banner message clean up beta banner implementation (#411) * add tests for beta banner (#411)
53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
from peeringdb_server.models import REFTAG_MAP
|
|
from django.core.management import call_command
|
|
from .util import ClientCase
|
|
|
|
import io
|
|
import sys
|
|
|
|
|
|
class TestWhois(ClientCase):
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
super(TestWhois, cls).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="POC-{}".format(visibility),
|
|
email="{}@localhost".format(visibility),
|
|
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
|