Framework tests skipped when they're not present

Author: ElSaico

Diff for: tests/test_dj.py

@@ -1,13 +1,18 @@
 import unittest
 
-from django.http import Http404
-from django.core.exceptions import ObjectDoesNotExist
+try:
+    from django.http import Http404
+    from django.core.exceptions import ObjectDoesNotExist
 
-# Ugh. Settings for Django.
-from django.conf import settings
-settings.configure(DEBUG=True)
+    # Ugh. Settings for Django.
+    from django.conf import settings
+    settings.configure(DEBUG=True)
+
+    from restless.dj import DjangoResource
+except ImportError:
+    settings = None
+    DjangoResource = object
 
-from restless.dj import DjangoResource
 from restless.exceptions import Unauthorized
 from restless.preparers import FieldsPreparer
 from restless.resources import skip_prepare
@@ -125,6 +130,7 @@ def detail(self, pk):
         raise Http404("Model with pk {0} not found.".format(pk))
 
 
+@unittest.skipIf(not settings, "Django is not available")
 class DjangoResourceTestCase(unittest.TestCase):
     def setUp(self):
         super(DjangoResourceTestCase, self).setUp()

Diff for: tests/test_fl.py

@@ -1,9 +1,13 @@
 import unittest
 
-# Ugh. Globals for Flask.
-import flask
+try:
+    # Ugh. Globals for Flask.
+    import flask
+    from restless.fl import FlaskResource
+except ImportError:
+    flask = None
+    FlaskResource = object
 
-from restless.fl import FlaskResource
 from restless.utils import json
 
 from .fakes import FakeHttpRequest
@@ -32,6 +36,7 @@ def create(self):
         self.fake_db.append(self.data)
 
 
+@unittest.skipIf(not flask, 'Flask is not available')
 class FlaskResourceTestCase(unittest.TestCase):
     def setUp(self):
         super(FlaskResourceTestCase, self).setUp()

Diff for: tests/test_pyr.py

@@ -1,8 +1,12 @@
 import unittest
 
-from pyramid import testing
+try:
+    from pyramid import testing
+    from restless.pyr import PyramidResource
+except ImportError:
+    testing = None
+    PyramidResource = object
 
-from restless.pyr import PyramidResource
 from restless.utils import json
 
 from .fakes import FakeHttpRequest, FakeHttpResponse
@@ -36,6 +40,8 @@ def is_authenticated(self):
 
         return True
 
+
+@unittest.skipIf(not testing, 'Pyramid is not available')
 class PyramidResourceTestCase(unittest.TestCase):
     def setUp(self):
         self.config = testing.setUp()

Diff for: tests/test_tnd.py

@@ -2,12 +2,10 @@
 import socket
 import six
 
-from restless.tnd import TornadoResource, _BridgeMixin
 from restless.utils import json
-from tornado import testing, web, httpserver, gen, version_info
-from tornado.iostream import IOStream
 from restless.constants import UNAUTHORIZED
 
+
 def _newer_or_equal_(v):
     for i in six.moves.xrange(min(len(v), len(version_info))):
         expected, tnd = v[i], version_info[i]
@@ -19,15 +17,38 @@ def _newer_or_equal_(v):
             return False
     return True
 
+
 def _equal_(v):
     for i in six.moves.xrange(min(len(v), len(version_info))):
         if v[i] != version_info[i]:
             return False
     return True
 
+try:
+    from restless.tnd import TornadoResource, _BridgeMixin
+    from tornado import testing, web, httpserver, gen, version_info
+    from tornado.iostream import IOStream
+    if _newer_or_equal_((4, 0, 0, 0)):
+        from tornado.http1connection import HTTP1Connection
+except ImportError:
+    class testing:
+        AsyncHTTPTestCase = object
+
+    class web:
+        @staticmethod
+        def Application(*args, **kw): return False
+
+    class gen:
+        @staticmethod
+        def coroutine(fn): return fn
+
+    class TornadoResource:
+        @staticmethod
+        def as_list(): pass
+
+        @staticmethod
+        def as_detail(): pass
 
-if _newer_or_equal_((4, 0, 0, 0)):
-    from tornado.http1connection import HTTP1Connection
 
 class TndBaseTestResource(TornadoResource):
     """
@@ -89,6 +110,7 @@ def create(self):
 ], debug=True)
 
 
+@unittest.skipIf(not app, 'Tornado is not available')
 class BaseHTTPTestCase(testing.AsyncHTTPTestCase):
     """
     base of test case
@@ -97,6 +119,7 @@ def get_app(self):
         return app
 
 
+@unittest.skipIf(not app, 'Tornado is not available')
 class TndResourceTestCase(BaseHTTPTestCase):
     """
     """
@@ -148,6 +171,7 @@ def test_not_authenticated(self):
         self.assertEqual(resp.code, UNAUTHORIZED)
 
 
+@unittest.skipIf(not app, 'Tornado is not available')
 class BaseTestCase(unittest.TestCase):
     """
     test case that export the wrapped tornado.web.RequestHandler.
@@ -176,6 +200,7 @@ def init_request_handler(self, rh_cls, view_type):
         self.new_handler = rq(app, fake_request)
 
 
+@unittest.skipIf(not app, 'Tornado is not available')
 class InternalTestCase(BaseTestCase):
     """
     test-cases that check internal structure of the wrapped
@@ -213,6 +238,7 @@ def test_var(self):
         self.assertTrue(hasattr(self.new_handler.resource_handler, 'application'))
 
 
+@unittest.skipIf(not app, 'Tornado is not available')
 class TndDeleteTestResource(TndBasicTestResource):
     """
     testing inherited resource
@@ -224,6 +250,7 @@ def delete_list(self):
         self.fake_db = {}
 
 
+@unittest.skipIf(not app, 'Tornado is not available')
 class FuncTrimTestCase(BaseTestCase):
     """
     test-cases that make sure we removed unnecessary handler functions
@@ -258,6 +285,7 @@ def test_inheritance_resource_detail(self):
         self.assertNotIn('put', self.new_handler.__class__.__dict__)
 
 
+@unittest.skipIf(not app, 'Tornado is not available')
 class TndAsyncResourceTestCase(BaseHTTPTestCase):
     """
     test asynchronous view_method