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, TimeEntry


class TaskTest(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 1")

    def testDisplay(self):
        self.assertEqual(str(self.task), "Test Task 1")
        self.task.name = ''
        self.assertEqual(str(self.task), "Unknown")

    def testBadCreation(self):
        with self.assertRaises(IntegrityError):
            Task.objects.create(name='Bad Test')

    def testTotalTime(self):
        self.assertEqual(self.task.total_time, 0)
        TimeEntry.objects.create(task=self.task, type=TimeEntry.TYPE_ON, date=now(), segment=10)
        self.assertEqual(self.task.total_time, 15)
        TimeEntry.objects.create(task=self.task, type=TimeEntry.TYPE_OFF, date=now(), segment=11)
        self.assertEqual(self.task.total_time, 15)
        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")

    def testBasicResponse(self):
        resp = self.client.get('/task/%d/' % self.task.pk)
        self.assertEqual(resp.status_code, 200)
        self.assertTrue('object' in resp.context)