CentOS compatibility: fix tests

Signed-off-by: Markus Lehtonen <markus.lehtonen@linux.intel.com>

diff --git a/tests/01_test_help.py b/tests/01_test_help.py
index b0bb9a1353f78fb8618028a3d16ef036079550b7..828ee3b198b223b4d8784716b90735ff244adfe0 100644 (file)
--- a/tests/01_test_help.py
+++ b/tests/01_test_help.py
@@ -4,7 +4,11 @@
 
 from . import context
 
-import unittest
+# Try unittest2 for CentOS
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 
 class TestHelp(unittest.TestCase):
     """Test help output of gbp commands"""

diff --git a/tests/02_test_upstream_source_tar_unpack.py b/tests/02_test_upstream_source_tar_unpack.py
index 02b50439b2302d4cd218a8a81473c192bcc86280..05821448f12e72b70849e22e842e045f8f8ed593 100644 (file)
--- a/tests/02_test_upstream_source_tar_unpack.py
+++ b/tests/02_test_upstream_source_tar_unpack.py
@@ -6,7 +6,11 @@ from . import context
 
 import os
 import tarfile
-import unittest
+# Try unittest2 for CentOS
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 
 import six
 

diff --git a/tests/05_test_detection.py b/tests/05_test_detection.py
index 126a299241c629834d3fa98424a8efddc5b2e8d7..c3f729bacf05086e2e12d53856bf3fe6c8310912 100644 (file)
--- a/tests/05_test_detection.py
+++ b/tests/05_test_detection.py
@@ -4,7 +4,11 @@
 
 from . import context
 
-import unittest
+# Try unittest2 for CentOS
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 
 from gbp.scripts import buildpackage
 from gbp.deb import (DebianPkgPolicy, orig_file)

diff --git a/tests/06_test_upstream_source.py b/tests/06_test_upstream_source.py
index c6df6f14f6eaae3bdf4e14a745e3e662e2766995..6b2a1a1efdce930f96f95716dd20a678cd125297 100644 (file)
--- a/tests/06_test_upstream_source.py
+++ b/tests/06_test_upstream_source.py
@@ -9,7 +9,11 @@ import os
 import shutil
 import tarfile
 import tempfile
-import unittest
+# Try unittest2 for CentOS
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 import zipfile
 
 from gbp.pkg import UpstreamSource

diff --git a/tests/08_test_patch.py b/tests/08_test_patch.py
index df1d43d4d4f8f5f60ded1e03a3293afdf25e611a..dd1532e21f81bb75676147b57b064ef34f1d6abe 100644 (file)
--- a/tests/08_test_patch.py
+++ b/tests/08_test_patch.py
@@ -5,7 +5,11 @@
 from . import context
 
 import os
-import unittest
+# Try unittest2 for CentOS
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 
 from gbp.patch_series import Patch
 

