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 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 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(self):
# double check that data is setup correctly before
# running pdbwipe
assert get_user_model().objects.all().count() == 2
for reftag, cls in 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 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
def test_run(self):
call_command("pdb_generate_test_data", limit=2, commit=True)
for reftag, cls in list(REFTAG_MAP.items()):
self.assertGreater(cls.objects.count(), 0)
for instance in cls.objects.all():
instance.full_clean()
def test_run(self):
call_command("pdb_generate_test_data", limit=2, commit=True)
for reftag, cls in REFTAG_MAP.items():
self.assertGreater(cls.objects.count(), 0)
for instance in cls.objects.all():
instance.full_clean()