fix scribe server version test

lbryio · Jun 9, 2022 · d624ce7 · d624ce7
1 parent ad00dde
commit d624ce7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/integration/blockchain/test_network.py b/tests/integration/blockchain/test_network.py
@@ -3,9 +3,10 @@
 
 from unittest.mock import Mock
 
-from scribe.blockchain.network import LBCRegTest
+from scribe.hub import HUB_PROTOCOL_VERSION
 from scribe.hub.udp import StatusServer
 from scribe.hub.session import LBRYElectrumX
+from scribe.blockchain.network import LBCRegTest
 
 from lbry.wallet.network import Network
 from lbry.wallet.orchstr8 import Conductor
@@ -35,7 +36,7 @@ async def test_server_features(self):
  'payment_address': '',
  'donation_address': '',
  'daily_fee': '0',
- 'server_version': scribe.__version__,
+ 'server_version': HUB_PROTOCOL_VERSION,
  'trending_algorithm': 'fast_ar',
  }, await self.ledger.network.get_server_features())
  # await self.conductor.spv_node.stop()
@@ -65,7 +66,7 @@ async def test_server_features(self):
  'payment_address': payment_address,
  'donation_address': donation_address,
  'daily_fee': '42',
- 'server_version': scribe.__version__,
+ 'server_version': HUB_PROTOCOL_VERSION,
  'trending_algorithm': 'fast_ar',
  }, await self.ledger.network.get_server_features())
 

diff --git a/tests/integration/blockchain/test_wallet_server_sessions.py b/tests/integration/blockchain/test_wallet_server_sessions.py
@@ -1,6 +1,5 @@
 import asyncio
 
-import scribe
 from scribe.hub import HUB_PROTOCOL_VERSION
 from scribe.hub.session import LBRYElectrumX