From fd84ced5c61ee41c58eb16f4e6a48ed42c3029ce Mon Sep 17 00:00:00 2001
From: Kyle Moore <moore.kyle39@gmail.com>
Date: Mon, 21 Dec 2020 21:06:16 -0500
Subject: [PATCH] Add nested function support for '.' separator syntax

---
 lambda_local/main.py             |  8 ++++++--
 tests/test_direct_invocations.py | 32 ++++++++++++++++++++++++++++++++
 2 files changed, 38 insertions(+), 2 deletions(-)

diff --git a/lambda_local/main.py b/lambda_local/main.py
index 738079f..acf507e 100644
--- a/lambda_local/main.py
+++ b/lambda_local/main.py
@@ -137,8 +137,12 @@ def load_source(request_id, path, function_name):
     else:
         raise Exception("unsupported python version")
 
-    func = getattr(mod, function_name)
-    return func
+    object_names = function_name.split('.')
+    object = mod
+    for name in object_names:
+        object = getattr(object, name)
+
+    return object
 
 
 def execute(func, event, context):
diff --git a/tests/test_direct_invocations.py b/tests/test_direct_invocations.py
index 1a418fd..91e8727 100644
--- a/tests/test_direct_invocations.py
+++ b/tests/test_direct_invocations.py
@@ -26,6 +26,15 @@ def my_failing_lambda_function(event, context):
     raise Exception('Oh no')
 
 
+class MyLambdaClass:
+    def my_lambda_method(self, event, context):
+        print("Hello World from My Lambda Method!")
+        return 42
+
+
+my_lambda_instance = MyLambdaClass()
+
+
 def test_function_call_for_pytest():
     (result, error_type) = lambda_call(
         my_lambda_function, {}, Context(1))
@@ -86,3 +95,26 @@ def test_check_command_line_error():
 
     os.remove(request_file)
     assert p.exitcode == 1
+
+
+def test_check_command_line_nested_function_name():
+    request = json.dumps({})
+    request_file = 'check_command_line_event.json'
+    with open(request_file, "w") as f:
+        f.write(request)
+
+    args = argparse.Namespace(event=request_file,
+                              file='tests/test_direct_invocations.py',
+                              function='my_lambda_instance.my_lambda_method',
+                              timeout=1,
+                              environment_variables='',
+                              library=None,
+                              version_name='',
+                              arn_string=''
+                              )
+    p = Process(target=lambda_run, args=(args,))
+    p.start()
+    p.join()
+
+    os.remove(request_file)
+    assert p.exitcode == 0