diff tests/test-wireproto.py @ 52521:b52f2b365eff

typing: add type hints to `ipeercapabilities.capabilities()`
author Matt Harbison <matt_harbison@yahoo.com>
date Mon, 16 Dec 2024 03:04:44 -0500
parents 5cf81d8d7de1
children
line wrap: on
line diff
--- a/tests/test-wireproto.py	Mon Dec 16 02:41:24 2024 -0500
+++ b/tests/test-wireproto.py	Mon Dec 16 03:04:44 2024 -0500
@@ -1,5 +1,9 @@
 import sys
 
+from typing import (
+    Set,
+)
+
 from mercurial import (
     error,
     pycompat,
@@ -55,8 +59,8 @@
     def close(self):
         pass
 
-    def capabilities(self):
-        return [b'batch']
+    def capabilities(self) -> Set[bytes]:
+        return {b'batch'}
 
     def _call(self, cmd, **args):
         args = pycompat.byteskwargs(args)