feat: add more test cases

Author: sunxilin

packages/example_extensions/pcm_file_reader_python/extension.py

@@ -26,7 +26,7 @@ def __init__(self, name: str) -> None:
         self._audio_frame_name: str = "pcm_frame"
         self._pcm_data: bytes = b""
         self._is_running: bool = False
-        self._send_task: asyncio.Task | None = None
+        self._send_task: asyncio.Task[None] | None = None
 
     async def on_init(self, ten_env: AsyncTenEnv) -> None:
         ten_env.log(LogLevel.DEBUG, "on_init")
@@ -123,11 +123,11 @@ async def _send_audio_frames(self, ten_env: AsyncTenEnv) -> None:
         """Send audio frames at specified interval."""
         # Calculate frame size: samples_per_frame * bytes_per_sample * channels
         # samples_per_frame = sample_rate * frame_interval_ms / 1000
-        samples_per_frame = (
-            self._sample_rate * self._frame_interval_ms // 1000
-        )
+        samples_per_frame = self._sample_rate * self._frame_interval_ms // 1000
         frame_size = (
-            samples_per_frame * self._bytes_per_sample * self._number_of_channels
+            samples_per_frame
+            * self._bytes_per_sample
+            * self._number_of_channels
         )
 
         ten_env.log(
@@ -180,7 +180,9 @@ async def _send_audio_frames(self, ten_env: AsyncTenEnv) -> None:
             )
 
             offset += current_frame_size
-            timestamp += self._frame_interval_ms * 1000  # Convert to microseconds
+            timestamp += (
+                self._frame_interval_ms * 1000
+            )  # Convert to microseconds
 
             # Wait for next frame interval
             if self._is_running and offset < len(self._pcm_data):
@@ -211,7 +213,15 @@ async def on_cmd(self, ten_env: AsyncTenEnv, cmd: Cmd) -> None:
         cmd_name = cmd.get_name()
         ten_env.log(LogLevel.DEBUG, f"on_cmd name {cmd_name}")
 
-        cmd_result = CmdResult.create(StatusCode.OK, cmd)
-        cmd_result.set_property_string("detail", "ok")
-
-        await ten_env.return_result(cmd_result)
+        if cmd_name == "get_status":
+            cmd_result = CmdResult.create(StatusCode.OK, cmd)
+            cmd_result.set_property_string(
+                "status", "running" if self._is_running else "stopped"
+            )
+            await ten_env.return_result(cmd_result)
+        else:
+            cmd_result = CmdResult.create(StatusCode.ERROR, cmd)
+            cmd_result.set_property_string(
+                "detail", f"Unknown command: {cmd_name}"
+            )
+            await ten_env.return_result(cmd_result)

packages/example_extensions/pcm_file_reader_python/manifest.json

@@ -35,26 +35,33 @@
     ],
     "api": {
         "property": {
-            "audio_file_path": {
-                "type": "string"
-            },
-            "sample_rate": {
-                "type": "int32"
-            },
-            "bytes_per_sample": {
-                "type": "int32"
-            },
-            "number_of_channels": {
-                "type": "int32"
-            },
-            "frame_interval_ms": {
-                "type": "int32"
+            "properties": {
+                "audio_file_path": {
+                    "type": "string"
+                },
+                "sample_rate": {
+                    "type": "int32"
+                },
+                "bytes_per_sample": {
+                    "type": "int32"
+                },
+                "number_of_channels": {
+                    "type": "int32"
+                },
+                "frame_interval_ms": {
+                    "type": "int32"
+                }
             }
         },
         "audio_frame_out": [
             {
                 "name": "pcm_frame"
             }
+        ],
+        "cmd_in": [
+            {
+                "name": "get_status"
+            }
         ]
     }
 }

tests/ten_runtime/integration/python/BUILD.gn

@@ -68,6 +68,7 @@ group("python") {
         "trigger_life_cycle_python",
         "two_async_exts_one_group_python",
         "two_async_exts_python",
+        "two_exts_graph_test_python",
         "websocket_server_python",
       ]
 

tests/ten_runtime/integration/python/two_exts_graph_test_python/BUILD.gn

