gavindiaz/pumpfun-bonkfun-bot_github
1
0
Fork 0
mirror of https://github.com/chainstacklabs/pumpfun-bonkfun-bot.git synced 2026-08-17 01:08:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: geyser auth type in tests

Browse Source
This commit is contained in:
smypmsa
2025-05-06 09:50:21 +00:00
parent fe17df2670
commit d979427662
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/compare_listeners.py
+2 -1
View File
@@ -74,6 +74,7 @@ async def run_comparison(test_duration: int = 300):
wss_endpoint = os.environ.get("SOLANA_NODE_WSS_ENDPOINT") wss_endpoint = os.environ.get("SOLANA_NODE_WSS_ENDPOINT")
geyser_endpoint = os.environ.get("GEYSER_ENDPOINT") geyser_endpoint = os.environ.get("GEYSER_ENDPOINT")
geyser_api_token = os.environ.get("GEYSER_API_TOKEN") geyser_api_token = os.environ.get("GEYSER_API_TOKEN")
geyser_auth_type = os.environ.get("GEYSER_AUTH_TYPE", "x-token")
if not wss_endpoint: if not wss_endpoint:
logger.error("SOLANA_NODE_WSS_ENDPOINT environment variable is not set") logger.error("SOLANA_NODE_WSS_ENDPOINT environment variable is not set")
@@ -98,7 +99,7 @@ async def run_comparison(test_duration: int = 300):
# Initialize Geyser listener if credentials are available # Initialize Geyser listener if credentials are available
if geyser_endpoint and geyser_api_token: if geyser_endpoint and geyser_api_token:
logger.info(f"Connecting to Geyser API: {geyser_endpoint}") logger.info(f"Connecting to Geyser API: {geyser_endpoint}")
geyser_listener = GeyserListener(geyser_endpoint, geyser_api_token, PumpAddresses.PROGRAM) geyser_listener = GeyserListener(geyser_endpoint, geyser_api_token, geyser_auth_type, PumpAddresses.PROGRAM)
geyser_callback = TimingTokenCallback("GeyserListener") geyser_callback = TimingTokenCallback("GeyserListener")
listener_tasks.append( listener_tasks.append(
tests/test_geyser_listener.py
+3 -2
View File
@@ -53,6 +53,7 @@ async def test_geyser_listener(
"""Test the Geyser listener functionality""" """Test the Geyser listener functionality"""
geyser_endpoint = os.environ.get("GEYSER_ENDPOINT") geyser_endpoint = os.environ.get("GEYSER_ENDPOINT")
geyser_api_token = os.environ.get("GEYSER_API_TOKEN") geyser_api_token = os.environ.get("GEYSER_API_TOKEN")
geyser_auth_type = os.environ.get("GEYSER_AUTH_TYPE", "x-token")
if not geyser_endpoint: if not geyser_endpoint:
logger.error("GEYSER_ENDPOINT environment variable is not set") logger.error("GEYSER_ENDPOINT environment variable is not set")
@@ -61,9 +62,9 @@ async def test_geyser_listener(
if not geyser_api_token: if not geyser_api_token:
logger.error("GEYSER_API_TOKEN environment variable is not set") logger.error("GEYSER_API_TOKEN environment variable is not set")
return [] return []
logger.info(f"Connecting to Geyser API: {geyser_endpoint}") logger.info(f"Connecting to Geyser API: {geyser_endpoint}")
listener = GeyserListener(geyser_endpoint, geyser_api_token, PumpAddresses.PROGRAM) listener = GeyserListener(geyser_endpoint, geyser_api_token, geyser_auth_type, PumpAddresses.PROGRAM)
callback = TestTokenCallback() callback = TestTokenCallback()
if match_string: if match_string: