test: refactor tests to use Unit_Test

sablier-labs · Jan 10, 2025 · b0f7f74 · b0f7f74
1 parent 612bb77
commit b0f7f74
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 11 deletions.
diff --git a/tests/unit/adminable/transferAdmin.t.sol b/tests/unit/adminable/transferAdmin.t.sol
@@ -4,16 +4,14 @@ pragma solidity >=0.8.22 <0.9.0;
 import { IAdminable } from "src/interfaces/IAdminable.sol";
 import { Errors } from "src/libraries/Errors.sol";
 
-import { CommonBase } from "../../Base.t.sol";
+import { Unit_Test } from "../Unit.t.sol";
 import { AdminableMock } from "../../mocks/AdminableMock.sol";
 
-contract TransferAdmin_Unit_Concrete_Test is CommonBase {
+contract TransferAdmin_Unit_Concrete_Test is Unit_Test {
     AdminableMock internal adminableMock;
-    address internal admin;
-    address internal eve;
 
     function setUp() public virtual override {
-        CommonBase.setUp();
+        Unit_Test.setUp();
 
         admin = createUser("admin");
         eve = createUser("eve");

diff --git a/tests/unit/batch/batch.t.sol b/tests/unit/batch/batch.t.sol
@@ -1,17 +1,17 @@
 // SPDX-License-Identifier: UNLICENSED
 pragma solidity >=0.8.22;
 
-import { CommonBase } from "../../Base.t.sol";
+import { Unit_Test } from "../Unit.t.sol";
 import { BatchMock } from "../../mocks/BatchMock.sol";
 
-contract Batch_Unit_Concrete_Test is CommonBase {
+contract Batch_Unit_Concrete_Test is Unit_Test {
     BatchMock internal batchMock;
     bytes[] internal calls;
     uint256 internal newNumber = 100;
     bytes[] internal results;
 
     function setUp() public virtual override {
-        CommonBase.setUp();
+        Unit_Test.setUp();
 
         batchMock = new BatchMock();
     }

diff --git a/tests/unit/no-delegate-call/noDelegateCall.t.sol b/tests/unit/no-delegate-call/noDelegateCall.t.sol
@@ -3,14 +3,14 @@ pragma solidity >=0.8.22;
 
 import { Errors } from "src/libraries/Errors.sol";
 
-import { CommonBase } from "../../Base.t.sol";
+import { Unit_Test } from "../Unit.t.sol";
 import { NoDelegateCallMock } from "../../mocks/NoDelegateCallMock.sol";
 
-contract NoDelegateCall_Unit_Concrete_Test is CommonBase {
+contract NoDelegateCall_Unit_Concrete_Test is Unit_Test {
     NoDelegateCallMock internal noDelegateCallMock;
 
     function setUp() public virtual override {
-        CommonBase.setUp();
+        Unit_Test.setUp();
 
         noDelegateCallMock = new NoDelegateCallMock();
     }