Fix type errors in server and tests

- Added type casting in `src/rendezqueue/server.py` for `self.server` and unpacking `server_address`.
- Added type casting in `test/server_test.py` for `sxpb.loads` result.
- Resolved all type errors reported by `ty check`.

Co-authored-by: google-labs-jules[bot] <161369871+google-labs-jules[bot]@users.noreply.github.com>

Author: gendeux

server_port.txt

@@ -1 +1 @@
-42641
+38351

src/rendezqueue/server.py

@@ -3,6 +3,13 @@
 import argparse
 from urllib.parse import urlparse
 from rendezqueue.impl import RendezqueueImpl
+from typing import cast
+
+class RendezqueueServer(socketserver.ThreadingMixIn, http.server.HTTPServer):
+    def __init__(self, server_address, RequestHandlerClass, impl, http_path="/"):
+        self.impl = impl
+        self.http_path = http_path
+        super().__init__(server_address, RequestHandlerClass)
 
 class RendezqueueHandler(http.server.BaseHTTPRequestHandler):
     def do_OPTIONS(self):
@@ -16,8 +23,11 @@ def do_OPTIONS(self):
         self.end_headers()
 
     def do_POST(self):
+        # Cast self.server to RendezqueueServer to access custom attributes
+        server = cast(RendezqueueServer, self.server)
+
         parsed_path = urlparse(self.path)
-        if parsed_path.path != self.server.http_path:
+        if parsed_path.path != server.http_path:
             self.send_error(404, "Not Found")
             return
 
@@ -29,7 +39,7 @@ def do_POST(self):
         content_length = int(self.headers.get("Content-Length", 0))
         body = self.rfile.read(content_length).decode("utf-8")
 
-        result = self.server.impl.tryswap_string(body)
+        result = server.impl.tryswap_string(body)
 
         if isinstance(result, int):
             self.send_response(result)
@@ -43,12 +53,6 @@ def do_POST(self):
             self.end_headers()
             self.wfile.write(result.encode("utf-8"))
 
-class RendezqueueServer(socketserver.ThreadingMixIn, http.server.HTTPServer):
-    def __init__(self, server_address, RequestHandlerClass, impl, http_path="/"):
-        self.impl = impl
-        self.http_path = http_path
-        super().__init__(server_address, RequestHandlerClass)
-
 def run():
     parser = argparse.ArgumentParser(description="Rendezqueue Server")
     parser.add_argument("--http_host", default="127.0.0.1", help="Host to bind to")
@@ -68,7 +72,9 @@ def run():
         http_path=args.http_path
     )
 
-    host, port = server.server_address
+    # server_address might be (host, port) or more depending on family
+    # For AF_INET it is (host, port)
+    host, port = server.server_address[:2]
     print(f"Server running at http://{host}:{port}{args.http_path}")
 
     if args.port_filepath:

test/server_test.py

@@ -2,7 +2,7 @@
 import sxpb
 import os
 from rendezqueue.impl import RendezqueueImpl
-from typing import Any, Dict, List
+from typing import Any, Dict, List, cast
 
 def normalize_list(obj: Any) -> List[Any]:
     if obj is None:
@@ -17,7 +17,8 @@ def test_sxpb_cases(self):
         with open(sxpb_path, "r") as f:
             data = f.read()
 
-        test_cases = sxpb.loads(data)
+        # sxpb.loads returns Any, but we know it's a list of dicts based on our file format
+        test_cases = cast(List[Dict[str, Any]], sxpb.loads(data))
 
         for case in test_cases:
             case_name = case.get("name", "Unknown")
@@ -85,11 +86,6 @@ def _verify_response(self, actual, expected, case_name):
             # If ttl not expected, it implies it should be None or not present in JSON output?
             # But dataclass has it. The test logic: "ttl should not be present in response when values are present"
             # This logic is usually enforced during encoding.
-            # Impl tryswap returns the object with ttl set potentially.
-            # But wait, my manual test verified logic.
-            # Let's check logic in swapstore.
-            # If match found, ttl is NOT returned in TrySwapResponse.
-            # Wait, let's check my swapstore.py
             pass
 
         if "values" in expected: