mirror of
https://github.com/peeringdb/peeringdb.git
synced 2024-05-11 05:55:09 +00:00
* stub in poetry for pipenv * re-add tester image * add pre-commit / formatting * fix ghactions * revert test data whitespace, exclude tests/data * revert ws * decruft, rm tox/pipenv * install dev packages for base image * add lgtm config to force to py3
162 lines
4.9 KiB
Python
162 lines
4.9 KiB
Python
import json
|
|
|
|
import pytest
|
|
import reversion
|
|
from django.conf import settings
|
|
from django.contrib.auth.models import Group
|
|
from django.db import IntegrityError
|
|
from django.test import TestCase
|
|
|
|
import peeringdb_server.models as models
|
|
|
|
|
|
class VeriQueueTests(TestCase):
|
|
"""
|
|
Test VerificationQueue creation and resolve
|
|
"""
|
|
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
"""
|
|
Test that verification queue items are created for all entities
|
|
for which it is enabled
|
|
"""
|
|
|
|
cls.guest_group = Group.objects.create(name="guest", id=settings.GUEST_GROUP_ID)
|
|
cls.user_group = Group.objects.create(name="user", id=settings.USER_GROUP_ID)
|
|
|
|
settings.USER_GROUP_ID = cls.user_group.id
|
|
settings.GUEST_GROUP_ID = cls.guest_group.id
|
|
|
|
cls.inst = {}
|
|
org = models.Organization.objects.create(name="Test", status="pending")
|
|
for model in models.QUEUE_ENABLED:
|
|
if model == models.Organization:
|
|
continue
|
|
if model == models.User:
|
|
cls.inst["user"] = model.objects.create_user(
|
|
"test", "test@localhost", "test"
|
|
)
|
|
cls.inst["user"].set_unverified()
|
|
else:
|
|
kwargs = {
|
|
"org": org,
|
|
"name": "Test %s" % model.handleref.tag,
|
|
"status": "pending",
|
|
}
|
|
if model.handleref.tag == "net":
|
|
kwargs.update(asn=1)
|
|
cls.inst[model.handleref.tag] = model.objects.create(**kwargs)
|
|
|
|
def test_get_for_entity(self):
|
|
"""
|
|
Test VerificationQueueItem.get_for_entity
|
|
"""
|
|
|
|
# test verification queue items were created for all queue enabled
|
|
# entities
|
|
for k, v in list(self.inst.items()):
|
|
vqi = models.VerificationQueueItem.get_for_entity(v)
|
|
self.assertEqual(vqi.item, v)
|
|
|
|
def test_deskpro_tickets(self):
|
|
"""
|
|
Test that tickets were created for the facility, ix and network
|
|
"""
|
|
user = self.inst["user"]
|
|
qs = models.DeskProTicket.objects
|
|
|
|
for tag in ["fac", "net", "ix"]:
|
|
inst = self.inst[tag]
|
|
vqi = models.VerificationQueueItem.get_for_entity(inst)
|
|
vqi.user = user
|
|
vqi.save()
|
|
self.assertEqual(
|
|
qs.filter(
|
|
subject=f"[{settings.RELEASE_ENV}] {vqi.content_type} - {inst}"
|
|
).exists(),
|
|
True,
|
|
)
|
|
|
|
def test_approve(self):
|
|
"""
|
|
Test VerificationqueueItem.approve
|
|
"""
|
|
|
|
ix = self.inst.get("ix")
|
|
vqi = models.VerificationQueueItem.get_for_entity(ix)
|
|
|
|
vqi.approve()
|
|
|
|
# after approval ix should be status 'ok'
|
|
ix.refresh_from_db()
|
|
self.assertEqual(ix.status, "ok")
|
|
|
|
# check that the status in the archive is correct (#558)
|
|
|
|
version = (
|
|
reversion.models.Version.objects.get_for_object(ix)
|
|
.order_by("-revision_id")
|
|
.first()
|
|
)
|
|
self.assertEqual(
|
|
json.loads(version.serialized_data)[0]["fields"]["status"], "ok"
|
|
)
|
|
|
|
# after approval vqi should no longer exist
|
|
with pytest.raises(models.VerificationQueueItem.DoesNotExist):
|
|
vqi.refresh_from_db()
|
|
|
|
def test_user_approve(self):
|
|
"""
|
|
Test VerificationqueueItem.approve when approving users
|
|
"""
|
|
|
|
# test that approving a user also moves them in the correct usergroup
|
|
|
|
user = self.inst.get("user")
|
|
vqi = models.VerificationQueueItem.get_for_entity(user)
|
|
|
|
vqi.approve()
|
|
|
|
# after approval user should be status 'ok'
|
|
|
|
user.refresh_from_db()
|
|
self.assertEqual(user.status, "ok")
|
|
|
|
# after approval user should be in 'users' group
|
|
self.assertEqual(user.groups.filter(name="user").exists(), True)
|
|
self.assertEqual(user.groups.filter(name="guest").exists(), False)
|
|
|
|
def test_deny(self):
|
|
"""
|
|
Test VerificationqueueItem.deny
|
|
"""
|
|
|
|
fac = self.inst.get("fac")
|
|
vqi = models.VerificationQueueItem.get_for_entity(fac)
|
|
|
|
vqi.deny()
|
|
|
|
# after denial fac should no longer exist
|
|
with pytest.raises(models.Facility.DoesNotExist):
|
|
fac.refresh_from_db()
|
|
|
|
# after denial vqi should no longer exist
|
|
with pytest.raises(models.VerificationQueueItem.DoesNotExist):
|
|
vqi.refresh_from_db()
|
|
|
|
def test_unique(self):
|
|
"""
|
|
Test that only one verification queue item can exist for an entity
|
|
"""
|
|
|
|
fac = self.inst.get("fac")
|
|
vqi = models.VerificationQueueItem.get_for_entity(fac)
|
|
|
|
with pytest.raises(IntegrityError):
|
|
models.VerificationQueueItem.objects.create(
|
|
content_type=models.ContentType.objects.get_for_model(type(fac)),
|
|
object_id=fac.id,
|
|
)
|