forked from ChipaDevTeam/PocketOptionAPI
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_fixed_connection.py
More file actions
157 lines (125 loc) · 5 KB
/
test_fixed_connection.py
File metadata and controls
157 lines (125 loc) · 5 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
"""
Test script to verify the fixed connection issue in the new async API
"""
import asyncio
import sys
from loguru import logger
from pocketoptionapi_async import AsyncPocketOptionClient
# Configure logging
logger.remove()
logger.add(
sys.stdout,
format="<green>{time:HH:mm:ss}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>",
)
async def test_connection_fix():
"""Test the fixed connection with proper handshake sequence"""
print("Testing Fixed Connection Issue")
print("=" * 60)
# Test with complete SSID format (like from browser)
complete_ssid = r'42["auth",{"session":"test_session_12345","isDemo":1,"uid":12345,"platform":1,"isFastHistory":true}]'
print("📝 Using complete SSID format:")
print(f" {complete_ssid[:50]}...")
print()
try:
# Create client
client = AsyncPocketOptionClient(
ssid=complete_ssid,
is_demo=True,
persistent_connection=False, # Use regular connection for testing
auto_reconnect=True,
)
print(" Client created successfully")
print(f"🔍 Session ID: {client.session_id}")
print(f"👤 UID: {client.uid}")
print(f"Demo mode: {client.is_demo}")
print(f"🏷️ Platform: {client.platform}")
print()
# Test connection
print("Testing connection with improved handshake...")
try:
success = await client.connect()
if success:
print(" CONNECTION SUCCESSFUL!")
print(f"📊 Connection info: {client.connection_info}")
print(
f"Connected to: {client.connection_info.region if client.connection_info else 'Unknown'}"
)
# Test basic functionality
print("\n📋 Testing basic functionality...")
try:
balance = await client.get_balance()
if balance:
print(f"Balance: ${balance.balance}")
else:
print(" No balance data received (expected with test SSID)")
except Exception as e:
print(f" Balance request failed (expected): {e}")
print("\n All connection tests passed!")
else:
print("Connection failed")
except Exception as e:
# This is expected with test SSID, but we should see proper handshake messages
print(f" Connection attempt result: {str(e)[:100]}...")
if "handshake" in str(e).lower() or "authentication" in str(e).lower():
print(
" Handshake sequence is working (authentication failed as expected with test SSID)"
)
else:
print("Unexpected connection error")
finally:
await client.disconnect()
print("🛑 Disconnected")
except Exception as e:
print(f"Test error: {e}")
return False
return True
async def test_old_vs_new_comparison():
"""Compare the handshake behavior with old API patterns"""
print("\n" + "=" * 60)
print("Connection Pattern Comparison")
print("=" * 60)
print("📋 OLD API Handshake Pattern:")
print(' 1. Server sends: 0{"sid":"..."}')
print(" 2. Client sends: 40")
print(' 3. Server sends: 40{"sid":"..."}')
print(" 4. Client sends: SSID message")
print(' 5. Server sends: 451-["successauth",...]')
print()
print("📋 NEW API Handshake Pattern (FIXED):")
print(" 1. Wait for server message with '0' and 'sid'")
print(" 2. Send '40' response")
print(" 3. Wait for server message with '40' and 'sid'")
print(" 4. Send SSID authentication")
print(" 5. Wait for authentication response")
print()
print("Key Fixes Applied:")
print(" Proper message sequence waiting (like old API)")
print(" Handshake completion before background tasks")
print(" Authentication event handling")
print(" Timeout handling for server responses")
print()
async def main():
"""Main test function"""
print("Testing Fixed Async API Connection")
print("Goal: Verify connection works like old API")
print()
# Test the fixed connection
success = await test_connection_fix()
# Show comparison
await test_old_vs_new_comparison()
print("=" * 60)
if success:
print(" CONNECTION FIX VERIFICATION COMPLETE")
print(
"📝 The new async API now follows the same handshake pattern as the old API"
)
print("Key improvements:")
print(" • Proper server response waiting")
print(" • Sequential handshake messages")
print(" • Authentication event handling")
print(" • Error handling with timeouts")
else:
print("CONNECTION FIX NEEDS MORE WORK")
print("=" * 60)
if __name__ == "__main__":
asyncio.run(main())