Deploy the linked list helper only for tests

rocket-pool · Aug 30, 2024 · 30e1cda · 30e1cda
1 parent d965f20
commit 30e1cda
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/test/_helpers/deployment.js b/test/_helpers/deployment.js
@@ -1,5 +1,5 @@
 /*** Dependencies ********************/
-import { LinkedListStorage, RocketStorage } from '../_utils/artifacts';
+import { RocketStorage } from '../_utils/artifacts';
 
 const hre = require('hardhat');
 const pako = require('pako');
@@ -100,13 +100,18 @@ const contracts = {
     rocketDAOProtocolProposal:                artifacts.require('RocketDAOProtocolProposal.sol'),
     // Utils
     addressQueueStorage:                      artifacts.require('AddressQueueStorage.sol'),
-    LinkedListStorage:                        artifacts.require('LinkedListStorageHelper.sol'),
     addressSetStorage:                        artifacts.require('AddressSetStorage.sol'),
 };
 
 // Development helper contracts
 const revertOnTransfer = artifacts.require('RevertOnTransfer.sol');
 const rocketNodeDepositLEB4 = artifacts.require('RocketNodeDepositLEB4.sol');
+if (network.name !== 'live' && network.name !== 'goerli') {
+    // the linked list storage helper needs to be added as a network contract
+    contracts.linkedListStorage = artifacts.require('LinkedListStorageHelper.sol');
+} else {
+    contracts.linkedListStorage = artifacts.require('LinkedListStorage.sol');
+}
 
 // Instance contract ABIs
 const abis = {

diff --git a/test/util/util-tests.js b/test/util/util-tests.js
@@ -22,7 +22,7 @@ export default function() {
 
         });
 
-        it.only(printTitle('random', 'pack/unpack shouldnt change values'), async () => {
+        it(printTitle('random', 'pack/unpack shouldnt change values'), async () => {
             const linkedListStorage = await LinkedListStorage.deployed();            
             let item = {
                 receiver: random,
@@ -79,7 +79,7 @@ export default function() {
             assert.equal(listLength, 0)
         });
 
-        it.only(printTitle('random', 'can remove the only queue item'), async () => {
+        it(printTitle('random', 'can remove the only queue item'), async () => {
             const linkedListStorage = await LinkedListStorage.deployed();
             let itemIn = {
                 receiver: random,