ExternalVendorCode/tahoe-lafs
1
0
Fork 0
mirror of https://github.com/tahoe-lafs/tahoe-lafs.git synced 2025-01-02 19:26:44 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix types.

Browse Source
This commit is contained in:
Itamar Turner-Trauring 2023-06-14 14:02:24 -04:00
parent b45ee20ba8
commit 8493b42024
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/allmydata/test/cli/wormholetesting.py
View File

@ -63,7 +63,7 @@ class MemoryWormholeServer(object):
specific application id and relay URL combination. specific application id and relay URL combination.
""" """
_apps: dict[ApplicationKey, _WormholeApp] = field(default=Factory(dict)) _apps: dict[ApplicationKey, _WormholeApp] = field(default=Factory(dict))
_waiters: dict[ApplicationKey, Deferred] = field(default=Factory(dict)) _waiters: dict[ApplicationKey, Deferred[IWormhole]] = field(default=Factory(dict))
def create( def create(
self, self,
@ -130,7 +130,7 @@ class TestingHelper(object):
key = (relay_url, appid) key = (relay_url, appid)
if key in self._server._waiters: if key in self._server._waiters:
raise ValueError(f"There is already a waiter for {key}") raise ValueError(f"There is already a waiter for {key}")
d = Deferred() d : Deferred[IWormhole] = Deferred()
self._server._waiters[key] = d self._server._waiters[key] = d
wormhole = await d wormhole = await d
return wormhole return wormhole
@ -166,7 +166,7 @@ class _WormholeApp(object):
appid/relay_url scope. appid/relay_url scope.
""" """
wormholes: dict[WormholeCode, IWormhole] = field(default=Factory(dict)) wormholes: dict[WormholeCode, IWormhole] = field(default=Factory(dict))
_waiting: dict[WormholeCode, List[Deferred]] = field(default=Factory(dict)) _waiting: dict[WormholeCode, List[Deferred[_MemoryWormhole]]] = field(default=Factory(dict))
_counter: Iterator[int] = field(default=Factory(count)) _counter: Iterator[int] = field(default=Factory(count))
def allocate_code(self, wormhole: IWormhole, code: Optional[WormholeCode]) -> WormholeCode: def allocate_code(self, wormhole: IWormhole, code: Optional[WormholeCode]) -> WormholeCode:
@ -192,13 +192,13 @@ class _WormholeApp(object):
return code return code
def wait_for_wormhole(self, code: WormholeCode) -> Awaitable[_MemoryWormhole]: def wait_for_wormhole(self, code: WormholeCode) -> Deferred[_MemoryWormhole]:
""" """
Return a ``Deferred`` which fires with the next wormhole to be associated Return a ``Deferred`` which fires with the next wormhole to be associated
with the given code. This is used to let the first end of a wormhole with the given code. This is used to let the first end of a wormhole
rendezvous with the second end. rendezvous with the second end.
""" """
d = Deferred() d : Deferred[_MemoryWormhole] = Deferred()
self._waiting.setdefault(code, []).append(d) self._waiting.setdefault(code, []).append(d)
return d return d
@ -242,8 +242,8 @@ class _MemoryWormhole(object):
_view: _WormholeServerView _view: _WormholeServerView
_code: Optional[WormholeCode] = None _code: Optional[WormholeCode] = None
_payload: DeferredQueue = field(default=Factory(DeferredQueue)) _payload: DeferredQueue[WormholeMessage] = field(default=Factory(DeferredQueue))
_waiting_for_code: list[Deferred] = field(default=Factory(list)) _waiting_for_code: list[Deferred[WormholeCode]] = field(default=Factory(list))
def allocate_code(self) -> None: def allocate_code(self) -> None:
if self._code is not None: if self._code is not None:
@ -265,7 +265,7 @@ class _MemoryWormhole(object):
def when_code(self) -> Deferred[WormholeCode]: def when_code(self) -> Deferred[WormholeCode]:
if self._code is None: if self._code is None:
d = Deferred() d : Deferred[WormholeCode] = Deferred()
self._waiting_for_code.append(d) self._waiting_for_code.append(d)
return d return d
return succeed(self._code) return succeed(self._code)