16d8fb9fea
Update API code to use viewsets. Update links and URLs to use keyword arguments. Signed-off-by: Roberto Rosario <Roberto.Rosario@mayan-edms.com>
42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from ..models import Message
|
|
|
|
from .literals import (
|
|
TEST_MESSAGE_LABEL, TEST_MESSAGE_LABEL_EDITED, TEST_MESSAGE_TEXT,
|
|
TEST_MESSAGE_TEXT_EDITED
|
|
)
|
|
|
|
|
|
class MOTDTestMixin(object):
|
|
def _create_test_message(self):
|
|
self.test_message = Message.objects.create(
|
|
label=TEST_MESSAGE_LABEL, message=TEST_MESSAGE_TEXT
|
|
)
|
|
|
|
def _request_message_create_view(self):
|
|
return self.post(
|
|
viewname='motd:message_create', data={
|
|
'label': TEST_MESSAGE_LABEL, 'message': TEST_MESSAGE_TEXT
|
|
}
|
|
)
|
|
|
|
def _request_message_delete_view(self):
|
|
return self.post(
|
|
viewname='motd:message_delete',
|
|
kwargs={'message_id': self.test_message.pk}
|
|
)
|
|
|
|
def _request_message_edit_view(self):
|
|
return self.post(
|
|
viewname='motd:message_edit',
|
|
kwargs={'message_id': self.test_message.pk},
|
|
data={
|
|
'label': TEST_MESSAGE_LABEL_EDITED,
|
|
'message': TEST_MESSAGE_TEXT_EDITED
|
|
}
|
|
)
|
|
|
|
def _request_message_list_view(self):
|
|
return self.get(viewname='motd:message_list')
|