diff --git a/tests/11_test_dch_main.py b/tests/11_test_dch_main.py
index ac747e64f7c6c10102080e115233f20f1a95634d..3bc177ef126fb6e2c6294d11adb5a4e145d3cf49 100644 (file)
--- a/tests/11_test_dch_main.py
+++ b/tests/11_test_dch_main.py
@@ -8,7 +8,11 @@ from .testutils import (DebianGitTestRepo, OsReleaseFile,
 
 from gbp.scripts import dch
 
-import unittest
+# Try unittest2 for CentOS
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 import os
 import re
 

diff --git a/tests/12_test_deb.py b/tests/12_test_deb.py
index d46f987c2597f88daa85d69efb3a6a88f572971b..2be30976d1efbd3b681dfabddd0c93fd9c0feb01 100644 (file)
--- a/tests/12_test_deb.py
+++ b/tests/12_test_deb.py
@@ -4,7 +4,12 @@
 
 from . import context
 
-import os, tempfile, unittest
+import os, tempfile
+# Try unittest2 for CentOS
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 
 import gbp.deb
 

diff --git a/tests/13_test_gbp_pq.py b/tests/13_test_gbp_pq.py
index dfcf87a66b3b9166cc4800cecf2123b585e88210..bf1ca102138df0d2944d6e4bc43eafe5a6930c4e 100644 (file)
--- a/tests/13_test_gbp_pq.py
+++ b/tests/13_test_gbp_pq.py
@@ -20,7 +20,11 @@ from . import testutils
 
 import os
 import logging
-import unittest
+# Try unittest2 for CentOS
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 
 from gbp.scripts.pq import generate_patches, export_patches
 import gbp.scripts.common.pq as pq

diff --git a/tests/16_test_supercommand.py b/tests/16_test_supercommand.py
index f72bb5622f63c8fa358c92e5a590d3991ded4914..296ef8ed7d49121e8e9a68a20a392da82420b22c 100644 (file)
--- a/tests/16_test_supercommand.py
+++ b/tests/16_test_supercommand.py
@@ -16,7 +16,11 @@
 """Test L{gbp} command wrapper"""
 
 import sys
-import unittest
+# Try unittest2 for CentOS
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 import gbp.scripts.supercommand
 
 class TestSuperCommand(unittest.TestCase):

diff --git a/tests/18_test_Config.py b/tests/18_test_Config.py
index 3f60e4523bc2dd3a145699913e99e4ff60b4ca72..45d97f17b7ea96aa87a399c1e3ccd2afbb24626f 100644 (file)
--- a/tests/18_test_Config.py
+++ b/tests/18_test_Config.py
@@ -2,7 +2,11 @@
 
 import os
 import sys
-import unittest
+# Try unittest2 for CentOS
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 from gbp.config import GbpOptionParser, GbpOptionGroup
 from .testutils import GbpLogTester
 

diff --git a/tests/19_test_gbp_scripts_config.py b/tests/19_test_gbp_scripts_config.py
index f503ef54115e784e9ac52b4a736ad34b51dd724a..c984329aa6366454525daeee5f71e31ea8e09bb0 100644 (file)
--- a/tests/19_test_gbp_scripts_config.py
+++ b/tests/19_test_gbp_scripts_config.py
@@ -17,7 +17,11 @@
 
 import os
 import sys
-import unittest
+# Try unittest2 for CentOS
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 import gbp.scripts.config
 
 class TestGbpConfigCommand(unittest.TestCase):

diff --git a/tests/component/rpm/test_import_srpm.py b/tests/component/rpm/test_import_srpm.py
index 51b45e133adc65769f50dca87e7d198abdb3c281..04b763b6975f1e46ac9ef61b2566a668cef2ed96 100644 (file)
--- a/tests/component/rpm/test_import_srpm.py
+++ b/tests/component/rpm/test_import_srpm.py
@@ -54,8 +54,8 @@ class TestImportPacked(ComponentTestBase):
         eq_(mock_import(['--no-pristine-tar', srpm]), 0)
         # Check repository state
         repo = GitRepository('gbp-test')
-        files =  {'Makefile', 'README', 'bar.tar.gz', 'dummy.sh', 'foo.txt',
-                  'gbp-test.spec', 'my.patch', 'mydir/myfile.txt'}
+        files =  set(['Makefile', 'README', 'bar.tar.gz', 'dummy.sh', 'foo.txt',
+                  'gbp-test.spec', 'my.patch', 'mydir/myfile.txt'])
         self._check_repo_state(repo, 'master', ['master', 'upstream'], files)
         # Four commits: upstream, packaging files, one patch and the removal
         # of imported patches
@@ -67,9 +67,9 @@ class TestImportPacked(ComponentTestBase):
         eq_(mock_import(['--no-pristine-tar', srpm]), 0)
         # Check repository state
         repo = GitRepository('gbp-test2')
-        files = {'Makefile', 'README', 'bar.tar.gz', 'dummy.sh', 'foo.txt',
+        files = set(['Makefile', 'README', 'bar.tar.gz', 'dummy.sh', 'foo.txt',
                  'gbp-test2.spec', 'gbp-test2-alt.spec', 'my.patch',
-                 'mydir/myfile.txt'}
+                 'mydir/myfile.txt'])
         self._check_repo_state(repo, 'master', ['master', 'upstream'], files)
 
         # Four commits: upstream, packaging files, one patch and the removal
@@ -85,8 +85,8 @@ class TestImportPacked(ComponentTestBase):
         eq_(mock_import(['--no-pristine-tar', '--orphan-packaging', srpm]), 0)
         # Check repository state
         repo = GitRepository('gbp-test2')
-        files = {'bar.tar.gz', 'foo.txt', 'gbp-test2.spec',
-                 'gbp-test2-alt.spec', 'my.patch', 'my2.patch', 'my3.patch'}
+        files = set(['bar.tar.gz', 'foo.txt', 'gbp-test2.spec',
+                 'gbp-test2-alt.spec', 'my.patch', 'my2.patch', 'my3.patch'])
         self._check_repo_state(repo, 'master', ['master', 'upstream'], files)
         # Only one commit: the packaging files
         eq_(len(repo.get_commits()), 1)
@@ -96,8 +96,8 @@ class TestImportPacked(ComponentTestBase):
         srpm = os.path.join(DATA_DIR, 'gbp-test-native-1.0-1.src.rpm')
         eq_(mock_import(['--native', srpm]), 0)
         # Check repository state
-        files = {'.gbp.conf', 'Makefile', 'README', 'dummy.sh',
-                 'packaging/gbp-test-native.spec'}
+        files = set(['.gbp.conf', 'Makefile', 'README', 'dummy.sh',
+                 'packaging/gbp-test-native.spec'])
         repo = GitRepository('gbp-test-native')
         self._check_repo_state(repo, 'master', ['master'], files)
         # Only one commit: the imported source tarball
@@ -147,8 +147,8 @@ class TestImportPacked(ComponentTestBase):
         eq_(len(repo.get_commits(until='upstream')), 1)
         # Import new version
         eq_(mock_import(['--no-pristine-tar', srpms[2]]), 0)
-        files = {'Makefile', 'README', 'bar.tar.gz', 'dummy.sh', 'foo.txt',
-                 'gbp-test.spec', 'my.patch', 'mydir/myfile.txt'}
+        files = set(['Makefile', 'README', 'bar.tar.gz', 'dummy.sh', 'foo.txt',
+                 'gbp-test.spec', 'my.patch', 'mydir/myfile.txt'])
         self._check_repo_state(repo, 'master', ['master', 'upstream'], files)
         eq_(len(repo.get_commits()), 11)
         eq_(len(repo.get_commits(until='upstream')), 2)
@@ -276,10 +276,10 @@ class TestImportPacked(ComponentTestBase):
                     srpm]), 0)
         # Check repository state
         repo = GitRepository('gbp-test2')
