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)
71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
from django.core.management import call_command
|
|
from django.contrib.auth import get_user_model
|
|
from django.conf import settings
|
|
|
|
from peeringdb_server.models import REFTAG_MAP, UTC
|
|
|
|
from .util import ClientCase, Group
|
|
|
|
|
|
class TestWipe(ClientCase):
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
super(TestWipe, cls).setUpTestData()
|
|
cls.superuser = get_user_model().objects.create_user(
|
|
"superuser", "superuser@localhost", "superuser", is_superuser=True
|
|
)
|
|
call_command("pdb_generate_test_data", limit=2, commit=True)
|
|
|
|
def test_run(self):
|
|
# double check that data is setup correctly before
|
|
# running pdbwipe
|
|
|
|
assert get_user_model().objects.all().count() == 2
|
|
|
|
for reftag, cls in list(REFTAG_MAP.items()):
|
|
assert cls.objects.all().count() > 1
|
|
|
|
assert Group.objects.all().count() > 2
|
|
|
|
# run pdb_wipe
|
|
settings.TUTORIAL_MODE = True
|
|
call_command("pdb_wipe", commit=True)
|
|
settings.TUTORIAL_MODE = False
|
|
|
|
# all entities should be gone
|
|
for reftag, cls in list(REFTAG_MAP.items()):
|
|
assert cls.objects.all().count() == 0
|
|
|
|
# only the user and guest groups should be left
|
|
assert Group.objects.all().count() == 2
|
|
|
|
# only the superuser should be left
|
|
assert get_user_model().objects.all().count() == 1
|
|
assert get_user_model().objects.first().is_superuser == True
|
|
|
|
def test_run_with_sync(self):
|
|
"""
|
|
Test running `pdb_wipe` and sync data from
|
|
test.peeringdb.com
|
|
"""
|
|
|
|
dates = {}
|
|
|
|
for reftag, cls in REFTAG_MAP.items():
|
|
assert cls.objects.all().count() > 1
|
|
dates[reftag] = cls.objects.all().first().created.replace(tzinfo=UTC())
|
|
|
|
settings.TUTORIAL_MODE = True
|
|
call_command(
|
|
"pdb_wipe",
|
|
commit=True,
|
|
load_data=True,
|
|
load_data_url="https://test.peeringdb.com/api",
|
|
)
|
|
settings.TUTORIAL_MODE = False
|
|
|
|
for reftag, cls in REFTAG_MAP.items():
|
|
created = cls.objects.all().first().created.replace(tzinfo=UTC())
|
|
assert created != dates[reftag]
|
|
assert cls.objects.all().count() > 1
|