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)