-        files = {'Makefile', 'README', 'dummy.sh', 'packaging/bar.tar.gz',
+        files = set(['Makefile', 'README', 'dummy.sh', 'packaging/bar.tar.gz',
                  'packaging/foo.txt', 'packaging/gbp-test2.spec',
                  'packaging/gbp-test2-alt.spec', 'packaging/my.patch',
-                 'packaging/my2.patch', 'packaging/my3.patch'}
+                 'packaging/my2.patch', 'packaging/my3.patch'])
         self._check_repo_state(repo, 'pack', ['pack', 'orig'], files)
         eq_(len(repo.get_commits()), 2)
         # Check packaging dir

diff --git a/tests/test_import_orig.py b/tests/test_import_orig.py
index d596d1dffea3b3f79a845f456f9fcfe9f856ca7c..aa39e38064bb533f88fb201aa0d10ccc64d1f7d9 100644 (file)
--- a/tests/test_import_orig.py
+++ b/tests/test_import_orig.py
@@ -21,7 +21,11 @@ import glob
 import os
 import tarfile
 import tempfile
-import unittest
+# Try unittest2 for CentOS
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 
 from gbp.errors import GbpError
 from gbp.pkg import UpstreamSource

diff --git a/tests/testutils/debiangittestrepo.py b/tests/testutils/debiangittestrepo.py
index 61456f1e03d9bb732c72deac2c6d7807ffc6da95..cec45fbe543bb0b6da9fe855416910f50a15239b 100644 (file)
--- a/tests/testutils/debiangittestrepo.py
+++ b/tests/testutils/debiangittestrepo.py
@@ -3,7 +3,11 @@
 from .. import context
 
 import os
-import unittest
+# Try unittest2 for CentOS
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 
 import gbp.deb.git