def test_add_reboot_binary_removed(self): id, sample = self.add_task() task = Task() task.load_from_db(id) task.create_empty() os.remove(task.targets[0].copied_binary) newid = task.add_reboot(id) assert newid is None
def test_add_reboot(self): id, sample = self.add_task(owner="MrDoge") sid = self.db.add_submit(None, None, None) task = Task() task.load_from_db(id) task.create_empty() newid = task.add_reboot(id, owner="Doge", submit_id=sid) task_path = cwd(analysis=newid) db_task = self.db.view_task(newid) assert newid is not None assert os.path.exists(task_path) assert db_task.targets[0].category == "file" assert db_task.package == "reboot" assert db_task.owner == "Doge" assert db_task.priority == 1 assert db_task.custom == "%s" % id assert db_task.memory == False assert db_task.targets[0].target == sample assert db_task.submit_id == sid assert len(task.targets) == 1 assert isinstance(task.targets[0], Target)