Nucypher
/
nucypher
mirror of https://github.com/nucypher/nucypher.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Add is_ritual_active to MockCoordinatorAgent.

pull/3345/head
derekpierre 2023-11-15 09:44:15 -05:00 committed by Derek Pierre
parent d6a1e3f01f
commit 7e6bafeeb1
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/mock/coordinator.py
View File

@ -246,6 +246,10 @@ class MockCoordinatorAgent(MockContractAgent):
else:
raise RuntimeError(f"Ritual {ritual_id} is in an unknown state") # :-(
def is_ritual_active(self, ritual_id: int) -> bool:
result = self.get_ritual_status(ritual_id)
return result == self.RitualStatus.ACTIVE
def get_ritual_id_from_public_key(self, public_key: DkgPublicKey) -> int:
for i, ritual in enumerate(self.rituals):
if bytes(ritual.public_key) == bytes(public_key):