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)