From 067db035910c8284d508401c34ca4cb5928c2b52 Mon Sep 17 00:00:00 2001 From: Anike Arni Date: Fri, 17 Mar 2017 11:18:39 -0300 Subject: [#1012] Generates recovery code in soledad client with @tayanefernandes --- service/test/unit/test_account_recovery.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) (limited to 'service/test/unit/test_account_recovery.py') diff --git a/service/test/unit/test_account_recovery.py b/service/test/unit/test_account_recovery.py index 026d161f..af14814a 100644 --- a/service/test/unit/test_account_recovery.py +++ b/service/test/unit/test_account_recovery.py @@ -26,8 +26,11 @@ class AccountRecoveryTest(unittest.TestCase): @inlineCallbacks def test_update_recovery_code(self): - mock_session = Mock() - account_recovery = AccountRecovery(mock_session) + generated_code = '4645a2f8997e5d0d' + mock_bonafide_session = Mock() + mock_soledad = Mock() + mock_soledad.create_recovery_code.return_value = generated_code + account_recovery = AccountRecovery(mock_bonafide_session, mock_soledad) yield account_recovery.update_recovery_code() - mock_session.update_recovery_code.assert_called_once_with('123') + mock_bonafide_session.update_recovery_code.assert_called_once_with(generated_code) -- cgit v1.2.3