diff options
author | vrs <none> | 2018-12-08 22:20:05 +0100 |
---|---|---|
committer | Patrick Totzke <patricktotzke@gmail.com> | 2018-12-10 10:25:49 +0000 |
commit | b981bfe0c61d9aa55652b4b0a01b846aaa9c993a (patch) | |
tree | b16298ac338ef19758af553b845ddc05b6124fc1 /tests/commands/utils_tests.py | |
parent | fdac378674a3e90413fcfba3dd2c9cc0debbb9cc (diff) |
attach accounts to envelopes
simplifies a few tests, fixes #1177
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) |