workpeg 0.3.0__tar.gz → 0.3.1__tar.gz

{workpeg-0.3.0/src/workpeg.egg-info → workpeg-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: workpeg
-Version: 0.3.0
+Version: 0.3.1
 Summary: Workpeg function runtime and SDK
 Author-email: Workpeg <support@workpeg.com>
 License: MIT

{workpeg-0.3.0 → workpeg-0.3.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "workpeg"
-version = "0.3.0"
+version = "0.3.1"
 description = "Workpeg function runtime and SDK"
 readme = "README.md"
 requires-python = ">=3.10"

{workpeg-0.3.0 → workpeg-0.3.1}/src/workpeg/run.py

@@ -12,7 +12,12 @@ def resolve_runtime(with_runtime: str | None, config: dict) -> str:
     return config.get("runtime", {}).get("default") or "cracker"
 
 
-def run_with_docker(path: str = ".", build_first: bool = True, network: str | None = None) -> None:
+def run_with_docker(
+    path: str = ".",
+    build_first: bool = True,
+    network: str | None = None,
+    expose: bool = True,
+) -> None:
     project_path = Path(path).resolve()
     cfg = load_config(project_path / "workpeg.json")
     port = cfg.get("runtime", {}).get("docker", {}).get("port", 8000)
@@ -21,10 +26,10 @@ def run_with_docker(path: str = ".", build_first: bool = True, network: str | No
         cfg.get("build", {}).get("image") or project_path.name
     )
 
-    cmd = [
-        "docker", "run", "--rm",
-        "-p", f"{port}:{port}",
-    ]
+    cmd = ["docker", "run", "--rm"]
+
+    if expose:
+        cmd += ["-p", f"{port}:{port}"]
 
     if network:
         cmd += ["--network", network]

{workpeg-0.3.0 → workpeg-0.3.1/src/workpeg.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: workpeg
-Version: 0.3.0
+Version: 0.3.1
 Summary: Workpeg function runtime and SDK
 Author-email: Workpeg <support@workpeg.com>
 License: MIT

{workpeg-0.3.0 → workpeg-0.3.1}/tests/test_run.py

@@ -155,3 +155,79 @@ def test_run_main_with_docker_and_network(monkeypatch):
         "build_first": True,
         "network": "workpeg_net",
     }
+
+# def run_with_docker(
+#     path: str = ".",
+#     build_first: bool = True,
+#     network: str | None = None,
+#     expose: bool = True,
+# ) -> None:
+#     project_path = Path(path).resolve()
+#     cfg = load_config(project_path / "workpeg.json")
+#     port = cfg.get("runtime", {}).get("docker", {}).get("port", 8000)
+
+#     image = build_image(path) if build_first else (
+#         cfg.get("build", {}).get("image") or project_path.name
+#     )
+
+#     cmd = [
+#         "docker", "run", "--rm",
+#         "-p", f"{port}:{port}" if expose else "",
+#     ]
+
+#     if network:
+#         cmd += ["--network", network]
+
+#     cmd.append(image)
+
+#     subprocess.run(cmd, check=True)
+
+
+def test_run_with_docker_no_expose(monkeypatch, tmp_path):
+    project = tmp_path / "demo"
+    project.mkdir()
+    (project / "workpeg.json").write_text(
+        '{"runtime": {"docker": {"port": 9000}}}'
+    )
+
+    calls = []
+
+    def fake_run(cmd, check):
+        calls.append((cmd, check))
+
+    monkeypatch.setattr(run.subprocess, "run", fake_run)
+
+    run.run_with_docker(path=str(project), build_first=False, expose=False)
+
+    assert len(calls) == 1
+    cmd, check = calls[0]
+    assert cmd == [
+        "docker", "run", "--rm",
+        "demo",
+    ]
+    assert check is True
+
+
+def test_run_with_docker_expose(monkeypatch, tmp_path):
+    project = tmp_path / "demo"
+    project.mkdir()
+    (project / "workpeg.json").write_text(
+        '{"runtime": {"docker": {"port": 9000}}}'
+    )
+
+    calls = []
+
+    def fake_run(cmd, check):
+        calls.append((cmd, check))
+
+    monkeypatch.setattr(run.subprocess, "run", fake_run)
+
+    run.run_with_docker(path=str(project), build_first=False, expose=True)
+
+    cmd, check = calls[0]
+    assert cmd == [
+        "docker", "run", "--rm",
+        "-p", "9000:9000",
+        "demo",
+    ]
+    assert check is True