From 1af90e116c88e1bdaa6143cf1eaaedb741359dfe Mon Sep 17 00:00:00 2001 From: niv vaknin Date: Sun, 29 Dec 2024 11:17:20 +0200 Subject: [PATCH] Fix testcase --- Quorum/tests/checks/test_price_feed.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/Quorum/tests/checks/test_price_feed.py b/Quorum/tests/checks/test_price_feed.py index 9e98ea4..e6dc266 100644 --- a/Quorum/tests/checks/test_price_feed.py +++ b/Quorum/tests/checks/test_price_feed.py @@ -53,7 +53,12 @@ def test_source_code_clean(): } """ cleaned = Checks.price_feed.remove_solidity_comments(code) - expected = """contract AaveV2Ethereum_ReserveFactorUpdatesMidJuly_20240711 is IProposalGenericExecutor { + expected = """pragma solidity ^0.8.0; + +import {IProposalGenericExecutor} from 'aave-helpers/interfaces/IProposalGenericExecutor.sol'; +import {AaveV2Ethereum, AaveV2EthereumAssets, ILendingPoolConfigurator} from 'aave-address-book/AaveV2Ethereum.sol'; + +contract AaveV2Ethereum_ReserveFactorUpdatesMidJuly_20240711 is IProposalGenericExecutor { ILendingPoolConfigurator public constant POOL_CONFIGURATOR = ILendingPoolConfigurator(AaveV2Ethereum.POOL_CONFIGURATOR); @@ -73,4 +78,4 @@ def test_source_code_clean(): POOL_CONFIGURATOR.setReserveFactor(AaveV2EthereumAssets.WETH_UNDERLYING, WETH_RF); } }""" - assert cleaned.replace(" ", "") == expected.replace(" ", "") \ No newline at end of file + assert cleaned.strip() == expected.strip()