From f4e5e3339827e48287c0471aa7ac2294b725474b Mon Sep 17 00:00:00 2001
From: bun919tw <bun919tw@gmail.com>
Date: Tue, 12 Oct 2021 17:06:55 +0800
Subject: [PATCH] contracts/FlashloanLender.sol: add transferOwnership

---
 contracts/FlashloanLender.sol |  5 +++++
 tests/Tokens/flashloanTest.js | 21 ++++++++++++++-------
 2 files changed, 19 insertions(+), 7 deletions(-)

diff --git a/contracts/FlashloanLender.sol b/contracts/FlashloanLender.sol
index 7af8fcc7a..de62e6014 100644
--- a/contracts/FlashloanLender.sol
+++ b/contracts/FlashloanLender.sol
@@ -60,6 +60,11 @@ contract FlashloanLender is ERC3156FlashLenderInterface {
         return CCollateralCapErc20(cToken).flashLoan(receiver, msg.sender, amount, data);
     }
 
+    function transferOwnership(address newOwner) external onlyOwner {
+        require(newOwner != address(0), "new owner cannot be zero address");
+        owner = newOwner;
+    }
+
     function updateUnderlyingMapping(CToken[] calldata cTokens) external onlyOwner returns (bool) {
         uint256 cTokenLength = cTokens.length;
         for (uint256 i = 0; i < cTokenLength; i++) {
diff --git a/tests/Tokens/flashloanTest.js b/tests/Tokens/flashloanTest.js
index 527d2601d..fa404f146 100644
--- a/tests/Tokens/flashloanTest.js
+++ b/tests/Tokens/flashloanTest.js
@@ -2,7 +2,7 @@ const {
   etherUnsigned,
   etherMantissa,
   mergeInterface,
-  unlockedAccount
+  address,
 } = require('../Utils/Ethereum');
 
 const {
@@ -25,6 +25,7 @@ describe('Flashloan test', function () {
   beforeEach(async () => {
     admin = saddle.accounts[0];
     nonAdmin = saddle.accounts[1];
+    other = saddle.accounts[2];
     cToken = await makeCToken({kind: 'ccollateralcap', supportMarket: true});
     flashloanReceiver = await makeFlashloanReceiver()
     flashloanLender = await deploy('FlashloanLender', [cToken.comptroller._address, admin]);
@@ -40,7 +41,7 @@ describe('Flashloan test', function () {
 
     await send(cToken.underlying, 'harnessSetBalance', [flashloanReceiver._address, receiverBalance]);
   });
-  
+
   describe('test FlashLoanLender interface', ()=>{
     let unsupportedCToken;
     beforeEach(async () => {
@@ -60,25 +61,31 @@ describe('Flashloan test', function () {
       await expect(call(flashloanLender, 'flashFee', [unsupportedCToken.underlying._address, borrowAmount])).rejects.toRevert('revert cannot find cToken of this underlying in the mapping');
       expect(await call(flashloanLender, 'flashFee', [cToken.underlying._address, borrowAmount])).toEqualNumber(totalFee);
     });
-    it("test updateUnderlyingMapping", async () => 
+    it("test updateUnderlyingMapping", async () =>
     {
       const borrowAmount = 10_000;
       const totalFee = 3
       await send(flashloanLender, 'updateUnderlyingMapping', [[unsupportedCToken._address]]);
       expect(await call(flashloanLender, 'flashFee', [cToken.underlying._address, borrowAmount])).toEqualNumber(totalFee);
     });
-    it("test removeUnderlyingMapping", async () => 
+    it("test removeUnderlyingMapping", async () =>
     {
       const borrowAmount = 10_000;
       await send(flashloanLender, 'removeUnderlyingMapping', [[cToken._address]]);
       await expect(call(flashloanLender, 'flashFee', [cToken.underlying._address, borrowAmount])).rejects.toRevert('revert cannot find cToken of this underlying in the mapping');
     });
-    it("test onlyOwner", async () => 
+    it("test onlyOwner", async () =>
     {
       await expect(send(flashloanLender, 'updateUnderlyingMapping', [[unsupportedCToken._address]], {from:nonAdmin})).rejects.toRevert("revert not owner");
     });
-    
-
+    it("test transferOwnership", async () =>
+    {
+      await expect(send(flashloanLender, 'transferOwnership', [other], {from:nonAdmin})).rejects.toRevert("revert not owner");
+    });
+    it("test transferOwnership with zero address", async () =>
+    {
+      await expect(send(flashloanLender, 'transferOwnership', [address(0)], {from:admin})).rejects.toRevert("revert new owner cannot be zero address");
+    });
   })
 
   describe('internal cash equal underlying balance', () => {