From 458d63a6c5783bfd8ad9b49ed7479e62683bd5a0 Mon Sep 17 00:00:00 2001 From: Vitor Hideyoshi Nakazone Batista Date: Tue, 21 May 2024 18:36:35 -0300 Subject: [PATCH] Tests VirusTotalService --- .../test_virus_total_service.py | 33 +++++++++++++++++++ 1 file changed, 33 insertions(+) create mode 100644 tests/virus_checker_service/test_virus_total_service.py diff --git a/tests/virus_checker_service/test_virus_total_service.py b/tests/virus_checker_service/test_virus_total_service.py new file mode 100644 index 0000000..ec93eab --- /dev/null +++ b/tests/virus_checker_service/test_virus_total_service.py @@ -0,0 +1,33 @@ +from io import BytesIO +from unittest import TestCase +from unittest.mock import Mock + +from storage_service.service.virus_checker.virus_total_service import VirusTotalService + + +class TestVirusTotalService(TestCase): + def test_check_virus_invalid(self): + mock_virus_checker = Mock() + mock_virus_checker.request.side_effect = [ + Mock(data={"id": "file_id"}), + Mock(json=Mock(return_value={"data": {"attributes": {"stats": {"malicious": 1, "suspicious": 1, "harmless": 1}}}})), + ] + + virus_total_service = VirusTotalService(mock_virus_checker) + + result = virus_total_service.check_virus(BytesIO(b"file_data")) + + self.assertFalse(result) + + def test_check_virus_valid(self): + mock_virus_checker = Mock() + mock_virus_checker.request.side_effect = [ + Mock(data={"id": "file_id"}), + Mock(json=Mock(return_value={"data": {"attributes": {"stats": {"malicious": 0, "suspicious": 0, "harmless": 0}}}})), + ] + + virus_total_service = VirusTotalService(mock_virus_checker) + + result = virus_total_service.check_virus(BytesIO(b"file_data")) + + self.assertTrue(result)