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)
64 lines
1.6 KiB
Python
64 lines
1.6 KiB
Python
import io
|
|
import datetime
|
|
import pytest
|
|
|
|
from .util import ClientCase, Group
|
|
|
|
from django.core.management import call_command
|
|
from django.conf import settings
|
|
from django.contrib.auth import get_user_model
|
|
|
|
from peeringdb_server.models import REFTAG_MAP, UTC
|
|
|
|
|
|
DATE_PAST = datetime.datetime(year=2019, month=11, day=1)
|
|
|
|
|
|
def setup_data():
|
|
|
|
call_command("pdb_generate_test_data", limit=3, commit=True)
|
|
|
|
date_past = DATE_PAST.replace(tzinfo=UTC())
|
|
|
|
# one object of each type moved to the past
|
|
|
|
for tag in ["fac", "net", "org", "ix"]:
|
|
for obj in REFTAG_MAP[tag].objects.all():
|
|
obj.created = obj.updated = date_past
|
|
obj.save()
|
|
break
|
|
|
|
# create users
|
|
|
|
User = get_user_model()
|
|
|
|
for i in range(1, 7):
|
|
User.objects.create_user(
|
|
"user_{}".format(i), "user_{}@localhost".format(i), "secret"
|
|
)
|
|
|
|
# move users 4, 5, 6 to the past
|
|
|
|
User.objects.filter(pk__in=[4, 5, 6]).update(created=date_past)
|
|
|
|
# verify all users except 1 and 4
|
|
|
|
for user in User.objects.exclude(pk__in=[1, 4]):
|
|
user.set_verified()
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_generate_for_past_date(db, data_stats_past):
|
|
output = io.StringIO()
|
|
setup_data()
|
|
call_command("pdb_stats", date=DATE_PAST.strftime("%Y%m%d"), stdout=output)
|
|
assert output.getvalue() == data_stats_past.txt
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_generate_for_current_date(db, data_stats_current):
|
|
output = io.StringIO()
|
|
setup_data()
|
|
call_command("pdb_stats", stdout=output)
|
|
assert data_stats_current.txt in output.getvalue()
|