def setUp(self):
yield super(TestPayoutManager, self).setUp()
fake_tc = MockObject()
fake_tc.add_listener = lambda *_: None
fake_response_peer = MockObject()
fake_response_peer.public_key = MockObject()
fake_response_peer.public_key.key_to_bin = lambda: 'a' * 64
fake_dht = MockObject()
fake_dht.connect_peer = lambda *_: succeed([fake_response_peer])
self.payout_manager = PayoutManager(fake_tc, fake_dht)
def test_update_payout_balance(self):
"""
Test whether the balance of peers is correctly updated
"""
fake_download, dl_state = TestLaunchManyCore.create_fake_download_and_state()
dl_state.get_status = lambda: DLSTATUS_DOWNLOADING
fake_tc = MockObject()
fake_tc.add_listener = lambda *_: None
self.lm.payout_manager = PayoutManager(fake_tc, None)
self.lm.state_cb_count = 4
self.lm.downloads = {'aaaa': fake_download}
self.lm.sesscb_states_callback([dl_state])
self.assertTrue(self.lm.payout_manager.tribler_peers)
def test_update_payout_balance(self):
"""
Test whether the balance of peers is correctly updated
"""
fake_download, dl_state = TestLaunchManyCore.create_fake_download_and_state(
)
dl_state.get_status = lambda: DLSTATUS_DOWNLOADING
fake_tc = MockObject()
fake_tc.add_listener = lambda *_: None
self.lm.payout_manager = PayoutManager(fake_tc, None)
self.lm.state_cb_count = 4
self.lm.downloads = {'aaaa': fake_download}
self.lm.sesscb_states_callback([dl_state])
self.assertTrue(self.lm.payout_manager.tribler_peers)
