| import asyncio |
| import websockets |
| import json |
| import requests |
| import uuid |
| import time |
| import urllib.parse |
| from identity import MEPIdentity |
|
|
| HUB_URL = "http://localhost:8000" |
|
|
| async def test(): |
| provider = MEPIdentity(f"test_provider_{uuid.uuid4().hex[:6]}.pem") |
| consumer = MEPIdentity(f"test_consumer_{uuid.uuid4().hex[:6]}.pem") |
| requests.post(f'{HUB_URL}/register', json={'pubkey': provider.pub_pem}) |
| requests.post(f'{HUB_URL}/register', json={'pubkey': consumer.pub_pem}) |
| ts = str(int(time.time())) |
| sig = provider.sign(provider.node_id, ts) |
| sig_safe = urllib.parse.quote(sig) |
| async with websockets.connect(f'ws://localhost:8000/ws/{provider.node_id}?timestamp={ts}&signature={sig_safe}') as ws: |
| submit_payload = json.dumps({'consumer_id': consumer.node_id, 'payload': 'Test payload', 'bounty': 1.0}) |
| submit_headers = consumer.get_auth_headers(submit_payload) |
| submit_headers["Content-Type"] = "application/json" |
| requests.post(f'{HUB_URL}/tasks/submit', data=submit_payload, headers=submit_headers) |
| |
| msg = await asyncio.wait_for(ws.recv(), timeout=2.0) |
| data = json.loads(msg) |
| print('Received:', data) |
| |
| if data['event'] == 'rfc': |
| task_id = data['data']['id'] |
| bid_payload = json.dumps({'task_id': task_id, 'provider_id': provider.node_id}) |
| bid_headers = provider.get_auth_headers(bid_payload) |
| bid_headers["Content-Type"] = "application/json" |
| resp = requests.post(f'{HUB_URL}/tasks/bid', data=bid_payload, headers=bid_headers) |
| print('Bid response:', resp.json()) |
| |
| complete_payload = json.dumps({ |
| 'task_id': task_id, |
| 'provider_id': provider.node_id, |
| 'result_payload': 'Done!' |
| }) |
| complete_headers = provider.get_auth_headers(complete_payload) |
| complete_headers["Content-Type"] = "application/json" |
| complete_resp = requests.post(f'{HUB_URL}/tasks/complete', data=complete_payload, headers=complete_headers) |
| print('Complete response:', complete_resp.json()) |
|
|
| if __name__ == '__main__': |
| asyncio.run(test()) |
|
|
