This commit is contained in:
@@ -0,0 +1,32 @@
|
||||
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()
|
||||
Reference in New Issue
Block a user