From f5d0ce14bbb4313f90aafd5c752345bb6284669c Mon Sep 17 00:00:00 2001
From: David Racero <canillas.mail@gmail.com>
Date: Wed, 2 Dec 2020 09:28:55 +0100
Subject: [PATCH] Fix verification tasks

---
 tasks/verifications/1_general.ts | 12 +++++++++---
 tasks/verifications/2_tokens.ts  | 10 ++++++++--
 2 files changed, 17 insertions(+), 5 deletions(-)

diff --git a/tasks/verifications/1_general.ts b/tasks/verifications/1_general.ts
index f0f8374b..16e7fb6d 100644
--- a/tasks/verifications/1_general.ts
+++ b/tasks/verifications/1_general.ts
@@ -1,6 +1,11 @@
 import { zeroAddress } from 'ethereumjs-util';
 import { task } from 'hardhat/config';
-import { loadPoolConfig, ConfigNames, getWethAddress } from '../../helpers/configuration';
+import {
+  loadPoolConfig,
+  ConfigNames,
+  getWethAddress,
+  getTreasuryAddress,
+} from '../../helpers/configuration';
 import { ZERO_ADDRESS } from '../../helpers/constants';
 import {
   getAaveProtocolDataProvider,
@@ -34,6 +39,7 @@ task('verify:general', 'Deploy oracles for dev enviroment')
       ProviderRegistry,
       MarketId,
     } = poolConfig as ICommonConfiguration;
+    const treasuryAddress = await getTreasuryAddress(poolConfig);
 
     const registryAddress = getParamPerNetwork(ProviderRegistry, network);
     const addressesProvider = await getLendingPoolAddressesProvider();
@@ -146,7 +152,7 @@ task('verify:general', 'Deploy oracles for dev enviroment')
     await verifyContract(aDAI, [
       lendingPoolProxy.address,
       DAI,
-      ZERO_ADDRESS,
+      treasuryAddress,
       'Aave interest bearing DAI',
       'aDAI',
       ZERO_ADDRESS,
@@ -176,7 +182,7 @@ task('verify:general', 'Deploy oracles for dev enviroment')
     await verifyContract(aUNI, [
       lendingPoolProxy.address,
       UNI,
-      ZERO_ADDRESS,
+      treasuryAddress,
       'Aave interest bearing UNI',
       'aUNI',
       ZERO_ADDRESS,
diff --git a/tasks/verifications/2_tokens.ts b/tasks/verifications/2_tokens.ts
index 8ec27d9f..f50c3cee 100644
--- a/tasks/verifications/2_tokens.ts
+++ b/tasks/verifications/2_tokens.ts
@@ -1,5 +1,10 @@
 import { task } from 'hardhat/config';
-import { loadPoolConfig, ConfigNames, getWethAddress } from '../../helpers/configuration';
+import {
+  loadPoolConfig,
+  ConfigNames,
+  getWethAddress,
+  getTreasuryAddress,
+} from '../../helpers/configuration';
 import { ZERO_ADDRESS } from '../../helpers/constants';
 import {
   getAddressById,
@@ -18,6 +23,7 @@ task('verify:tokens', 'Deploy oracles for dev enviroment')
     const network = localDRE.network.name as eEthereumNetwork;
     const poolConfig = loadPoolConfig(pool);
     const { ReserveAssets, ReservesConfig } = poolConfig as ICommonConfiguration;
+    const treasuryAddress = await getTreasuryAddress(poolConfig);
 
     const addressesProvider = await getLendingPoolAddressesProvider();
     const lendingPoolProxy = await getLendingPool();
@@ -82,7 +88,7 @@ task('verify:tokens', 'Deploy oracles for dev enviroment')
       await verifyContract(aToken, [
         lendingPoolProxy.address,
         tokenAddress,
-        ZERO_ADDRESS,
+        treasuryAddress,
         `Aave interest bearing ${token}`,
         `a${token}`,
         ZERO_ADDRESS,