| Javi Merino | c26e1c7 | 2014-04-08 16:39:01 +0100 | [diff] [blame^] | 1 | #!/usr/bin/python |
| 2 | |
| 3 | import unittest |
| 4 | import os, sys |
| 5 | import shutil, tempfile |
| 6 | |
| 7 | TESTS_DIRECTORY = os.path.dirname(os.path.realpath(__file__)) |
| 8 | sys.path.append(os.path.join(TESTS_DIRECTORY, "..", "cr2")) |
| 9 | |
| 10 | class SetupDirectory(unittest.TestCase): |
| 11 | def __init__(self, files_to_copy, *args, **kwargs): |
| 12 | self.files_to_copy = files_to_copy |
| 13 | super(SetupDirectory, self).__init__(*args, **kwargs) |
| 14 | |
| 15 | def setUp(self): |
| 16 | self.out_dir = tempfile.mkdtemp() |
| 17 | os.chdir(self.out_dir) |
| 18 | |
| 19 | for fname in self.files_to_copy: |
| 20 | src_fname = os.path.join(TESTS_DIRECTORY, fname) |
| 21 | shutil.copy(src_fname, self.out_dir) |
| 22 | |
| 23 | def tearDown(self): |
| 24 | shutil.rmtree(self.out_dir) |