nikdoof/django-ett
1
0
Fork 0
mirror of https://github.com/nikdoof/django-ett.git synced 2025-12-17 19:59:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add total_time to tasks, and archive flag

Browse Source
This commit is contained in:
Andrew Williams
2012-10-19 23:18:11 +01:00
parent 2dca6210af
commit 829eac4767
2 changed files with 20 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
django_ett/etasks/tests.py
View File

@@ -1,9 +1,10 @@
from __future__ import with_statement
from django.test import TestCase
from django.db import IntegrityError
from django.utils.timezone import now
from django.contrib.auth.models import User
from .models import Task
from .models import Task, TimeEntry
class TaskTest(TestCase):
@@ -21,12 +22,20 @@ class TaskTest(TestCase):
with self.assertRaises(IntegrityError):
Task.objects.create(name='Bad Test')
def testTotalTime(self):
self.assertEqual(self.task.total_time, 0)
entry = TimeEntry.objects.create(task=self.task, type=TimeEntry.TYPE_ON, date=now(), segment=10)
self.assertEqual(self.task.total_time, 15)
entry2 = TimeEntry.objects.create(task=self.task, type=TimeEntry.TYPE_OFF, date=now(), segment=11)
self.assertEqual(self.task.total_time, 15)
entry3 = TimeEntry.objects.create(task=self.task, type=TimeEntry.TYPE_ON, date=now(), segment=12)
self.assertEqual(self.task.total_time, 30)
class TaskDetailViewTest(TestCase):
def setUp(self):
self.user, created = User.objects.get_or_create(username='TestUser')
self.task = Task.objects.create(user=self.user, name="Test Task 2")
self.task = Task.objects.create(user=self.user, name="Test Task 2")
def testBasicResponse(self):
resp = self.client.get('/task/%d/' % self.task.pk)