Extreme do/undo amounts tests

pkg/vault/test/foundry/LinearBaseExtremeAmounts.t.sol

@@ -0,0 +1,28 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+pragma solidity ^0.8.24;
+
+import { IVault } from "@balancer-labs/v3-interfaces/contracts/vault/IVault.sol";
+import { ArrayHelpers } from "@balancer-labs/v3-solidity-utils/contracts/test/ArrayHelpers.sol";
+import { CastingHelpers } from "@balancer-labs/v3-solidity-utils/contracts/helpers/CastingHelpers.sol";
+
+import { PoolMock } from "../../contracts/test/PoolMock.sol";
+import { BaseExtremeAmountsTest } from "./utils/BaseExtremeAmountsTest.sol";
+
+contract LinearBaseExtremeAmountsTest is BaseExtremeAmountsTest {
+    using ArrayHelpers for *;
+    using CastingHelpers for *;
+
+    function setUp() public virtual override {
+        BaseExtremeAmountsTest.setUp();
+    }
+
+    function _createPool(address[] memory tokens, string memory label) internal override returns (address) {
+        address newPool = address(new PoolMock(IVault(address(vault)), "ERC20 Pool - DAI/USDC", "ERC20_POOL_DAI_USDC"));
+        vm.label(newPool, label);
+
+        factoryMock.registerTestPool(newPool, vault.buildTokenConfig(tokens.asIERC20()), poolHooksContract, lp);
+
+        return address(newPool);
+    }
+}

pkg/vault/test/foundry/StablePoolExtremeAmountsWithMaxSwapFee.t.sol

@@ -0,0 +1,55 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+pragma solidity ^0.8.24;
+
+import { StablePoolFactory } from "@balancer-labs/v3-pool-stable/contracts/StablePoolFactory.sol";
+import { StablePool } from "@balancer-labs/v3-pool-stable/contracts/StablePool.sol";
+
+import { IBasePool } from "@balancer-labs/v3-interfaces/contracts/vault/IBasePool.sol";
+import { IVault } from "@balancer-labs/v3-interfaces/contracts/vault/IVault.sol";
+import { PoolRoleAccounts } from "@balancer-labs/v3-interfaces/contracts/vault/VaultTypes.sol";
+import { ArrayHelpers } from "@balancer-labs/v3-solidity-utils/contracts/test/ArrayHelpers.sol";
+import { CastingHelpers } from "@balancer-labs/v3-solidity-utils/contracts/helpers/CastingHelpers.sol";
+import { ProtocolFeeControllerMock } from "@balancer-labs/v3-vault/contracts/test/ProtocolFeeControllerMock.sol";
+
+import { PoolMock } from "../../contracts/test/PoolMock.sol";
+import { BaseExtremeAmountsTest } from "./utils/BaseExtremeAmountsTest.sol";
+
+contract StablePoolExtremeAmountsWithMaxSwapFeeTest is BaseExtremeAmountsTest {
+    using ArrayHelpers for *;
+    using CastingHelpers for *;
+
+    uint256 internal constant DEFAULT_AMP_FACTOR = 200;
+
+    function setUp() public virtual override {
+        BaseExtremeAmountsTest.setUp();
+    }
+
+    function _initMaxBPTAmount() internal pure override returns (uint256) {
+        return 1e12 * 1e18;
+    }
+
+    function _createPool(address[] memory tokens, string memory label) internal override returns (address) {
+        StablePoolFactory factory = new StablePoolFactory(IVault(address(vault)), 365 days, "Factory v1", "Pool v1");
+
+        PoolRoleAccounts memory roleAccounts;
+
+        StablePool newPool = StablePool(
+            factory.create(
+                "Stable Pool",
+                "STABLE",
+                vault.buildTokenConfig(tokens.asIERC20()),
+                DEFAULT_AMP_FACTOR,
+                roleAccounts,
+                MAX_SWAP_FEE, // Set max swap fee
+                address(0),
+                false, // Do not enable donations
+                false, // Do not disable unbalanced add/remove liquidity
+                ZERO_BYTES32
+            )
+        );
+        vm.label(address(newPool), label);
+
+        return address(newPool);
+    }
+}

