add warmup context

Author: YunchuWang

azure/functions/_abc.py

@@ -47,6 +47,11 @@ def message(self) -> str:
         pass
 
 
+class WarmUpContext(abc.ABC):
+    """Warmup context object."""
+    pass
+
+
 class TraceContext(abc.ABC):
     """Trace context object."""
 
@@ -126,6 +131,12 @@ def retry_context(self) -> RetryContext:
         """Context for retries to the function."""
         pass
 
+    @property
+    @abc.abstractmethod
+    def warmup_context(self) -> WarmUpContext:
+        """Context for warmup to the function."""
+        pass
+
 
 class HttpRequest(abc.ABC):
     """HTTP request object."""

tests/test_http_asgi.py

@@ -5,7 +5,7 @@
 import unittest
 
 import azure.functions as func
-from azure.functions._abc import TraceContext, RetryContext
+from azure.functions._abc import TraceContext, RetryContext, WarmUpContext
 from azure.functions._http_asgi import (
     AsgiMiddleware
 )
@@ -114,15 +114,17 @@ def _generate_func_context(
         function_name='httptrigger',
         function_directory='/home/roger/wwwroot/httptrigger',
         trace_context=TraceContext,
-        retry_context=RetryContext
+        retry_context=RetryContext,
+        warmup_context=WarmUpContext
     ) -> func.Context:
         class MockContext(func.Context):
-            def __init__(self, ii, fn, fd, tc, rc):
+            def __init__(self, ii, fn, fd, tc, rc, wc):
                 self._invocation_id = ii
                 self._function_name = fn
                 self._function_directory = fd
                 self._trace_context = tc
                 self._retry_context = rc
+                self._warmup_context = wc
 
             @property
             def invocation_id(self):
@@ -144,8 +146,12 @@ def trace_context(self):
             def retry_context(self):
                 return self._retry_context
 
+            @property
+            def warmup_context(self):
+                return self._warmup_context
+
         return MockContext(invocation_id, function_name, function_directory,
-                           trace_context, retry_context)
+                           trace_context, retry_context, warmup_context)
 
     def test_middleware_calls_app(self):
         app = MockAsgiApplication()

tests/test_http_wsgi.py

@@ -5,7 +5,7 @@
 from io import StringIO, BytesIO
 
 import azure.functions as func
-from azure.functions._abc import TraceContext, RetryContext
+from azure.functions._abc import TraceContext, RetryContext, WarmUpContext
 from azure.functions._http import HttpResponseHeaders
 from azure.functions._http_wsgi import (
     WsgiRequest,
@@ -223,15 +223,17 @@ def _generate_func_context(
         function_name='httptrigger',
         function_directory='/home/roger/wwwroot/httptrigger',
         trace_context=TraceContext,
-        retry_context=RetryContext
+        retry_context=RetryContext,
+        warmup_context=WarmUpContext
     ) -> func.Context:
         class MockContext(func.Context):
-            def __init__(self, ii, fn, fd, tc, rc):
+            def __init__(self, ii, fn, fd, tc, rc, wc):
                 self._invocation_id = ii
                 self._function_name = fn
                 self._function_directory = fd
                 self._trace_context = tc
                 self._retry_context = rc
+                self._warmup_context = wc
 
             @property
             def invocation_id(self):
@@ -253,8 +255,12 @@ def trace_context(self):
             def retry_context(self):
                 return self._retry_context
 
+            @property
+            def warmup_context(self):
+                return self._warmup_context
+
         return MockContext(invocation_id, function_name, function_directory,
-                           trace_context, retry_context)
+                           trace_context, retry_context, warmup_context)
 
     def _generate_wsgi_app(self,
                            status='200 OK',