mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-04 15:28:27 +01:00
Ruslan Kuprieiev
8f568ab197
* state: only load when needed * state: don't use separate load/dump * checkout: group link removal logic together * state: use diskcache.Cache * staging: use pbar and save to state * state: don't pass the whole repo to __init__ * staging: explicitly pass state object * fs: get rid of odb property * state: don't close the connection to cache * repo: close: close state
74 lines
1.9 KiB
Python
74 lines
1.9 KiB
Python
import os
|
|
|
|
from dvc.hash_info import HashInfo
|
|
from dvc.path_info import PathInfo
|
|
from dvc.state import State
|
|
from dvc.utils import file_md5
|
|
|
|
|
|
def test_state(tmp_dir, dvc):
|
|
tmp_dir.gen("foo", "foo content")
|
|
path = tmp_dir / "foo"
|
|
path_info = PathInfo(path)
|
|
hash_info = HashInfo("md5", file_md5(path, dvc.fs))
|
|
|
|
state = State(dvc.root_dir, dvc.tmp_dir)
|
|
|
|
state.save(path_info, dvc.fs, hash_info)
|
|
assert state.get(path_info, dvc.fs) == hash_info
|
|
|
|
path.unlink()
|
|
path.write_text("1")
|
|
|
|
assert state.get(path_info, dvc.fs) is None
|
|
|
|
hash_info = HashInfo("md5", file_md5(path, dvc.fs))
|
|
state.save(path_info, dvc.fs, hash_info)
|
|
|
|
assert state.get(path_info, dvc.fs) == hash_info
|
|
|
|
|
|
def test_state_overflow(tmp_dir, dvc):
|
|
# NOTE: trying to add more entries than state can handle,
|
|
# to see if it will clean up and vacuum successfully
|
|
dvc.config["state"]["row_limit"] = 10
|
|
|
|
path = tmp_dir / "dir"
|
|
path.mkdir()
|
|
for i in range(20):
|
|
(path / str(i)).write_text(str(i))
|
|
|
|
dvc.add("dir")
|
|
|
|
|
|
def mock_get_inode(inode):
|
|
def get_inode_mocked(_):
|
|
return inode
|
|
|
|
return get_inode_mocked
|
|
|
|
|
|
def test_remove_links(tmp_dir, dvc):
|
|
tmp_dir.dvc_gen({"foo": "foo_content", "bar": "bar_content"})
|
|
|
|
assert len(dvc.state.links) == 2
|
|
|
|
dvc.state.remove_links(["foo", "bar"], dvc.fs)
|
|
|
|
assert len(dvc.state.links) == 0
|
|
|
|
|
|
def test_get_unused_links(tmp_dir, dvc):
|
|
tmp_dir.dvc_gen({"foo": "foo_content", "bar": "bar_content"})
|
|
|
|
links = [os.path.join(dvc.root_dir, link) for link in ["foo", "bar"]]
|
|
assert set(dvc.state.get_unused_links([], dvc.fs)) == {"foo", "bar"}
|
|
assert set(dvc.state.get_unused_links(links[:1], dvc.fs)) == {"bar"}
|
|
assert set(dvc.state.get_unused_links(links, dvc.fs)) == set()
|
|
assert set(
|
|
dvc.state.get_unused_links(
|
|
(links[:1] + [os.path.join(dvc.root_dir, "not-existing-file")]),
|
|
dvc.fs,
|
|
)
|
|
) == {"bar"}
|