pkg/vault/test/foundry/StablePoolExtremeAmountsWithMinSwapFee.t.sol

@@ -0,0 +1,55 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+pragma solidity ^0.8.24;
+
+import { StablePoolFactory } from "@balancer-labs/v3-pool-stable/contracts/StablePoolFactory.sol";
+import { StablePool } from "@balancer-labs/v3-pool-stable/contracts/StablePool.sol";
+
+import { IBasePool } from "@balancer-labs/v3-interfaces/contracts/vault/IBasePool.sol";
+import { IVault } from "@balancer-labs/v3-interfaces/contracts/vault/IVault.sol";
+import { PoolRoleAccounts } from "@balancer-labs/v3-interfaces/contracts/vault/VaultTypes.sol";
+import { ArrayHelpers } from "@balancer-labs/v3-solidity-utils/contracts/test/ArrayHelpers.sol";
+import { CastingHelpers } from "@balancer-labs/v3-solidity-utils/contracts/helpers/CastingHelpers.sol";
+import { ProtocolFeeControllerMock } from "@balancer-labs/v3-vault/contracts/test/ProtocolFeeControllerMock.sol";
+
+import { PoolMock } from "../../contracts/test/PoolMock.sol";
+import { BaseExtremeAmountsTest } from "./utils/BaseExtremeAmountsTest.sol";
+
+contract StablePoolExtremeAmountsWithMinSwapFeeTest is BaseExtremeAmountsTest {
+    using ArrayHelpers for *;
+    using CastingHelpers for *;
+
+    uint256 internal constant DEFAULT_AMP_FACTOR = 200;
+
+    function setUp() public virtual override {
+        BaseExtremeAmountsTest.setUp();
+    }
+
+    function _initMaxBPTAmount() internal pure override returns (uint256) {
+        return 1e12 * 1e18;
+    }
+
+    function _createPool(address[] memory tokens, string memory label) internal override returns (address) {
+        StablePoolFactory factory = new StablePoolFactory(IVault(address(vault)), 365 days, "Factory v1", "Pool v1");
+
+        PoolRoleAccounts memory roleAccounts;
+
+        StablePool newPool = StablePool(
+            factory.create(
+                "Stable Pool",
+                "STABLE",
+                vault.buildTokenConfig(tokens.asIERC20()),
+                DEFAULT_AMP_FACTOR,
+                roleAccounts,
+                MIN_SWAP_FEE, // Set min swap fee
+                address(0),
+                false, // Do not enable donations
+                false, // Do not disable unbalanced add/remove liquidity
+                ZERO_BYTES32
+            )
+        );
+        vm.label(address(newPool), label);
+
+        return address(newPool);
+    }
+}

pkg/vault/test/foundry/WeightedPoolExtremeAmountsWithMaxSwapFeeTest.t copy.sol

