diff options
Diffstat (limited to 'tests/commands/utils_tests.py')
-rw-r--r-- | tests/commands/utils_tests.py | 10 |
1 files changed, 4 insertions, 6 deletions
diff --git a/tests/commands/utils_tests.py b/tests/commands/utils_tests.py index 639fb1cd..0642ad32 100644 --- a/tests/commands/utils_tests.py +++ b/tests/commands/utils_tests.py @@ -183,9 +183,8 @@ class TestSetEncrypt(unittest.TestCase): envelope['To'] = 'ambig@example.com' gpg_key = crypto.get_key(FPR) account = _Account(encrypt_to_self=True, gpg_key=gpg_key) - with mock.patch('alot.commands.thread.settings.get_account_by_address', - mock.Mock(return_value=account)): - await utils.update_keys(ui, envelope) + envelope.account = account + await utils.update_keys(ui, envelope) self.assertTrue(envelope.encrypt) self.assertIn(FPR, envelope.encrypt_keys) self.assertEqual(gpg_key, envelope.encrypt_keys[FPR]) @@ -198,8 +197,7 @@ class TestSetEncrypt(unittest.TestCase): envelope['To'] = 'ambig@example.com' gpg_key = crypto.get_key(FPR) account = _Account(encrypt_to_self=False, gpg_key=gpg_key) - with mock.patch('alot.commands.thread.settings.get_account_by_address', - mock.Mock(return_value=account)): - await utils.update_keys(ui, envelope) + envelope.account = account + await utils.update_keys(ui, envelope) self.assertTrue(envelope.encrypt) self.assertNotIn(FPR, envelope.encrypt_keys) |