test_application_publisher_in_progress_notify.py

from portality import models
from portality import constants
from portality.bll import exceptions
from doajtest.helpers import DoajTestCase
from doajtest.fixtures import ApplicationFixtureFactory
import time

from portality.events.consumers.application_publisher_inprogress_notify import ApplicationPublisherInprogressNotify


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

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

    def test_should_consume(self):
        source = ApplicationFixtureFactory.make_application_source()

        event = models.Event(constants.EVENT_APPLICATION_STATUS, context={"application": {}, "old_status": "pending", "new_status": "in progress"})
        assert ApplicationPublisherInprogressNotify.should_consume(event)

        event = models.Event(constants.EVENT_APPLICATION_STATUS,
                             context={"application": {}, "old_status": "in progress", "new_status": "in progress"})
        assert not ApplicationPublisherInprogressNotify.should_consume(event)

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

        event = models.Event(constants.EVENT_APPLICATION_STATUS)
        assert not ApplicationPublisherInprogressNotify.should_consume(event)

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

        acc = models.Account()
        acc.set_id("publisher")
        acc.set_email("test@example.com")
        acc.save()

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

        event = models.Event(constants.EVENT_APPLICATION_STATUS,
                             context={"application": app.data, "old_status": "pending",
                                      "new_status": "in progress"})

        # event = models.Event(constants.EVENT_APPLICATION_STATUS, context={"application": "abcdefghijk", "old_status": "in progress", "new_status": "revisions_required"})
        ApplicationPublisherInprogressNotify.consume(event)

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

        n = ns[0]
        assert n.who == "publisher", "Expected: {}, Received: {}".format("publisher", n.who)
        assert n.created_by == ApplicationPublisherInprogressNotify.ID, "Expected: {}, Received: {}".format(ApplicationPublisherInprogressNotify.ID, n.created_by)
        assert n.classification == constants.NOTIFICATION_CLASSIFICATION_STATUS_CHANGE, "Expected: {}, Received: {}".format(constants.NOTIFICATION_CLASSIFICATION_STATUS_CHANGE, n.classification)
        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": {"dummy" : "data"}})
        with self.assertRaises(exceptions.NoSuchObjectException):
            ApplicationPublisherInprogressNotify.consume(event)