wasi-nn: add CLI tests

These tests are Rust versions of what `ci/run-wasi-nn-example.sh`
executed. Now that all of the test-checking (ML backend? artifacts?) can
be done via Rust, the Bash script is no longer needed; this change
removes it.

This change also switches over CI to run all of the new wasi-nn tests by
modifying the `main.yml` workflow and the `ci/run-tests.sh` script.

.github/workflows/main.yml

@@ -389,6 +389,12 @@ jobs:
     - run: echo CARGO_BUILD_TARGET=${{ matrix.target }} >> $GITHUB_ENV
       if: matrix.target != ''
 
+    # Install OpenVINO for testing wasmtime-wasi-nn.
+    - uses: abrown/install-openvino-action@v6
+      with:
+        version: 2022.3.0
+        apt: true
+
     # Fix an ICE for now in gcc when compiling zstd with debuginfo (??)
     - run: echo CFLAGS=-g0 >> $GITHUB_ENV
       if: matrix.target == 'x86_64-pc-windows-gnu'
@@ -479,32 +485,6 @@ jobs:
     # Windows fails GitHub Actions will confusingly mark the failed Windows job
     # as cancelled instead of failed.
 
-  # Build and test the wasi-nn module.
-  test_wasi_nn:
-    needs: determine
-    if: needs.determine.outputs.run-full
-    name: Test wasi-nn module
-    runs-on: ubuntu-latest
-    steps:
-      - uses: actions/checkout@v3
-        with:
-          submodules: true
-      - uses: ./.github/actions/install-rust
-      - run: rustup target add wasm32-wasi
-      - uses: abrown/install-openvino-action@v6
-        with:
-          version: 2022.3.0
-          apt: true
-      - run: ./ci/run-wasi-nn-example.sh
-        env:
-          RUST_BACKTRACE: 1
-
-      # common logic to cancel the entire run if this job fails
-      - run: gh run cancel ${{ github.run_id }}
-        if: failure() && github.event_name != 'pull_request'
-        env:
-          GH_TOKEN: ${{ github.token }}
-
   build-preview1-component-adapter:
     name: Build wasi-preview1-component-adapter
     needs: determine
@@ -752,7 +732,6 @@ jobs:
       - checks
       - checks_winarm64
       - fuzz_targets
-      - test_wasi_nn
       - bench
       - meta_deterministic_check
       - verify-publish

ci/run-tests.sh

@@ -5,8 +5,10 @@ cargo test \
     --features wasi-threads \
     --features wasi-http \
     --features component-model \
+    --features wasmtime-wasi-nn/test-check \
     --workspace \
-    --exclude 'wasmtime-wasi-*' \
+    --exclude 'wasmtime-wasi-http' \
+    --exclude 'wasmtime-wasi-crypto' \
     --exclude wasi-tests \
     --exclude wasi-http-tests \
     --exclude command-tests \

tests/all/cli_tests.rs

@@ -597,6 +597,44 @@ fn run_simple_with_wasi_threads() -> Result<()> {
     Ok(())
 }
 
+#[cfg(feature = "wasi-nn")]
+#[test]
+fn image_classification_with_wasi_nn() -> Result<()> {
+    wasmtime_wasi_nn::test_check!();
+    let wasm =
+        wasmtime_wasi_nn::test_check::cargo_build("crates/wasi-nn/examples/image-classification");
+    let artifacts_dir = wasmtime_wasi_nn::test_check::artifacts_dir();
+    let artifacts_dir = artifacts_dir.display();
+    let stdout = run_wasmtime(&[
+        "run",
+        "--wasi-modules=experimental-wasi-nn",
+        &format!("--mapdir=fixture::{artifacts_dir}"),
+        &format!("{}", wasm.display()),
+    ])?;
+    assert!(stdout.contains("InferenceResult(963"));
+    Ok(())
+}
+
+#[cfg(feature = "wasi-nn")]
+#[test]
+fn image_classification_with_wasi_nn_and_named_models() -> Result<()> {
+    wasmtime_wasi_nn::test_check!();
+    let wasm = wasmtime_wasi_nn::test_check::cargo_build(
+        "crates/wasi-nn/examples/image-classification-named",
+    );
+    let artifacts_dir = wasmtime_wasi_nn::test_check::artifacts_dir();
+    let artifacts_dir = artifacts_dir.display();
+    let stdout = run_wasmtime(&[
+        "run",
+        "--wasi-modules=experimental-wasi-nn",
+        &format!("--mapdir=fixture::{artifacts_dir}"),
+        &format!("--wasi-nn-graph=openvino::{artifacts_dir}"),
+        &format!("{}", wasm.display()),
+    ])?;
+    assert!(stdout.contains("InferenceResult(963"));
+    Ok(())
+}
+
 #[test]
 fn wasm_flags() -> Result<()> {
     // Any argument after the wasm module should be interpreted as for the