@@ -0,0 +1,83 @@
+#
+# Copyright © 2025 Agora
+# This file is part of TEN Framework, an open source project.
+# Licensed under the Apache License, Version 2.0, with certain conditions.
+# Refer to the "LICENSE" file in the root directory for more information.
+#
+import("//build/ten_runtime/feature/test.gni")
+import("//build/ten_runtime/ten.gni")
+
+ten_package_test_prepare_app("two_exts_graph_test_python_app") {
+  src_app = "default_app_python"
+  src_app_language = "python"
+  generated_app_src_root_dir_name = "two_exts_graph_test_python_app"
+
+  replace_paths_after_install_app = [
+    "two_exts_graph_test_python_app/manifest.json",
+    "two_exts_graph_test_python_app/property.json",
+  ]
+
+  test_files = exec_script(
+          "//.gnfiles/build/scripts/glob_file.py",
+          [
+            "--dir",
+            rebase_path(
+                "//tests/ten_runtime/integration/python/two_exts_graph_test_python/two_exts_graph_test_python_app/tests/**/*"),
+            "--dir-base",
+            rebase_path(
+                "//tests/ten_runtime/integration/python/two_exts_graph_test_python/two_exts_graph_test_python_app/tests"),
+            "--recursive",
+            "--only-output-file",
+          ],
+          "json")
+
+  replace_paths_after_install_all = []
+  foreach(test_file, test_files) {
+    test_file_rel_path = test_file.relative_path
+    replace_paths_after_install_all +=
+        [ "two_exts_graph_test_python_app/tests/${test_file_rel_path}" ]
+  }
+
+  if (ten_enable_ten_manager) {
+    deps = [
+      "//core/src/ten_manager",
+      "//packages/core_apps/default_app_python:upload_default_app_python_to_server",
+      "//packages/core_systems/pytest_ten:upload_pytest_ten_to_server",
+      "//packages/example_extensions/pcm_file_reader_python:upload_pcm_file_reader_python_to_server",
+      "//packages/example_extensions/simple_echo_python:upload_simple_echo_python_to_server",
+      "//packages/example_extensions/simple_http_server_cpp:upload_simple_http_server_cpp_to_server",
+    ]
+  }
+}
+
+ten_package_test_prepare_auxiliary_resources(
+    "two_exts_graph_test_python_test_files") {
+  resources = [
+    "__init__.py",
+    "test_case.py",
+  ]
+
+  utils_files = exec_script("//.gnfiles/build/scripts/glob_file.py",
+                            [
+                              "--dir",
+                              rebase_path("//tests/utils/**/*"),
+                              "--dir-base",
+                              rebase_path("//tests/utils"),
+                              "--recursive",
+                              "--only-output-file",
+                            ],
+                            "json")
+
+  foreach(utils_file, utils_files) {
+    utils_file_rel_path = utils_file.relative_path
+    resources +=
+        [ "//tests/utils/${utils_file_rel_path}=>utils/${utils_file_rel_path}" ]
+  }
+}
+
+group("two_exts_graph_test_python") {
+  deps = [
+    ":two_exts_graph_test_python_app",
+    ":two_exts_graph_test_python_test_files",
+  ]
+}

tests/ten_runtime/integration/python/two_exts_graph_test_python/__init__.py

