mirror of
https://github.com/peeringdb/peeringdb.git
synced 2024-05-11 05:55:09 +00:00
f575278455
* Version History with revert and rollback functionality (#531) * fix twentyc.rpc requirement to allow >= 0.3.5
127 lines
3.8 KiB
Python
127 lines
3.8 KiB
Python
import pytest
|
|
|
|
from django.test import TestCase
|
|
from django.contrib.auth.models import Group
|
|
|
|
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
|
|
"""
|
|
|
|
guest_group = Group.objects.create(name="guest")
|
|
user_group = Group.objects.create(name="user")
|
|
|
|
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 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=u"[test]{} - {}".format(
|
|
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")
|
|
|
|
# after approval vqi should no longer exist
|
|
with self.assertRaises(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 self.assertRaises(models.Facility.DoesNotExist):
|
|
fac.refresh_from_db()
|
|
|
|
# after denial vqi should no longer exist
|
|
with self.assertRaises(models.VerificationQueueItem.DoesNotExist):
|
|
vqi.refresh_from_db()
|