diff --git a/python/lib/scheduler/dmod/test/test_rsa_key_pair.py b/python/lib/scheduler/dmod/test/test_rsa_key_pair.py index 6615ed9b5..e71f2f79b 100644 --- a/python/lib/scheduler/dmod/test/test_rsa_key_pair.py +++ b/python/lib/scheduler/dmod/test/test_rsa_key_pair.py @@ -1,4 +1,5 @@ import unittest +import tempfile from pathlib import Path from tempfile import TemporaryDirectory from ..scheduler.rsa_key_pair import RsaKeyPair @@ -9,7 +10,8 @@ class TestRsaKeyPair(unittest.TestCase): def setUp(self) -> None: self.rsa_key_pairs: Dict[int, RsaKeyPair] = dict() - self.rsa_key_pairs[1] = RsaKeyPair(directory='.', name='id_rsa_1') + self.tempdir = tempfile.TemporaryDirectory() + self.rsa_key_pairs[1] = RsaKeyPair(directory=self.tempdir.name, name='id_rsa_1') self.serial_rsa_key_pairs: Dict[int, dict] = dict() self.serial_rsa_key_pairs[1] = self.rsa_key_pairs[1].to_dict() @@ -18,6 +20,7 @@ def setUp(self) -> None: def tearDown(self) -> None: self.rsa_key_pairs[1].private_key_file.unlink(missing_ok=True) self.rsa_key_pairs[1].public_key_file.unlink(missing_ok=True) + self.tempdir.cleanup() def test_generate_key_pair_1_a(self): """