@@ -0,0 +1,104 @@
+"""
+Test two_exts_graph_test_python.
+"""
+
+import subprocess
+import os
+import sys
+from sys import stdout
+from .utils import build_config, build_pkg, fs_utils
+
+
+def test_two_exts_graph_test_python():
+    """Test client and app server."""
+    base_path = os.path.dirname(os.path.abspath(__file__))
+    root_dir = os.path.join(base_path, "../../../../../")
+
+    my_env = os.environ.copy()
+
+    # Set the required environment variables for the test.
+    my_env["PYTHONMALLOC"] = "malloc"
+    my_env["PYTHONDEVMODE"] = "1"
+
+    app_dir_name = "two_exts_graph_test_python_app"
+    app_root_path = os.path.join(base_path, "two_exts_graph_test_python_app")
+    app_language = "python"
+
+    build_config_args = build_config.parse_build_config(
+        os.path.join(root_dir, "tgn_args.txt"),
+    )
+
+    # Before starting, cleanup the old app package.
+    fs_utils.remove_tree(app_root_path)
+
+    print(f'Assembling and building package "{app_dir_name}".')
+
+    rc = build_pkg.prepare_and_build_app(
+        build_config_args,
+        root_dir,
+        base_path,
+        app_dir_name,
+        app_language,
+    )
+    if rc != 0:
+        assert False, "Failed to build package."
+
+    # Step 1: Bootstrap Python dependencies (update pyproject.toml and sync)
+    print("Bootstrapping Python dependencies...")
+    rc = build_pkg.bootstrap_python_dependencies(
+        app_root_path, my_env, log_level=1
+    )
+    if rc != 0:
+        assert False, "Failed to bootstrap Python dependencies."
+
+    # Step 2: Setup AddressSanitizer if needed
+    if sys.platform == "linux":
+        if build_config_args.enable_sanitizer:
+            libasan_path = os.path.join(
+                base_path,
+                (
+                    "two_exts_graph_test_python_app/ten_packages/system/"
+                    "ten_runtime/lib/libasan.so"
+                ),
+            )
+
+            if os.path.exists(libasan_path):
+                print("Using AddressSanitizer library.")
+                my_env["LD_PRELOAD"] = libasan_path
+
+    tests_dir = os.path.join(app_root_path, "tests")
+
+    # Step 3:
+    #
+    # Run the test using uv run pytest.
+    # When sanitizer is enabled, we need to bypass `uv run` because `uv` itself
+    # may trigger memory leak reports (false positives from the tool itself),
+    # causing the test to fail.
+    if sys.platform == "linux" and build_config_args.enable_sanitizer:
+        print("Starting pytest with python from venv (bypassing uv run)...")
+        venv_path = os.path.join(tests_dir, ".venv")
+        python_exe = os.path.join(venv_path, "bin", "python")
+        uv_run_pytest_cmd = [python_exe, "-m", "pytest", "-s"]
+        my_env["VIRTUAL_ENV"] = venv_path
+    else:
+        uv_run_pytest_cmd = [
+            "uv",
+            "run",
+            "pytest",
+            "-s",
+        ]
+
+    try:
+        tester_process = subprocess.Popen(
+            uv_run_pytest_cmd,
+            stdout=stdout,
+            stderr=subprocess.STDOUT,
+            env=my_env,
+            cwd=tests_dir,
+        )
+
+        tester_rc = tester_process.wait()
+        assert tester_rc == 0
+    finally:
+        if build_config_args.ten_enable_tests_cleanup is True:
+            fs_utils.remove_tree(app_root_path)

tests/ten_runtime/integration/python/two_exts_graph_test_python/test_case.py

@@ -0,0 +1,39 @@
+{
+  "type": "app",
+  "name": "default_app_python",
+  "version": "0.11.51",
+  "dependencies": [
+    {
+      "type": "system",
+      "name": "ten_runtime",
+      "version": "0.11.51"
+    },
+    {
+      "type": "system",
+      "name": "ten_runtime_python",
+      "version": "0.11.51"
+    },
+    {
+      "type": "extension",
+      "name": "simple_http_server_cpp",
+      "version": "0.11.51"
+    },
+    {
+      "type": "extension",
+      "name": "pcm_file_reader_python",
+      "version": "0.11.51"
+    },
+    {
+      "type": "extension",
+      "name": "simple_echo_python",
+      "version": "0.11.51"
+    }
+  ],
+  "dev_dependencies": [
+    {
+      "type": "system",
+      "name": "pytest_ten",
+      "version": "0.11.51"
+    }
+  ]
+}

tests/ten_runtime/integration/python/two_exts_graph_test_python/two_exts_graph_test_python_app/manifest.json

@@ -0,0 +1 @@
+{}

tests/ten_runtime/integration/python/two_exts_graph_test_python/two_exts_graph_test_python_app/property.json

@@ -0,0 +1,21 @@
+#!/bin/bash
+
+set -e
+
+cd "$(dirname "${BASH_SOURCE[0]}")/../.."
+
+export PYTHONPATH=ten_packages/system/ten_runtime_python/lib:ten_packages/system/ten_runtime_python/interface:ten_packages/system/pytest_ten:.
+
+# If the Python app imports some modules that are compiled with a different
+# version of libstdc++ (ex: PyTorch), the Python app may encounter confusing
+# errors. To solve this problem, we can preload the correct version of
+# libstdc++.
+#
+# export LD_PRELOAD=/lib/x86_64-linux-gnu/libstdc++.so.6
+#
+# Another solution is to make sure the module 'ten_runtime_python' is imported
+# _after_ the module that requires another version of libstdc++ is imported.
+#
+# Refer to https://github.com/pytorch/pytorch/issues/102360?from_wecom=1#issuecomment-1708989096
+
+python -m pytest -s tests/ "$@"