From bd01fe1b72bb8bba436c77fd63572e3569d902af Mon Sep 17 00:00:00 2001
From: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
Date: Thu, 17 Sep 2015 02:57:28 -0400
Subject: [PATCH] Add 1 seconds delay to avoid version revert tests failing on
 MySQL. This is because MySQL doesn't store milliseconds in the DB. Version
 revert uses timestamp as the original document and the new version are
 created within the same second the revert code doesn't work. This delay is a
 hack and a better solution to differentiate versions by serial or similar
 needs to be implemented.

---
 mayan/apps/documents/tests/test_api.py    |  9 ++++-
 mayan/apps/documents/tests/test_models.py | 49 +++++++++++++++++++++++
 2 files changed, 56 insertions(+), 2 deletions(-)

diff --git a/mayan/apps/documents/tests/test_api.py b/mayan/apps/documents/tests/test_api.py
index 880c9551ea..9b00e9df19 100644
--- a/mayan/apps/documents/tests/test_api.py
+++ b/mayan/apps/documents/tests/test_api.py
@@ -2,6 +2,8 @@
 
 from __future__ import unicode_literals
 
+import time
+
 from json import loads
 
 from django.contrib.auth.models import User
@@ -205,13 +207,16 @@ class DocumentAPITestCase(APITestCase):
                 file_object=File(file_object),
             )
 
+        # Needed by MySQL as milliseconds value is not store in timestamp field
+        time.sleep(1)
+
         with open(TEST_DOCUMENT_PATH) as file_object:
             document.new_version(file_object=File(file_object))
 
-        document_version = document.versions.first()
-
         self.assertEqual(document.versions.count(), 2)
 
+        document_version = document.versions.first()
+
         self.client.post(
             reverse(
                 'rest_api:documentversion-revert', args=(document_version.pk,)
diff --git a/mayan/apps/documents/tests/test_models.py b/mayan/apps/documents/tests/test_models.py
index 61f573c6a7..00d49ea5fb 100644
--- a/mayan/apps/documents/tests/test_models.py
+++ b/mayan/apps/documents/tests/test_models.py
@@ -1,5 +1,7 @@
 from __future__ import unicode_literals
 
+import time
+
 from django.core.files import File
 from django.test import TestCase, override_settings
 
@@ -126,3 +128,50 @@ class MultiPageTiffTestCase(TestCase):
             '40adaa9d658b65c70a7f002dfe084a8354bb77c0dfbf1993e31fb024a285fb1d'
         )
         self.assertEqual(self.document.page_count, 2)
+
+
+@override_settings(OCR_AUTO_OCR=False)
+class DocumentVersionTestCase(TestCase):
+    def setUp(self):
+        self.document_type = DocumentType.objects.create(
+            label=TEST_DOCUMENT_TYPE
+        )
+
+        with open(TEST_SMALL_DOCUMENT_PATH) as file_object:
+            self.document = self.document_type.new_document(
+                file_object=File(file_object)
+            )
+
+    def tearDown(self):
+        self.document_type.delete()
+
+    def test_add_new_version(self):
+        self.assertEqual(self.document.versions.count(), 1)
+
+        with open(TEST_DOCUMENT_PATH) as file_object:
+            self.document.new_version(
+                file_object=File(file_object)
+            )
+
+        self.assertEqual(self.document.versions.count(), 2)
+
+        self.assertEqual(
+            self.document.checksum,
+            'c637ffab6b8bb026ed3784afdb07663fddc60099853fae2be93890852a69ecf3'
+        )
+
+    def test_revert_version(self):
+        self.assertEqual(self.document.versions.count(), 1)
+
+        time.sleep(1)
+
+        with open(TEST_DOCUMENT_PATH) as file_object:
+            self.document.new_version(
+                file_object=File(file_object)
+            )
+
+        self.assertEqual(self.document.versions.count(), 2)
+
+        self.document.versions.first().revert()
+
+        self.assertEqual(self.document.versions.count(), 1)