Add initial support for asyncio nested calls

timothycrosley · timothycrosley · commit ef01ea457e08 · 2016-05-22T15:51:33.000-07:00
diff --git a/hug/interface.py b/hug/interface.py
@@ -49,6 +49,8 @@
 
     def asyncio_call(function, *args, **kwargs):
         loop = asyncio.get_event_loop()
+        if loop.is_running():
+            return await function(*args, **kwargs)
         function = ensure_future(function(*args, **kwargs), loop=loop)
         loop.run_until_complete(function)
         return function.result()
diff --git a/tests/test_async.py b/tests/test_async.py
@@ -38,22 +38,38 @@ async def hello_world():
 
 
 def test_basic_call_on_method_async():
+    """Test to ensure the most basic call still works if applied to a method"""
+    class API(object):
 
+        @hug.call()
+        async def hello_world(self=None):
+            return "Hello World!"
+
+    api_instance = API()
+    assert api_instance.hello_world.interface.http
+    assert loop.run_until_complete(api_instance.hello_world()) == "Hello World!"
+    assert hug.test.get(api, '/hello_world').data == "Hello World!"
+
+
+def test_nested_call_on_method_async():
     """Test to ensure the most basic call still works if applied to a method"""
     class API(object):
 
         @hug.call()
         async def hello_world(self=None):
             return "Hello World!"
 
+        @hug.local()
+        async def
+
     api_instance = API()
     assert api_instance.hello_world.interface.http
     assert loop.run_until_complete(api_instance.hello_world()) == "Hello World!"
     assert hug.test.get(api, '/hello_world').data == "Hello World!"
 
 
 def test_basic_call_on_method_through_api_instance_async():
-
+    """Test to ensure instance method calling via async works as expected"""
     class API(object):
 
         def hello_world(self):
@@ -70,7 +86,7 @@ async def hello_world():
 
 
 def test_basic_call_on_method_registering_without_decorator_async():
-
+    """Test to ensure async methods can be used without decorator"""
     class API(object):
 
         def __init__(self):
@@ -83,3 +99,5 @@ async def hello_world_method(self):
 
     assert loop.run_until_complete(api_instance.hello_world_method()) == "Hello World!"
     assert hug.test.get(api, '/hello_world_method').data == "Hello World!"
+
+
