Fix handling of multiple Out params (#317)

Fixes: #296

Author: elprans

azure/functions_worker/dispatcher.py

@@ -295,7 +295,7 @@ async def _handle__invocation_request(self, req):
             output_data = []
             if fi.output_types:
                 for out_name, out_type_info in fi.output_types.items():
-                    val = args[name].get()
+                    val = args[out_name].get()
                     if val is None:
                         # TODO: is the "Out" parameter optional?
                         # Can "None" be marshaled into protos.TypedData?
@@ -308,7 +308,7 @@ async def _handle__invocation_request(self, req):
 
                     output_data.append(
                         protos.ParameterBinding(
-                            name=name,
+                            name=out_name,
                             data=rpc_val))
 
             return_value = None

tests/queue_functions/put_queue_multiple_out/function.json

@@ -0,0 +1,24 @@
+{
+  "scriptFile": "main.py",
+  "disabled": false,
+
+  "bindings": [
+    {
+      "type": "httpTrigger",
+      "direction": "in",
+      "name": "req"
+    },
+    {
+      "name": "resp",
+      "type": "http",
+      "direction": "out"
+    },
+    {
+      "direction": "out",
+      "name": "msg",
+      "queueName": "testqueue-return-multiple-outparam",
+      "connection": "AzureWebJobsStorage",
+      "type": "queue"
+    }
+  ]
+}

tests/queue_functions/put_queue_multiple_out/main.py

@@ -0,0 +1,8 @@
+import azure.functions as func
+
+
+def main(req: func.HttpRequest, resp: func.Out[func.HttpResponse],
+         msg: func.Out[func.QueueMessage]) -> None:
+    data = req.get_body().decode()
+    msg.set(func.QueueMessage(body=data))
+    resp.set(func.HttpResponse(body='HTTP response: {}'.format(data)))

tests/test_queue_functions.py

@@ -61,3 +61,9 @@ def test_queue_return_multiple(self):
 
         # wait for queue_trigger to process the queue item
         time.sleep(1)
+
+    def test_queue_return_multiple_outparam(self):
+        r = self.webhost.request('POST', 'put_queue_multiple_out',
+                                 data='foo')
+        self.assertTrue(200 <= r.status_code < 300)
+        self.assertEqual(r.text, 'HTTP response: foo')