summaryrefslogtreecommitdiff
path: root/keymanager/tests/test_migrator.py
diff options
context:
space:
mode:
Diffstat (limited to 'keymanager/tests/test_migrator.py')
-rw-r--r--keymanager/tests/test_migrator.py176
1 files changed, 0 insertions, 176 deletions
diff --git a/keymanager/tests/test_migrator.py b/keymanager/tests/test_migrator.py
deleted file mode 100644
index 64cd8e10..00000000
--- a/keymanager/tests/test_migrator.py
+++ /dev/null
@@ -1,176 +0,0 @@
-# -*- coding: utf-8 -*-
-# test_migrator.py
-# Copyright (C) 2015 LEAP
-#
-# This program is free software: you can redistribute it and/or modify
-# it under the terms of the GNU General Public License as published by
-# the Free Software Foundation, either version 3 of the License, or
-# (at your option) any later version.
-#
-# This program 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 General Public License for more details.
-#
-# You should have received a copy of the GNU General Public License
-# along with this program. If not, see <http://www.gnu.org/licenses/>.
-
-
-"""
-Tests for the migrator.
-"""
-
-
-from collections import namedtuple
-from mock import Mock
-from twisted.internet.defer import succeed, inlineCallbacks
-
-from leap.keymanager.migrator import KeyDocumentsMigrator, KEY_ID_KEY
-from leap.keymanager.documents import (
- TAGS_PRIVATE_INDEX,
- KEYMANAGER_ACTIVE_TAG,
- KEYMANAGER_KEY_TAG,
- KEYMANAGER_DOC_VERSION,
-
- KEY_ADDRESS_KEY,
- KEY_UIDS_KEY,
- KEY_VERSION_KEY,
- KEY_FINGERPRINT_KEY,
- KEY_VALIDATION_KEY,
- KEY_LAST_AUDITED_AT_KEY,
- KEY_ENCR_USED_KEY,
- KEY_SIGN_USED_KEY,
-)
-from leap.keymanager.validation import ValidationLevels
-
-from common import (
- KeyManagerWithSoledadTestCase,
- ADDRESS,
- ADDRESS_2,
- KEY_FINGERPRINT,
-)
-
-
-class OpenPGPCryptoTestCase(KeyManagerWithSoledadTestCase):
- @inlineCallbacks
- def test_simple_migration(self):
- get_from_index = self._soledad.get_from_index
- delete_doc = self._soledad.delete_doc
- put_doc = self._soledad.put_doc
-
- def my_get_from_index(*args):
- docs = []
- if (args[0] == TAGS_PRIVATE_INDEX and
- args[2] == '0'):
- SoledadDocument = namedtuple("SoledadDocument", ["content"])
- if args[1] == KEYMANAGER_KEY_TAG:
- docs = [SoledadDocument({
- KEY_ADDRESS_KEY: [ADDRESS],
- KEY_ID_KEY: KEY_FINGERPRINT[-16:],
- KEY_FINGERPRINT_KEY: KEY_FINGERPRINT,
- KEY_VALIDATION_KEY: str(ValidationLevels.Weak_Chain),
- KEY_LAST_AUDITED_AT_KEY: 0,
- KEY_ENCR_USED_KEY: True,
- KEY_SIGN_USED_KEY: False,
- })]
- if args[1] == KEYMANAGER_ACTIVE_TAG:
- docs = [SoledadDocument({
- KEY_ID_KEY: KEY_FINGERPRINT[-16:],
- })]
- return succeed(docs)
-
- self._soledad.get_from_index = my_get_from_index
- self._soledad.delete_doc = Mock(return_value=succeed(None))
- self._soledad.put_doc = Mock(return_value=succeed(None))
-
- try:
- migrator = KeyDocumentsMigrator(self._soledad)
- yield migrator.migrate()
- call_list = self._soledad.put_doc.call_args_list
- finally:
- self._soledad.get_from_index = get_from_index
- self._soledad.delete_doc = delete_doc
- self._soledad.put_doc = put_doc
-
- self.assertEqual(len(call_list), 2)
- active = call_list[0][0][0]
- key = call_list[1][0][0]
-
- self.assertTrue(KEY_ID_KEY not in active.content)
- self.assertEqual(active.content[KEY_VERSION_KEY],
- KEYMANAGER_DOC_VERSION)
- self.assertEqual(active.content[KEY_FINGERPRINT_KEY], KEY_FINGERPRINT)
- self.assertEqual(active.content[KEY_VALIDATION_KEY],
- str(ValidationLevels.Weak_Chain))
- self.assertEqual(active.content[KEY_LAST_AUDITED_AT_KEY], 0)
- self.assertEqual(active.content[KEY_ENCR_USED_KEY], True)
- self.assertEqual(active.content[KEY_SIGN_USED_KEY], False)
-
- self.assertTrue(KEY_ID_KEY not in key.content)
- self.assertTrue(KEY_ADDRESS_KEY not in key.content)
- self.assertTrue(KEY_VALIDATION_KEY not in key.content)
- self.assertTrue(KEY_LAST_AUDITED_AT_KEY not in key.content)
- self.assertTrue(KEY_ENCR_USED_KEY not in key.content)
- self.assertTrue(KEY_SIGN_USED_KEY not in key.content)
- self.assertEqual(key.content[KEY_UIDS_KEY], [ADDRESS])
-
- @inlineCallbacks
- def test_two_active_docs(self):
- get_from_index = self._soledad.get_from_index
- delete_doc = self._soledad.delete_doc
- put_doc = self._soledad.put_doc
-
- def my_get_from_index(*args):
- docs = []
- if (args[0] == TAGS_PRIVATE_INDEX and
- args[2] == '0'):
- SoledadDocument = namedtuple("SoledadDocument", ["content"])
- if args[1] == KEYMANAGER_KEY_TAG:
- validation = str(ValidationLevels.Provider_Trust)
- docs = [SoledadDocument({
- KEY_ADDRESS_KEY: [ADDRESS, ADDRESS_2],
- KEY_ID_KEY: KEY_FINGERPRINT[-16:],
- KEY_FINGERPRINT_KEY: KEY_FINGERPRINT,
- KEY_VALIDATION_KEY: validation,
- KEY_LAST_AUDITED_AT_KEY: 1984,
- KEY_ENCR_USED_KEY: True,
- KEY_SIGN_USED_KEY: False,
- })]
- if args[1] == KEYMANAGER_ACTIVE_TAG:
- docs = [
- SoledadDocument({
- KEY_ADDRESS_KEY: ADDRESS,
- KEY_ID_KEY: KEY_FINGERPRINT[-16:],
- }),
- SoledadDocument({
- KEY_ADDRESS_KEY: ADDRESS_2,
- KEY_ID_KEY: KEY_FINGERPRINT[-16:],
- }),
- ]
- return succeed(docs)
-
- self._soledad.get_from_index = my_get_from_index
- self._soledad.delete_doc = Mock(return_value=succeed(None))
- self._soledad.put_doc = Mock(return_value=succeed(None))
-
- try:
- migrator = KeyDocumentsMigrator(self._soledad)
- yield migrator.migrate()
- call_list = self._soledad.put_doc.call_args_list
- finally:
- self._soledad.get_from_index = get_from_index
- self._soledad.delete_doc = delete_doc
- self._soledad.put_doc = put_doc
-
- self.assertEqual(len(call_list), 3)
- for active in [call[0][0] for call in call_list][:2]:
- self.assertTrue(KEY_ID_KEY not in active.content)
- self.assertEqual(active.content[KEY_VERSION_KEY],
- KEYMANAGER_DOC_VERSION)
- self.assertEqual(active.content[KEY_FINGERPRINT_KEY],
- KEY_FINGERPRINT)
- self.assertEqual(active.content[KEY_VALIDATION_KEY],
- str(ValidationLevels.Weak_Chain))
- self.assertEqual(active.content[KEY_LAST_AUDITED_AT_KEY], 0)
- self.assertEqual(active.content[KEY_ENCR_USED_KEY], False)
- self.assertEqual(active.content[KEY_SIGN_USED_KEY], False)