peeringdb-peeringdb/tests/test_veriqueue.py

import json
import pytest

from django.test import TestCase
from django.contrib.auth.models import Group
from django.db import IntegrityError

import peeringdb_server.models as models

import reversion


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 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"[test]{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,
            )