refactor: add getter for the query library

balancer · Oct 27, 2023 · e8bd2ea · e8bd2ea
1 parent df4ec3a
commit e8bd2ea
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/pkg/interfaces/contracts/standalone-utils/IBalancerRelayer.sol b/pkg/interfaces/contracts/standalone-utils/IBalancerRelayer.sol
@@ -24,6 +24,8 @@ import "../vault/IVault.sol";
 interface IBalancerRelayer {
     function getLibrary() external view returns (address);
 
+    function getQueryLibrary() external view returns (address);
+
     function getVault() external view returns (IVault);
 
     function multicall(bytes[] calldata data) external payable returns (bytes[] memory results);

diff --git a/pkg/standalone-utils/contracts/relayer/BalancerRelayer.sol b/pkg/standalone-utils/contracts/relayer/BalancerRelayer.sol
@@ -82,6 +82,10 @@ contract BalancerRelayer is IBalancerRelayer, Version, ReentrancyGuard {
         return _library;
     }
 
+    function getQueryLibrary() external view override returns (address) {
+        return _queryLibrary;
+    }
+
     function multicall(bytes[] calldata data) external payable override nonReentrant returns (bytes[] memory results) {
         uint256 numData = data.length;
 

diff --git a/pkg/standalone-utils/test/BaseRelayerLibrary.test.ts b/pkg/standalone-utils/test/BaseRelayerLibrary.test.ts
@@ -9,7 +9,7 @@ import { deploy, deployedAt } from '@balancer-labs/v2-helpers/src/contract';
 import * as expectEvent from '@balancer-labs/v2-helpers/src/test/expectEvent';
 import { actionId } from '@balancer-labs/v2-helpers/src/models/misc/actions';
 
-import { ANY_ADDRESS, MAX_UINT256 } from '@balancer-labs/v2-helpers/src/constants';
+import { ANY_ADDRESS, MAX_UINT256, ZERO_ADDRESS } from '@balancer-labs/v2-helpers/src/constants';
 import Vault from '@balancer-labs/v2-helpers/src/models/vault/Vault';
 import { BigNumberish, bn, fp } from '@balancer-labs/v2-helpers/src/numbers';
 import { toChainedReference } from './helpers/chainedReferences';
@@ -48,6 +48,10 @@ describe('BaseRelayerLibrary', function () {
       expect(await relayer.getLibrary()).to.equal(relayerLibrary.address);
     });
 
+    it('returns the query library address', async () => {
+      expect(await relayer.getQueryLibrary()).not.to.equal(ZERO_ADDRESS);
+    });
+
     it('returns the vault address', async () => {
       expect(await relayer.getVault()).to.equal(vault.address);
     });