fixing format
This commit is contained in:
@@ -1,7 +1,7 @@
|
|||||||
import socket
|
|
||||||
import asyncio
|
import asyncio
|
||||||
import aioice
|
import socket
|
||||||
|
|
||||||
|
import aioice
|
||||||
from bedrock_agentcore import BedrockAgentCoreApp
|
from bedrock_agentcore import BedrockAgentCoreApp
|
||||||
|
|
||||||
app = BedrockAgentCoreApp()
|
app = BedrockAgentCoreApp()
|
||||||
@@ -10,7 +10,7 @@ app = BedrockAgentCoreApp()
|
|||||||
def test_udp():
|
def test_udp():
|
||||||
"""Test UDP connectivity using STUN server"""
|
"""Test UDP connectivity using STUN server"""
|
||||||
stun_server = ("stun.l.google.com", 19302)
|
stun_server = ("stun.l.google.com", 19302)
|
||||||
msg = b'\x00\x01\x00\x00' + b'\x21\x12\xa4\x42' + b'\x00' * 12
|
msg = b"\x00\x01\x00\x00" + b"\x21\x12\xa4\x42" + b"\x00" * 12
|
||||||
|
|
||||||
sock = None
|
sock = None
|
||||||
try:
|
try:
|
||||||
@@ -65,6 +65,7 @@ async def _async_turn_test(turn_server, turn_port, username, password, turn_tran
|
|||||||
finally:
|
finally:
|
||||||
await connection.close()
|
await connection.close()
|
||||||
|
|
||||||
|
|
||||||
def test_turn_with_auth(server, port, username, password, transport):
|
def test_turn_with_auth(server, port, username, password, transport):
|
||||||
"""Sync wrapper for aioice TURN test."""
|
"""Sync wrapper for aioice TURN test."""
|
||||||
return asyncio.run(_async_turn_test(server, port, username, password, transport))
|
return asyncio.run(_async_turn_test(server, port, username, password, transport))
|
||||||
@@ -75,27 +76,38 @@ def comprehensive_network_test():
|
|||||||
results = {}
|
results = {}
|
||||||
|
|
||||||
# Test basic UDP connectivity
|
# Test basic UDP connectivity
|
||||||
results['udp_stun'] = test_udp()
|
results["udp_stun"] = test_udp()
|
||||||
|
|
||||||
# TURN test list
|
# TURN test list
|
||||||
turn_servers = [
|
turn_servers = [
|
||||||
(
|
(
|
||||||
"turn.cloudflare.com", # cleaned
|
"turn.cloudflare.com", # cleaned
|
||||||
3478,
|
3478,
|
||||||
"g001676bf83775dd93ddd77c69c80da1d13027af179ef51b4af7e3567a5028fd",
|
"g0c069ee067cb5585f7a4799950312a355f2cb9c20e23d21bd1ecb1f93c1e8b0",
|
||||||
"a77444990e4fe3b82fd0fbe795f7499409edfbc3b170cea4b28155021ab6623c",
|
"d40a0992b4d08175af51f32f46a3732725a69045422f96b388420c98dd13ced3",
|
||||||
"udp"
|
"udp",
|
||||||
|
),
|
||||||
|
(
|
||||||
|
"turn.cloudflare.com", # cleaned
|
||||||
|
5349,
|
||||||
|
"g0c069ee067cb5585f7a4799950312a355f2cb9c20e23d21bd1ecb1f93c1e8b0",
|
||||||
|
"d40a0992b4d08175af51f32f46a3732725a69045422f96b388420c98dd13ced3",
|
||||||
|
"tcp",
|
||||||
|
),
|
||||||
|
(
|
||||||
|
"turn.cloudflare.com", # cleaned
|
||||||
|
443,
|
||||||
|
"g0c069ee067cb5585f7a4799950312a355f2cb9c20e23d21bd1ecb1f93c1e8b0",
|
||||||
|
"d40a0992b4d08175af51f32f46a3732725a69045422f96b388420c98dd13ced3",
|
||||||
|
"tcp",
|
||||||
),
|
),
|
||||||
]
|
]
|
||||||
|
|
||||||
results['turn_tests'] = []
|
results["turn_tests"] = []
|
||||||
|
|
||||||
for host, port, username, password, transport in turn_servers:
|
for host, port, username, password, transport in turn_servers:
|
||||||
success = test_turn_with_auth(host, port, username, password, transport)
|
success = test_turn_with_auth(host, port, username, password, transport)
|
||||||
results['turn_tests'].append({
|
results["turn_tests"].append({"server": f"{host}:{port}", "success": success})
|
||||||
"server": f"{host}:{port}",
|
|
||||||
"success": success
|
|
||||||
})
|
|
||||||
|
|
||||||
return results
|
return results
|
||||||
|
|
||||||
@@ -124,9 +136,9 @@ def test_tcp_connectivity(host, port):
|
|||||||
def my_agent(payload):
|
def my_agent(payload):
|
||||||
network_results = comprehensive_network_test()
|
network_results = comprehensive_network_test()
|
||||||
|
|
||||||
udp_ok = network_results.get('udp_stun', False)
|
udp_ok = network_results.get("udp_stun", False)
|
||||||
turn_ok = any(t['success'] for t in network_results['turn_tests'])
|
turn_ok = any(t["success"] for t in network_results["turn_tests"])
|
||||||
tcp_ok = network_results.get('tcp_test', False)
|
tcp_ok = network_results.get("tcp_test", False)
|
||||||
|
|
||||||
connectivity_status = []
|
connectivity_status = []
|
||||||
if udp_ok:
|
if udp_ok:
|
||||||
@@ -139,7 +151,9 @@ def my_agent(payload):
|
|||||||
return {
|
return {
|
||||||
"result": f"Hello {payload.get('name', 'World')}!",
|
"result": f"Hello {payload.get('name', 'World')}!",
|
||||||
"network_test_results": network_results,
|
"network_test_results": network_results,
|
||||||
"connectivity_status": ", ".join(connectivity_status) if connectivity_status else "No connectivity",
|
"connectivity_status": ", ".join(connectivity_status)
|
||||||
|
if connectivity_status
|
||||||
|
else "No connectivity",
|
||||||
"webrtc_feasible": udp_ok or turn_ok,
|
"webrtc_feasible": udp_ok or turn_ok,
|
||||||
"turn_available": turn_ok,
|
"turn_available": turn_ok,
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user