diff options
| author |   spisarski <s.pisarski@cablelabs.com> | 2017-07-27 10:27:14 -0600 |
|---|---|---|
| committer | spisarski <s.pisarski@cablelabs.com> | 2017-07-27 10:27:14 -0600 |
| commit | ca276f452540f68c08cb3df9049e9e7876364dac (patch) | |
| tree | ce26c5fb721f10402669bf761c4e69317d3f3eca /snaps/tests | |
| parent | fdcd5600bade294ae2a2207ef01da92f22f64327 (diff) | |
Ensure library and tests close all necessary resources.
The SNAPS-OO library and tests had left open files, ssh, and scp
connections. These have all now been wrapped with try/finally
blocks.
JIRA: SNAPS-152
Change-Id: I43e09978b5c075bd78ff3279c0799556b8758878
Signed-off-by: spisarski <s.pisarski@cablelabs.com>
Diffstat (limited to 'snaps/tests')
| -rw-r--r-- | snaps/tests/file_utils_tests.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/snaps/tests/file_utils_tests.py b/snaps/tests/file_utils_tests.py index f3a622a..ef8b4ae 100644 --- a/snaps/tests/file_utils_tests.py +++ b/snaps/tests/file_utils_tests.py @@ -37,10 +37,14 @@ class FileUtilsTests(unittest.TestCase): os.makedirs(self.test_dir) self.tmpFile = self.test_dir + '/bar.txt' + self.tmp_file_opened = None if not os.path.exists(self.tmpFile): - open(self.tmpFile, 'wb') + self.tmp_file_opened = open(self.tmpFile, 'wb') def tearDown(self): + if self.tmp_file_opened: + self.tmp_file_opened.close() + if os.path.exists(self.test_dir) and os.path.isdir(self.test_dir): shutil.rmtree(self.tmp_dir) |