(test): Add unit test for size of ToS constructor address lists

smartcontractkit · Jan 19, 2024 · 19e29d9 · 19e29d9
1 parent 48413a2
commit 19e29d9
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/contracts/src/v0.8/functions/dev/v1_X/accessControl/TermsOfServiceAllowList.sol b/contracts/src/v0.8/functions/dev/v1_X/accessControl/TermsOfServiceAllowList.sol
@@ -39,6 +39,8 @@ contract TermsOfServiceAllowList is ITermsOfServiceAllowList, IAccessController,
   // |                       Initialization                         |
   // ================================================================
 
+  /// @dev Initial allowed senders & blocked senders have a combined limit of around 500k addresses
+  /// Beyond that, migrations will need to handled in batches
   constructor(
     TermsOfServiceAllowListConfig memory config,
     address[] memory initialAllowedSenders,

diff --git a/contracts/src/v0.8/functions/tests/v1_X/FunctionsTermsOfServiceAllowList.t.sol b/contracts/src/v0.8/functions/tests/v1_X/FunctionsTermsOfServiceAllowList.t.sol
@@ -10,10 +10,33 @@ import "forge-std/Vm.sol";
 
 /// @notice #constructor
 contract FunctionsTermsOfServiceAllowList_Constructor is FunctionsRoutesSetup {
+  uint256 internal constant initalSendersEntries = 100000; // ~500k limit, keeping lower for test speed
+  address[] internal s_initialAllowedSenders;
+  address[] internal s_initialBlockedSenders;
+
+  function setUp() public virtual override {
+    FunctionsRoutesSetup.setUp();
+
+    for (uint256 i = 1; i <= initalSendersEntries; ++i) {
+      s_initialAllowedSenders.push(vm.addr(i));
+    }
+  }
+
   function test_Constructor_Success() public {
     assertEq(s_termsOfServiceAllowList.typeAndVersion(), "Functions Terms of Service Allow List v1.1.0");
     assertEq(s_termsOfServiceAllowList.owner(), OWNER_ADDRESS);
   }
+
+  function test_Constructor_InitializationSuccess() public {
+
+    TermsOfServiceAllowList _termsOfServiceAllowList = new TermsOfServiceAllowList(
+      getTermsOfServiceConfig(),
+      s_initialAllowedSenders,
+      s_initialBlockedSenders
+    );
+
+    assertEq(_termsOfServiceAllowList.getAllowedSendersCount(), initalSendersEntries);
+  }
 }
 
 /// @notice #getConfig