304 lines
15 KiB
Python
304 lines
15 KiB
Python
from datetime import datetime
|
|
from decimal import Decimal
|
|
import unittest
|
|
import warnings
|
|
|
|
from monero.wallet import Wallet
|
|
from monero.account import Account
|
|
from monero.address import address
|
|
from monero.numbers import PaymentID
|
|
from monero.transaction import IncomingPayment, Transaction
|
|
|
|
class FiltersTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
class MockBackend(object):
|
|
def __init__(self):
|
|
self.transfers = []
|
|
tx = Transaction(
|
|
timestamp=datetime(2018, 1, 29, 15, 0, 25),
|
|
height=1087606,
|
|
hash='a0b876ebcf7c1d499712d84cedec836f9d50b608bb22d6cb49fd2feae3ffed14',
|
|
fee=Decimal('0.00352891'))
|
|
pm = IncomingPayment(
|
|
amount=Decimal('1'),
|
|
local_address=address('Bf6ngv7q2TBWup13nEm9AjZ36gLE6i4QCaZ7XScZUKDUeGbYEHmPRdegKGwLT8tBBK7P6L32RELNzCR6QzNFkmogDjvypyV'),
|
|
payment_id=PaymentID('0166d8da6c0045c51273dd65d6f63734beb8a84e0545a185b2cfd053fced9f5d'),
|
|
transaction=tx)
|
|
self.transfers.append(pm)
|
|
tx = Transaction(
|
|
timestamp=datetime(2018, 1, 29, 14, 57, 47),
|
|
height=1087601,
|
|
hash='f34b495cec77822a70f829ec8a5a7f1e727128d62e6b1438e9cb7799654d610e',
|
|
fee=Decimal('0.008661870000'))
|
|
pm = IncomingPayment(
|
|
amount=Decimal('3.000000000000'),
|
|
local_address=address('BhE3cQvB7VF2uuXcpXp28Wbadez6GgjypdRS1F1Mzqn8Advd6q8VfaX8ZoEDobjejrMfpHeNXoX8MjY8q8prW1PEALgr1En'),
|
|
payment_id=PaymentID('f75ad90e25d71a12'),
|
|
transaction=tx)
|
|
self.transfers.append(pm)
|
|
tx = Transaction(
|
|
timestamp=datetime(2018, 1, 29, 13, 17, 18),
|
|
height=1087530,
|
|
hash='5c3ab739346e9d98d38dc7b8d36a4b7b1e4b6a16276946485a69797dbf887cd8',
|
|
fee=Decimal('0.000962550000'))
|
|
pm = IncomingPayment(
|
|
amount=Decimal('10.000000000000'),
|
|
local_address=address('9tQoHWyZ4yXUgbz9nvMcFZUfDy5hxcdZabQCxmNCUukKYicXegsDL7nQpcUa3A1pF6K3fhq3scsyY88tdB1MqucULcKzWZC'),
|
|
payment_id=PaymentID('f75ad90e25d71a12'),
|
|
transaction=tx)
|
|
self.transfers.append(pm)
|
|
tx = Transaction(
|
|
timestamp=datetime(2018, 1, 29, 13, 17, 18),
|
|
height=1087530,
|
|
hash='4ea70add5d0c7db33557551b15cd174972fcfc73bf0f6a6b47b7837564b708d3',
|
|
fee=Decimal('0.000962550000'))
|
|
pm = IncomingPayment(
|
|
amount=Decimal('4.000000000000'),
|
|
local_address=address('9tQoHWyZ4yXUgbz9nvMcFZUfDy5hxcdZabQCxmNCUukKYicXegsDL7nQpcUa3A1pF6K3fhq3scsyY88tdB1MqucULcKzWZC'),
|
|
payment_id=PaymentID('f75ad90e25d71a12'),
|
|
transaction=tx)
|
|
self.transfers.append(pm)
|
|
tx = Transaction(
|
|
timestamp=datetime(2018, 1, 29, 13, 17, 18),
|
|
height=1087530,
|
|
hash='e9a71c01875bec20812f71d155bfabf42024fde3ec82475562b817dcc8cbf8dc',
|
|
fee=Decimal('0.000962550000'))
|
|
pm = IncomingPayment(
|
|
amount=Decimal('2.120000000000'),
|
|
local_address=address('9tQoHWyZ4yXUgbz9nvMcFZUfDy5hxcdZabQCxmNCUukKYicXegsDL7nQpcUa3A1pF6K3fhq3scsyY88tdB1MqucULcKzWZC'),
|
|
payment_id=PaymentID('cb248105ea6a9189'),
|
|
transaction=tx)
|
|
self.transfers.append(pm)
|
|
tx = Transaction(
|
|
timestamp=datetime(2018, 1, 29, 14, 57, 47),
|
|
height=1087601,
|
|
hash='5ef7ead6a041101ed326568fbb59c128403cba46076c3f353cd110d969dac808',
|
|
fee=Decimal('0.000962430000'))
|
|
pm = IncomingPayment(
|
|
amount=Decimal('7.000000000000'),
|
|
local_address=address('BhE3cQvB7VF2uuXcpXp28Wbadez6GgjypdRS1F1Mzqn8Advd6q8VfaX8ZoEDobjejrMfpHeNXoX8MjY8q8prW1PEALgr1En'),
|
|
payment_id=PaymentID('0000000000000000'),
|
|
transaction=tx)
|
|
self.transfers.append(pm)
|
|
tx = Transaction(
|
|
timestamp=datetime(2018, 1, 29, 13, 17, 18),
|
|
height=1087530,
|
|
hash='cc44568337a186c2e1ccc080b43b4ae9db26a07b7afd7edeed60ce2fc4a6477f',
|
|
fee=Decimal('0.000962550000'))
|
|
pm = IncomingPayment(
|
|
amount=Decimal('10.000000000000'),
|
|
local_address=address('9tQoHWyZ4yXUgbz9nvMcFZUfDy5hxcdZabQCxmNCUukKYicXegsDL7nQpcUa3A1pF6K3fhq3scsyY88tdB1MqucULcKzWZC'),
|
|
payment_id=PaymentID('0000000000000000'),
|
|
transaction=tx)
|
|
self.transfers.append(pm)
|
|
tx = Transaction(
|
|
timestamp=datetime(2018, 1, 29, 21, 13, 28),
|
|
height=None,
|
|
hash='d29264ad317e8fdb55ea04484c00420430c35be7b3fe6dd663f99aebf41a786c',
|
|
fee=Decimal('0.000961950000'))
|
|
pm = IncomingPayment(
|
|
amount=Decimal('3.140000000000'),
|
|
local_address=address('9tQoHWyZ4yXUgbz9nvMcFZUfDy5hxcdZabQCxmNCUukKYicXegsDL7nQpcUa3A1pF6K3fhq3scsyY88tdB1MqucULcKzWZC'),
|
|
payment_id=PaymentID('03f6649304ea4cb2'),
|
|
transaction=tx)
|
|
self.transfers.append(pm)
|
|
|
|
def height(self):
|
|
return 1087607
|
|
|
|
def accounts(self):
|
|
return [Account(self, 0)]
|
|
|
|
def transfers_in(self, account, pmtfilter):
|
|
return list(pmtfilter.filter(self.transfers))
|
|
|
|
self.wallet = Wallet(MockBackend())
|
|
|
|
def test_filter_none(self):
|
|
pmts = self.wallet.incoming()
|
|
self.assertEqual(len(pmts), 7)
|
|
|
|
def test_filter_payment_id(self):
|
|
pmts = self.wallet.incoming(payment_id='cb248105ea6a9189')
|
|
self.assertEqual(len(pmts), 1)
|
|
self.assertEqual(
|
|
pmts[0].transaction.hash,
|
|
'e9a71c01875bec20812f71d155bfabf42024fde3ec82475562b817dcc8cbf8dc')
|
|
pmts = self.wallet.incoming(payment_id='f75ad90e25d71a12')
|
|
self.assertEqual(len(pmts), 3)
|
|
pmts = self.wallet.incoming(payment_id=('cb248105ea6a9189', 'f75ad90e25d71a12'))
|
|
self.assertEqual(len(pmts), 4)
|
|
self.assertEqual(
|
|
pmts,
|
|
self.wallet.incoming(payment_id=(PaymentID('cb248105ea6a9189'), 'f75ad90e25d71a12')))
|
|
|
|
def test_filter_address(self):
|
|
pmts = self.wallet.incoming(
|
|
local_address='BhE3cQvB7VF2uuXcpXp28Wbadez6GgjypdRS1F1Mzqn8Advd6q8VfaX8ZoEDobjejrMfpHeNXoX8MjY8q8prW1PEALgr1En')
|
|
self.assertEqual(len(pmts), 2)
|
|
self.assertEqual(
|
|
pmts,
|
|
self.wallet.incoming(
|
|
local_address=address('BhE3cQvB7VF2uuXcpXp28Wbadez6GgjypdRS1F1Mzqn8Advd6q8VfaX8ZoEDobjejrMfpHeNXoX8MjY8q8prW1PEALgr1En')))
|
|
pmts = self.wallet.incoming(
|
|
local_address=(
|
|
'BhE3cQvB7VF2uuXcpXp28Wbadez6GgjypdRS1F1Mzqn8Advd6q8VfaX8ZoEDobjejrMfpHeNXoX8MjY8q8prW1PEALgr1En',
|
|
'Bf6ngv7q2TBWup13nEm9AjZ36gLE6i4QCaZ7XScZUKDUeGbYEHmPRdegKGwLT8tBBK7P6L32RELNzCR6QzNFkmogDjvypyV'))
|
|
self.assertEqual(len(pmts), 3)
|
|
|
|
def test_filter_mempool(self):
|
|
with warnings.catch_warnings(record=True) as w:
|
|
warnings.simplefilter('always')
|
|
pmts = self.wallet.incoming()
|
|
self.assertEqual(len(pmts), 7)
|
|
for p in pmts:
|
|
self.assertGreater(self.wallet.confirmations(p.transaction), 0)
|
|
pmts = self.wallet.incoming(unconfirmed=True)
|
|
self.assertEqual(len(pmts), 8)
|
|
pmts = self.wallet.incoming(unconfirmed=True, confirmed=False)
|
|
self.assertEqual(len(pmts), 1)
|
|
self.assertEqual(
|
|
pmts[0].transaction.hash,
|
|
'd29264ad317e8fdb55ea04484c00420430c35be7b3fe6dd663f99aebf41a786c')
|
|
self.assertEqual(self.wallet.confirmations(pmts[0]), 0)
|
|
self.assertEqual(self.wallet.confirmations(pmts[0].transaction), 0)
|
|
self.assertEqual(len(w), 0)
|
|
pmts = self.wallet.incoming(unconfirmed=True, confirmed=False, min_height=1)
|
|
self.assertEqual(len(pmts), 0)
|
|
self.assertEqual(len(w), 1)
|
|
self.assertIs(w[0].category, RuntimeWarning)
|
|
pmts = self.wallet.incoming(unconfirmed=True, confirmed=False, max_height=99999999999999)
|
|
self.assertEqual(len(pmts), 0)
|
|
self.assertEqual(len(w), 2)
|
|
self.assertIs(w[1].category, RuntimeWarning)
|
|
pmts = self.wallet.incoming(payment_id='03f6649304ea4cb2')
|
|
self.assertEqual(len(pmts), 0)
|
|
pmts = self.wallet.incoming(unconfirmed=True, payment_id='03f6649304ea4cb2')
|
|
self.assertEqual(len(pmts), 1)
|
|
pmts = self.wallet.incoming(
|
|
local_address='9tQoHWyZ4yXUgbz9nvMcFZUfDy5hxcdZabQCxmNCUukKYicXegsDL7nQpcUa3A1pF6K3fhq3scsyY88tdB1MqucULcKzWZC')
|
|
self.assertEqual(len(pmts), 4)
|
|
pmts = self.wallet.incoming(
|
|
unconfirmed=True,
|
|
local_address='9tQoHWyZ4yXUgbz9nvMcFZUfDy5hxcdZabQCxmNCUukKYicXegsDL7nQpcUa3A1pF6K3fhq3scsyY88tdB1MqucULcKzWZC')
|
|
self.assertEqual(len(pmts), 5)
|
|
pmts = self.wallet.incoming(
|
|
local_address='9tQoHWyZ4yXUgbz9nvMcFZUfDy5hxcdZabQCxmNCUukKYicXegsDL7nQpcUa3A1pF6K3fhq3scsyY88tdB1MqucULcKzWZC',
|
|
payment_id='03f6649304ea4cb2')
|
|
self.assertEqual(len(pmts), 0)
|
|
pmts = self.wallet.incoming(
|
|
unconfirmed=True,
|
|
local_address='9tQoHWyZ4yXUgbz9nvMcFZUfDy5hxcdZabQCxmNCUukKYicXegsDL7nQpcUa3A1pF6K3fhq3scsyY88tdB1MqucULcKzWZC',
|
|
payment_id='03f6649304ea4cb2')
|
|
self.assertEqual(len(pmts), 1)
|
|
self.assertEqual(len(w), 2)
|
|
|
|
def test_filter_mempool_absent(self):
|
|
with warnings.catch_warnings(record=True) as w:
|
|
warnings.simplefilter('always')
|
|
pmts = self.wallet.incoming()
|
|
self.assertEqual(len(pmts), 7)
|
|
for p in pmts:
|
|
self.assertGreater(self.wallet.confirmations(p.transaction), 0)
|
|
pmts = self.wallet.incoming(unconfirmed=False)
|
|
self.assertEqual(len(pmts), 7)
|
|
pmts = self.wallet.incoming(confirmed=True)
|
|
self.assertEqual(len(pmts), 7)
|
|
pmts = self.wallet.incoming(confirmed=True, unconfirmed=False)
|
|
self.assertEqual(len(pmts), 7)
|
|
self.assertEqual(len(w), 0)
|
|
|
|
def test_filter_mempool_present(self):
|
|
with warnings.catch_warnings(record=True) as w:
|
|
warnings.simplefilter('always')
|
|
pmts = self.wallet.incoming(unconfirmed=True)
|
|
self.assertEqual(len(pmts), 8)
|
|
pmts = self.wallet.incoming(unconfirmed=True, confirmed=False)
|
|
self.assertEqual(len(pmts), 1)
|
|
self.assertEqual(
|
|
pmts[0].transaction.hash,
|
|
'd29264ad317e8fdb55ea04484c00420430c35be7b3fe6dd663f99aebf41a786c')
|
|
self.assertEqual(self.wallet.confirmations(pmts[0]), 0)
|
|
self.assertEqual(self.wallet.confirmations(pmts[0].transaction), 0)
|
|
self.assertEqual(len(w), 0)
|
|
|
|
def test_filter_mempool_filter_height(self):
|
|
with warnings.catch_warnings(record=True) as w:
|
|
warnings.simplefilter('always')
|
|
# mempool is always excluded and warnings are generated
|
|
pmts = self.wallet.incoming(unconfirmed=True, confirmed=False, min_height=1)
|
|
self.assertEqual(len(pmts), 0)
|
|
self.assertEqual(len(w), 1)
|
|
self.assertIs(w[0].category, RuntimeWarning)
|
|
pmts = self.wallet.incoming(unconfirmed=True, confirmed=False, max_height=99999999999999)
|
|
self.assertEqual(len(pmts), 0)
|
|
self.assertEqual(len(w), 2)
|
|
self.assertIs(w[1].category, RuntimeWarning)
|
|
|
|
def test_filter_mempool_filter_payment_id(self):
|
|
with warnings.catch_warnings(record=True) as w:
|
|
warnings.simplefilter('always')
|
|
# mempool excluded
|
|
pmts = self.wallet.incoming(payment_id='03f6649304ea4cb2')
|
|
self.assertEqual(len(pmts), 0)
|
|
# mempool included
|
|
pmts = self.wallet.incoming(unconfirmed=True, payment_id='03f6649304ea4cb2')
|
|
self.assertEqual(len(pmts), 1)
|
|
self.assertEqual(
|
|
pmts[0].transaction.hash,
|
|
'd29264ad317e8fdb55ea04484c00420430c35be7b3fe6dd663f99aebf41a786c')
|
|
self.assertEqual(len(w), 0)
|
|
|
|
def test_filter_mempool_filter_address(self):
|
|
with warnings.catch_warnings(record=True) as w:
|
|
warnings.simplefilter('always')
|
|
# mempool excluded
|
|
pmts = self.wallet.incoming(
|
|
local_address='9tQoHWyZ4yXUgbz9nvMcFZUfDy5hxcdZabQCxmNCUukKYicXegsDL7nQpcUa3A1pF6K3fhq3scsyY88tdB1MqucULcKzWZC')
|
|
self.assertEqual(len(pmts), 4)
|
|
# mempool included
|
|
pmts = self.wallet.incoming(
|
|
unconfirmed=True,
|
|
local_address='9tQoHWyZ4yXUgbz9nvMcFZUfDy5hxcdZabQCxmNCUukKYicXegsDL7nQpcUa3A1pF6K3fhq3scsyY88tdB1MqucULcKzWZC')
|
|
self.assertEqual(len(pmts), 5)
|
|
self.assertEqual(len(w), 0)
|
|
|
|
def test_filter_mempool_filter_address_and_payment_id(self):
|
|
with warnings.catch_warnings(record=True) as w:
|
|
warnings.simplefilter('always')
|
|
# mempool excluded
|
|
pmts = self.wallet.incoming(
|
|
local_address='9tQoHWyZ4yXUgbz9nvMcFZUfDy5hxcdZabQCxmNCUukKYicXegsDL7nQpcUa3A1pF6K3fhq3scsyY88tdB1MqucULcKzWZC',
|
|
payment_id='03f6649304ea4cb2')
|
|
self.assertEqual(len(pmts), 0)
|
|
# mempool included
|
|
pmts = self.wallet.incoming(
|
|
unconfirmed=True,
|
|
local_address='9tQoHWyZ4yXUgbz9nvMcFZUfDy5hxcdZabQCxmNCUukKYicXegsDL7nQpcUa3A1pF6K3fhq3scsyY88tdB1MqucULcKzWZC',
|
|
payment_id='03f6649304ea4cb2')
|
|
self.assertEqual(len(pmts), 1)
|
|
self.assertEqual(
|
|
pmts[0].transaction.hash,
|
|
'd29264ad317e8fdb55ea04484c00420430c35be7b3fe6dd663f99aebf41a786c')
|
|
self.assertEqual(len(w), 0)
|
|
|
|
def test_filter_mempool_filter_txid(self):
|
|
with warnings.catch_warnings(record=True) as w:
|
|
warnings.simplefilter('always')
|
|
# mempool excluded
|
|
pmts = self.wallet.incoming(
|
|
tx_id='d29264ad317e8fdb55ea04484c00420430c35be7b3fe6dd663f99aebf41a786c')
|
|
self.assertEqual(len(pmts), 0)
|
|
# mempool included
|
|
pmts = self.wallet.incoming(
|
|
unconfirmed=True,
|
|
tx_id='d29264ad317e8fdb55ea04484c00420430c35be7b3fe6dd663f99aebf41a786c')
|
|
self.assertEqual(len(pmts), 1)
|
|
self.assertEqual(
|
|
pmts[0].transaction.hash,
|
|
'd29264ad317e8fdb55ea04484c00420430c35be7b3fe6dd663f99aebf41a786c')
|
|
self.assertEqual(len(w), 0)
|
|
|
|
def test_filter_excessive(self):
|
|
self.assertRaises(ValueError, self.wallet.incoming, excessive_argument='foo')
|