Tests: fix Python 2.6 compatibility

ronnix · ronnix · commit 9fba74208f6a · 2014-07-05T02:06:13.000+02:00
diff --git a/fabtools/tests/test_system.py b/fabtools/tests/test_system.py
@@ -1,23 +1,18 @@
-try:
-    import unittest2 as unittest
-except ImportError:
-    import unittest
-
 from mock import patch
 
+import pytest
 
-class TestUnsupportedFamily(unittest.TestCase):
 
-    def test_unsupported_system(self):
+def test_unsupported_system():
 
-        from fabtools.system import UnsupportedFamily
+    from fabtools.system import UnsupportedFamily
 
-        with self.assertRaises(UnsupportedFamily) as cm:
+    with pytest.raises(UnsupportedFamily) as excinfo:
 
-            with patch('fabtools.system.distrib_id') as mock_distrib_id:
-                mock_distrib_id.return_value = 'foo'
+        with patch('fabtools.system.distrib_id') as mock_distrib_id:
+            mock_distrib_id.return_value = 'foo'
 
-                raise UnsupportedFamily(supported=['debian', 'redhat'])
+            raise UnsupportedFamily(supported=['debian', 'redhat'])
 
-        exception_msg = str(cm.exception)
-        self.assertEquals(exception_msg, "Unsupported system foo (supported families: debian, redhat)")
+    exception_msg = str(excinfo.value)
+    assert exception_msg == "Unsupported system foo (supported families: debian, redhat)"
