@@ -19,6 +19,7 @@ import { registerNode } from '../../test/_helpers/node';
1919import { userDeposit } from '../../test/_helpers/deposit' ;
2020import { deployMegapool , getMegapoolForNode , getValidatorInfo , nodeDeposit } from '../../test/_helpers/megapool' ;
2121import { decompressABI } from '../../test/_utils/contract' ;
22+ import { setDAOProtocolBootstrapSetting } from '../../test/dao/scenario-dao-protocol-bootstrap' ;
2223
2324const helpers = require ( '@nomicfoundation/hardhat-network-helpers' ) ;
2425const hre = require ( 'hardhat' ) ;
@@ -57,6 +58,9 @@ export default function() {
5758 } ) ;
5859
5960 it ( printTitle ( 'upgrade' , 'updates expected settings' ) , async ( ) => {
61+ // Disable node deposits ahead of upgrade
62+ await setDAOProtocolBootstrapSetting ( RocketDAOProtocolSettingsNode , 'node.deposit.enabled' , false , { from : owner } )
63+
6064 await executeUpgrade ( owner , upgradeContract , rocketStorageAddress ) ;
6165 const upgradeTime = await helpers . time . latest ( ) ;
6266
@@ -130,6 +134,9 @@ export default function() {
130134 // Check rocketMegapool ABI exists
131135 const rocketMegapoolAbi = await rocketStorage . getString ( ethers . solidityPackedKeccak256 ( [ 'string' , 'string' ] , [ 'contract.abi' , 'rocketMegapool' ] ) ) ;
132136 assert . notEqual ( rocketMegapoolAbi . length , 0 ) ;
137+
138+ // Check node deposits are enabled
139+ assert . equal ( await rocketDAOProtocolSettingsNode . getDepositEnabled ( ) , true ) ;
133140 } ) ;
134141
135142 it ( printTitle ( 'node' , 'can create megapool and deposit from a node registered before upgrade' ) , async ( ) => {
0 commit comments