File size: 3,327 Bytes
7bec764 a25490a 7bec764 a25490a 7bec764 a25490a 7bec764 a25490a 7bec764 a25490a 7bec764 a25490a 7bec764 a25490a 7bec764 a25490a 7bec764 a25490a 7bec764 a25490a 7bec764 a25490a 7bec764 a25490a 7bec764 a25490a 7bec764 a25490a | 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 | import asyncio
import websockets
import json
import requests
import uuid
import time
import urllib.parse
from identity import MEPIdentity
HUB_URL = "http://localhost:8000"
WS_URL = "ws://localhost:8000"
async def test_direct_message():
print("=== Testing MEP Direct Messaging (Zero Bounty) ===")
# 1. Start Alice (Provider)
alice = MEPIdentity(f"alice_{uuid.uuid4().hex[:6]}.pem")
# 2. Start Bob (Consumer)
bob = MEPIdentity(f"bob_{uuid.uuid4().hex[:6]}.pem")
requests.post(f"{HUB_URL}/register", json={"pubkey": alice.pub_pem})
requests.post(f"{HUB_URL}/register", json={"pubkey": bob.pub_pem})
print(f"✅ Registered Alice ({alice.node_id}) and Bob ({bob.node_id})")
async def alice_listen():
ts = str(int(time.time()))
sig = alice.sign(alice.node_id, ts)
sig_safe = urllib.parse.quote(sig)
async with websockets.connect(f"{WS_URL}/ws/{alice.node_id}?timestamp={ts}&signature={sig_safe}") as ws:
print("👧 Alice: Online and listening...")
msg = await asyncio.wait_for(ws.recv(), timeout=5)
data = json.loads(msg)
print("👧 Alice: Received DIRECT MESSAGE!")
print(f"👧 Alice: Payload: {data['data']['payload']}")
print(f"👧 Alice: Bounty: {data['data']['bounty']} SECONDS")
# Alice replies for free
payload_str = json.dumps({
"task_id": data['data']['id'],
"provider_id": alice.node_id,
"result_payload": "Yes Bob, I am available for a meeting tomorrow at 2 PM. Free of charge! 🐱"
})
headers = alice.get_auth_headers(payload_str)
headers["Content-Type"] = "application/json"
requests.post(f"{HUB_URL}/tasks/complete", data=payload_str, headers=headers)
print("👧 Alice: Sent reply!")
async def bob_listen():
ts = str(int(time.time()))
sig = bob.sign(bob.node_id, ts)
sig_safe = urllib.parse.quote(sig)
async with websockets.connect(f"{WS_URL}/ws/{bob.node_id}?timestamp={ts}&signature={sig_safe}") as ws:
# Bob submits a direct task to Alice with 0 bounty
await asyncio.sleep(1) # Let Alice connect first
print("👦 Bob: Sending Direct Message to Alice (0.0 SECONDS)...")
payload_str = json.dumps({
"consumer_id": bob.node_id,
"payload": "Hey Alice, are you free for a meeting tomorrow at 2 PM?",
"bounty": 0.0,
"target_node": alice.node_id
})
headers = bob.get_auth_headers(payload_str)
headers["Content-Type"] = "application/json"
requests.post(f"{HUB_URL}/tasks/submit", data=payload_str, headers=headers)
# Bob waits for Alice's reply
msg = await asyncio.wait_for(ws.recv(), timeout=5)
data = json.loads(msg)
print(f"👦 Bob: Received reply from {data['data']['provider_id']}:")
print(f"👦 Bob: \"{data['data']['result_payload']}\"")
await asyncio.gather(alice_listen(), bob_listen())
print("=== Direct Messaging Test Complete! ===")
if __name__ == "__main__":
asyncio.run(test_direct_message())
|