Added test for ext_keymanager_fetch_key monkey patch.

- Issue #319

1 files changed, 76 insertions, 0 deletions

diff --git a/service/test/unit/support/test_ext_keymanager_fetch_key.py b/service/test/unit/support/test_ext_keymanager_fetch_key.py
new file mode 100644
index 00000000..8998198d
--- /dev/null
+++ b/service/test/unit/support/test_ext_keymanager_fetch_key.py
@@ -0,0 +1,76 @@
+#
+# Copyright (c) 2014 ThoughtWorks, Inc.
+#
+# Pixelated is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Pixelated is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with Pixelated. If not, see <http://www.gnu.org/licenses/>.
+import unittest
+from mock import MagicMock, patch
+
+from leap.keymanager import KeyManager
+from leap.keymanager.keys import KEY_ADDRESS_KEY, KEY_TYPE_KEY, KEY_ID_KEY, KEY_FINGERPRINT_KEY, KEY_DATA_KEY, KEY_PRIVATE_KEY, KEY_LENGTH_KEY, KEY_EXPIRY_DATE_KEY, KEY_FIRST_SEEN_AT_KEY, KEY_LAST_AUDITED_AT_KEY, KEY_VALIDATION_KEY, KEY_TAGS_KEY
+from leap.keymanager.openpgp import OpenPGPKey
+from leap.keymanager.errors import KeyNotFound
+import pixelated.support.ext_keymanager_fetch_key
+from requests.exceptions import HTTPError
+
+
+class TestDoc(object):
+    def __init__(self, encryption_key):
+        self.content = encryption_key
+
+sample_key = {
+    KEY_ADDRESS_KEY: 'foo@bar.de',
+    KEY_TYPE_KEY: 'type',
+    KEY_ID_KEY: 'key_id',
+    KEY_FINGERPRINT_KEY: 'fingerprint',
+    KEY_DATA_KEY: 'key_data',
+    KEY_PRIVATE_KEY: None,
+    KEY_LENGTH_KEY: 'length',
+    KEY_EXPIRY_DATE_KEY: 'expiry_date',
+    KEY_FIRST_SEEN_AT_KEY: 'first_seen_at',
+    KEY_LAST_AUDITED_AT_KEY: 'last_audited_at',
+    KEY_VALIDATION_KEY: 'validation',
+    KEY_TAGS_KEY: 'tags',
+}
+
+
+class TestExtKeyManagerFetchKey(unittest.TestCase):
+
+    @patch('leap.keymanager.requests')
+    def test_retrieves_key(self, requests_mock):
+        nickserver_url = 'http://some/nickserver/uri'
+        soledad = MagicMock()
+        soledad.get_from_index.side_effect = [[], [TestDoc(sample_key)]]
+
+        km = KeyManager('me@bar.de', nickserver_url, soledad, ca_cert_path='some path')
+
+        result = km.get_key('foo@bar.de', OpenPGPKey)
+
+        self.assertEqual(str(OpenPGPKey('foo@bar.de', key_id='key_id')), str(result))
+
+    @patch('leap.keymanager.requests')
+    def test_http_error_500(self, requests_mock):
+        def do_request(one, data=None, verify=None):
+            response = MagicMock()
+            response.raise_for_status = MagicMock()
+            response.raise_for_status.side_effect = HTTPError
+            return response
+
+        nickserver_url = 'http://some/nickserver/uri'
+        soledad = MagicMock()
+        soledad.get_from_index.side_effect = [[], []]
+        requests_mock.get.side_effect = do_request
+
+        km = KeyManager('me@bar.de', nickserver_url, soledad, ca_cert_path='some path')
+
+        self.assertRaises(KeyNotFound, km.get_key, 'foo@bar.de', OpenPGPKey)