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:
@@ -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()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user