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