Modernize MOTD app

Update API code to use viewsets. Update links and URLs to use
keyword arguments.

Signed-off-by: Roberto Rosario <Roberto.Rosario@mayan-edms.com>
This commit is contained in:
Roberto Rosario
2019-01-19 01:00:58 -04:00
parent 9ed93b54af
commit 16d8fb9fea
13 changed files with 281 additions and 120 deletions
+11 -13
View File
@@ -7,14 +7,12 @@ from django.utils import timezone
from ..models import Message
from .literals import TEST_LABEL, TEST_MESSAGE
from .mixins import MOTDTestMixin
class MOTDTestCase(TestCase):
class MOTDTestCase(MOTDTestMixin, TestCase):
def setUp(self):
self.motd = Message.objects.create(
label=TEST_LABEL, message=TEST_MESSAGE
)
self._create_test_message()
def test_basic(self):
queryset = Message.objects.get_for_now()
@@ -22,25 +20,25 @@ class MOTDTestCase(TestCase):
self.assertEqual(queryset.exists(), True)
def test_start_datetime(self):
self.motd.start_datetime = timezone.now() - timedelta(days=1)
self.motd.save()
self.test_message.start_datetime = timezone.now() - timedelta(days=1)
self.test_message.save()
queryset = Message.objects.get_for_now()
self.assertEqual(queryset.first(), self.motd)
self.assertEqual(queryset.first(), self.test_message)
def test_end_datetime(self):
self.motd.start_datetime = timezone.now() - timedelta(days=2)
self.motd.end_datetime = timezone.now() - timedelta(days=1)
self.motd.save()
self.test_message.start_datetime = timezone.now() - timedelta(days=2)
self.test_message.end_datetime = timezone.now() - timedelta(days=1)
self.test_message.save()
queryset = Message.objects.get_for_now()
self.assertEqual(queryset.exists(), False)
def test_enable(self):
self.motd.enabled = False
self.motd.save()
self.test_message.enabled = False
self.test_message.save()
queryset = Message.objects.get_for_now()