def tests_init_with_stratum(self): manager = HathorManager(self.clock, tx_storage=self.tx_storage, stratum_port=50505) manager.start() manager.stop() del manager
def test_manager_connections(self): tx_storage = TransactionMemoryStorage() tmpdir = tempfile.mkdtemp() wallet = Wallet(directory=tmpdir) wallet.unlock(b'teste') manager = HathorManager(self.clock, tx_storage=tx_storage, wallet=wallet) endpoint = 'tcp://127.0.0.1:8005' manager.connections.connect_to(endpoint, use_ssl=True) self.assertFalse(endpoint in manager.connections.connecting_peers) self.assertFalse(endpoint in manager.connections.handshaking_peers) self.assertFalse(endpoint in manager.connections.connected_peers) manager.stop() manager.stop() shutil.rmtree(tmpdir)
def test_wrong_stop(self): manager = HathorManager(self.clock, tx_storage=self.tx_storage) with self.assertRaises(Exception): manager.stop() manager.start() manager.stop() with self.assertRaises(Exception): manager.stop()
class _ResourceTest(unittest.TestCase): def _manager_kwargs(self): peer_id = PeerId() network = 'testnet' wallet = self._create_test_wallet() tx_storage = getattr(self, 'tx_storage', None) if tx_storage is None: if self.use_memory_storage: from hathor.transaction.storage.memory_storage import TransactionMemoryStorage tx_storage = TransactionMemoryStorage() else: from hathor.transaction.storage.rocksdb_storage import TransactionRocksDBStorage directory = tempfile.mkdtemp() self.tmpdirs.append(directory) tx_storage = TransactionRocksDBStorage(directory) assert ( hasattr(self, '_enable_sync_v1') and hasattr(self, '_enable_sync_v2') and (self._enable_sync_v1 or self._enable_sync_v2) ), ( 'Please set both `_enable_sync_v1` and `_enable_sync_v2` on the class. ' 'Also they can\'t both be False. ' 'This is by design so we don\'t forget to test for multiple sync versions.' ) return dict( peer_id=peer_id, network=network, wallet=wallet, tx_storage=tx_storage, wallet_index=True, enable_sync_v1=self._enable_sync_v1, enable_sync_v2=self._enable_sync_v2, ) def setUp(self): super().setUp() self.reactor = self.clock self.manager = HathorManager(self.reactor, **self._manager_kwargs()) self.manager.allow_mining_without_peers() _set_test_mode(TestMode.TEST_ALL_WEIGHT) self.manager.start() def tearDown(self): return self.manager.stop()
class _ResourceTest(unittest.TestCase): def _manager_kwargs(self): peer_id = PeerId() network = 'testnet' wallet = self._create_test_wallet() tx_storage = getattr(self, 'tx_storage', None) return dict( peer_id=peer_id, network=network, wallet=wallet, tx_storage=tx_storage, wallet_index=True ) def setUp(self): super().setUp() self.reactor = self.clock self.manager = HathorManager(self.reactor, **self._manager_kwargs()) self.manager.allow_mining_without_peers() self.manager.test_mode = TestMode.TEST_ALL_WEIGHT self.manager.start() def tearDown(self): return self.manager.stop()