mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-05 23:09:32 +01:00
141 lines
3.6 KiB
Python
141 lines
3.6 KiB
Python
import json
|
|
import logging
|
|
import os
|
|
import time
|
|
|
|
import mock
|
|
import pytest
|
|
|
|
from dvc import __version__
|
|
from dvc.updater import Updater
|
|
|
|
|
|
@pytest.fixture
|
|
def tmp_global_dir(tmp_path):
|
|
"""
|
|
Fixture to prevent modifying the actual global config
|
|
"""
|
|
with mock.patch("dvc.config.Config.get_dir", return_value=str(tmp_path)):
|
|
yield
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def mock_env(monkeypatch):
|
|
monkeypatch.delenv("CI", raising=False)
|
|
monkeypatch.setenv("DVC_TEST", "False")
|
|
|
|
|
|
@pytest.fixture
|
|
def updater(tmp_path, tmp_global_dir):
|
|
return Updater(tmp_path)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_tty(mocker):
|
|
return mocker.patch("sys.stdout.isatty", return_value=True)
|
|
|
|
|
|
@mock.patch("requests.get")
|
|
def test_fetch(mock_get, updater):
|
|
mock_get.return_value.status_code = 200
|
|
mock_get.return_value.json.return_value = {"version": __version__}
|
|
|
|
assert not os.path.exists(updater.updater_file)
|
|
|
|
updater.fetch(detach=False)
|
|
|
|
mock_get.assert_called_once_with(Updater.URL, timeout=Updater.TIMEOUT_GET)
|
|
assert os.path.isfile(updater.updater_file)
|
|
|
|
with open(updater.updater_file) as fobj:
|
|
info = json.load(fobj)
|
|
|
|
assert info["version"] == __version__
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"config, result",
|
|
[
|
|
({}, True),
|
|
({"check_update": "true"}, True),
|
|
({"check_update": "false"}, False),
|
|
],
|
|
)
|
|
def test_is_enabled(dvc, updater, config, result):
|
|
with dvc.config.edit(validate=False) as conf:
|
|
conf["core"] = config
|
|
|
|
assert result == updater.is_enabled()
|
|
|
|
|
|
@pytest.mark.parametrize("result", [True, False])
|
|
@mock.patch("dvc.updater.Updater._check")
|
|
def test_check_update_respect_config(mock_check, result, updater, mocker):
|
|
mocker.patch.object(updater, "is_enabled", return_value=result)
|
|
updater.check()
|
|
assert result == mock_check.called
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"current,latest,notify",
|
|
[
|
|
("0.0.2", "0.0.2", False),
|
|
("0.0.2", "0.0.3", True),
|
|
("0.0.2", "0.0.1", False),
|
|
],
|
|
ids=["uptodate", "behind", "ahead"],
|
|
)
|
|
def test_check_updates(mock_tty, updater, caplog, current, latest, notify):
|
|
updater.current = current
|
|
with open(updater.updater_file, "w+") as f:
|
|
json.dump({"version": latest}, f)
|
|
|
|
caplog.clear()
|
|
with caplog.at_level(logging.INFO, logger="dvc.updater"):
|
|
updater.check()
|
|
|
|
if notify:
|
|
assert f"Update available {current} -> {latest}" in caplog.text
|
|
else:
|
|
assert not caplog.text
|
|
|
|
|
|
def test_check_refetches_each_day(mock_tty, updater, caplog, mocker):
|
|
updater.current = "0.0.8"
|
|
with open(updater.updater_file, "w+") as f:
|
|
json.dump({"version": "0.0.9"}, f)
|
|
fetch = mocker.patch.object(updater, "fetch")
|
|
|
|
time_value = time.time() + 24 * 60 * 60 + 10
|
|
mock_time = mocker.patch("time.time", return_value=time_value)
|
|
|
|
caplog.clear()
|
|
with caplog.at_level(logging.INFO, logger="dvc.updater"):
|
|
updater.check()
|
|
assert not caplog.text
|
|
fetch.assert_called_once()
|
|
mock_time.assert_called()
|
|
|
|
|
|
def test_check_fetches_on_invalid_data_format(
|
|
mock_tty, updater, caplog, mocker
|
|
):
|
|
updater.current = "0.0.5"
|
|
with open(updater.updater_file, "w+") as f:
|
|
f.write('"{"version: "0.0.6"')
|
|
fetch = mocker.patch.object(updater, "fetch")
|
|
caplog.clear()
|
|
with caplog.at_level(logging.INFO, logger="dvc.updater"):
|
|
updater.check()
|
|
assert not caplog.text
|
|
fetch.assert_called_once()
|
|
|
|
|
|
@mock.patch("dvc.updater.Updater._check")
|
|
def test_check(mock_check, updater):
|
|
updater.check()
|
|
updater.check()
|
|
updater.check()
|
|
|
|
assert mock_check.call_count == 3
|