Queue remove: Can remove successed,failed tasks msgs

Followed from #7592

1. Seperate remove method to a new file.
2. Add queued, failed, processed flags to remove.
3. Add new unit tests for queue status --queue/fail/success
4. Implement methods to remove done tasks.
5. add some new unit and functional test for celery_queue.remove
6. bump into dvc-task version 0.0.13
This commit is contained in:
karajan1001
2022-07-12 21:39:49 +09:00
committed by Peter Rowlands (변기호)
parent ff3b23c1b6
commit b37524603a
10 changed files with 568 additions and 109 deletions
+11 -19
View File
@@ -15,23 +15,9 @@ def test_experiments_remove(dvc, scm, mocker):
"queue",
"remove",
"--all",
]
)
assert cli_args.func == CmdQueueRemove
cmd = cli_args.func(cli_args)
m = mocker.patch(
"dvc.repo.experiments.queue.celery.LocalCeleryQueue.clear",
return_value={},
)
assert cmd.run() == 0
m.assert_called_once_with()
cli_args = parse_args(
[
"queue",
"remove",
"--queued",
"--success",
"--failed",
"exp1",
"exp2",
]
@@ -39,13 +25,19 @@ def test_experiments_remove(dvc, scm, mocker):
assert cli_args.func == CmdQueueRemove
cmd = cli_args.func(cli_args)
m = mocker.patch(
remove_mocker = mocker.patch(
"dvc.repo.experiments.queue.celery.LocalCeleryQueue.remove",
return_value={},
)
assert cmd.run() == 0
m.assert_called_once_with(revs=["exp1", "exp2"])
remove_mocker.assert_called_once_with(
revs=["exp1", "exp2"],
_all=True,
success=True,
failed=True,
queued=True,
)
def test_experiments_kill(dvc, scm, mocker):