Skip to content

Commit 57db56d

Browse files
committed
chore(protocol-contracts): make OperatorStaking and Rewarder owned by ProtocolStaking owner
1 parent 2293f82 commit 57db56d

4 files changed

Lines changed: 8 additions & 23 deletions

File tree

protocol-contracts/staking/contracts/OperatorRewarder.sol

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ contract OperatorRewarder is Ownable {
5353
_;
5454
}
5555

56-
constructor(address owner, ProtocolStaking protocolStaking_, OperatorStaking operatorStaking_) Ownable(owner) {
56+
constructor(ProtocolStaking protocolStaking_, OperatorStaking operatorStaking_) Ownable(protocolStaking_.owner()) {
5757
_token = IERC20(protocolStaking_.stakingToken());
5858
_protocolStaking = protocolStaking_;
5959
_operatorStaking = operatorStaking_;

protocol-contracts/staking/contracts/OperatorStaking.sol

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -68,20 +68,18 @@ contract OperatorStaking is ERC20, Ownable, ReentrancyGuardTransient {
6868
* @param name The name of the ERC20 token.
6969
* @param symbol The symbol of the ERC20 token.
7070
* @param protocolStaking_ The ProtocolStaking contract address.
71-
* @param owner The owner address.
7271
*/
7372
constructor(
7473
string memory name,
7574
string memory symbol,
76-
ProtocolStaking protocolStaking_,
77-
address owner
78-
) ERC20(name, symbol) Ownable(owner) {
75+
ProtocolStaking protocolStaking_
76+
) ERC20(name, symbol) Ownable(protocolStaking_.owner()) {
7977
_asset = IERC20(protocolStaking_.stakingToken());
8078
_protocolStaking = protocolStaking_;
8179

8280
IERC20(asset()).approve(address(protocolStaking_), type(uint256).max);
8381

84-
address rewarder_ = address(new OperatorRewarder(owner, protocolStaking_, this));
82+
address rewarder_ = address(new OperatorRewarder(protocolStaking_, this));
8583
protocolStaking_.setRewardsRecipient(rewarder_);
8684
_rewarder = rewarder_;
8785

protocol-contracts/staking/test/OperatorRewarder.test.ts

Lines changed: 2 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -25,12 +25,7 @@ describe('OperatorRewarder', function () {
2525
ethers.parseEther('0.5'), // reward rate
2626
]),
2727
);
28-
const operatorStaking = await ethers.deployContract('$OperatorStaking', [
29-
'OPStake',
30-
'OP',
31-
protocolStaking,
32-
admin.address,
33-
]);
28+
const operatorStaking = await ethers.deployContract('$OperatorStaking', ['OPStake', 'OP', protocolStaking]);
3429
const mock = await ethers.getContractAt('OperatorRewarder', await operatorStaking.rewarder());
3530
await expect(mock.token()).to.eventually.eq(token.target);
3631

@@ -321,11 +316,7 @@ describe('OperatorRewarder', function () {
321316
await this.operatorStaking.connect(this.staker1).deposit(ethers.parseEther('1'), this.staker1);
322317
await timeIncreaseNoMine(9);
323318

324-
const newRewarder = await ethers.deployContract('OperatorRewarder', [
325-
this.admin,
326-
this.protocolStaking,
327-
this.operatorStaking,
328-
]);
319+
const newRewarder = await ethers.deployContract('OperatorRewarder', [this.protocolStaking, this.operatorStaking]);
329320

330321
this.tx = this.operatorStaking.connect(this.admin).setRewarder(newRewarder);
331322
});

protocol-contracts/staking/test/OperatorStaking.test.ts

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ describe('OperatorStaking', function () {
2424
0n, // reward rate
2525
]),
2626
);
27-
const mock = await ethers.deployContract('$OperatorStaking', ['OPStake', 'OP', protocolStaking, admin.address]);
27+
const mock = await ethers.deployContract('$OperatorStaking', ['OPStake', 'OP', protocolStaking]);
2828

2929
await Promise.all(
3030
[staker1, staker2].flatMap(account => [
@@ -379,11 +379,7 @@ describe('OperatorStaking', function () {
379379

380380
describe('with new rewarder', async function () {
381381
beforeEach(async function () {
382-
const newRewarder = await ethers.deployContract('OperatorRewarder', [
383-
this.admin,
384-
this.protocolStaking,
385-
this.mock,
386-
]);
382+
const newRewarder = await ethers.deployContract('OperatorRewarder', [this.protocolStaking, this.mock]);
387383
const oldRewarder = await ethers.getContractAt('OperatorRewarder', await this.mock.rewarder());
388384

389385
await this.protocolStaking.connect(this.admin).addEligibleAccount(this.mock);

0 commit comments

Comments
 (0)