コード例 #1
0
def test_list_files_single_file():
    client = mock.MagicMock()
    client.get_file_properties = mock.MagicMock()
    client.get_file_properties.return_value = 'fp'

    i = 0
    for file in ops.list_files(client, 'a', 'b/c', True):
        i += 1
        assert file == 'fp'
    assert i == 1
コード例 #2
0
def test_list_files_directory(patched_cisf):
    _file = azure.storage.file.models.File(name='name')
    client = mock.MagicMock()
    client.list_directories_and_files.return_value = [_file]
    client.get_file_properties.return_value = _file

    i = 0
    for file in ops.list_files(client, 'dir', '', True):
        i += 1
        assert file.name == 'name'
    assert i == 1

    _dir = azure.storage.file.models.Directory(name='dirname')
    _file = azure.storage.file.models.File(name='dirname/name')
    client = mock.MagicMock()
    client.list_directories_and_files.side_effect = [[_dir, _file]]
    client.get_file_properties.side_effect = [_file]

    i = 0
    for file in ops.list_files(client, '', '', True):
        i += 1
        assert file.name == _file.name
        assert type(file) == azure.storage.file.models.File
    assert i == 1