diff options
Diffstat (limited to 'src/mailman_pgp/workflows/tests/test_base.py')
| -rw-r--r-- | src/mailman_pgp/workflows/tests/test_base.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/src/mailman_pgp/workflows/tests/test_base.py b/src/mailman_pgp/workflows/tests/test_base.py index a52571c..c99c6d6 100644 --- a/src/mailman_pgp/workflows/tests/test_base.py +++ b/src/mailman_pgp/workflows/tests/test_base.py @@ -88,6 +88,7 @@ class TestPubkeyMixin(unittest.TestCase): pubkey=self.sender_key.pubkey, pubkey_pre_confirmed=True) workflow.run_thru('pubkey_checks') + next(workflow) with patch.object(workflow, '_step_do_subscription') as step: next(workflow) step.assert_called_once_with() @@ -112,7 +113,7 @@ class TestPubkeyMixin(unittest.TestCase): self.sender_key.pubkey.fingerprint) with patch.object(receive_workflow, - '_step_send_confirm_request') as step: + '_step_send_key_confirm_request') as step: next(receive_workflow) step.assert_called_once_with() @@ -147,6 +148,7 @@ class TestPubkeyMixin(unittest.TestCase): receive_workflow.token = workflow.token receive_workflow.restore() receive_workflow.run_thru('receive_key') + next(receive_workflow) with patch.object(receive_workflow, '_step_do_subscription') as step: next(receive_workflow) step.assert_called_once_with() @@ -182,8 +184,8 @@ class TestPubkeyMixin(unittest.TestCase): receive_workflow = ConfirmSubscriptionPolicy(self.mlist) receive_workflow.token = workflow.token receive_workflow.restore() - receive_workflow.run_thru('receive_confirmation') - + receive_workflow.run_thru('receive_key_confirmation') + next(receive_workflow) with patch.object(receive_workflow, '_step_do_subscription') as step: next(receive_workflow) step.assert_called_once_with() |