async def setUp(self):
await 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: b'a' * 64
fake_dht = MockObject()
fake_dht.connect_peer = lambda *_: succeed([fake_response_peer])
self.payout_manager = PayoutManager(fake_tc, fake_dht)
async def test_update_payout_balance(self):
"""
Test whether the balance of peers is correctly updated
"""
fake_download, dl_state = create_fake_download_and_state()
dl_state.get_status = lambda: DLSTATUS_DOWNLOADING
self.session.dlmgr = DownloadManager(self.session)
fake_tc = MockObject()
fake_tc.add_listener = lambda *_: None
self.session.payout_manager = PayoutManager(fake_tc, None)
self.session.dlmgr.initialize()
self.session.dlmgr.state_cb_count = 4
self.session.dlmgr.downloads = {b'aaaa': fake_download}
await self.session.dlmgr.sesscb_states_callback([dl_state])
self.assertTrue(self.session.payout_manager.tribler_peers)
