♻️ Make Pod mothership internal (#1101)

Vectorized · Oct 2, 2024 · 68989f9 · 68989f9
1 parent 0411fd1
commit 68989f9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/accounts/Pod.sol b/src/accounts/Pod.sol
@@ -43,7 +43,7 @@ abstract contract Pod is Receiver {
     /*.•°:°.´+˚.*°.˚:*.´•*.+°.•°:´*.´•*.•°.•°:°.´:•˚°.*°.˚:*.´+°.•*/
 
     /// @dev Returns the mothership contract.
-    function mothership() public view virtual returns (address result) {
+    function _mothership() internal view virtual returns (address result) {
         /// @solidity memory-safe-assembly
         assembly {
             result := shr(96, sload(_MOTHERSHIP_SLOT))
@@ -78,7 +78,7 @@ abstract contract Pod is Receiver {
     /// @dev Requires that the caller is the mothership.
     /// This is called in the `onlyMothership` modifier.
     function _checkMothership() internal view virtual {
-        if (msg.sender != mothership()) revert CallerNotMothership();
+        if (msg.sender != _mothership()) revert CallerNotMothership();
     }
 
     /// @dev Requires that the caller is the mothership.

diff --git a/test/utils/mocks/MockPod.sol b/test/utils/mocks/MockPod.sol
@@ -7,6 +7,10 @@ import {Brutalizer} from "../Brutalizer.sol";
 /// @dev WARNING! This mock is strictly intended for testing purposes only.
 /// Do NOT copy anything here into production code unless you really know what you are doing.
 contract MockPod is Pod, Brutalizer {
+    function mothership() public view returns (address) {
+        return _mothership();
+    }
+
     function initializeMothership(address initialMothership) public {
         _initializeMothership(_brutalized(initialMothership));
     }