def test_to_row_long_form_not_absolute(self): file_info = api.FileInfo(TEST_DBFS_PATH, False, 1) row = file_info.to_row(is_long_form=True, is_absolute=False) assert len(row) == 3 assert row[0] == 'file' assert row[1] == 1 assert TEST_DBFS_PATH.basename == row[2]
def test_to_row_not_long_form_not_absolute(self): file_info = api.FileInfo(TEST_DBFS_PATH, False, 1, DUMMY_TIME) row = file_info.to_row(is_long_form=False, is_absolute=False) assert len(row) == 1 assert TEST_DBFS_PATH.basename == row[0]
# limitations under the License. from base64 import b64encode import os import requests import mock import pytest import databricks_cli.dbfs.api as api from databricks_cli.dbfs.dbfs_path import DbfsPath from databricks_cli.dbfs.exceptions import LocalFileExistsException TEST_DBFS_PATH = DbfsPath('dbfs:/test') TEST_FILE_JSON = {'path': '/test', 'is_dir': False, 'file_size': 1} TEST_FILE_INFO = api.FileInfo(TEST_DBFS_PATH, False, 1) def get_resource_does_not_exist_exception(): response = requests.Response() response._content = '{"error_code": "' + api.DbfsErrorCodes.RESOURCE_DOES_NOT_EXIST + '"}' # NOQA return requests.exceptions.HTTPError(response=response) class TestFileInfo(object): def test_to_row_not_long_form_not_absolute(self): file_info = api.FileInfo(TEST_DBFS_PATH, False, 1) row = file_info.to_row(is_long_form=False, is_absolute=False) assert len(row) == 1 assert TEST_DBFS_PATH.basename == row[0]
import mock import pytest import databricks_cli.dbfs.api as api from databricks_cli.dbfs.dbfs_path import DbfsPath from databricks_cli.dbfs.exceptions import LocalFileExistsException TEST_DBFS_PATH = DbfsPath('dbfs:/test') DUMMY_TIME = 1613158406000 TEST_FILE_JSON = { 'path': '/test', 'is_dir': False, 'file_size': 1, 'modification_time': DUMMY_TIME } TEST_FILE_INFO = api.FileInfo(TEST_DBFS_PATH, False, 1, DUMMY_TIME) def get_resource_does_not_exist_exception(): response = requests.Response() response._content = ('{"error_code": "' + api.DbfsErrorCodes.RESOURCE_DOES_NOT_EXIST + '"}').encode() # NOQA return requests.exceptions.HTTPError(response=response) def get_partial_delete_exception( message="[...] operation has deleted 10 files [...]"): response = requests.Response() response.status_code = 503 response._content = ('{{"error_code": "{}","message": "{}"}}'.format(