Python FileInfo Examples

Example #1

File: test_api.py Project: phi-dbq/databricks-cli

 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]

Example #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]

Example #3

File: test_api.py Project: phi-dbq/databricks-cli

# 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]

Example #4

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(