Decouple TACoChildApplication and infraction collector

theref · Aug 9, 2024 · 4ca9272 · 4ca9272
1 parent 904b811
commit 4ca9272
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 16 deletions.
diff --git a/contracts/contracts/coordination/InfractionCollector.sol b/contracts/contracts/coordination/InfractionCollector.sol
@@ -52,7 +52,7 @@ contract InfractionCollector is OwnableUpgradeable {
             if (participant.transcript.length == 0) {
                 // Transcript TX wasn't posted
                 // Penalize the staking provider
-                tacoChildApplication.penalize(stakingProviders[i]);
+                // tacoChildApplication.penalize(stakingProviders[i]);
                 infractions[ritualId][stakingProviders[i]][
                     InfractionType.MISSING_TRANSCRIPT
                 ] = true;

diff --git a/contracts/contracts/coordination/TACoChildApplication.sol b/contracts/contracts/coordination/TACoChildApplication.sol
@@ -31,7 +31,7 @@ contract TACoChildApplication is ITACoRootToChild, ITACoChildApplication, Initia
 
     ITACoChildToRoot public immutable rootApplication;
     address public coordinator;
-    address public infractionCollector;
+    address public adjudicator;
 
     uint96 public immutable minimumAuthorization;
 
@@ -61,21 +61,18 @@ contract TACoChildApplication is ITACoRootToChild, ITACoChildApplication, Initia
     /**
      * @notice Initialize function for using with OpenZeppelin proxy
      */
-    function initialize(address _coordinator, address _infractionCollector) external initializer {
+    function initialize(address _coordinator, address _adjudicator) external initializer {
+        require(coordinator == address(0) || adjudicator == address(0), "Contracts already set");
         require(
-            coordinator == address(0) || infractionCollector == address(0),
-            "Contracts already set"
-        );
-        require(
-            _coordinator != address(0) && _infractionCollector != address(0),
+            _coordinator != address(0) && _adjudicator != address(0),
             "Contracts must be specified"
         );
         require(
             address(Coordinator(_coordinator).application()) == address(this),
             "Invalid coordinator"
         );
         coordinator = _coordinator;
-        infractionCollector = _infractionCollector;
+        adjudicator = _adjudicator;
     }
 
     function authorizedStake(address _stakingProvider) external view returns (uint96) {
@@ -203,10 +200,7 @@ contract TACoChildApplication is ITACoRootToChild, ITACoChildApplication, Initia
      * @param _stakingProvider Staking provider address
      */
     function penalize(address _stakingProvider) external override {
-        require(
-            msg.sender == address(infractionCollector),
-            "Only infractionCollector allowed to penalize"
-        );
+        require(msg.sender == address(adjudicator), "Only adjudicator allowed to penalize");
         rootApplication.penalize(_stakingProvider);
         emit Penalized(_stakingProvider);
     }

diff --git a/scripts/lynx/deploy_child.py b/scripts/lynx/deploy_child.py
@@ -51,8 +51,7 @@ def main():
     ritual_token = deployer.deploy(project.LynxRitualToken)
 
     coordinator = deployer.deploy(project.Coordinator)
-    infraction = deployer.deploy(project.InfractionCollector)
-    deployer.transact(taco_child_application.initialize, coordinator.address, infraction.address)
+    deployer.transact(taco_child_application.initialize, coordinator.address)
 
     global_allow_list = deployer.deploy(project.GlobalAllowList)
 

diff --git a/tests/test_child_application.py b/tests/test_child_application.py
@@ -331,7 +331,7 @@ def test_penalize(accounts, root_application, child_application, coordinator):
     ) = accounts[0:]
 
     # Penalize can be done only from adjudicator address
-    with ape.reverts("Only infractionCollector allowed to penalize"):
+    with ape.reverts("Only adjudicator allowed to penalize"):
         child_application.penalize(staking_provider, sender=staking_provider)
 
     tx = child_application.penalize(staking_provider, sender=creator)