toArray function tests

Author: NouDaimon

contracts/data/EnumerableMapAddressToAddressMock.sol

@@ -32,4 +32,12 @@ contract EnumerableMapAddressToAddressMock {
     function remove(address key) external returns (bool) {
         return map.remove(key);
     }
+
+    function toArray()
+        external
+        view
+        returns (address[] memory keys, address[] memory values)
+    {
+        (keys, values) = map.toArray();
+    }
 }

contracts/data/EnumerableMapUintToAddressMock.sol

@@ -32,4 +32,12 @@ contract EnumerableMapUintToAddressMock {
     function remove(uint256 key) external returns (bool) {
         return map.remove(key);
     }
+
+    function toArray()
+        external
+        view
+        returns (uint256[] memory keys, address[] memory values)
+    {
+        (keys, values) = map.toArray();
+    }
 }

test/data/EnumerableMap.ts

@@ -187,6 +187,19 @@ describe('EnumerableMap', () => {
             .false;
         });
       });
+
+      describe('#toArray()', () => {
+        it('returns arrays of keys and values in map', async () => {
+          await instance['set(address,address)'](addressOne, addressFour);
+          await instance['set(address,address)'](addressTwo, addressFive);
+          await instance['set(address,address)'](addressThree, addressSix);
+
+          const [keys, values] = await instance.callStatic['toArray()']();
+
+          expect(keys).to.deep.equal([addressOne, addressTwo, addressThree]);
+          expect(values).to.deep.equal([addressFour, addressFive, addressSix]);
+        });
+      });
     });
   });
 
@@ -365,6 +378,19 @@ describe('EnumerableMap', () => {
             .false;
         });
       });
+
+      describe('#toArray()', () => {
+        it('returns arrays of keys and values in map', async () => {
+          await instance['set(uint256,address)'](uintOne, addressOne);
+          await instance['set(uint256,address)'](uintTwo, addressTwo);
+          await instance['set(uint256,address)'](uintThree, addressThree);
+
+          const [keys, values] = await instance.callStatic['toArray()']();
+
+          expect(keys).to.deep.equal([uintOne, uintTwo, uintThree]);
+          expect(values).to.deep.equal([addressOne, addressTwo, addressThree]);
+        });
+      });
     });
   });
 });