Changed SimplePriceOracle so ETH price can be set on testnets (#151)

* Changed SimplePriceOracle so ETH price can be set on testnets * fix lens test
compound-finance · Nov 16, 2021 · ecc4ab8 · ecc4ab8
1 parent 9e93e85
commit ecc4ab8
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 7 deletions.
diff --git a/contracts/SimplePriceOracle.sol b/contracts/SimplePriceOracle.sol
@@ -7,16 +7,22 @@ contract SimplePriceOracle is PriceOracle {
     mapping(address => uint) prices;
     event PricePosted(address asset, uint previousPriceMantissa, uint requestedPriceMantissa, uint newPriceMantissa);
 
-    function getUnderlyingPrice(CToken cToken) public view returns (uint) {
+    function _getUnderlyingAddress(CToken cToken) private view returns (address) {
+        address asset;
         if (compareStrings(cToken.symbol(), "cETH")) {
-            return 1e18;
+            asset = 0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE;
         } else {
-            return prices[address(CErc20(address(cToken)).underlying())];
+            asset = address(CErc20(address(cToken)).underlying());
         }
+        return asset;
+    }
+
+    function getUnderlyingPrice(CToken cToken) public view returns (uint) {
+        return prices[_getUnderlyingAddress(cToken)];
     }
 
     function setUnderlyingPrice(CToken cToken, uint underlyingPriceMantissa) public {
-        address asset = address(CErc20(address(cToken)).underlying());
+        address asset = _getUnderlyingAddress(cToken);
         emit PricePosted(asset, prices[asset], underlyingPriceMantissa, underlyingPriceMantissa);
         prices[asset] = underlyingPriceMantissa;
     }

diff --git a/networks/ropsten.json b/networks/ropsten.json
@@ -61,7 +61,8 @@
         "cZRX": "0x6B8b0D7875B4182Fb126877023fB93b934dD302A",
         "cWBTC": "0x541c9cB0E97b77F142684cc33E8AC9aC17B1990F",
         "USDC": "0x07865c6e87b9f70255377e024ace6630c1eaa37f",
-        "ComptrollerRopsten": "0x1435b33741523dd5BB99f8FD8Dc00cF926fE96Fb"
+        "ComptrollerRopsten": "0x1435b33741523dd5BB99f8FD8Dc00cF926fE96Fb",
+        "SimplePriceOracle": "0x7BBF806F69ea21Ea9B8Af061AD0C1c41913063A1"
     },
     "Blocks": {
         "ComptrollerG8": 10279074,

diff --git a/tests/Lens/CompoundLensTest.js b/tests/Lens/CompoundLensTest.js
@@ -245,7 +245,7 @@ describe('CompoundLens', () => {
       ).toEqual(
         {
           cToken: cEth._address,
-          underlyingPrice: "1000000000000000000",
+          underlyingPrice: "0",
         }
       );
     });
@@ -264,7 +264,7 @@ describe('CompoundLens', () => {
         },
         {
           cToken: cEth._address,
-          underlyingPrice: "1000000000000000000",
+          underlyingPrice: "0",
         }
       ]);
     });