From ae87c52ca57543eca81cd0b88307090180920fa7 Mon Sep 17 00:00:00 2001
From: Andrey Antukh <niwi@niwi.be>
Date: Wed, 23 Oct 2013 20:10:09 +0200
Subject: [PATCH] Add task diff for testing.

---
 greenmine/projects/tasks/serializers.py     | 11 +++++------
 greenmine/projects/tasks/tests/tests_api.py | 12 ++++++++++++
 2 files changed, 17 insertions(+), 6 deletions(-)

diff --git a/greenmine/projects/tasks/serializers.py b/greenmine/projects/tasks/serializers.py
index 4efacd0a..8d0dd105 100644
--- a/greenmine/projects/tasks/serializers.py
+++ b/greenmine/projects/tasks/serializers.py
@@ -48,12 +48,11 @@ class TaskSerializer(serializers.ModelSerializer):
         diff_list = []
         current = None
 
-        if obj:
-            for version in reversed(list(reversion.get_for_object(obj))):
-                if current:
-                    task_diff = self.get_task_diff(current, version)
-                    diff_list.append(task_diff)
+        for version in reversed(list(reversion.get_for_object(obj))):
+            if current:
+                task_diff = self.get_task_diff(current, version)
+                diff_list.append(task_diff)
 
-                current = version
+            current = version
 
         return diff_list
diff --git a/greenmine/projects/tasks/tests/tests_api.py b/greenmine/projects/tasks/tests/tests_api.py
index b9ee2522..00a15415 100644
--- a/greenmine/projects/tasks/tests/tests_api.py
+++ b/greenmine/projects/tasks/tests/tests_api.py
@@ -6,6 +6,8 @@ from django import test
 from django.core import mail
 from django.core.urlresolvers import reverse
 
+import reversion
+
 from greenmine.base.users.tests import create_user
 from greenmine.projects.tests import create_project, add_membership
 from greenmine.projects.milestones.tests import create_milestone
@@ -93,6 +95,16 @@ class TasksTestCase(test.TestCase):
         response = self.client.login(username=self.user1.username,
                                      password=self.user1.username)
         self.assertTrue(response)
+
+        # Change task for generate history/diff.
+        with reversion.create_revision():
+            self.task1.tags = ["LL"]
+            self.task1.save()
+
+        with reversion.create_revision():
+            self.task1.tags = ["LLKK"]
+            self.task1.save()
+
         response = self.client.get(reverse("tasks-detail", args=(self.task1.id,)))
         self.assertEqual(response.status_code, 200)
         self.client.logout()