[feat] remove the keytype support

- Resolves: #8031

1 files changed, 2 insertions, 3 deletions

diff --git a/src/leap/keymanager/tests/__init__.py b/src/leap/keymanager/tests/__init__.py
index 2a6a3f1..a20e1fd 100644
--- a/src/leap/keymanager/tests/__init__.py
+++ b/src/leap/keymanager/tests/__init__.py
@@ -27,7 +27,6 @@ from twisted.trial import unittest
 from leap.common.testing.basetest import BaseLeapTest
 from leap.soledad.client import Soledad
 from leap.keymanager import KeyManager
-from leap.keymanager.openpgp import OpenPGPKey
 
 PATH = os.path.dirname(os.path.realpath(__file__))
 
@@ -56,7 +55,7 @@ class KeyManagerWithSoledadTestCase(unittest.TestCase, BaseLeapTest):
         km = self._key_manager()
 
         # wait for the indexes to be ready for the tear down
-        d = km._wrapper_map[OpenPGPKey].deferred_init
+        d = km._openpgp.deferred_init
         d.addCallback(lambda _: self.delete_all_keys(km))
         d.addCallback(lambda _: self.tearDownEnv())
         d.addCallback(lambda _: self._soledad.close())
@@ -66,7 +65,7 @@ class KeyManagerWithSoledadTestCase(unittest.TestCase, BaseLeapTest):
         def delete_keys(keys):
             deferreds = []
             for key in keys:
-                d = km._wrapper_map[key.__class__].delete_key(key)
+                d = km._openpgp.delete_key(key)
                 deferreds.append(d)
             return gatherResults(deferreds)