mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-10 15:28:34 +01:00
38 lines
959 B
Python
38 lines
959 B
Python
import os
|
|
import stat
|
|
import shutil
|
|
import filecmp
|
|
|
|
from dvc.main import main
|
|
from dvc.data_cloud import file_md5
|
|
from dvc.stage import Stage, CmdOutputNoCacheError, CmdOutputOutsideOfRepoError
|
|
from dvc.stage import CmdOutputDoesNotExistError, CmdOutputIsNotFileError
|
|
from dvc.project import StageNotFoundError
|
|
|
|
from tests.test_repro import TestRepro
|
|
|
|
|
|
class TestCheckout(TestRepro):
|
|
def setUp(self):
|
|
super(TestCheckout, self).setUp()
|
|
|
|
self.orig = 'orig'
|
|
shutil.copy(self.FOO, self.orig)
|
|
os.chmod(self.FOO, stat.S_IWRITE)
|
|
os.unlink(self.FOO)
|
|
|
|
def test(self):
|
|
self.dvc.checkout()
|
|
self._test_checkout()
|
|
|
|
def _test_checkout(self):
|
|
self.assertTrue(os.path.isfile(self.FOO))
|
|
self.assertTrue(filecmp.cmp(self.FOO, self.orig))
|
|
|
|
|
|
class TestCmdCheckout(TestCheckout):
|
|
def test(self):
|
|
ret = main(['checkout'])
|
|
self.assertEqual(ret, 0)
|
|
self._test_checkout()
|