diff --git a/src/AvailDepository.sol b/src/AvailDepository.sol index 507f94a..d7debf0 100644 --- a/src/AvailDepository.sol +++ b/src/AvailDepository.sol @@ -50,7 +50,8 @@ contract AvailDepository is PausableUpgradeable, AccessControlDefaultAdminRulesU external initializer { - if (governance == address(0) || pauser == address(0) || depositor == address(0) || newDepository == bytes32(0)) { + if (governance == address(0) || pauser == address(0) || depositor == address(0) || newDepository == bytes32(0)) + { revert ZeroAddress(); } depository = newDepository; diff --git a/test/DeqRouterTest.t.sol b/test/DeqRouterTest.t.sol index 89aff34..116cab9 100644 --- a/test/DeqRouterTest.t.sol +++ b/test/DeqRouterTest.t.sol @@ -61,10 +61,19 @@ contract DeqRouterTest is Test { assertEq(address(newDeqRouter.avail()), address(newAvail)); } - function testRevertZeroAddress_initialize(address newAvail, address newGovernance, address newPauser, address newSwapRouter, address newStakedAvail) external { + function testRevertZeroAddress_initialize( + address newAvail, + address newGovernance, + address newPauser, + address newSwapRouter, + address newStakedAvail + ) external { vm.assume(newAvail != address(0)); DeqRouter newDeqRouter = new DeqRouter(IERC20(newAvail)); - vm.assume(newGovernance == address(0) || newPauser == address(0) || newSwapRouter == address(0) || newStakedAvail == address(0)); + vm.assume( + newGovernance == address(0) || newPauser == address(0) || newSwapRouter == address(0) + || newStakedAvail == address(0) + ); vm.expectRevert(IDeqRouter.ZeroAddress.selector); newDeqRouter.initialize(newGovernance, newPauser, newSwapRouter, IStakedAvail(newStakedAvail)); }