Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

tests/plugins/test_google_meet_node.py

@@ -0,0 +1,187 @@
+"""Tests for the google_meet node primitive.
+
+Covers protocol helpers, the file-backed registry, the server's
+token-and-dispatch machinery, a mocked client, and the CLI plumbing.
+We never open a real socket — websockets.serve / websockets.sync.client
+are fully mocked.
+"""
+
+from __future__ import annotations
+
+import argparse
+import asyncio
+import json
+from pathlib import Path
+
+import pytest
+
+
+@pytest.fixture(autouse=True)
+def _isolate_home(tmp_path, monkeypatch):
+    hermes_home = tmp_path / ".hermes"
+    hermes_home.mkdir()
+    monkeypatch.setenv("HERMES_HOME", str(hermes_home))
+    yield hermes_home
+
+
+# ---------------------------------------------------------------------------
+# protocol.py
+# ---------------------------------------------------------------------------
+
+def test_protocol_encode_decode_roundtrip():
+    from plugins.google_meet.node import protocol
+
+    msg = protocol.make_request("ping", "tok", {"x": 1}, req_id="abc")
+    raw = protocol.encode(msg)
+    out = protocol.decode(raw)
+    assert out == msg
+    assert out["type"] == "ping"
+    assert out["id"] == "abc"
+    assert out["token"] == "tok"
+    assert out["payload"] == {"x": 1}
+
+
+# ---------------------------------------------------------------------------
+# registry.py
+# ---------------------------------------------------------------------------
+
+def test_registry_add_get_roundtrip_persists(tmp_path):
+    from plugins.google_meet.node.registry import NodeRegistry
+
+    p = tmp_path / "nodes.json"
+    r = NodeRegistry(path=p)
+    r.add("mac", "ws://mac.local:18789", "deadbeef")
+
+    # Second instance sees it.
+    r2 = NodeRegistry(path=p)
+    entry = r2.get("mac")
+    assert entry is not None
+    assert entry["name"] == "mac"
+    assert entry["url"] == "ws://mac.local:18789"
+    assert entry["token"] == "deadbeef"
+    assert "added_at" in entry
+
+
+# ---------------------------------------------------------------------------
+# server.py — token + dispatch
+# ---------------------------------------------------------------------------
+
+def test_server_ensure_token_generates_and_persists(tmp_path):
+    from plugins.google_meet.node.server import NodeServer
+
+    p = tmp_path / "tok.json"
+    s1 = NodeServer(token_path=p)
+    t1 = s1.ensure_token()
+    assert isinstance(t1, str) and len(t1) == 32
+
+    # Reuse on a fresh instance.
+    s2 = NodeServer(token_path=p)
+    t2 = s2.ensure_token()
+    assert t1 == t2
+
+    data = json.loads(p.read_text(encoding="utf-8"))
+    assert data["token"] == t1
+    assert "generated_at" in data
+
+
+def _run(coro):
+    return asyncio.new_event_loop().run_until_complete(coro) if False else asyncio.run(coro)
+
+
+# ---------------------------------------------------------------------------
+# client.py
+# ---------------------------------------------------------------------------
+
+class _FakeWS:
+    """Minimal context-manager stand-in for websockets.sync.client.connect."""
+
+    def __init__(self, reply_builder):
+        self._reply_builder = reply_builder
+        self.sent = []
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc, tb):
+        return False
+
+    def send(self, raw):
+        self.sent.append(raw)
+
+    def recv(self, timeout=None):
+        return self._reply_builder(self.sent[-1])
+
+
+def _install_fake_ws(monkeypatch, reply_builder):
+    fake_ws_holder = {}
+
+    def _connect(url, **kwargs):
+        ws = _FakeWS(reply_builder)
+        fake_ws_holder["ws"] = ws
+        fake_ws_holder["url"] = url
+        fake_ws_holder["kwargs"] = kwargs
+        return ws
+
+    # Patch the concrete import site inside client._rpc
+    import websockets.sync.client as wsc  # type: ignore
+    monkeypatch.setattr(wsc, "connect", _connect)
+    return fake_ws_holder
+
+
+def test_client_rpc_sends_correct_envelope_and_parses_response(monkeypatch):
+    from plugins.google_meet.node.client import NodeClient
+    from plugins.google_meet.node import protocol
+
+    def reply(raw_out):
+        req = protocol.decode(raw_out)
+        return protocol.encode(protocol.make_response(req["id"], {"ok": True, "echo": req["type"]}))
+
+    holder = _install_fake_ws(monkeypatch, reply)
+
+    c = NodeClient("ws://remote:1", "tok123")
+    out = c._rpc("ping", {"hello": 1})
+    assert out == {"ok": True, "echo": "ping"}
+
+    sent = json.loads(holder["ws"].sent[0])
+    assert sent["type"] == "ping"
+    assert sent["token"] == "tok123"
+    assert sent["payload"] == {"hello": 1}
+    assert sent["id"]  # non-empty
+    assert holder["url"] == "ws://remote:1"
+
+
+# ---------------------------------------------------------------------------
+# cli.py
+# ---------------------------------------------------------------------------
+
+def _build_parser():
+    from plugins.google_meet.node.cli import register_cli
+
+    parser = argparse.ArgumentParser(prog="meet-node-test")
+    register_cli(parser)
+    return parser
+
+
+def test_cli_approve_list_remove(capsys):
+    from plugins.google_meet.node.registry import NodeRegistry
+
+    p = _build_parser()
+
+    args = p.parse_args(["approve", "mac", "ws://mac:1", "tok"])
+    rc = args.func(args)
+    assert rc == 0
+    assert NodeRegistry().get("mac") is not None
+
+    args = p.parse_args(["list"])
+    rc = args.func(args)
+    assert rc == 0
+    out = capsys.readouterr().out
+    assert "mac" in out
+    assert "ws://mac:1" in out
+
+    args = p.parse_args(["remove", "mac"])
+    rc = args.func(args)
+    assert rc == 0
+    assert NodeRegistry().get("mac") is None
+
+