diff --git a/modules/sdk-core/src/bitgo/defi/defiVault.ts b/modules/sdk-core/src/bitgo/defi/defiVault.ts index ec9641ec75..59a4f8e281 100644 --- a/modules/sdk-core/src/bitgo/defi/defiVault.ts +++ b/modules/sdk-core/src/bitgo/defi/defiVault.ts @@ -65,6 +65,7 @@ export class DefiVault implements IDefiVault { } const raw = await this.bitgo .get(this.bitgo.microservicesUrl(`/api/defi-service/v1/vaults/${params.vaultId}`)) + .set('enterprise-id', this.wallet.toJSON().enterprise) .result(); return decodeWithCodec(GetVaultResponse, raw, 'getVaultConfig'); } diff --git a/modules/sdk-core/test/unit/bitgo/defi/defiVault.ts b/modules/sdk-core/test/unit/bitgo/defi/defiVault.ts index 4e91d40699..900e895884 100644 --- a/modules/sdk-core/test/unit/bitgo/defi/defiVault.ts +++ b/modules/sdk-core/test/unit/bitgo/defi/defiVault.ts @@ -14,6 +14,7 @@ describe('DefiVault', function () { function mockRequest(result: any) { return { send: sinon.stub().returnsThis(), + set: sinon.stub().returnsThis(), query: sinon.stub().returnsThis(), result: sinon.stub().resolves(result), }; @@ -71,6 +72,7 @@ describe('DefiVault', function () { id: 'test-wallet-id', coin: 'eth', keys: ['user-key', 'backup-key', 'bitgo-key'], + enterprise: 'test-enterprise-id', }; wallet = new Wallet(mockBitGo, mockBaseCoin, mockWalletData); @@ -110,6 +112,7 @@ describe('DefiVault', function () { result.should.deepEqual(vaultConfig); mockBitGo.get.calledWith('https://bitgo.com/api/defi-service/v1/vaults/vlt-concrete-1').should.be.true(); + req.set.calledWith('enterprise-id', 'test-enterprise-id').should.be.true(); }); it('should throw if vaultId is missing', async function () {