254 lines
12 KiB
Python
254 lines
12 KiB
Python
import os.path
|
|
import shutil
|
|
import unittest
|
|
from io import BytesIO, StringIO
|
|
from os import path
|
|
from unittest import TestCase
|
|
|
|
from webdav3.client import Client
|
|
from webdav3.exceptions import MethodNotSupported
|
|
|
|
|
|
class ClientTestCase(TestCase):
|
|
remote_path_file = 'test_dir/test.txt'
|
|
remote_path_file2 = 'test_dir2/test.txt'
|
|
remote_inner_path_file = 'test_dir/inner/test.txt'
|
|
remote_path_dir = 'test_dir'
|
|
remote_path_dir2 = 'test_dir2'
|
|
remote_inner_path_dir = 'test_dir/inner'
|
|
local_base_dir = 'tests/'
|
|
local_file = 'test.txt'
|
|
local_file_path = local_base_dir + 'test.txt'
|
|
local_path_dir = local_base_dir + 'res/test_dir'
|
|
|
|
def setUp(self):
|
|
options = {
|
|
'webdav_hostname': 'http://localhost:8585',
|
|
'webdav_login': 'alice',
|
|
'webdav_password': 'secret1234'
|
|
}
|
|
self.client = Client(options)
|
|
if path.exists(path=self.local_path_dir):
|
|
shutil.rmtree(path=self.local_path_dir)
|
|
|
|
def tearDown(self):
|
|
if path.exists(path=self.local_path_dir):
|
|
shutil.rmtree(path=self.local_path_dir)
|
|
if self.client.check(remote_path=self.remote_path_dir):
|
|
self.client.clean(remote_path=self.remote_path_dir)
|
|
if self.client.check(remote_path=self.remote_path_dir2):
|
|
self.client.clean(remote_path=self.remote_path_dir2)
|
|
|
|
def test_list(self):
|
|
self._prepare_for_downloading()
|
|
file_list = self.client.list()
|
|
self.assertIsNotNone(file_list, 'List of files should not be None')
|
|
self.assertGreater(file_list.__len__(), 0, 'Expected that amount of files more then 0')
|
|
|
|
def test_free(self):
|
|
with self.assertRaises(MethodNotSupported):
|
|
self.assertGreater(self.client.free(), 0, 'Expected that free space on WebDAV server is more then 0 bytes')
|
|
|
|
def test_check(self):
|
|
self.assertTrue(self.client.check(), 'Expected that root directory is exist')
|
|
|
|
def test_mkdir(self):
|
|
if self.client.check(remote_path=self.remote_path_dir):
|
|
self.client.clean(remote_path=self.remote_path_dir)
|
|
self.client.mkdir(remote_path=self.remote_path_dir)
|
|
self.assertTrue(self.client.check(remote_path=self.remote_path_dir), 'Expected the directory is created.')
|
|
|
|
def test_download_to(self):
|
|
self._prepare_for_downloading()
|
|
buff = BytesIO()
|
|
self.client.download_from(buff=buff, remote_path=self.remote_path_file)
|
|
self.assertEqual(buff.getvalue(), b'test content for testing of webdav client')
|
|
|
|
def test_download(self):
|
|
self._prepare_for_downloading()
|
|
self.client.download(local_path=self.local_path_dir, remote_path=self.remote_path_dir)
|
|
self.assertTrue(path.exists(self.local_path_dir), 'Expected the directory is downloaded.')
|
|
self.assertTrue(path.isdir(self.local_path_dir), 'Expected this is a directory.')
|
|
self.assertTrue(path.exists(self.local_path_dir + os.path.sep + self.local_file),
|
|
'Expected the file is downloaded')
|
|
self.assertTrue(path.isfile(self.local_path_dir + os.path.sep + self.local_file),
|
|
'Expected this is a file')
|
|
|
|
def test_download_sync(self):
|
|
self._prepare_for_downloading()
|
|
|
|
def callback():
|
|
self.assertTrue(path.exists(self.local_path_dir + os.path.sep + self.local_file),
|
|
'Expected the file is downloaded')
|
|
self.assertTrue(path.isfile(self.local_path_dir + os.path.sep + self.local_file),
|
|
'Expected this is a file')
|
|
|
|
self.client.download_sync(local_path=self.local_path_dir + os.path.sep + self.local_file,
|
|
remote_path=self.remote_path_file, callback=callback)
|
|
self.assertTrue(path.exists(self.local_path_dir + os.path.sep + self.local_file),
|
|
'Expected the file has already been downloaded')
|
|
|
|
def test_download_async(self):
|
|
self._prepare_for_downloading()
|
|
|
|
def callback():
|
|
self.assertTrue(path.exists(self.local_path_dir + os.path.sep + self.local_file),
|
|
'Expected the file is downloaded')
|
|
self.assertTrue(path.isfile(self.local_path_dir + os.path.sep + self.local_file),
|
|
'Expected this is a file')
|
|
|
|
self.client.download_async(local_path=self.local_path_dir + os.path.sep + self.local_file,
|
|
remote_path=self.remote_path_file, callback=callback)
|
|
self.assertFalse(path.exists(self.local_path_dir + os.path.sep + self.local_file),
|
|
'Expected the file has not been downloaded yet')
|
|
|
|
def test_upload_from(self):
|
|
self._prepare_for_uploading()
|
|
buff = StringIO()
|
|
buff.write(u'test content for testing of webdav client')
|
|
self.client.upload_to(buff=buff, remote_path=self.remote_path_file)
|
|
self.assertTrue(self.client.check(self.remote_path_file), 'Expected the file is uploaded.')
|
|
|
|
def test_upload(self):
|
|
self._prepare_for_uploading()
|
|
self.client.upload(remote_path=self.remote_path_file, local_path=self.local_path_dir)
|
|
self.assertTrue(self.client.check(self.remote_path_dir), 'Expected the directory is created.')
|
|
self.assertTrue(self.client.check(self.remote_path_file), 'Expected the file is uploaded.')
|
|
|
|
def test_upload_file(self):
|
|
self._prepare_for_uploading()
|
|
self.client.upload_file(remote_path=self.remote_path_file, local_path=self.local_file_path)
|
|
self.assertTrue(self.client.check(remote_path=self.remote_path_file), 'Expected the file is uploaded.')
|
|
|
|
def test_upload_sync(self):
|
|
self._prepare_for_uploading()
|
|
|
|
def callback():
|
|
self.assertTrue(self.client.check(self.remote_path_dir), 'Expected the directory is created.')
|
|
self.assertTrue(self.client.check(self.remote_path_file), 'Expected the file is uploaded.')
|
|
|
|
self.client.upload(remote_path=self.remote_path_file, local_path=self.local_path_dir)
|
|
|
|
def test_upload_async(self):
|
|
self._prepare_for_uploading()
|
|
|
|
def callback():
|
|
self.assertTrue(self.client.check(self.remote_path_dir), 'Expected the directory is created.')
|
|
self.assertTrue(self.client.check(self.remote_path_file), 'Expected the file is uploaded.')
|
|
|
|
self.client.upload(remote_path=self.remote_path_file, local_path=self.local_path_dir)
|
|
|
|
def test_copy(self):
|
|
self._prepare_for_downloading()
|
|
self.client.mkdir(remote_path=self.remote_path_dir2)
|
|
self.client.copy(remote_path_from=self.remote_path_file, remote_path_to=self.remote_path_file2)
|
|
self.assertTrue(self.client.check(remote_path=self.remote_path_file2))
|
|
|
|
def test_move(self):
|
|
self._prepare_for_downloading()
|
|
self.client.mkdir(remote_path=self.remote_path_dir2)
|
|
self.client.move(remote_path_from=self.remote_path_file, remote_path_to=self.remote_path_file2)
|
|
self.assertFalse(self.client.check(remote_path=self.remote_path_file))
|
|
self.assertTrue(self.client.check(remote_path=self.remote_path_file2))
|
|
|
|
def test_clean(self):
|
|
self._prepare_for_downloading()
|
|
self.client.clean(remote_path=self.remote_path_dir)
|
|
self.assertFalse(self.client.check(remote_path=self.remote_path_file))
|
|
self.assertFalse(self.client.check(remote_path=self.remote_path_dir))
|
|
|
|
def test_info(self):
|
|
self._prepare_for_downloading()
|
|
result = self.client.info(remote_path=self.remote_path_file)
|
|
self.assertEqual(result['size'], '41')
|
|
self.assertTrue('created' in result)
|
|
self.assertTrue('modified' in result)
|
|
|
|
def test_directory_is_dir(self):
|
|
self._prepare_for_downloading()
|
|
self.assertTrue(self.client.is_dir(self.remote_path_dir), 'Should return True for directory')
|
|
|
|
def test_file_is_not_dir(self):
|
|
self._prepare_for_downloading()
|
|
self.assertFalse(self.client.is_dir(self.remote_path_file), 'Should return False for file')
|
|
|
|
def test_get_property_of_non_exist(self):
|
|
self._prepare_for_downloading()
|
|
result = self.client.get_property(remote_path=self.remote_path_file, option={'name': 'aProperty'})
|
|
self.assertEqual(result, None, 'For not found property should return value as None')
|
|
|
|
def test_set_property(self):
|
|
self._prepare_for_downloading()
|
|
self.client.set_property(remote_path=self.remote_path_file, option={
|
|
'namespace': 'test',
|
|
'name': 'aProperty',
|
|
'value': 'aValue'
|
|
})
|
|
result = self.client.get_property(remote_path=self.remote_path_file,
|
|
option={'namespace': 'test', 'name': 'aProperty'})
|
|
self.assertEqual(result, 'aValue', 'Property value should be set')
|
|
|
|
def test_set_property_batch(self):
|
|
self._prepare_for_downloading()
|
|
self.client.set_property_batch(remote_path=self.remote_path_file, option=[
|
|
{
|
|
'namespace': 'test',
|
|
'name': 'aProperty',
|
|
'value': 'aValue'
|
|
},
|
|
{
|
|
'namespace': 'test',
|
|
'name': 'aProperty2',
|
|
'value': 'aValue2'
|
|
}
|
|
])
|
|
result = self.client.get_property(remote_path=self.remote_path_file,
|
|
option={'namespace': 'test', 'name': 'aProperty'})
|
|
self.assertEqual(result, 'aValue', 'First property value should be set')
|
|
result = self.client.get_property(remote_path=self.remote_path_file,
|
|
option={'namespace': 'test', 'name': 'aProperty2'})
|
|
self.assertEqual(result, 'aValue2', 'Second property value should be set')
|
|
|
|
def test_pull(self):
|
|
self._prepare_for_downloading(True)
|
|
self.client.pull(self.remote_path_dir, self.local_path_dir)
|
|
self.assertTrue(path.exists(self.local_path_dir), 'Expected the directory is downloaded.')
|
|
self.assertTrue(path.exists(self.local_path_dir + os.path.sep + 'inner'), 'Expected the directory is downloaded.')
|
|
self.assertTrue(path.exists(self.local_path_dir + os.path.sep + 'inner'), 'Expected the directory is downloaded.')
|
|
self.assertTrue(path.isdir(self.local_path_dir), 'Expected this is a directory.')
|
|
self.assertTrue(path.isdir(self.local_path_dir + os.path.sep + 'inner'), 'Expected this is a directory.')
|
|
self.assertTrue(path.exists(self.local_path_dir + os.path.sep + self.local_file),
|
|
'Expected the file is downloaded')
|
|
self.assertTrue(path.isfile(self.local_path_dir + os.path.sep + self.local_file),
|
|
'Expected this is a file')
|
|
|
|
def test_push(self):
|
|
self._prepare_for_uploading()
|
|
self.client.push(self.remote_path_dir, self.local_path_dir)
|
|
self.assertTrue(self.client.check(self.remote_path_dir), 'Expected the directory is created.')
|
|
self.assertTrue(self.client.check(self.remote_path_file), 'Expected the file is uploaded.')
|
|
|
|
def _prepare_for_downloading(self, inner_dir=False):
|
|
if not self.client.check(remote_path=self.remote_path_dir):
|
|
self.client.mkdir(remote_path=self.remote_path_dir)
|
|
if not self.client.check(remote_path=self.remote_path_file):
|
|
self.client.upload_file(remote_path=self.remote_path_file, local_path=self.local_file_path)
|
|
if not path.exists(self.local_path_dir):
|
|
os.makedirs(self.local_path_dir)
|
|
if inner_dir:
|
|
if not self.client.check(remote_path=self.remote_inner_path_dir):
|
|
self.client.mkdir(remote_path=self.remote_inner_path_dir)
|
|
if not self.client.check(remote_path=self.remote_inner_path_file):
|
|
self.client.upload_file(remote_path=self.remote_inner_path_file, local_path=self.local_file_path)
|
|
|
|
def _prepare_for_uploading(self):
|
|
if not self.client.check(remote_path=self.remote_path_dir):
|
|
self.client.mkdir(remote_path=self.remote_path_dir)
|
|
if not path.exists(path=self.local_path_dir):
|
|
os.makedirs(self.local_path_dir)
|
|
if not path.exists(path=self.local_path_dir + os.sep + self.local_file):
|
|
shutil.copy(src=self.local_file_path, dst=self.local_path_dir + os.sep + self.local_file)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|