mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
import os
|
|
|
|
from dvc.state import State
|
|
from dvc.utils import file_md5
|
|
from dvc.main import main
|
|
|
|
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)
|