| 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() |
|
|
