Do not raise on not 2XX error codes

Also accept unicode ascii keys

2 files changed, 12 insertions, 7 deletions

diff --git a/keymanager/src/leap/keymanager/__init__.py b/keymanager/src/leap/keymanager/__init__.py
index 05cbccaa..2f39ad9e 100644
--- a/keymanager/src/leap/keymanager/__init__.py
+++ b/keymanager/src/leap/keymanager/__init__.py
@@ -127,11 +127,16 @@ class KeyManager(object):
             self._ca_cert_path is not None,
             'We need the CA certificate path!')
         res = self._fetcher.get(uri, data=data, verify=self._ca_cert_path)
-        # assert that the response is valid
-        res.raise_for_status()
-        leap_assert(
-            res.headers['content-type'].startswith('application/json'),
-            'Content-type is not JSON.')
+        # Nickserver now returns 404 for key not found and 500 for
+        # other cases (like key too small), so we are skipping this
+        # check for the time being
+        # res.raise_for_status()
+
+        # Responses are now text/plain, although it's json anyway, but
+        # this will fail when it shouldn't
+        # leap_assert(
+        #     res.headers['content-type'].startswith('application/json'),
+        #     'Content-type is not JSON.')
         return res
 
     def _put(self, uri, data=None):
diff --git a/keymanager/src/leap/keymanager/openpgp.py b/keymanager/src/leap/keymanager/openpgp.py
index aa04ed09..7946db85 100644
--- a/keymanager/src/leap/keymanager/openpgp.py
+++ b/keymanager/src/leap/keymanager/openpgp.py
@@ -301,9 +301,9 @@ class OpenPGPScheme(EncryptionScheme):
         Put key contained in ascii-armored C{key_data} in local storage.
 
         :param key_data: The key data to be stored.
-        :type key_data: str
+        :type key_data: str or unicode
         """
-        leap_assert_type(key_data, str)
+        leap_assert_type(key_data, (str, unicode))
         # TODO: add more checks for correct key data.
         leap_assert(key_data is not None, 'Data does not represent a key.')