afonso/core-extra
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

daemon/test: improved MoveNodesRequest to be hashable using dataclass itself, fixed grpc test not properly using a wrapped MoveNodesRequest object

Browse source
This commit is contained in:
Blake Harnden 2022-04-28 23:20:17 -07:00
parent aa8ea40ce6
commit 2ab2c27d49
2 changed files with 7 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

20
daemon/core/api/grpc/wrappers.py
View file

@ -1195,13 +1195,13 @@ class EmanePathlossesRequest:
)
@dataclass
@dataclass(frozen=True)
class MoveNodesRequest:
session_id: int
node_id: int
source: str = None
position: Position = None
geo: Geo = None
source: str = field(compare=False, default=None)
position: Position = field(compare=False, default=None)
geo: Geo = field(compare=False, default=None)
def to_proto(self) -> core_pb2.MoveNodesRequest:
position = self.position.to_proto() if self.position else None
@ -1213,15 +1213,3 @@ class MoveNodesRequest:
position=position,
geo=geo,
)
def __members(self) -> Tuple[int, int]:
return self.session_id, self.node_id
def __eq__(self, other: "MoveNodesRequest") -> bool:
if type(other) is type(self):
return self.__members() == other.__members()
else:
return False
def __hash__(self):
return hash(self.__members())