feat(rstest): add preserveNewUrl option to keep new URL untouched (#12632)

* feat(rstest): add preserveNewUrl option to keep new URL untouched

* fix: format Rust code with rustfmt

Author: fi3ework

crates/node_binding/napi-binding.d.ts

@@ -2790,6 +2790,7 @@ export interface RawRstestPluginOptions {
   importMetaPathName: boolean
   hoistMockModule: boolean
   manualMockRoot: string
+  preserveNewUrl?: Array<string>
 }
 
 export interface RawRuleSetCondition {

crates/rspack_binding_api/src/rstest.rs

@@ -12,6 +12,11 @@ pub struct RawRstestPluginOptions {
   pub hoist_mock_module: bool,
   // Root of the manual mock directory.
   pub manual_mock_root: String,
+  // Preserve `new URL("*.<ext>", import.meta.url)` expressions for specified extensions
+  // instead of transforming them to asset imports.
+  // This allows rstest to dynamically load modules (e.g., wasm) at runtime.
+  // Example: [".wasm"] to preserve wasm URL expressions.
+  pub preserve_new_url: Option<Vec<String>>,
 }
 
 impl From<RawRstestPluginOptions> for RstestPluginOptions {
@@ -21,6 +26,7 @@ impl From<RawRstestPluginOptions> for RstestPluginOptions {
       hoist_mock_module: value.hoist_mock_module,
       import_meta_path_name: value.import_meta_path_name,
       manual_mock_root: value.manual_mock_root,
+      preserve_new_url: value.preserve_new_url.unwrap_or_default(),
     }
   }
 }

crates/rspack_plugin_rstest/src/lib.rs

@@ -4,5 +4,6 @@ mod mock_module_id_dependency;
 mod module_path_name_dependency;
 mod parser_plugin;
 mod plugin;
+mod url_dependency;
 
 pub use plugin::*;

crates/rspack_plugin_rstest/src/plugin.rs

@@ -19,6 +19,7 @@ use crate::{
   mock_method_dependency::MockMethodDependencyTemplate,
   mock_module_id_dependency::MockModuleIdDependencyTemplate,
   module_path_name_dependency::ModulePathNameDependencyTemplate, parser_plugin::RstestParserPlugin,
+  url_dependency::RstestUrlDependencyTemplate,
 };
 
 #[derive(Debug)]
@@ -27,6 +28,7 @@ pub struct RstestPluginOptions {
   pub hoist_mock_module: bool,
   pub import_meta_path_name: bool,
   pub manual_mock_root: String,
+  pub preserve_new_url: Vec<String>,
 }
 
 #[derive(Debug)]
@@ -148,6 +150,15 @@ async fn compilation_stage_9999(
     Arc::new(ImportDependencyTemplate::default()),
   );
 
+  if !self.options.preserve_new_url.is_empty() {
+    compilation.set_dependency_template(
+      RstestUrlDependencyTemplate::template_type(),
+      Arc::new(RstestUrlDependencyTemplate::new(
+        self.options.preserve_new_url.clone(),
+      )),
+    );
+  }
+
   Ok(())
 }
 

crates/rspack_plugin_rstest/src/url_dependency.rs

@@ -0,0 +1,57 @@
+use rspack_cacheable::cacheable;
+use rspack_core::{
+  DependencyCodeGeneration, DependencyTemplate, DependencyTemplateType, DependencyType,
+  ModuleDependency, TemplateContext, TemplateReplaceSource,
+};
+use rspack_plugin_javascript::dependency::{URLDependency, URLDependencyTemplate};
+
+#[cacheable]
+#[derive(Debug, Default)]
+pub struct RstestUrlDependencyTemplate {
+  /// List of extensions to preserve (e.g., [".wasm", ".node"])
+  preserve_extensions: Vec<String>,
+}
+
+impl RstestUrlDependencyTemplate {
+  pub fn new(preserve_extensions: Vec<String>) -> Self {
+    Self {
+      preserve_extensions,
+    }
+  }
+
+  pub fn template_type() -> DependencyTemplateType {
+    DependencyTemplateType::Dependency(DependencyType::NewUrl)
+  }
+}
+
+impl DependencyTemplate for RstestUrlDependencyTemplate {
+  fn render(
+    &self,
+    dep: &dyn DependencyCodeGeneration,
+    source: &mut TemplateReplaceSource,
+    code_generatable_context: &mut TemplateContext,
+  ) {
+    let dep = dep
+      .as_any()
+      .downcast_ref::<URLDependency>()
+      .expect("RstestUrlDependencyTemplate should be used for URLDependency");
+
+    // Strip query string and fragment from request path before checking extension
+    let request = dep.request();
+    let request_path = request.split(&['?', '#'][..]).next().unwrap_or(request);
+
+    let should_preserve = request_path.rsplit('.').next().is_some_and(|ext| {
+      self.preserve_extensions.iter().any(|preserve_ext| {
+        // Support both ".ext" and "ext" formats
+        let preserve_ext = preserve_ext.trim_start_matches('.');
+        ext.eq_ignore_ascii_case(preserve_ext)
+      })
+    });
+
+    if should_preserve {
+      return;
+    }
+
+    URLDependencyTemplate::default().render(dep, source, code_generatable_context);
+  }
+}

tests/rspack-test/configCases/rstest/new-url-wasm/index.js

@@ -0,0 +1,4 @@
+const wasmUrl = new URL("./test.wasm", import.meta.url);
+const pngUrl = new URL("./test.png", import.meta.url);
+
+console.log(wasmUrl, pngUrl);

tests/rspack-test/configCases/rstest/new-url-wasm/rspack.config.js

@@ -0,0 +1,38 @@
+const {
+	experiments: { RstestPlugin }
+} = require("@rspack/core");
+
+/** @type {import("@rspack/core").Configuration} */
+module.exports = [
+	{
+		entry: "./index.js",
+		target: "node",
+		node: {
+			__filename: false,
+			__dirname: false
+		},
+		output: {
+			filename: "bundle.js"
+		},
+		plugins: [
+			new RstestPlugin({
+				injectModulePathName: true,
+				hoistMockModule: true,
+				importMetaPathName: true,
+				manualMockRoot: __dirname,
+				preserveNewUrl: [".wasm"]
+			})
+		]
+	},
+	{
+		entry: {
+			main: "./test.js"
+		},
+		output: {
+			filename: "[name].js"
+		},
+		externalsPresets: {
+			node: true
+		}
+	}
+];

tests/rspack-test/configCases/rstest/new-url-wasm/test.config.js

@@ -0,0 +1,7 @@
+/** @type {import("../../../..").TConfigCaseConfig} */
+module.exports = {
+	findBundle: function (i) {
+		if (i === 0) return ["main.js"];
+		return ["main.js"];
+	}
+};

tests/rspack-test/configCases/rstest/new-url-wasm/test.js

@@ -0,0 +1,12 @@
+const fs = require("fs");
+const path = require("path");
+
+const content = fs.readFileSync(path.resolve(__dirname, "bundle.js"), "utf-8");
+
+it("should keep wasm new URL untouched in rstest", () => {
+	expect(content).toContain('new URL("./test.wasm", import.meta.url)');
+});
+
+it("should keep non-wasm new URL behavior", () => {
+	expect(content).toContain("/* asset import */");
+});