@@ -0,0 +1,52 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+pragma solidity ^0.8.24;
+
+import { WeightedPoolFactory } from "@balancer-labs/v3-pool-weighted/contracts/WeightedPoolFactory.sol";
+import { WeightedPool } from "@balancer-labs/v3-pool-weighted/contracts/WeightedPool.sol";
+
+import { IVault } from "@balancer-labs/v3-interfaces/contracts/vault/IVault.sol";
+import { PoolRoleAccounts } from "@balancer-labs/v3-interfaces/contracts/vault/VaultTypes.sol";
+import { ArrayHelpers } from "@balancer-labs/v3-solidity-utils/contracts/test/ArrayHelpers.sol";
+import { CastingHelpers } from "@balancer-labs/v3-solidity-utils/contracts/helpers/CastingHelpers.sol";
+import { ProtocolFeeControllerMock } from "@balancer-labs/v3-vault/contracts/test/ProtocolFeeControllerMock.sol";
+
+import { PoolMock } from "../../contracts/test/PoolMock.sol";
+import { BaseExtremeAmountsTest } from "./utils/BaseExtremeAmountsTest.sol";
+
+contract WeightedPoolExtremeAmountsWithMaxSwapFeeTest is BaseExtremeAmountsTest {
+    using ArrayHelpers for *;
+    using CastingHelpers for *;
+
+    function setUp() public virtual override {
+        BaseExtremeAmountsTest.setUp();
+    }
+
+    function _createPool(address[] memory tokens, string memory label) internal override returns (address) {
+        WeightedPoolFactory factory = new WeightedPoolFactory(
+            IVault(address(vault)),
+            365 days,
+            "Factory v1",
+            "Pool v1"
+        );
+        PoolRoleAccounts memory roleAccounts;
+
+        WeightedPool newPool = WeightedPool(
+            factory.create(
+                "50/50 Weighted Pool",
+                "50_50WP",
+                vault.buildTokenConfig(tokens.asIERC20()),
+                [uint256(50e16), uint256(50e16)].toMemoryArray(),
+                roleAccounts,
+                MAX_SWAP_FEE,
+                address(0),
+                false,
+                false,
+                bytes32(0)
+            )
+        );
+        vm.label(address(newPool), label);
+
+        return address(newPool);
+    }
+}

pkg/vault/test/foundry/WeightedPoolExtremeAmountsWithMinSwapFee.t.sol

@@ -0,0 +1,52 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+pragma solidity ^0.8.24;
+
+import { WeightedPoolFactory } from "@balancer-labs/v3-pool-weighted/contracts/WeightedPoolFactory.sol";
+import { WeightedPool } from "@balancer-labs/v3-pool-weighted/contracts/WeightedPool.sol";
+
+import { IVault } from "@balancer-labs/v3-interfaces/contracts/vault/IVault.sol";
+import { PoolRoleAccounts } from "@balancer-labs/v3-interfaces/contracts/vault/VaultTypes.sol";
+import { ArrayHelpers } from "@balancer-labs/v3-solidity-utils/contracts/test/ArrayHelpers.sol";
+import { CastingHelpers } from "@balancer-labs/v3-solidity-utils/contracts/helpers/CastingHelpers.sol";
+import { ProtocolFeeControllerMock } from "@balancer-labs/v3-vault/contracts/test/ProtocolFeeControllerMock.sol";
+
+import { PoolMock } from "../../contracts/test/PoolMock.sol";
+import { BaseExtremeAmountsTest } from "./utils/BaseExtremeAmountsTest.sol";
+
+contract WeightedPoolExtremeAmountsWithMinSwapFeeTest is BaseExtremeAmountsTest {
+    using ArrayHelpers for *;
+    using CastingHelpers for *;
+
+    function setUp() public virtual override {
+        BaseExtremeAmountsTest.setUp();
+    }
+
+    function _createPool(address[] memory tokens, string memory label) internal override returns (address) {
+        WeightedPoolFactory factory = new WeightedPoolFactory(
+            IVault(address(vault)),
+            365 days,
+            "Factory v1",
+            "Pool v1"
+        );
+        PoolRoleAccounts memory roleAccounts;
+
+        WeightedPool newPool = WeightedPool(
+            factory.create(
+                "50/50 Weighted Pool",
+                "50_50WP",
+                vault.buildTokenConfig(tokens.asIERC20()),
+                [uint256(50e16), uint256(50e16)].toMemoryArray(),
+                roleAccounts,
+                MIN_SWAP_FEE,
+                address(0),
+                false,
+                false,
+                bytes32(0)
+            )
+        );
+        vm.label(address(newPool), label);
+
+        return address(newPool);
+    }
+}