Allow remote module to transmit exception to Ubersmith

Author: Frédéric Guillot

tests/test_api.py

@@ -66,6 +66,26 @@ def test_execute_method_returns_string(self):
 
         self.router.invoke_method.assert_called_with(module=self.module2, method='remote_method', params=[], env={'variable1': 'value1'}, callback={})
         assert_that(json.loads(output.data.decode(output.charset)), is_('simple string'))
+        assert_that(output.status_code, is_(200))
+
+    def test_execute_method_raise_an_exception(self):
+        self.router.invoke_method.side_effect = Exception('Some Error')
+        output = self.api_client.post(self.generate_module_path('module2'),
+                                      headers={'Content-Type': 'application/json'},
+                                      data=json.dumps(
+                                              {
+                                                  "method": "remote_method",
+                                                  "params": [],
+                                                  "env": {
+                                                      "variable1": "value1"
+                                                  },
+                                                  "callback": {}
+                                              }
+                                      ))
+
+        self.router.invoke_method.assert_called_with(module=self.module2, method='remote_method', params=[], env={'variable1': 'value1'}, callback={})
+        assert_that(json.loads(output.data.decode(output.charset)), is_('Some Error'))
+        assert_that(output.status_code, is_(500))
 
     def test_execute_method_returns_list(self):
         self.router.invoke_method.return_value = ['a', 'b', 'c']
@@ -84,6 +104,7 @@ def test_execute_method_returns_list(self):
 
         self.router.invoke_method.assert_called_with(module=self.module2, method='remote_method', params=[], env={'variable1': 'value1'}, callback={})
         assert_that(json.loads(output.data.decode(output.charset)), is_(['a', 'b', 'c']))
+        assert_that(output.status_code, is_(200))
 
     def test_invoking_unknown_module_returns_a_404(self):
         output = self.api_client.post(self.generate_module_path('new_module'),
@@ -106,6 +127,7 @@ def test_listing_unknown_module_returns_a_404(self):
 
         assert_that(output.status_code, is_(404))
 
+
 class NoTrailingSlashApiTest(ApiTest):
     def generate_module_path(self, module_name):
         return '/{0}'.format(module_name)

ubersmith_remote_module_server/api.py

@@ -41,8 +41,12 @@ def list_implemented_methods(self, module):
     def handle_remote_invocation(self, module):
         logging.debug("Handle remote invocation for {module}".format(module=module))
         data = request.get_json()
-        output = self.router.invoke_method(module=module, **data)
-        return json_response(output, 200)
+        try:
+            output = self.router.invoke_method(module=module, **data)
+            return json_response(output, 200)
+        except Exception as e:
+            return json_response(str(e), 500)
+
 
 def json_response(data, code):
     json_data = json.dumps(data, indent=None)