ExternalVendorCode/tahoe-lafs
1
0
Fork 0
mirror of https://github.com/tahoe-lafs/tahoe-lafs.git synced 2024-12-21 13:57:51 +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.
"""
_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(
self,
@ -130,7 +130,7 @@ class TestingHelper(object):
key = (relay_url, appid)
if key in self._server._waiters:
raise ValueError(f"There is already a waiter for {key}")
d = Deferred()
d : Deferred[IWormhole] = Deferred()
self._server._waiters[key] = d
wormhole = await d
return wormhole
@ -166,7 +166,7 @@ class _WormholeApp(object):
appid/relay_url scope.
"""
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))
def allocate_code(self, wormhole: IWormhole, code: Optional[WormholeCode]) -> WormholeCode:
@ -192,13 +192,13 @@ class _WormholeApp(object):
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
with the given code. This is used to let the first end of a wormhole
rendezvous with the second end.
"""
d = Deferred()
d : Deferred[_MemoryWormhole] = Deferred()
self._waiting.setdefault(code, []).append(d)
return d
@ -242,8 +242,8 @@ class _MemoryWormhole(object):
_view: _WormholeServerView
_code: Optional[WormholeCode] = None
_payload: DeferredQueue = field(default=Factory(DeferredQueue))
_waiting_for_code: list[Deferred] = field(default=Factory(list))
_payload: DeferredQueue[WormholeMessage] = field(default=Factory(DeferredQueue))
_waiting_for_code: list[Deferred[WormholeCode]] = field(default=Factory(list))
def allocate_code(self) -> None:
if self._code is not None:
@ -265,7 +265,7 @@ class _MemoryWormhole(object):
def when_code(self) -> Deferred[WormholeCode]:
if self._code is None:
d = Deferred()
d : Deferred[WormholeCode] = Deferred()
self._waiting_for_code.append(d)
return d
return succeed(self._code)