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)