chore: initial commit

scripts/send_test_requests.py

@@ -0,0 +1,72 @@
+#!/usr/bin/env python3
+"""
+Quick smoke sender for the temp print server.
+Requires python -m pip install grpcio grpcio-tools.
+"""
+import subprocess
+import sys
+import tempfile
+from pathlib import Path
+import grpc
+
+
+def generate_stubs(proto_root: Path, out_dir: Path):
+    protos = [
+        "v1/primitives/bool.proto",
+        "v1/primitives/int.proto",
+        "v1/primitives/string.proto",
+        "v1/collections/list.proto",
+        "v1/collections/common.proto",
+        "v1/collections/element_kind.proto",
+    ]
+    args = [
+        sys.executable,
+        "-m",
+        "grpc_tools.protoc",
+        f"-I{proto_root}",
+        f"--python_out={out_dir}",
+        f"--grpc_python_out={out_dir}",
+    ] + protos
+    subprocess.check_call(args, cwd=proto_root)
+
+
+def main():
+    repo_root = Path(__file__).resolve().parent.parent
+    proto_root = repo_root / "proto"
+    with tempfile.TemporaryDirectory() as td:
+        out_dir = Path(td)
+        generate_stubs(proto_root, out_dir)
+        sys.path.insert(0, str(out_dir))
+
+        from v1.primitives import bool_pb2_grpc as bool_grpc, bool_pb2
+        from v1.primitives import int_pb2_grpc as int_grpc, int_pb2
+        from v1.primitives import string_pb2_grpc as str_grpc, string_pb2
+        from v1.collections import list_pb2_grpc as list_grpc, list_pb2
+
+        channel = grpc.insecure_channel("localhost:50051")
+
+        bool_stub = bool_grpc.BoolCacheServiceStub(channel)
+        int_stub = int_grpc.IntCacheServiceStub(channel)
+        str_stub = str_grpc.StringCacheServiceStub(channel)
+        list_stub = list_grpc.ListCacheServiceStub(channel)
+
+        print("Sending Bool Set")
+        bool_stub.Set(bool_pb2.BoolSetRequest(id="b1", value=True))
+
+        print("Sending Int Set/Get/Delete")
+        int_stub.Set(int_pb2.IntSetRequest(id="i1", value=42))
+        int_resp = int_stub.Get(int_pb2.IntGetRequest(id="i1"))
+        print("Int get response:", int_resp.value)
+        int_stub.Delete(int_pb2.IntDeleteRequest(id="i1"))
+
+        print("Sending String Set")
+        str_stub.Set(string_pb2.StringSetRequest(id="s1", value="hello"))
+
+        print("Sending List Set")
+        list_stub.Set(list_pb2.ListSetRequest(id="l1", elements=[]))
+
+        print("Done. Check server stdout for printed messages.")
+
+
+if __name__ == "__main__":
+    main()