def test_list(self):
#Arrange
client = DropboxConnection.get_client('auth_token', 'test_1')
#Arrange input files
input_file_1_info = ('input_file.1.dmp', 100)
input_file_1 = MagicMock()
input_file_1.name = input_file_1_info[0]
input_file_1.size = input_file_1_info[1]
result_file_1_info = ('result_file.1.dmp', 100)
result_file_1 = MagicMock()
result_file_1.name = result_file_1_info[0]
result_file_1.size = result_file_1_info[1]
#Arrange response object
response = MagicMock()
response.entries = [input_file_1, result_file_1]
#Mock calls
client._client.files_list_folder = MagicMock(return_value=response)
#Act
files, sizes = client.list(file_name_prefix='input_file')
#Assert
self.assertCountEqual(list(files), [Path(input_file_1_info[0])])
self.assertCountEqual(list(sizes), [input_file_1_info[1]])
def download(self):
#Arrange
auth_token = None
client = DropboxConnection.get_client(auth_token, 'test_1')
#Act
client.download(small_file)
client.download(cnn_model_file)
def upload(self):
#Arrange
auth_token = None
client = DropboxConnection.get_client(auth_token, 'test_1')
#Act
client.upload(ut_constants.DATA_STORE / cnn_model_file)
client.upload(ut_constants.DROPBOX_STORE / small_file)
def test_list_no_files(self):
#Arrange
client = DropboxConnection.get_client('auth_token', 'test_1')
#Arrange response object
response = MagicMock()
response.entries = None
#Mock calls
client._client.files_list_folder = MagicMock(return_value=response)
#Act
files, size = client.list()
#Assert
self.assertCountEqual(files, [])
def test_get_client(self):
#Act
client = DropboxConnection.get_client('auth_token', remote_dir)
#Assert
self.assertIsNotNone(client)