diff options
Diffstat (limited to 'src/leap/mail/tests/__init__.py')
| -rw-r--r-- | src/leap/mail/tests/__init__.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/src/leap/mail/tests/__init__.py b/src/leap/mail/tests/__init__.py index 8094c11..5493d43 100644 --- a/src/leap/mail/tests/__init__.py +++ b/src/leap/mail/tests/__init__.py @@ -26,7 +26,6 @@ from twisted.trial import unittest from leap.soledad.client import Soledad from leap.keymanager import KeyManager -from leap.keymanager.openpgp import OpenPGPKey from leap.common.testing.basetest import BaseLeapTest @@ -97,8 +96,8 @@ class TestCaseWithKeyManager(unittest.TestCase, BaseLeapTest): self._km._async_client.request = Mock(return_value="") self._km._async_client_pinned.request = Mock(return_value="") - d1 = self._km.put_raw_key(PRIVATE_KEY, OpenPGPKey, ADDRESS) - d2 = self._km.put_raw_key(PRIVATE_KEY_2, OpenPGPKey, ADDRESS_2) + d1 = self._km.put_raw_key(PRIVATE_KEY, ADDRESS) + d2 = self._km.put_raw_key(PRIVATE_KEY_2, ADDRESS_2) return gatherResults([d1, d2]) def tearDown(self): |