From f118e3b1adc8b5247b205181adb1e617959d6bee Mon Sep 17 00:00:00 2001
From: qedk <1994constant@gmail.com>
Date: Thu, 21 Mar 2024 04:34:45 +0400
Subject: [PATCH] feat: add functions to interface, remove proxy admin
 deployment from tests

---
 src/interfaces/IAvailBridge.sol | 16 ++++++++++++++++
 test/AvailBridgeTest.t.sol      |  5 +----
 2 files changed, 17 insertions(+), 4 deletions(-)

diff --git a/src/interfaces/IAvailBridge.sol b/src/interfaces/IAvailBridge.sol
index 5d4b005..d4c31a5 100644
--- a/src/interfaces/IAvailBridge.sol
+++ b/src/interfaces/IAvailBridge.sol
@@ -58,4 +58,20 @@ interface IAvailBridge {
     error InvalidMessage();
     error UnlockFailed();
     error WithdrawFailed();
+
+    function setPaused(bool status) external;
+    // no good way to have this function be in the import, so it's commented out
+    // function updateVectorx(address newVectorx) external;
+    function updateTokens(bytes32[] calldata assetIds, address[] calldata tokenAddresses) external;
+    function updateFeePerByte(uint256 newFeePerByte) external;
+    function updateFeeRecipient(address newFeeRecipient) external;
+    function withdrawFees() external;
+    function receiveMessage(Message calldata message, MerkleProofInput calldata input) external;
+    function receiveAVAIL(Message calldata message, MerkleProofInput calldata input) external;
+    function receiveETH(Message calldata message, MerkleProofInput calldata input) external;
+    function receiveERC20(Message calldata message, MerkleProofInput calldata input) external;
+    function sendMessage(bytes32 recipient, bytes calldata data) external payable;
+    function sendAVAIL(bytes32 recipient, uint256 amount) external;
+    function sendETH(bytes32 recipient) external payable;
+    function sendERC20(bytes32 assetId, bytes32 recipient, uint256 amount) external;
 }
diff --git a/test/AvailBridgeTest.t.sol b/test/AvailBridgeTest.t.sol
index 9f0d77a..121d6b2 100644
--- a/test/AvailBridgeTest.t.sol
+++ b/test/AvailBridgeTest.t.sol
@@ -3,7 +3,6 @@ pragma solidity ^0.8.25;
 
 import {TransparentUpgradeableProxy} from
     "lib/openzeppelin-contracts/contracts/proxy/transparent/TransparentUpgradeableProxy.sol";
-import {ProxyAdmin} from "lib/openzeppelin-contracts/contracts/proxy/transparent/ProxyAdmin.sol";
 import {IAccessControl} from "lib/openzeppelin-contracts/contracts/access/IAccessControl.sol";
 import {Pausable} from "lib/openzeppelin-contracts/contracts/utils/Pausable.sol";
 import {IAvailBridge, AvailBridge} from "src/AvailBridge.sol";
@@ -18,7 +17,6 @@ contract AvailBridgeTest is Test, MurkyBase {
     AvailBridge public bridge;
     Avail public avail;
     VectorxMock public vectorx;
-    ProxyAdmin public admin;
     Sha2Merkle public sha2merkle;
     address public owner;
     address public pauser;
@@ -26,11 +24,10 @@ contract AvailBridgeTest is Test, MurkyBase {
 
     function setUp() external {
         vectorx = new VectorxMock();
-        admin = new ProxyAdmin(msg.sender);
         pauser = makeAddr("pauser");
         sha2merkle = new Sha2Merkle();
         address impl = address(new AvailBridge());
-        bridge = AvailBridge(address(new TransparentUpgradeableProxy(impl, address(admin), "")));
+        bridge = AvailBridge(address(new TransparentUpgradeableProxy(impl, msg.sender, "")));
         avail = new Avail(address(bridge));
         bridge.initialize(0, msg.sender, IAvail(address(avail)), msg.sender, pauser, IVectorx(vectorx));
         owner = msg.sender;