started to add coverage for circus.green

circus/tests/support.py

@@ -55,6 +55,8 @@ def resolve_name(name):
 
 class TestCircus(unittest.TestCase):
 
+    arbiter_factory = get_arbiter
+
     def setUp(self):
         self.arbiters = []
         self.files = []
@@ -99,14 +101,15 @@ def _create_circus(cls, callable, plugins=None, stats=False, **kw):
         worker.update(kw)
         debug = kw.get('debug', False)
 
+        fact = cls.arbiter_factory
         if stats:
-            arbiter = get_arbiter([worker], background=True, plugins=plugins,
-                                  stats_endpoint=DEFAULT_ENDPOINT_STATS,
-                                  statsd=True,
-                                  debug=debug, statsd_close_outputs=not debug)
+            arbiter = fact([worker], background=True, plugins=plugins,
+                           stats_endpoint=DEFAULT_ENDPOINT_STATS,
+                           statsd=True,
+                           debug=debug, statsd_close_outputs=not debug)
         else:
-            arbiter = get_arbiter([worker], background=True, plugins=plugins,
-                                  debug=debug)
+            arbiter = fact([worker], background=True, plugins=plugins,
+                           debug=debug)
         arbiter.start()
         return testfile, arbiter
 
@@ -175,6 +178,14 @@ def run_process(test_file):
     return 1
 
 
+def has_gevent():
+    try:
+        import gevent       # NOQA
+        return True
+    except ImportError:
+        return False
+
+
 class TimeoutException(Exception):
     pass
 

circus/tests/test_circusd.py

@@ -9,19 +9,12 @@
 from circus.arbiter import Arbiter
 from circus.util import REDIRECT_TO
 from circus import util
+from circus.tests.support import has_gevent
 
 
 CIRCUS_INI = os.path.join(os.path.dirname(__file__), 'circus.ini')
 
 
-def has_gevent():
-    try:
-        import gevent       # NOQA
-        return True
-    except ImportError:
-        return False
-
-
 class TestCircusd(TestCase):
 
     def setUp(self):

circus/tests/test_green.py

@@ -0,0 +1,10 @@
+from unittest2 import skipIf
+
+from circus.tests.support import has_gevent
+from circus.green import get_arbiter
+from circus.tests.test_arbiter import TestTrainer
+
+
+@skipIf(not has_gevent(), 'Tests for Gevent')
+class TestGreen(TestTrainer):
+    arbiter_factory = get_arbiter