Use Protocol for stubs

Author: andystaples

durabletask-azurefunctions/durabletask/azurefunctions/internal/azurefunctions_null_stub.py

@@ -1,38 +1,34 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 
-from durabletask.internal.ProtoTaskHubSidecarServiceStub import ProtoTaskHubSidecarServiceStub
+from durabletask.internal.proto_task_hub_sidecar_service_stub import ProtoTaskHubSidecarServiceStub
 
 
 class AzureFunctionsNullStub(ProtoTaskHubSidecarServiceStub):
     """A task hub sidecar stub class that implements all methods as no-ops."""
-
-    def __init__(self):
-        """Constructor.
-        """
-        self.Hello = lambda *args, **kwargs: None
-        self.StartInstance = lambda *args, **kwargs: None
-        self.GetInstance = lambda *args, **kwargs: None
-        self.RewindInstance = lambda *args, **kwargs: None
-        self.WaitForInstanceStart = lambda *args, **kwargs: None
-        self.WaitForInstanceCompletion = lambda *args, **kwargs: None
-        self.RaiseEvent = lambda *args, **kwargs: None
-        self.TerminateInstance = lambda *args, **kwargs: None
-        self.SuspendInstance = lambda *args, **kwargs: None
-        self.ResumeInstance = lambda *args, **kwargs: None
-        self.QueryInstances = lambda *args, **kwargs: None
-        self.PurgeInstances = lambda *args, **kwargs: None
-        self.GetWorkItems = lambda *args, **kwargs: None
-        self.CompleteActivityTask = lambda *args, **kwargs: None
-        self.CompleteOrchestratorTask = lambda *args, **kwargs: None
-        self.CompleteEntityTask = lambda *args, **kwargs: None
-        self.StreamInstanceHistory = lambda *args, **kwargs: None
-        self.CreateTaskHub = lambda *args, **kwargs: None
-        self.DeleteTaskHub = lambda *args, **kwargs: None
-        self.SignalEntity = lambda *args, **kwargs: None
-        self.GetEntity = lambda *args, **kwargs: None
-        self.QueryEntities = lambda *args, **kwargs: None
-        self.CleanEntityStorage = lambda *args, **kwargs: None
-        self.AbandonTaskActivityWorkItem = lambda *args, **kwargs: None
-        self.AbandonTaskOrchestratorWorkItem = lambda *args, **kwargs: None
-        self.AbandonTaskEntityWorkItem = lambda *args, **kwargs: None
+    Hello = lambda *args, **kwargs: None
+    StartInstance = lambda *args, **kwargs: None
+    GetInstance = lambda *args, **kwargs: None
+    RewindInstance = lambda *args, **kwargs: None
+    WaitForInstanceStart = lambda *args, **kwargs: None
+    WaitForInstanceCompletion = lambda *args, **kwargs: None
+    RaiseEvent = lambda *args, **kwargs: None
+    TerminateInstance = lambda *args, **kwargs: None
+    SuspendInstance = lambda *args, **kwargs: None
+    ResumeInstance = lambda *args, **kwargs: None
+    QueryInstances = lambda *args, **kwargs: None
+    PurgeInstances = lambda *args, **kwargs: None
+    GetWorkItems = lambda *args, **kwargs: None
+    CompleteActivityTask = lambda *args, **kwargs: None
+    CompleteOrchestratorTask = lambda *args, **kwargs: None
+    CompleteEntityTask = lambda *args, **kwargs: None
+    StreamInstanceHistory = lambda *args, **kwargs: None
+    CreateTaskHub = lambda *args, **kwargs: None
+    DeleteTaskHub = lambda *args, **kwargs: None
+    SignalEntity = lambda *args, **kwargs: None
+    GetEntity = lambda *args, **kwargs: None
+    QueryEntities = lambda *args, **kwargs: None
+    CleanEntityStorage = lambda *args, **kwargs: None
+    AbandonTaskActivityWorkItem = lambda *args, **kwargs: None
+    AbandonTaskOrchestratorWorkItem = lambda *args, **kwargs: None
+    AbandonTaskEntityWorkItem = lambda *args, **kwargs: None

durabletask/internal/ProtoTaskHubSidecarServiceStub.py

@@ -1,38 +1,34 @@
-from typing import Callable
+from typing import Any, Callable, Protocol
 
 
-class ProtoTaskHubSidecarServiceStub(object):
+class ProtoTaskHubSidecarServiceStub(Protocol):
     """A stub class roughly matching the TaskHubSidecarServiceStub generated from the .proto file.
     Used by Azure Functions during orchestration and entity executions to inject custom behavior,
     as no real sidecar stub is available.
     """
-
-    def __init__(self):
-        """Constructor.
-        """
-        self.Hello: Callable[..., None]
-        self.StartInstance: Callable[..., None]
-        self.GetInstance: Callable[..., None]
-        self.RewindInstance: Callable[..., None]
-        self.WaitForInstanceStart: Callable[..., None]
-        self.WaitForInstanceCompletion: Callable[..., None]
-        self.RaiseEvent: Callable[..., None]
-        self.TerminateInstance: Callable[..., None]
-        self.SuspendInstance: Callable[..., None]
-        self.ResumeInstance: Callable[..., None]
-        self.QueryInstances: Callable[..., None]
-        self.PurgeInstances: Callable[..., None]
-        self.GetWorkItems: Callable[..., None]
-        self.CompleteActivityTask: Callable[..., None]
-        self.CompleteOrchestratorTask: Callable[..., None]
-        self.CompleteEntityTask: Callable[..., None]
-        self.StreamInstanceHistory: Callable[..., None]
-        self.CreateTaskHub: Callable[..., None]
-        self.DeleteTaskHub: Callable[..., None]
-        self.SignalEntity: Callable[..., None]
-        self.GetEntity: Callable[..., None]
-        self.QueryEntities: Callable[..., None]
-        self.CleanEntityStorage: Callable[..., None]
-        self.AbandonTaskActivityWorkItem: Callable[..., None]
-        self.AbandonTaskOrchestratorWorkItem: Callable[..., None]
-        self.AbandonTaskEntityWorkItem: Callable[..., None]
+    Hello: Callable[..., Any]
+    StartInstance: Callable[..., Any]
+    GetInstance: Callable[..., Any]
+    RewindInstance: Callable[..., Any]
+    WaitForInstanceStart: Callable[..., Any]
+    WaitForInstanceCompletion: Callable[..., Any]
+    RaiseEvent: Callable[..., Any]
+    TerminateInstance: Callable[..., Any]
+    SuspendInstance: Callable[..., Any]
+    ResumeInstance: Callable[..., Any]
+    QueryInstances: Callable[..., Any]
+    PurgeInstances: Callable[..., Any]
+    GetWorkItems: Callable[..., Any]
+    CompleteActivityTask: Callable[..., Any]
+    CompleteOrchestratorTask: Callable[..., Any]
+    CompleteEntityTask: Callable[..., Any]
+    StreamInstanceHistory: Callable[..., Any]
+    CreateTaskHub: Callable[..., Any]
+    DeleteTaskHub: Callable[..., Any]
+    SignalEntity: Callable[..., Any]
+    GetEntity: Callable[..., Any]
+    QueryEntities: Callable[..., Any]
+    CleanEntityStorage: Callable[..., Any]
+    AbandonTaskActivityWorkItem: Callable[..., Any]
+    AbandonTaskOrchestratorWorkItem: Callable[..., Any]
+    AbandonTaskEntityWorkItem: Callable[..., Any]