mirror of https://github.com/snachodog/mybuddy.git
27 lines
1015 B
Python
27 lines
1015 B
Python
# -*- coding: utf-8 -*-
|
|
from django.test import TestCase
|
|
from django.utils import timezone
|
|
|
|
from core import utils
|
|
|
|
|
|
class UtilsTestCase(TestCase):
|
|
def test_duration_string(self):
|
|
duration = timezone.timedelta(hours=1, minutes=30, seconds=45)
|
|
self.assertEqual(
|
|
utils.duration_string(duration), "1 hour, 30 minutes, 45 seconds"
|
|
)
|
|
self.assertEqual(utils.duration_string(duration, "m"), "1 hour, 30 minutes")
|
|
self.assertEqual(utils.duration_string(duration, "h"), "1 hour")
|
|
self.assertRaises(TypeError, lambda: utils.duration_string("1 hour"))
|
|
|
|
def test_duration_parts(self):
|
|
duration = timezone.timedelta(hours=1, minutes=30, seconds=45)
|
|
self.assertEqual(utils.duration_parts(duration), (1, 30, 45))
|
|
self.assertRaises(TypeError, lambda: utils.duration_parts("1 hour"))
|
|
|
|
def test_random_color(self):
|
|
color = utils.random_color()
|
|
self.assertIsInstance(color, str)
|
|
self.assertIn(color, utils.COLORS)
|