From a0a49d1e10da362f4bab683ae357ed72ab40b2ef Mon Sep 17 00:00:00 2001 From: Anike Arni Date: Thu, 16 Mar 2017 18:32:13 -0300 Subject: [feature] generate recovery code with @tayane --- testing/tests/client/test_recovery_code.py | 38 ++++++++++++++++++++++++++++++ 1 file changed, 38 insertions(+) create mode 100644 testing/tests/client/test_recovery_code.py (limited to 'testing/tests/client/test_recovery_code.py') diff --git a/testing/tests/client/test_recovery_code.py b/testing/tests/client/test_recovery_code.py new file mode 100644 index 00000000..7bbccc41 --- /dev/null +++ b/testing/tests/client/test_recovery_code.py @@ -0,0 +1,38 @@ +# -*- CODing: utf-8 -*- +# test_recovery_code.py +# Copyright (C) 2017 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 . +""" +Tests for recovery code generation. +""" +import binascii + +from mock import patch +from twisted.trial import unittest +from leap.soledad.client._recovery_code import RecoveryCode + + +class RecoveryCodeTestCase(unittest.TestCase): + + @patch('leap.soledad.client._recovery_code.os.urandom') + def test_generate_recovery_code(self, mock_os_urandom): + generated_random_code = '123456' + mock_os_urandom.return_value = generated_random_code + recovery_code = RecoveryCode() + + code = recovery_code.generate() + + mock_os_urandom.assert_called_with(RecoveryCode.code_length) + self.assertEqual(binascii.hexlify(generated_random_code), code) -- cgit v1.2.3