Add convenience method for checking whether provider public key has a…

…lready been set.
nucypher · Sep 13, 2023 · afce141 · afce141
1 parent 3b70a7c
commit afce141
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/contracts/contracts/coordination/Coordinator.sol b/contracts/contracts/coordination/Coordinator.sol
@@ -177,6 +177,11 @@ contract Coordinator is AccessControlDefaultAdminRules, FlatRateFeeModel {
         revert("No keys found prior to the provided ritual");
     }
 
+    function isProviderPublicKeySet(address _provider) external view returns (bool) {
+        ParticipantKey[] storage participantHistory = participantKeysHistory[_provider];
+        return participantHistory.length > 0;
+    }
+
     function setTimeout(uint32 newTimeout) external onlyRole(DEFAULT_ADMIN_ROLE) {
         emit TimeoutChanged(timeout, newTimeout);
         timeout = newTimeout;

diff --git a/tests/test_coordinator.py b/tests/test_coordinator.py
@@ -152,7 +152,10 @@ def test_invalid_initiate_ritual(coordinator, nodes, accounts, initiator, global
 def initiate_ritual(coordinator, erc20, allow_logic, authority, nodes):
     for node in nodes:
         public_key = gen_public_key()
+        assert not coordinator.isProviderPublicKeySet(node)
         coordinator.setProviderPublicKey(public_key, sender=node)
+        assert coordinator.isProviderPublicKeySet(node)
+
     cost = coordinator.getRitualInitiationCost(nodes, DURATION)
     erc20.approve(coordinator.address, cost, sender=authority)
     tx = coordinator.initiateRitual(
@@ -212,7 +215,11 @@ def test_initiate_ritual(
 def test_provider_public_key(coordinator, nodes):
     selected_provider = nodes[0]
     public_key = gen_public_key()
+
+    assert not coordinator.isProviderPublicKeySet(selected_provider)
     tx = coordinator.setProviderPublicKey(public_key, sender=selected_provider)
+    assert coordinator.isProviderPublicKeySet(selected_provider)
+
     ritual_id = coordinator.numberOfRituals()
 
     events = coordinator.ParticipantPublicKeySet.from_receipt(tx)