class TestBitsplitDaemon(object):
"""
TEST BITSPLIT DAEMON
"""
def __init__(self):
self.daemon = None
def setup(self):
""" Create Daemon instance. """
self.daemon = Daemon()
def test_instantiation(self):
""" Test that it worked at least once. """
assert self.daemon is not None
def test_running(self):
""" Test that it worked at least once. """
self.daemon.tick = Mock()
self.daemon.run(count=1, delay=0)
assert self.daemon.tick.called
def test_tick(self):
""" Test that it worked at least once. """
self.daemon.tick()
def test_handle_incoming(self):
""" Test calls to the incoming distributions. """
old_func = Distribution.find_pending_incoming_funds
good_mock_distro = MagicMock()
good_mock_distro.has_incoming_funds = Mock(return_value=True)
bad_mock_distro = MagicMock()
bad_mock_distro.has_incoming_funds = Mock(return_value=False)
Distribution.find_pending_incoming_funds = Mock(return_value=[
good_mock_distro,
bad_mock_distro,
])
self.daemon.handle_incoming_distributions()
Distribution.find_pending_incoming_funds = old_func
assert good_mock_distro.set_pending_outgoing.called
assert not bad_mock_distro.set_pending_outgoing.called
def test_handle_outgoing(self):
""" Test calls to the outgoing distributions. """
old_func = Distribution.find_pending_outgoing_funds
good_mock_distro = MagicMock()
good_mock_distro.has_sent_all_transactions = Mock(return_value=True)
bad_mock_distro = MagicMock()
bad_mock_distro.has_sent_all_transactions = Mock(return_value=False)
Distribution.find_pending_outgoing_funds = Mock(return_value=[
good_mock_distro,
bad_mock_distro,
])
self.daemon.handle_outgoing_distributions()
Distribution.find_pending_outgoing_funds = old_func
assert good_mock_distro.set_pending_verification.called
assert not good_mock_distro.send_unsent_transactions.called
assert not bad_mock_distro.set_pending_verification.called
assert bad_mock_distro.send_unsent_transactions.called
def test_handle_verification(self):
""" Test calls to the distributions needing verification. """
old_func = Distribution.find_pending_verification
good_mock_distro = MagicMock()
good_mock_distro.is_complete = Mock(return_value=True)
bad_mock_distro = MagicMock()
bad_mock_distro.is_complete = Mock(return_value=False)
Distribution.find_pending_verification = Mock(return_value=[
good_mock_distro,
bad_mock_distro,
])
self.daemon.verify_completed_distributions()
Distribution.find_pending_verification = old_func
assert good_mock_distro.set_verified.called
assert not bad_mock_distro.set_verified.called
def setup(self):
""" Create Daemon instance. """
self.daemon = Daemon()
