121 lines
6.4 KiB
Python
121 lines
6.4 KiB
Python
|
from companion.pools.flexpool import FlexpoolHandler, Miner, Transaction
|
||
|
|
from datetime import datetime, timedelta
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
class TestFlexpoolHandler:
|
||
|
|
# def test_block(self, mocker):
|
||
|
|
def test_init(self):
|
||
|
|
handler = FlexpoolHandler()
|
||
|
|
assert handler.pool_name == 'flexpool'
|
||
|
|
|
||
|
|
def test_new_balance_with_notification(self, mocker):
|
||
|
|
"""Old balance is 1, new balance is 2, should send notification"""
|
||
|
|
notifier = mocker.Mock()
|
||
|
|
notifier.notify_balance = mocker.Mock()
|
||
|
|
handler = FlexpoolHandler(notifier=notifier)
|
||
|
|
miner = mocker.patch('flexpoolapi.miner')
|
||
|
|
miner().balance.return_value = 2
|
||
|
|
mocker.patch('companion.pools.flexpool.FlexpoolHandler._watch_miner_payments')
|
||
|
|
mocker.patch('companion.pools.flexpool.Miner.get_payements')
|
||
|
|
last_balance, last_transaction = handler.watch_miner(address='addr', last_balance=1)
|
||
|
|
assert last_balance == 2
|
||
|
|
notifier.notify_balance.assert_called_once()
|
||
|
|
|
||
|
|
def test_new_balance_after_payment_with_notification(self, mocker):
|
||
|
|
"""Old balance is 0, new balance is 1 (old > new), should send notification"""
|
||
|
|
notifier = mocker.Mock()
|
||
|
|
notifier.notify_balance = mocker.Mock()
|
||
|
|
handler = FlexpoolHandler(notifier=notifier)
|
||
|
|
miner = mocker.patch('flexpoolapi.miner')
|
||
|
|
miner().balance.return_value = 0
|
||
|
|
mocker.patch('companion.pools.flexpool.FlexpoolHandler._watch_miner_payments')
|
||
|
|
mocker.patch('companion.pools.flexpool.Miner.get_payements')
|
||
|
|
last_balance, last_transaction = handler.watch_miner(address='addr', last_balance=1)
|
||
|
|
assert last_balance == 0
|
||
|
|
notifier.notify_balance.assert_called_once()
|
||
|
|
|
||
|
|
def test_very_new_balance_with_notification(self, mocker):
|
||
|
|
"""Old balance doesn't exist, new balance is 1, should send notification"""
|
||
|
|
notifier = mocker.Mock()
|
||
|
|
notifier.notify_balance = mocker.Mock()
|
||
|
|
handler = FlexpoolHandler(notifier=notifier)
|
||
|
|
miner = mocker.patch('flexpoolapi.miner')
|
||
|
|
miner().balance.return_value = 1
|
||
|
|
mocker.patch('companion.pools.flexpool.FlexpoolHandler._watch_miner_payments')
|
||
|
|
mocker.patch('companion.pools.flexpool.Miner.get_payements')
|
||
|
|
last_balance, last_transaction = handler.watch_miner(address='addr')
|
||
|
|
assert last_balance == 1
|
||
|
|
notifier.notify_balance.assert_called_once()
|
||
|
|
|
||
|
|
def test_same_balance_without_notification(self, mocker):
|
||
|
|
"""Old balance and new balance are the same, should not send notification"""
|
||
|
|
notifier = mocker.Mock()
|
||
|
|
notifier.notify_balance = mocker.Mock()
|
||
|
|
handler = FlexpoolHandler(notifier=notifier)
|
||
|
|
miner = mocker.patch('flexpoolapi.miner')
|
||
|
|
miner().balance.return_value = 1
|
||
|
|
mocker.patch('companion.pools.flexpool.FlexpoolHandler._watch_miner_payments')
|
||
|
|
mocker.patch('companion.pools.flexpool.Miner.get_payements')
|
||
|
|
last_balance, last_transaction = handler.watch_miner(address='addr', last_balance=1)
|
||
|
|
assert last_balance == 1
|
||
|
|
notifier.notify_balance.assert_not_called()
|
||
|
|
|
||
|
|
def test_new_payment_with_notification(self, mocker):
|
||
|
|
"""One transaction saved (trx1), two transactions detected (trx1, trx2), should send notification"""
|
||
|
|
notifier = mocker.Mock()
|
||
|
|
notifier.notify_payment = mocker.Mock()
|
||
|
|
handler = FlexpoolHandler(notifier=notifier)
|
||
|
|
mocker.patch('flexpoolapi.miner')
|
||
|
|
mocker.patch('companion.pools.flexpool.FlexpoolHandler._watch_miner_balance')
|
||
|
|
get_payements = mocker.patch('companion.pools.flexpool.Miner.get_payements')
|
||
|
|
get_payements.return_value = [
|
||
|
|
Transaction(txid='trx1', amount=1, time=datetime.now(), duration=timedelta(minutes=1)),
|
||
|
|
Transaction(txid='trx2', amount=1, time=datetime.now(), duration=timedelta(minutes=1))
|
||
|
|
]
|
||
|
|
last_balance, last_transaction = handler.watch_miner(address='addr', last_transaction='trx1')
|
||
|
|
assert last_transaction == 'trx2'
|
||
|
|
notifier.notify_payment.assert_called_once()
|
||
|
|
|
||
|
|
def test_very_new_payment_with_notification(self, mocker):
|
||
|
|
"""No transaction saved, one transaction detected (trx1), should send notification"""
|
||
|
|
notifier = mocker.Mock()
|
||
|
|
notifier.notify_payment = mocker.Mock()
|
||
|
|
handler = FlexpoolHandler(notifier=notifier)
|
||
|
|
mocker.patch('flexpoolapi.miner')
|
||
|
|
mocker.patch('companion.pools.flexpool.FlexpoolHandler._watch_miner_balance')
|
||
|
|
get_payements = mocker.patch('companion.pools.flexpool.Miner.get_payements')
|
||
|
|
get_payements.return_value = [
|
||
|
|
Transaction(txid='trx1', amount=1, time=datetime.now(), duration=timedelta(minutes=1))
|
||
|
|
]
|
||
|
|
last_balance, last_transaction = handler.watch_miner(address='addr', last_transaction=None)
|
||
|
|
assert last_transaction == 'trx1'
|
||
|
|
notifier.notify_payment.assert_called_once()
|
||
|
|
|
||
|
|
def test_same_payment_without_notification(self, mocker):
|
||
|
|
"""One transaction saved (trx1), one transaction detected (trx1), should not send notification"""
|
||
|
|
notifier = mocker.Mock()
|
||
|
|
notifier.notify_payment = mocker.Mock()
|
||
|
|
handler = FlexpoolHandler(notifier=notifier)
|
||
|
|
mocker.patch('flexpoolapi.miner')
|
||
|
|
mocker.patch('companion.pools.flexpool.FlexpoolHandler._watch_miner_balance')
|
||
|
|
get_payements = mocker.patch('companion.pools.flexpool.Miner.get_payements')
|
||
|
|
get_payements.return_value = [
|
||
|
|
Transaction(txid='trx1', amount=1, time=datetime.now(), duration=timedelta(minutes=1))
|
||
|
|
]
|
||
|
|
last_balance, last_transaction = handler.watch_miner(address='addr', last_transaction='trx1')
|
||
|
|
assert last_transaction == 'trx1'
|
||
|
|
notifier.notify_payment.assert_not_called()
|
||
|
|
|
||
|
|
def test_zero_payment_without_notification(self, mocker):
|
||
|
|
"""Zero transaction saved, zero transaction detected, should not send notification"""
|
||
|
|
notifier = mocker.Mock()
|
||
|
|
notifier.notify_payment = mocker.Mock()
|
||
|
|
handler = FlexpoolHandler(notifier=notifier)
|
||
|
|
mocker.patch('flexpoolapi.miner')
|
||
|
|
mocker.patch('companion.pools.flexpool.FlexpoolHandler._watch_miner_balance')
|
||
|
|
get_payements = mocker.patch('companion.pools.flexpool.Miner.get_payements')
|
||
|
|
get_payements.return_value = []
|
||
|
|
last_balance, last_transaction = handler.watch_miner(address='addr', last_transaction=None)
|
||
|
|
assert last_transaction is None
|
||
|
|
notifier.notify_payment.assert_not_called()
|