Files
mayan-edms/mayan/apps/cabinets/tests/test_events.py
Roberto Rosario 0267c79b07 Backport test improvements
Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
2019-04-20 02:39:57 -04:00

39 lines
1.2 KiB
Python

from __future__ import unicode_literals
from actstream.models import Action
from mayan.apps.documents.tests.test_models import GenericDocumentTestCase
from ..events import (
event_cabinets_add_document, event_cabinets_remove_document
)
from .mixins import CabinetTestMixin
class CabinetsEventsTestCase(CabinetTestMixin, GenericDocumentTestCase):
def setUp(self):
super(CabinetsEventsTestCase, self).setUp()
self._create_test_cabinet()
def test_document_cabinet_add_event(self):
Action.objects.all().delete()
self.test_cabinet.add_document(document=self.test_document)
self.assertEqual(Action.objects.last().target, self.test_document)
self.assertEqual(
Action.objects.last().verb,
event_cabinets_add_document.id
)
def test_document_cabinet_remove_event(self):
self.test_cabinet.add_document(document=self.test_document)
Action.objects.all().delete()
self.test_cabinet.remove_document(document=self.test_document)
self.assertEqual(Action.objects.first().target, self.test_document)
self.assertEqual(
Action.objects.first().verb,
event_cabinets_remove_document.id
)