mirror of
https://github.com/peeringdb/peeringdb.git
synced 2024-05-11 05:55:09 +00:00
148 lines
4.6 KiB
Python
148 lines
4.6 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
|
|
|
|
def test_undelete_ixlan_netixlan_dupe_same_ix(self):
|
|
ixlan_c = REFTAG_MAP["ixlan"].objects.create(ix=self.ix_a, status="ok")
|
|
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=ixlan_c,
|
|
status="ok",
|
|
ipaddr4=netixlan_a.ipaddr4,
|
|
network=self.net_a,
|
|
speed=100,
|
|
)
|
|
|
|
assert ixlan_c.netixlan_set_active.count() == 1
|
|
|
|
self._undelete(self.ixlan_a)
|
|
|
|
assert self.ixlan_a.status == "ok"
|
|
assert self.ixlan_a.netixlan_set_active.count() == 1
|
|
assert ixlan_c.netixlan_set_active.count() == 0
|
|
|
|
def test_undelete_ixlan_netixlan_dupe_same_ix_ipv6(self):
|
|
ixlan_c = REFTAG_MAP["ixlan"].objects.create(ix=self.ix_a, status="ok")
|
|
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=ixlan_c,
|
|
status="ok",
|
|
ipaddr6=netixlan_a.ipaddr6,
|
|
network=self.net_a,
|
|
speed=100,
|
|
)
|
|
|
|
assert ixlan_c.netixlan_set_active.count() == 1
|
|
|
|
self._undelete(self.ixlan_a)
|
|
|
|
assert self.ixlan_a.status == "ok"
|
|
assert self.ixlan_a.netixlan_set_active.count() == 1
|
|
assert ixlan_c.netixlan_set_active.count() == 0
|