From 3d43125decd402925101ee1a921d7656e6efb416 Mon Sep 17 00:00:00 2001 From: b31ngd3v Date: Sat, 14 Jan 2023 05:18:37 +0530 Subject: [PATCH] test: improve test_csv2cve_valid_file for future failures --- test/test_csv2cve.py | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/test/test_csv2cve.py b/test/test_csv2cve.py index 8798aa33e7..244cba086b 100644 --- a/test/test_csv2cve.py +++ b/test/test_csv2cve.py @@ -30,16 +30,17 @@ async def test_csv2cve_valid_file(self, caplog): "Known CVEs in ('curl', '7.34.0'), ('kerberos', '1.15.1'), ('kerberos_5', '1.15.1'):", ) in caplog.record_tuples - for cve in [ - "3 CVE(s) in mit.kerberos v1.15.1", - # "58 CVE(s) in haxx.curl v7.34.0", Seems to be changing right now - "10 CVE(s) in mit.kerberos_5 v1.15.1", + for cve_count, product in [ + [3, "mit.kerberos v1.15.1"], + [60, "haxx.curl v7.34.0"], + [10, "mit.kerberos_5 v1.15.1"], ]: - assert ( - "cve_bin_tool.CVEScanner", - logging.INFO, - cve, - ) in caplog.record_tuples + retrieved_cve_count = 0 + for captured_line in caplog.record_tuples: + if captured_line[2].find(product) != -1: + retrieved_cve_count = int(captured_line[2].split()[0]) + + assert retrieved_cve_count >= cve_count @pytest.mark.asyncio @pytest.mark.parametrize(