From d9794276629c8ce3312306fdd61e3ee4997edc6a Mon Sep 17 00:00:00 2001 From: smypmsa Date: Tue, 6 May 2025 09:50:21 +0000 Subject: [PATCH] fix: geyser auth type in tests --- tests/compare_listeners.py | 3 ++- tests/test_geyser_listener.py | 5 +++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/compare_listeners.py b/tests/compare_listeners.py index 2c6b0bd..50e51d6 100644 --- a/tests/compare_listeners.py +++ b/tests/compare_listeners.py @@ -74,6 +74,7 @@ async def run_comparison(test_duration: int = 300): wss_endpoint = os.environ.get("SOLANA_NODE_WSS_ENDPOINT") geyser_endpoint = os.environ.get("GEYSER_ENDPOINT") geyser_api_token = os.environ.get("GEYSER_API_TOKEN") + geyser_auth_type = os.environ.get("GEYSER_AUTH_TYPE", "x-token") if not wss_endpoint: 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 if geyser_endpoint and geyser_api_token: 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") listener_tasks.append( diff --git a/tests/test_geyser_listener.py b/tests/test_geyser_listener.py index 7c9e030..43781d5 100644 --- a/tests/test_geyser_listener.py +++ b/tests/test_geyser_listener.py @@ -53,6 +53,7 @@ async def test_geyser_listener( """Test the Geyser listener functionality""" geyser_endpoint = os.environ.get("GEYSER_ENDPOINT") geyser_api_token = os.environ.get("GEYSER_API_TOKEN") + geyser_auth_type = os.environ.get("GEYSER_AUTH_TYPE", "x-token") if not geyser_endpoint: logger.error("GEYSER_ENDPOINT environment variable is not set") @@ -61,9 +62,9 @@ async def test_geyser_listener( if not geyser_api_token: logger.error("GEYSER_API_TOKEN environment variable is not set") return [] - + 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() if match_string: