@@ -18,6 +18,7 @@ import assert from 'assert';
1818import { registerNode } from '../../test/_helpers/node' ;
1919import { userDeposit } from '../../test/_helpers/deposit' ;
2020import { deployMegapool , getMegapoolForNode , getValidatorInfo , nodeDeposit } from '../../test/_helpers/megapool' ;
21+ import { setDAOProtocolBootstrapSetting } from '../../test/dao/scenario-dao-protocol-bootstrap' ;
2122
2223const helpers = require ( '@nomicfoundation/hardhat-network-helpers' ) ;
2324const hre = require ( 'hardhat' ) ;
@@ -56,6 +57,9 @@ export default function() {
5657 } ) ;
5758
5859 it ( printTitle ( 'upgrade' , 'updates expected settings' ) , async ( ) => {
60+ // Disable node deposits ahead of upgrade
61+ await setDAOProtocolBootstrapSetting ( RocketDAOProtocolSettingsNode , 'node.deposit.enabled' , false , { from : owner } )
62+
5963 await executeUpgrade ( owner , upgradeContract , rocketStorageAddress ) ;
6064 const upgradeTime = await helpers . time . latest ( ) ;
6165
@@ -125,6 +129,9 @@ export default function() {
125129 assertBN . equal ( split [ 1 ] , '0.09' . ether ) ; // Voter share
126130 assertBN . equal ( split [ 2 ] , '0' . ether ) ; // Protocol share
127131 assertBN . equal ( split [ 3 ] , '0.86' . ether ) ; // User share
132+
133+ // Check node deposits are enabled
134+ assert . equal ( await rocketDAOProtocolSettingsNode . getDepositEnabled ( ) , true ) ;
128135 } ) ;
129136
130137 it ( printTitle ( 'node' , 'can create megapool and deposit from a node registered before upgrade' ) , async ( ) => {
0 commit comments