foster/bitcoin-bitcoin-core
0
0
Fork 0
mirror of https://github.com/bitcoin/bitcoin.git synced 2025-02-14 11:26:09 -05:00
Code Issues Activity

tests: Use mocktime for wallet encryption timeout

Browse source
This commit is contained in:
Andrew Chow 2022-08-19 13:51:39 -04:00
parent 6b56873b41
commit 02dea9a47f
1 changed files with 7 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
test/functional/wallet_encryption.py
View file

@ -9,8 +9,7 @@ import time
from test_framework.test_framework import BitcoinTestFramework from test_framework.test_framework import BitcoinTestFramework
from test_framework.util import ( from test_framework.util import (
assert_raises_rpc_error, assert_raises_rpc_error,
assert_greater_than, assert_equal,
assert_greater_than_or_equal,
) )
@ -76,21 +75,18 @@ class WalletEncryptionTest(BitcoinTestFramework):
self.log.info('Check a timeout less than the limit') self.log.info('Check a timeout less than the limit')
MAX_VALUE = 100000000 MAX_VALUE = 100000000
expected_time = int(time.time()) + MAX_VALUE - 600 now = int(time.time())
self.nodes[0].setmocktime(now)
expected_time = now + MAX_VALUE - 600
self.nodes[0].walletpassphrase(passphrase2, MAX_VALUE - 600) self.nodes[0].walletpassphrase(passphrase2, MAX_VALUE - 600)
# give buffer for walletpassphrase, since it iterates over all encrypted keys
expected_time_with_buffer = time.time() + MAX_VALUE - 600
actual_time = self.nodes[0].getwalletinfo()['unlocked_until'] actual_time = self.nodes[0].getwalletinfo()['unlocked_until']
assert_greater_than_or_equal(actual_time, expected_time) assert_equal(actual_time, expected_time)
assert_greater_than(expected_time_with_buffer, actual_time)
self.log.info('Check a timeout greater than the limit') self.log.info('Check a timeout greater than the limit')
expected_time = int(time.time()) + MAX_VALUE - 1 expected_time = now + MAX_VALUE
self.nodes[0].walletpassphrase(passphrase2, MAX_VALUE + 1000) self.nodes[0].walletpassphrase(passphrase2, MAX_VALUE + 1000)
expected_time_with_buffer = time.time() + MAX_VALUE
actual_time = self.nodes[0].getwalletinfo()['unlocked_until'] actual_time = self.nodes[0].getwalletinfo()['unlocked_until']
assert_greater_than_or_equal(actual_time, expected_time) assert_equal(actual_time, expected_time)
assert_greater_than(expected_time_with_buffer, actual_time)
if __name__ == '__main__': if __name__ == '__main__':