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)