Type NodeInfo.role as NodeRole instead of raw int

The role legend comment ("0=voter, 1=standby, 2=spare") encoded an
enum that already exists in dqlitewire as NodeRole. Switch the field
annotation to NodeRole and construct VOTER instances via the enum.
IntEnum stays int-compatible so existing integer comparisons continue
to work.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>

Author: antoineleclair

src/dqliteclient/node_store.py

@@ -4,6 +4,8 @@
 from collections.abc import Sequence
 from dataclasses import dataclass
 
+from dqlitewire import NodeRole
+
 
 @dataclass(frozen=True, slots=True)
 class NodeInfo:
@@ -16,7 +18,7 @@ class NodeInfo:
 
     node_id: int
     address: str
-    role: int  # 0=voter, 1=standby, 2=spare
+    role: NodeRole
 
 
 class NodeStore(ABC):
@@ -44,7 +46,7 @@ class MemoryNodeStore(NodeStore):
     def __init__(self, initial_addresses: list[str] | None = None) -> None:
         if initial_addresses:
             self._nodes: tuple[NodeInfo, ...] = tuple(
-                NodeInfo(node_id=i + 1, address=addr, role=0)
+                NodeInfo(node_id=i + 1, address=addr, role=NodeRole.VOTER)
                 for i, addr in enumerate(initial_addresses)
             )
         else:

tests/test_node_store.py

@@ -77,3 +77,20 @@ async def test_node_info_is_hashable(self) -> None:
         info2 = NodeInfo(node_id=1, address="h:1", role=0)
         assert hash(info1) == hash(info2)
         assert {info1, info2} == {info1}
+
+    async def test_memory_store_seeds_with_noderole_voter(self) -> None:
+        from dqlitewire import NodeRole
+
+        store = MemoryNodeStore(initial_addresses=["a:9001", "b:9001"])
+        nodes = await store.get_nodes()
+        assert all(isinstance(n.role, NodeRole) for n in nodes)
+        assert all(n.role == NodeRole.VOTER for n in nodes)
+        # IntEnum comparison with raw int still works.
+        assert nodes[0].role == 0
+
+    async def test_node_info_role_accepts_noderole(self) -> None:
+        from dqlitewire import NodeRole
+
+        info = NodeInfo(node_id=1, address="h:1", role=NodeRole.STANDBY)
+        assert info.role is NodeRole.STANDBY
+        assert info.role == 1