def test__ensure_target_directory_unique_inside_target_dir(self):
"""
Should create directory with unique name inside target directory.
:return:
"""
import os.path
from artefact.localhost.file import FileCopy
expected_target_dir = "./test_target"
expected_unique_dir_name = "me_unique"
collector = FileCopy(parameters={}, parent=None)
collector._destination_directory = expected_target_dir
try:
with mock.patch(
'artefact.localhost.file.FileCopy._get_unique_directory_name',
MagicMock(return_value=expected_unique_dir_name)):
collector._ensure_target_directory()
self.assertTrue(
os.path.exists(
os.path.join(expected_target_dir,
expected_unique_dir_name)))
finally:
if os.path.exists(
os.path.join(expected_target_dir,
expected_unique_dir_name)):
os.rmdir(
os.path.join(expected_target_dir,
expected_unique_dir_name))
if os.path.exists(expected_target_dir):
os.rmdir(expected_target_dir)
def test__ensure_target_directory(self):
"""
Should return path of unique target directory
:return:
"""
import os.path
from artefact.localhost.file import FileCopy
expected_target_dir = "./test_target"
expected_unique_dir_name = "me_unique"
expected_return_value = './test_target/me_unique'
if platform.system() == "Windows":
expected_return_value = './test_target\\me_unique'
collector = FileCopy(parameters={}, parent=None)
collector._destination_directory = expected_target_dir
try:
with mock.patch(
'artefact.localhost.file.FileCopy._get_unique_directory_name',
MagicMock(return_value=expected_unique_dir_name)):
acutal_return_value = collector._ensure_target_directory()
self.assertEqual(expected_return_value, acutal_return_value)
finally:
if os.path.exists(
os.path.join(expected_target_dir,
expected_unique_dir_name)):
os.rmdir(
os.path.join(expected_target_dir,
expected_unique_dir_name))
if os.path.exists(expected_target_dir):
os.rmdir(expected_target_dir)