From b3c1e7257721dc5286219b14a3a57b409fd1370d Mon Sep 17 00:00:00 2001
From: Amir Bandeali <abandeali1@gmail.com>
Date: Wed, 20 Nov 2019 16:43:02 -0800
Subject: [PATCH] Warn if Exchange contract not registered in StakingProxy

---
 packages/migrations/src/test_contract_configs.ts | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/packages/migrations/src/test_contract_configs.ts b/packages/migrations/src/test_contract_configs.ts
index 305119999a..48ea63d067 100644
--- a/packages/migrations/src/test_contract_configs.ts
+++ b/packages/migrations/src/test_contract_configs.ts
@@ -262,6 +262,9 @@ async function testContractConfigsAsync(provider: SupportedProvider): Promise<vo
         const stakingLogicAddress = await stakingProxy.stakingContract().callAsync();
         warnIfMismatch(stakingLogicAddress, addresses.staking, 'Unexpected Staking contract attached to StakingProxy');
 
+        const isExchangeRegistered = await stakingContract.validExchanges(addresses.exchange).callAsync();
+        warnIfMismatch(isExchangeRegistered, true, 'Exchange not registered in StakingProxy');
+
         const zrxVaultAddress = await stakingContract.getZrxVault().callAsync();
         warnIfMismatch(zrxVaultAddress, addresses.zrxVault, 'Unexpected ZrxVault set in StakingProxy');