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()
