diff options
Diffstat (limited to 'src/mailman_pgp/model/tests/test_fs_key.py')
| -rw-r--r-- | src/mailman_pgp/model/tests/test_fs_key.py | 49 |
1 files changed, 35 insertions, 14 deletions
diff --git a/src/mailman_pgp/model/tests/test_fs_key.py b/src/mailman_pgp/model/tests/test_fs_key.py index 8828e4b..3a4f859 100644 --- a/src/mailman_pgp/model/tests/test_fs_key.py +++ b/src/mailman_pgp/model/tests/test_fs_key.py @@ -16,7 +16,7 @@ # this program. If not, see <http://www.gnu.org/licenses/>. """""" -from os.path import join, exists +from os.path import exists, join from tempfile import TemporaryDirectory from unittest import TestCase @@ -47,19 +47,7 @@ class TestFSKey(TestCase): key.load() self.assertEqual(key.key.fingerprint, key_data.fingerprint) - def test_reload_none(self): - key_name = 'something.asc' - key = FSKey(self.tmpdir.name, key_name, False) - key_data = load_key('rsa_1024.priv.asc') - with open(key.key_path, 'w') as key_file: - key_file.write(str(key_data)) - - self.assertIsNone(key.key) - key.reload() - self.assertIsNotNone(key.key) - self.assertEqual(key.key.fingerprint, key_data.fingerprint) - - def test_reload_not_none(self): + def test_reload(self): key_name = 'something.asc' key_path = join(self.tmpdir.name, key_name) key_data = load_key('rsa_1024.priv.asc') @@ -76,6 +64,18 @@ class TestFSKey(TestCase): self.assertIsNotNone(key.key) self.assertEqual(key.key.fingerprint, new_key_data.fingerprint) + def test_reload_none(self): + key_name = 'something.asc' + key = FSKey(self.tmpdir.name, key_name, False) + key_data = load_key('rsa_1024.priv.asc') + with open(key.key_path, 'w') as key_file: + key_file.write(str(key_data)) + + self.assertIsNone(key.key) + key.reload() + self.assertIsNotNone(key.key) + self.assertEqual(key.key.fingerprint, key_data.fingerprint) + def test_save(self): key_name = 'something.asc' key = FSKey(self.tmpdir.name, key_name) @@ -97,3 +97,24 @@ class TestFSKey(TestCase): key.delete() self.assertFalse(exists(key.key_path)) self.assertIsNotNone(key.key) + + def test_delete_none(self): + key = FSKey(self.tmpdir.name, 'something.asc') + key.delete() + + def test_shred(self): + key_name = 'something.asc' + key_path = join(self.tmpdir.name, key_name) + key_data = load_key('rsa_1024.priv.asc') + with open(key_path, 'w') as key_file: + key_file.write(str(key_data)) + + key = FSKey(self.tmpdir.name, key_name, True) + + key.shred() + self.assertFalse(exists(key.key_path)) + self.assertIsNotNone(key.key) + + def test_shred_none(self): + key = FSKey(self.tmpdir.name, 'something.asc') + key.shred() |