wehub-resource-sync
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
import os
|
|
import shutil
|
|
import tempfile
|
|
import unittest
|
|
|
|
from swift.utils import copy_files_by_pattern
|
|
|
|
|
|
class TestFileUtils(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self._tmp_dir = tempfile.TemporaryDirectory()
|
|
self.tmp_dir = self._tmp_dir.name
|
|
|
|
def tearDown(self):
|
|
shutil.rmtree(self.tmp_dir)
|
|
|
|
def test_copy_files(self):
|
|
os.makedirs(os.path.join(self.tmp_dir, 'source'))
|
|
os.makedirs(os.path.join(self.tmp_dir, 'source', 'subfolder'))
|
|
with open(os.path.join(self.tmp_dir, 'source', '1.txt'), 'w') as f:
|
|
f.write('')
|
|
with open(os.path.join(self.tmp_dir, 'source', 'subfolder', '2.txt'), 'w') as f:
|
|
f.write('')
|
|
copy_files_by_pattern(
|
|
os.path.join(self.tmp_dir, 'source'), os.path.join(self.tmp_dir, 'target'), ['*.txt', 'subfolder/*.txt'])
|
|
self.assertTrue(os.path.exists(os.path.join(self.tmp_dir, 'target', '1.txt')))
|
|
self.assertTrue(os.path.exists(os.path.join(self.tmp_dir, 'target', 'subfolder', '2.txt')))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|