inherit the protocol or not?

Author: unkcpz

src/plumpy/coordinator.py

@@ -2,13 +2,14 @@
 from __future__ import annotations
 
 from re import Pattern
-from typing import TYPE_CHECKING, Any, Callable, Hashable, Protocol
+from typing import TYPE_CHECKING, Any, Callable, Hashable, Protocol, runtime_checkable
 
 if TYPE_CHECKING:
     ID_TYPE = Hashable
     Receiver = Callable[..., Any]
 
 
+@runtime_checkable
 class Coordinator(Protocol):
     def hook_rpc_receiver(
         self,

tests/rmq/test_coordinator.py

@@ -0,0 +1,6 @@
+from plumpy.coordinator import Coordinator
+from . import RmqCoordinator
+
+
+def test_mock_coordinator():
+    assert isinstance(RmqCoordinator, Coordinator)

tests/test_coordinator.py

@@ -0,0 +1,6 @@
+from plumpy.coordinator import Coordinator
+from .utils import MockCoordinator
+
+
+def test_mock_coordinator():
+    assert isinstance(MockCoordinator, Coordinator)

tests/utils.py

@@ -13,6 +13,7 @@
 
 import plumpy
 from plumpy import persistence, process_states, processes, utils
+from plumpy.coordinator import Coordinator
 from plumpy.exceptions import CoordinatorConnectionError
 from plumpy.message import MessageBuilder
 from plumpy.rmq import TaskRejected
@@ -25,7 +26,7 @@
 Snapshot = collections.namedtuple('Snapshot', ['state', 'bundle', 'outputs'])
 
 
-class MockCoordinator:
+class MockCoordinator(Coordinator):
     def __init__(self):
         self._task_receivers = {}
         self._broadcast_receivers = {}