mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-08 07:18:27 +01:00
Mr. Outis
c6b8be8f8a
* import: Output as on optional argument Close #1335 * Use tempfile to prevent circular dependencies
40 lines
1.0 KiB
Python
40 lines
1.0 KiB
Python
import os
|
|
import stat
|
|
import shutil
|
|
import tempfile
|
|
from uuid import uuid4
|
|
|
|
from dvc.main import main
|
|
from dvc.command.imp import CmdImport
|
|
from dvc.exceptions import DvcException
|
|
|
|
from tests.basic_env import TestDvc
|
|
|
|
|
|
class TestCmdImport(TestDvc):
|
|
def test(self):
|
|
ret = main(['import', self.FOO, 'import'])
|
|
self.assertEqual(ret, 0)
|
|
self.assertTrue(os.path.exists('import.dvc'))
|
|
|
|
ret = main(['import', 'non-existing-file', 'import'])
|
|
self.assertNotEqual(ret, 0)
|
|
|
|
def test_unsupported(self):
|
|
ret = main(['import', 'unsupported://path', 'import_unsupported'])
|
|
self.assertNotEqual(ret, 0)
|
|
|
|
class TestDefaultOutput(TestDvc):
|
|
def test(self):
|
|
tmpdir = self.mkdtemp()
|
|
filename = str(uuid4())
|
|
tmpfile = os.path.join(tmpdir, filename)
|
|
|
|
with open(tmpfile, 'w') as fd:
|
|
fd.write('content')
|
|
|
|
ret = main(['import', tmpfile])
|
|
self.assertEqual(ret, 0)
|
|
self.assertTrue(os.path.exists(filename))
|
|
self.assertEqual(open(filename).read(), 'content')
|