mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
75 lines
2.1 KiB
Python
75 lines
2.1 KiB
Python
import os
|
|
import mock
|
|
|
|
from dvc.utils.compat import str
|
|
|
|
from dvc.state import State
|
|
from dvc.utils import file_md5
|
|
from dvc.main import main
|
|
from dvc.utils.fs import get_inode
|
|
|
|
from tests.basic_env import TestDvc
|
|
|
|
|
|
class TestState(TestDvc):
|
|
def test_update(self):
|
|
path = os.path.join(self.dvc.root_dir, self.FOO)
|
|
md5 = file_md5(path)[0]
|
|
|
|
state = State(self.dvc, self.dvc.config.config)
|
|
|
|
with state:
|
|
entry_md5 = state.update(path)
|
|
self.assertEqual(entry_md5, md5)
|
|
|
|
os.unlink(path)
|
|
with open(path, "a") as fd:
|
|
fd.write("1")
|
|
|
|
md5 = file_md5(path)[0]
|
|
entry_md5 = state.update(path)
|
|
self.assertEqual(entry_md5, md5)
|
|
|
|
|
|
class TestStateOverflow(TestDvc):
|
|
def test(self):
|
|
# NOTE: trying to add more entries than state can handle,
|
|
# to see if it will clean up and vacuum successfully
|
|
ret = main(["config", "state.row_limit", "10"])
|
|
self.assertEqual(ret, 0)
|
|
|
|
dname = "dir"
|
|
os.mkdir(dname)
|
|
for i in range(20):
|
|
with open(os.path.join(dname, str(i)), "w+") as fobj:
|
|
fobj.write(str(i))
|
|
|
|
ret = main(["add", "dir"])
|
|
self.assertEqual(ret, 0)
|
|
|
|
|
|
class TestGetStateRecordForInode(TestDvc):
|
|
@staticmethod
|
|
def mock_get_inode(special_path, special_value):
|
|
def get_inode_mocked(path):
|
|
if path == special_path:
|
|
return special_value
|
|
else:
|
|
return get_inode(path)
|
|
|
|
return get_inode_mocked
|
|
|
|
@mock.patch("dvc.state.get_inode", autospec=True)
|
|
def test_transforms_inode(self, get_inode_mock):
|
|
state = State(self.dvc, self.dvc.config.config)
|
|
inode = state.MAX_INT + 2
|
|
self.assertNotEqual(inode, state._to_sqlite(inode))
|
|
|
|
path = os.path.join(self.dvc.root_dir, self.FOO)
|
|
get_inode_mock.side_effect = self.mock_get_inode(path, inode)
|
|
|
|
with state:
|
|
state.update(path)
|
|
ret = state.get_state_record_for_inode(inode)
|
|
self.assertIsNotNone(ret)
|