1
0
mirror of https://github.com/peeringdb/peeringdb.git synced 2024-05-11 05:55:09 +00:00
Files
peeringdb-peeringdb/tests/test_undelete.py
Matt Griswold ba6f9b6432 Qu1003 (#621)
* 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)
2020-01-08 13:29:58 -06:00

106 lines
3.1 KiB
Python

import datetime
from .util import ClientCase, Group
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
from peeringdb_server.management.commands.pdb_undelete import Command
class TestUndelete(ClientCase):
@classmethod
def setUpTestData(cls):
super(TestUndelete, cls).setUpTestData()
call_command("pdb_generate_test_data", limit=2, commit=True)
cls.org_a = REFTAG_MAP["org"].objects.get(id=1)
cls.org_b = REFTAG_MAP["org"].objects.get(id=2)
cls.net_a = cls.org_a.net_set.first()
cls.ix_a = cls.org_a.ix_set.first()
cls.ix_b = cls.org_b.ix_set.first()
cls.fac_a = cls.org_a.fac_set.first()
cls.fac_b = cls.org_b.fac_set.first()
cls.ixlan_a = cls.ix_a.ixlan_set.first()
cls.ixlan_b = cls.ix_b.ixlan_set.first()
cls.date = datetime.date
@property
def _command(self):
command = Command()
command.commit = True
return command
def _undelete(self, obj):
command = self._command
command.date = obj.updated
command.undelete(obj.HandleRef.tag, obj.id)
obj.refresh_from_db()
def test_undelete_ixlan(self):
assert self.ixlan_a.netixlan_set_active.count() == 1
assert self.ixlan_a.ixpfx_set_active.count() == 2
self.ixlan_a.delete()
assert self.ixlan_a.status == "deleted"
self._undelete(self.ixlan_a)
assert self.ixlan_a.status == "ok"
assert self.ixlan_a.netixlan_set_active.count() == 1
assert self.ixlan_a.ixpfx_set_active.count() == 2
def test_undelete_fac(self):
assert self.fac_a.netfac_set_active.count() == 1
assert self.fac_a.ixfac_set_active.count() == 1
self.fac_a.delete()
assert self.fac_a.status == "deleted"
self._undelete(self.fac_a)
assert self.fac_a.status == "ok"
assert self.fac_a.netfac_set_active.count() == 1
assert self.fac_a.ixfac_set_active.count() == 1
def test_undelete_ixlan_netixlan_dupe_other(self):
netixlan_a = self.ixlan_a.netixlan_set.first()
self.ixlan_a.delete()
netixlan_c = REFTAG_MAP["netixlan"].objects.create(
asn=self.net_a.asn,
ixlan=self.ixlan_b,
status="ok",
ipaddr4=netixlan_a.ipaddr4,
network=self.net_a,
speed=100,
)
self._undelete(self.ixlan_a)
assert self.ixlan_a.status == "ok"
assert self.ixlan_a.netixlan_set_active.count() == 0
def test_undelete_ixlan_netixlan_dupe_other_ipv6(self):
netixlan_a = self.ixlan_a.netixlan_set.first()
self.ixlan_a.delete()
netixlan_c = REFTAG_MAP["netixlan"].objects.create(
asn=self.net_a.asn,
ixlan=self.ixlan_b,
status="ok",
ipaddr6=netixlan_a.ipaddr6,
network=self.net_a,
speed=100,
)
self._undelete(self.ixlan_a)
assert self.ixlan_a.status == "ok"
assert self.ixlan_a.netixlan_set_active.count() == 0