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
