def test_empty_reprocess(self): db.connect() mkdir(cwd(analysis=1)) process_task_range("1") assert os.path.exists(cwd("reports", "report.json", analysis=1)) obj = json.load(open(cwd("reports", "report.json", analysis=1), "rb")) assert "contact back" in obj["debug"]["errors"][0]
def test_process_task_range_range(self, p): mkdir(cwd(analysis=3)) for x in xrange(10, 101): mkdir(cwd(analysis=x)) process_task_range("3,5,10-100") assert p.call_count == 92 # 101-10+1 p.assert_any_call({ "id": 3, "category": "file", "target": "", "options": {}, "package": None, "custom": None, }) # We did not create an analysis directory for analysis=5. with pytest.raises(AssertionError): p.assert_any_call({ "id": 5, "category": "file", "target": "", "options": {}, "package": None, "custom": None, }) for x in xrange(10, 101): p.assert_any_call({ "id": x, "category": "file", "target": "", "options": {}, "package": None, "custom": None, })
def test_process_task_range_single(self, p): mkdir(cwd(analysis=1234)) process_task_range("1234") p.assert_called_once_with({ "id": 1234, "category": "file", "target": "", "options": {}, "package": None, "custom": None, })
def test_process_task_range_single(self, p): mkdir(cwd(analysis=1234)) process_task_range("1234") p.assert_called_once_with({ "id": 1234, "category": "file", "target": "", "options": {}, "package": None, "custom": None, })
def test_config_load_once(): set_cwd(tempfile.mkdtemp()) cuckoo_create() db.connect() t0 = db.add_path(__file__) t1 = db.add_path(__file__) shutil.copytree("tests/files/sample_analysis_storage", cwd(analysis=t0)) shutil.copytree("tests/files/sample_analysis_storage", cwd(analysis=t1)) with mock.patch("cuckoo.common.config.ConfigParser.ConfigParser") as p: process_task_range("%d,%d" % (t0, t1)) assert p.return_value.read.call_count == 2 p.return_value.read.assert_any_call(cwd("conf", "processing.conf")) p.return_value.read.assert_any_call(cwd("conf", "reporting.conf"))
def test_process_task_range_multi(self, p): mkdir(cwd(analysis=1234)) mkdir(cwd(analysis=2345)) process_task_range("1234,2345") assert p.call_count == 2 p.assert_any_call({ "id": 1234, "category": "file", "target": "", "options": {}, "package": None, "custom": None, }) p.assert_any_call({ "id": 2345, "category": "file", "target": "", "options": {}, "package": None, "custom": None, })
def test_process_task_range_multi(self, p): mkdir(cwd(analysis=1234)) mkdir(cwd(analysis=2345)) process_task_range("1234,2345") assert p.call_count == 2 p.assert_any_call({ "id": 1234, "category": "file", "target": "", "options": {}, "package": None, "custom": None, }) p.assert_any_call({ "id": 2345, "category": "file", "target": "", "options": {}, "package": None, "custom": None, })
def test_process_task_range_multi_db(self, p): mkdir(cwd(analysis=1234)) mkdir(cwd(analysis=2345)) p.return_value.view_task.return_value = {} process_task_range("2345") p.return_value.view_task.assert_called_once_with(2345)
def test_process_task_range_single_db(self, p): mkdir(cwd(analysis=1234)) p.return_value.view_task.return_value = {} process_task_range("1234") p.return_value.view_task.assert_called_once_with(1234)
def test_process_task_range_single_noanal(self, p): process_task_range("1234") p.assert_not_called()
def test_process_task_range_duplicate(self, p): process_task_range("3,3,42") assert p.return_value.view_task.call_count == 2 p.return_value.view_task.assert_any_call(3) p.return_value.view_task.assert_any_call(42)
def test_process_task_range_multi_db(self, p): mkdir(cwd(analysis=1234)) mkdir(cwd(analysis=2345)) p.return_value.view_task.return_value = {} process_task_range("2345") p.return_value.view_task.assert_called_once_with(2345)
def test_process_task_range_single_db(self, p): mkdir(cwd(analysis=1234)) p.return_value.view_task.return_value = {} process_task_range("1234") p.return_value.view_task.assert_called_once_with(1234)
def test_process_task_range_single_noanal(self, p): process_task_range("1234") p.assert_not_called()