File size: 1,392 Bytes
af26c53 a25490a af26c53 | 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 | import asyncio
import requests
import json
import websockets
from identity import MEPIdentity
HUB_URL = "http://localhost:8000"
async def test():
# 1. Generate identity automatically
bot = MEPIdentity("test_bot.pem")
print(f"✅ Generated Identity: {bot.node_id}")
# 2. Register
resp = requests.post(f"{HUB_URL}/register", json={"pubkey": bot.pub_pem})
print("Registration:", resp.json())
# 3. Connect WS with query params
import time
ts = str(int(time.time()))
sig = bot.sign(bot.node_id, ts)
# Add url encoding for signature to be safe in URL
import urllib.parse
sig_safe = urllib.parse.quote(sig)
ws_url = f"ws://localhost:8000/ws/{bot.node_id}?timestamp={ts}&signature={sig_safe}"
try:
async with websockets.connect(ws_url):
print("✅ WebSocket Authenticated!")
# 4. Submit Task
payload_dict = {"consumer_id": bot.node_id, "payload": "Test secure task", "bounty": 1.0}
payload_str = json.dumps(payload_dict)
headers = bot.get_auth_headers(payload_str)
r = requests.post(f"{HUB_URL}/tasks/submit", data=payload_str, headers=headers)
print("Submit Task:", r.json())
except Exception as e:
print("❌ Error:", e)
if __name__ == "__main__":
asyncio.run(test())
|