mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-06 15:28:28 +01:00
Ruslan Kuprieiev
d5dd0fd804
Since we now use file size to help detect file changes on filesystems with bad mtime resolution, we no longer need these explicit sleep() calls in our tests for mac os. Signed-off-by: Ruslan Kuprieiev <[email protected]>
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)
|