mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
queue status: use a short rev to represent tasks
1. use a short rev to represent tasks 2. Add a unit test for the result shown
This commit is contained in:
committed by
Peter Rowlands (변기호)
parent
8dfcb0a36d
commit
d7a917e2ab
@@ -23,7 +23,7 @@ class CmdQueueStatus(CmdBase):
|
||||
for exp in result:
|
||||
created = format_time(exp.get("timestamp"))
|
||||
td.append(
|
||||
[exp["rev"], exp.get("name", ""), created, exp["status"]]
|
||||
[exp["rev"][:7], exp.get("name", ""), created, exp["status"]]
|
||||
)
|
||||
td.render()
|
||||
|
||||
|
||||
@@ -99,7 +99,9 @@ def test_experiments_stop(dvc, scm, mocker):
|
||||
m.assert_called_once_with(kill=True)
|
||||
|
||||
|
||||
def test_experiments_status(dvc, scm, mocker):
|
||||
def test_experiments_status(dvc, scm, mocker, capsys, caplog):
|
||||
from datetime import datetime
|
||||
|
||||
cli_args = parse_args(
|
||||
[
|
||||
"queue",
|
||||
@@ -109,12 +111,31 @@ def test_experiments_status(dvc, scm, mocker):
|
||||
assert cli_args.func == CmdQueueStatus
|
||||
|
||||
cmd = cli_args.func(cli_args)
|
||||
status_result = [
|
||||
{
|
||||
"rev": "c61a525a4ff39007301b4516fb6e54b323a0587b",
|
||||
"name": "I40",
|
||||
"timestamp": datetime(2022, 6, 9, 20, 49, 48),
|
||||
"status": "Queued",
|
||||
},
|
||||
{
|
||||
"rev": "8da9c339da30636261a3491a90aafdb760a4168f",
|
||||
"name": "I60",
|
||||
"timestamp": datetime(2022, 6, 9, 20, 49, 43),
|
||||
"status": "Running",
|
||||
},
|
||||
]
|
||||
m = mocker.patch(
|
||||
"dvc.repo.experiments.queue.local.LocalCeleryQueue.status",
|
||||
return_value=status_result,
|
||||
)
|
||||
|
||||
assert cmd.run() == 0
|
||||
m.assert_called_once_with()
|
||||
log, _ = capsys.readouterr()
|
||||
assert "Rev Name Created Status" in log
|
||||
assert "c61a525 I40 Jun 09, 2022 Queued" in log
|
||||
assert "8da9c33 I60 Jun 09, 2022 Running" in log
|
||||
|
||||
|
||||
def test_queue_logs(dvc, scm, mocker):
|
||||
|
||||
Reference in New Issue
Block a user