From 67a65a2aa9ecd77de372511aa54541100923d3d3 Mon Sep 17 00:00:00 2001
From: Omer Katz <omer.drow@gamil.com>
Date: Thu, 26 Jun 2014 11:17:57 +0300
Subject: [PATCH] Installing unittest2 on Python 2.6.

---
 setup.py                 | 3 +++
 tests/test_connection.py | 7 ++++++-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 85707d00..f8f0d6da 100644
--- a/setup.py
+++ b/setup.py
@@ -58,6 +58,9 @@ if sys.version_info[0] == 3:
 else:
     extra_opts['tests_require'] = ['nose', 'coverage', 'blinker', 'django>=1.4.2', 'PIL', 'jinja2>=2.6', 'python-dateutil']
 
+    if sys.version_info[0] == 2 and sys.version_info[1] == 6:
+        extra_opts['tests_require'].append('unittest2')
+
 setup(name='mongoengine',
       version=VERSION,
       author='Harry Marr',
diff --git a/tests/test_connection.py b/tests/test_connection.py
index bf615ceb..a5b1b089 100644
--- a/tests/test_connection.py
+++ b/tests/test_connection.py
@@ -1,6 +1,11 @@
 import sys
 sys.path[0:0] = [""]
-import unittest
+
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
+
 import datetime
 
 import pymongo