Commit a1c2987e authored by Maurelian's avatar Maurelian

safe-tools: Make address label prefix configurable

parent ba0c2ac9
...@@ -231,7 +231,8 @@ contract LivenessGuard_FuzzOwnerManagement_Test is StdCheats, StdUtils, Liveness ...@@ -231,7 +231,8 @@ contract LivenessGuard_FuzzOwnerManagement_Test is StdCheats, StdUtils, Liveness
threshold = bound(threshold, 1, initialOwners); threshold = bound(threshold, 1, initialOwners);
// Generate the initial owners and keys and setup the safe // Generate the initial owners and keys and setup the safe
(address[] memory ownerAddrs, uint256[] memory ownerkeys) = SafeTestLib.makeAddrsAndKeys(initialOwners); (address[] memory ownerAddrs, uint256[] memory ownerkeys) =
SafeTestLib.makeAddrsAndKeys("safeTest", initialOwners);
// record the private keys for later use // record the private keys for later use
for (uint256 i = 0; i < ownerAddrs.length; i++) { for (uint256 i = 0; i < ownerAddrs.length; i++) {
privateKeys[ownerAddrs[i]] = ownerkeys[i]; privateKeys[ownerAddrs[i]] = ownerkeys[i];
......
...@@ -47,7 +47,7 @@ contract LivenessModule_TestInit is Test, SafeTestTools { ...@@ -47,7 +47,7 @@ contract LivenessModule_TestInit is Test, SafeTestTools {
vm.warp(initTime); vm.warp(initTime);
// Create a Safe with 10 owners // Create a Safe with 10 owners
(, uint256[] memory keys) = SafeTestLib.makeAddrsAndKeys(10); (, uint256[] memory keys) = SafeTestLib.makeAddrsAndKeys("moduleTest", 10);
safeInstance = _setupSafe(keys, 8); safeInstance = _setupSafe(keys, 8);
livenessGuard = new LivenessGuard(safeInstance.safe); livenessGuard = new LivenessGuard(safeInstance.safe);
......
...@@ -39,7 +39,7 @@ contract SafeSigners_Test is Test, SafeTestTools { ...@@ -39,7 +39,7 @@ contract SafeSigners_Test is Test, SafeTestTools {
// Limit the number of signatures to 25 // Limit the number of signatures to 25
uint256 numSigs = bound(_numSigs, 1, 25); uint256 numSigs = bound(_numSigs, 1, 25);
(, uint256[] memory keys) = SafeTestLib.makeAddrsAndKeys(numSigs); (, uint256[] memory keys) = SafeTestLib.makeAddrsAndKeys("getSigsTest", numSigs);
for (uint256 i = 0; i < keys.length; i++) { for (uint256 i = 0; i < keys.length; i++) {
if (sigType(keys[i]) == SigTypes.Contract) { if (sigType(keys[i]) == SigTypes.Contract) {
keys[i] = keys[i] =
......
...@@ -69,7 +69,13 @@ library SafeTestLib { ...@@ -69,7 +69,13 @@ library SafeTestLib {
} }
/// @dev Get arrays of addresses and private keys. The arrays are sorted by address, and the addresses are labelled /// @dev Get arrays of addresses and private keys. The arrays are sorted by address, and the addresses are labelled
function makeAddrsAndKeys(uint256 num) internal returns (address[] memory addrs, uint256[] memory keys) { function makeAddrsAndKeys(
string memory prefix,
uint256 num
)
internal
returns (address[] memory addrs, uint256[] memory keys)
{
keys = new uint256[](num); keys = new uint256[](num);
addrs = new address[](num); addrs = new address[](num);
for (uint256 i; i < num; i++) { for (uint256 i; i < num; i++) {
...@@ -79,7 +85,7 @@ library SafeTestLib { ...@@ -79,7 +85,7 @@ library SafeTestLib {
for (uint256 i; i < num; i++) { for (uint256 i; i < num; i++) {
addrs[i] = Vm(VM_ADDR).addr(keys[i]); addrs[i] = Vm(VM_ADDR).addr(keys[i]);
Vm(VM_ADDR).label(getAddr(keys[i]), string.concat("SAFETEST: Signer ", Vm(VM_ADDR).toString(i))); Vm(VM_ADDR).label(getAddr(keys[i]), string.concat(prefix, Vm(VM_ADDR).toString(i)));
} }
} }
...@@ -592,7 +598,7 @@ contract SafeTestTools { ...@@ -592,7 +598,7 @@ contract SafeTestTools {
} }
function _setupSafe() public returns (SafeInstance memory) { function _setupSafe() public returns (SafeInstance memory) {
(, uint256[] memory defaultPKs) = SafeTestLib.makeAddrsAndKeys(3); (, uint256[] memory defaultPKs) = SafeTestLib.makeAddrsAndKeys("default", 3);
return _setupSafe( return _setupSafe(
defaultPKs, defaultPKs,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment