def test_category_checks_modified_file(self, clean_init_storage, mocker): class mock_sample: sha256 = "123" analysis_man = AnalysisManager(task=mock_task(), error_queue=queue.Queue()) assert analysis_man.init_storage() is True mocker.patch("lib.cuckoo.core.scheduler.Database.view_sample", return_value=mock_sample()) assert analysis_man.category_checks() is False
def test_category_checks_no_store_file(self, clean_init_storage, grab_sample, mocker): class mock_sample: sha256 = "5dd87d3d6b9d8b4016e3c36b189234772661e690c21371f1eb8e018f0f0dec2b" sample_location = grab_sample(sample_hash="5dd87d3d6b9d8b4016e3c36b189234772661e690c21371f1eb8e018f0f0dec2b") mock_task_cat = mock_task() mock_task_cat.target = sample_location analysis_man = AnalysisManager(task=mock_task_cat, error_queue=queue.Queue()) assert analysis_man.init_storage() is True mocker.patch("lib.cuckoo.core.scheduler.Database.view_sample", return_value=mock_sample()) mocker.patch("lib.cuckoo.core.scheduler.AnalysisManager.store_file", return_value=False) assert analysis_man.category_checks() is False