Updated args_parser function and added unittest.

Author: pdthummar

azure_functions_worker/main.py

@@ -2,11 +2,10 @@
 # Licensed under the MIT License.
 """Main entrypoint."""
 
-
 import argparse
 
 
-def parse_args():
+def create_args_parser():
     parser = argparse.ArgumentParser(
         description='Python Azure Functions Worker')
     parser.add_argument('--host',
@@ -26,18 +25,19 @@ def parse_args():
                              'syslog, or a file path')
     parser.add_argument('--grpcMaxMessageLength', type=int,
                         dest='grpc_max_msg_len')
-    parser.add_argument('--functions-uri', dest='functions_uri',
+    parser.add_argument('--functions-uri', dest='functions_uri', type=str,
                         help='URI with IP Address and Port used to'
                              ' connect to the Host via gRPC.')
     parser.add_argument('--functions-request-id', dest='functions_request_id',
-                        help='Request ID used for gRPC communication '
-                             'with the Host.')
-    parser.add_argument('--functions-worker-id', dest='functions_worker_id',
+                        type=str, help='Request ID used for gRPC communication '
+                                       'with the Host.')
+    parser.add_argument('--functions-worker-id',
+                        dest='functions_worker_id', type=str,
                         help='Worker ID assigned to this language worker.')
     parser.add_argument('--functions-grpc-max-message-length', type=int,
                         dest='functions_grpc_max_msg_len',
                         help='Max grpc message length for Functions')
-    return parser.parse_args()
+    return parser
 
 
 def main():
@@ -49,7 +49,8 @@ def main():
     from ._thirdparty import aio_compat
     from .logging import error_logger, logger, format_exception
 
-    args = parse_args()
+    parser = create_args_parser()
+    args = parser.parse_args()
     logging.setup(log_level=args.log_level, log_destination=args.log_to)
 
     logger.info('Starting Azure Functions Python Worker.')

tests/unittests/test_main.py

@@ -0,0 +1,31 @@
+import unittest
+
+from azure_functions_worker.main import create_args_parser
+
+
+class TestMain(unittest.TestCase):
+
+    def test_args_parser(self):
+        parser = create_args_parser()
+        parsed = parser.parse_args(
+            ['--host', '127.0.0.1',
+             '--port', '50821',
+             '--workerId', 'e9efd817-47a1-45dc-9e20-e6f975d7a025',
+             '--requestId', 'cbef5957-cdb3-4462-9ee7-ac9f91be0a51',
+             '--grpcMaxMessageLength', '2147483647',
+             '--functions-uri', 'http://127.0.0.1:50821',
+             '--functions-worker-id', 'e9efd817-47a1-45dc-9e20-e6f975d7a025',
+             '--functions-request-id', 'cbef5957-cdb3-4462-9ee7-ac9f91be0a51',
+             '--functions-grpc-max-message-length', '2147483647'])
+        self.assertEqual(parsed.host, '127.0.0.1')
+        self.assertEqual(parsed.worker_id,
+                         'e9efd817-47a1-45dc-9e20-e6f975d7a025')
+        self.assertEqual(parsed.request_id,
+                         'cbef5957-cdb3-4462-9ee7-ac9f91be0a51')
+        self.assertEqual(parsed.grpc_max_msg_len, 2147483647)
+        self.assertEqual(parsed.functions_uri, 'http://127.0.0.1:50821')
+        self.assertEqual(parsed.functions_worker_id,
+                         'e9efd817-47a1-45dc-9e20-e6f975d7a025')
+        self.assertEqual(parsed.functions_request_id,
+                         'cbef5957-cdb3-4462-9ee7-ac9f91be0a51')
+        self.assertEqual(parsed.functions_grpc_max_msg_len, 2147483647)