test_application_publisher_assigned_notify.py

from portality import models
from portality import constants
from portality.bll import exceptions
from doajtest.helpers import DoajTestCase
from portality.events.consumers.application_publisher_assigned_notify import ApplicationPublisherAssignedNotify
from doajtest.fixtures import ApplicationFixtureFactory
import time


class TestApplicationPublisherAssignedNotify(DoajTestCase):
    def setUp(self):
        super(TestApplicationPublisherAssignedNotify, self).setUp()

    def tearDown(self):
        super(TestApplicationPublisherAssignedNotify, self).tearDown()

    def test_should_consume(self):
        source = ApplicationFixtureFactory.make_application_source()
        source["admin"]["application_type"] = constants.APPLICATION_TYPE_NEW_APPLICATION

        event = models.Event(constants.EVENT_APPLICATION_ASSED_ASSIGNED, context={"application" : source, "old_editor": "", "new_editor": source["admin"]["editor"]})
        assert ApplicationPublisherAssignedNotify.should_consume(event)

        event = models.Event(constants.EVENT_APPLICATION_ASSED_ASSIGNED,
                             context={"application": source, "old_editor": "editor"})
        assert not ApplicationPublisherAssignedNotify.should_consume(event)

        event = models.Event("test:event", context={"application" : source})
        assert not ApplicationPublisherAssignedNotify.should_consume(event)

        event = models.Event(constants.EVENT_APPLICATION_ASSED_ASSIGNED)
        assert not ApplicationPublisherAssignedNotify.should_consume(event)

    def test_consume_success(self):
        self._make_and_push_test_context("/")

        source = ApplicationFixtureFactory.make_application_source()
        app = models.Application(**source)
        # app.save()

        acc = models.Account()
        acc.set_id(app.owner)
        acc.set_email("test@example.com")
        acc.save(blocking=True)

        event = models.Event(constants.EVENT_APPLICATION_ASSED_ASSIGNED, context={"application" : app.data, "old_editor": "", "new_editor": app.editor})
        ApplicationPublisherAssignedNotify.consume(event)

        time.sleep(1)
        ns = models.Notification.all()
        assert len(ns) == 1

        n = ns[0]
        assert n.who == app.owner
        assert n.created_by == ApplicationPublisherAssignedNotify.ID
        assert n.classification == constants.NOTIFICATION_CLASSIFICATION_ASSIGN
        assert n.long is not None
        assert n.short is not None
        assert n.action is None
        assert not n.is_seen()

    def test_consume_fail(self):
        event = models.Event(constants.EVENT_APPLICATION_ASSED_ASSIGNED, context={"application": {"key" : "value"}})
        with self.assertRaises(exceptions.NoSuchObjectException):
            ApplicationPublisherAssignedNotify.consume(event)

        source = ApplicationFixtureFactory.make_application_source()
        del source["admin"]["owner"]
        app = models.Application(**source)
        # app.save(blocking=True)

        event = models.Event(constants.EVENT_APPLICATION_ASSED_ASSIGNED, context={"application": app.data})
        with self.assertRaises(exceptions.NoSuchPropertyException):
            ApplicationPublisherAssignedNotify.consume(event)