fix: Add functools.wraps decorator (GoogleCloudPlatform#179)

svleeuwen · anniefu · web-flow · commit f2285f96072d · 2022-05-10T13:37:32.000-07:00
* fix: Add functools.wraps decorator (GoogleCloudPlatform#178) To make sure function attributes are copied to `_http_view_func_wrapper` Co-authored-by: Annie Fu <16651409+anniefu@users.noreply.github.com>
diff --git a/src/functions_framework/__init__.py b/src/functions_framework/__init__.py
@@ -95,6 +95,7 @@ def setup_logging():
 
 
 def _http_view_func_wrapper(function, request):
+    @functools.wraps(function)
     def view_func(path):
         return function(request._get_current_object())
 
diff --git a/tests/test_view_functions.py b/tests/test_view_functions.py
@@ -33,6 +33,17 @@ def test_http_view_func_wrapper():
     assert function.calls == [pretend.call(request_object)]
 
 
+def test_http_view_func_wrapper_attribute_copied():
+    def function(_):
+        pass
+
+    function.attribute = "foo"
+    view_func = functions_framework._http_view_func_wrapper(function, pretend.stub())
+
+    assert view_func.__name__ == "function"
+    assert view_func.attribute == "foo"
+
+
 def test_event_view_func_wrapper(monkeypatch):
     data = pretend.stub()
     json = {
