39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
|
import django
|
||
|
|
import os
|
||
|
|
from django.test import TestCase
|
||
|
|
from django.db import models
|
||
|
|
from datetime import datetime
|
||
|
|
from zoneinfo import ZoneInfo
|
||
|
|
|
||
|
|
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "timetracker.settings")
|
||
|
|
django.setup()
|
||
|
|
from django.conf import settings
|
||
|
|
from games.models import Game, Edition, Purchase, Session
|
||
|
|
|
||
|
|
ZONEINFO = ZoneInfo(settings.TIME_ZONE)
|
||
|
|
|
||
|
|
|
||
|
|
class FormatDurationTest(TestCase):
|
||
|
|
def setUp(self) -> None:
|
||
|
|
return super().setUp()
|
||
|
|
|
||
|
|
def test_duration_format(self):
|
||
|
|
g = Game(name="The Test Game")
|
||
|
|
g.save()
|
||
|
|
e = Edition(game=g, name="The Test Game Edition")
|
||
|
|
e.save()
|
||
|
|
p = Purchase(
|
||
|
|
edition=e, date_purchased=datetime(2022, 9, 26, 14, 58, tzinfo=ZONEINFO)
|
||
|
|
)
|
||
|
|
p.save()
|
||
|
|
s = Session(
|
||
|
|
purchase=p,
|
||
|
|
timestamp_start=datetime(2022, 9, 26, 14, 58, tzinfo=ZONEINFO),
|
||
|
|
timestamp_end=datetime(2022, 9, 26, 17, 38, tzinfo=ZONEINFO),
|
||
|
|
)
|
||
|
|
s.save()
|
||
|
|
self.assertEqual(
|
||
|
|
s.duration_formatted(),
|
||
|
|
"02:40",
|
||
|
|
)
|