From adbd7e68706f1a847e7cddd86dbe6201c74e8cd5 Mon Sep 17 00:00:00 2001
From: Audrey Roy <audreyr@gmail.com>
Date: Mon, 19 Aug 2013 23:18:11 +0200
Subject: [PATCH] Better organization of cleanup tests involving Unicode
 filenames and dir names.

---
 tests/test_cleanup.py | 33 +++++++++++++++++++++------------
 1 file changed, 21 insertions(+), 12 deletions(-)

diff --git a/tests/test_cleanup.py b/tests/test_cleanup.py
index 54f95c002..62c05e4f9 100755
--- a/tests/test_cleanup.py
+++ b/tests/test_cleanup.py
@@ -11,7 +11,12 @@
 from __future__ import unicode_literals
 import os
 import shutil
-import unittest
+import sys
+
+if sys.version_info[:2] < (2, 7):
+    import unittest2 as unittest
+else:
+    import unittest
 
 from cookiecutter import cleanup, exceptions
 
@@ -28,17 +33,6 @@ def test_remove_repo(self):
         self.assertTrue(os.path.isfile('tests/fake-project/README.rst'))
         self.assertFalse(os.path.exists('tests/fake-repo'))
 
-    def test_remove_repo_unicode(self):
-        if os.path.supports_unicode_filenames:
-            success = cleanup.remove_repo(
-                repo_dir='tests/fake-repööö',
-                generated_project='fake-pröööject'
-            )
-            self.assertTrue(success)
-            self.assertTrue(os.path.isdir('tests/fake-pröööject'))
-            self.assertTrue(os.path.isfile('tests/fake-pröööject/README.rst'))
-            self.assertFalse(os.path.exists('tests/fake-repööö'))
-
     def test_remove_repo_bad(self):
         self.assertRaises(
             exceptions.MissingProjectDir,
@@ -52,6 +46,21 @@ def tearDown(self):
             os.mkdir('tests/fake-repo')
             shutil.move('tests/fake-project', 'tests/fake-repo/fake-project')
 
+
+class TestCleanupUnicode(unittest.TestCase):
+
+    def test_remove_repo_unicode(self):
+        if os.path.supports_unicode_filenames:
+            success = cleanup.remove_repo(
+                repo_dir='tests/fake-repööö',
+                generated_project='fake-pröööject'
+            )
+            self.assertTrue(success)
+            self.assertTrue(os.path.isdir('tests/fake-pröööject'))
+            self.assertTrue(os.path.isfile('tests/fake-pröööject/README.rst'))
+            self.assertFalse(os.path.exists('tests/fake-repööö'))
+
+    def tearDown(self):
         if os.path.supports_unicode_filenames:
             if not os.path.exists('tests/fake-repööö'):
                 os.mkdir('tests/fake-repööö')