Tests: initial "wasm-wasi-component" test

1 files changed, 60 insertions, 0 deletions

diff --git a/test/unit/applications/lang/wasm_component.py b/test/unit/applications/lang/wasm_component.py
new file mode 100644
index 00000000..6f7b5518
--- /dev/null
+++ b/test/unit/applications/lang/wasm_component.py
@@ -0,0 +1,60 @@
+from pathlib import Path
+import shutil
+import subprocess
+from urllib.parse import quote
+
+from unit.applications.proto import ApplicationProto
+from unit.option import option
+
+
+class ApplicationWasmComponent(ApplicationProto):
+    @staticmethod
+    def prepare_env(script):
+        try:
+            subprocess.check_output(['cargo', 'component', '--help'])
+        except (subprocess.CalledProcessError, FileNotFoundError):
+            return None
+
+        temp_dir = Path(f'{option.temp_dir}/wasm_component/')
+
+        if not temp_dir.exists():
+            temp_dir.mkdir()
+
+        app_path = f'{temp_dir}/{script}'
+
+        shutil.copytree(f'{option.test_dir}/wasm_component/{script}', app_path)
+
+        try:
+            output = subprocess.check_output(
+                ['cargo', 'component', 'build', '--release'],
+                cwd=app_path,
+                stderr=subprocess.STDOUT,
+            )
+        except KeyboardInterrupt:
+            raise
+
+        except subprocess.CalledProcessError:
+            return None
+
+        return output
+
+    def load(self, script, **kwargs):
+        self.prepare_env(script)
+
+        component_path = f'{option.temp_dir}/wasm_component/{script}/target/wasm32-wasi/release/test_wasi_component.wasm'
+
+        self._load_conf(
+            {
+                "listeners": {
+                    "*:8080": {"pass": f"applications/{quote(script, '')}"}
+                },
+                "applications": {
+                    script: {
+                        "type": "wasm-wasi-component",
+                        "processes": {"spare": 0},
+                        "component": component_path,
+                    }
+                },
+            },
+            **kwargs,
+        )