mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
Paweł Redzyński
52ffae81c4
* imp: refactor import message * import: reword error message Signed-off-by: Ruslan Kuprieiev <[email protected]>
68 lines
2.0 KiB
Python
68 lines
2.0 KiB
Python
import os
|
|
from uuid import uuid4
|
|
|
|
from dvc import logger
|
|
from dvc.exceptions import DvcException
|
|
from dvc.main import main
|
|
from mock import patch
|
|
from tests.basic_env import TestDvc
|
|
|
|
try:
|
|
from StringIO import StringIO
|
|
except ImportError:
|
|
from io import StringIO
|
|
|
|
|
|
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')
|
|
|
|
|
|
class TestFailedImportMessage(TestDvc):
|
|
color_patch = patch.object(logger, 'colorize', new=lambda x, color='': x)
|
|
|
|
def test(self):
|
|
self.color_patch.start()
|
|
self._test()
|
|
self.color_patch.stop()
|
|
|
|
@patch('dvc.command.imp.urlparse')
|
|
def _test(self, imp_urlparse_patch):
|
|
logger.logger.handlers[1].stream = StringIO()
|
|
page_address = 'http://somesite.com/file_name'
|
|
|
|
def dvc_exception(*args, **kwargs):
|
|
raise DvcException('message')
|
|
|
|
imp_urlparse_patch.side_effect = dvc_exception
|
|
main(['import', page_address])
|
|
self.assertIn('Error: failed to import '
|
|
'http://somesite.com/file_name. You could also try '
|
|
'downloading it manually and adding it with `dvc add` '
|
|
'command.',
|
|
logger.logger.handlers[1].stream.getvalue())
|