add tests for js directive

src/cli/javascript/mod.rs

@@ -14,7 +14,7 @@ mod runtime;
 pub use request_filter::RequestFilter;
 pub use runtime::Runtime;
 
-use crate::core::{blueprint, HttpIO};
+use crate::core::{blueprint, HttpIO, WorkerIO};
 
 pub fn init_http(
     http: Arc<impl HttpIO>,
@@ -25,6 +25,13 @@ pub fn init_http(
     Arc::new(RequestFilter::new(http, script_io))
 }
 
+pub fn init_worker_io<T, V>(script: blueprint::Script) -> Arc<dyn WorkerIO<T, V> + Send + Sync>
+where
+    Runtime: WorkerIO<T, V>,
+{
+    (Arc::new(Runtime::new(script))) as _
+}
+
 fn create_header_map(
     headers: BTreeMap<String, String>,
 ) -> anyhow::Result<reqwest::header::HeaderMap> {

tests/core/parse.rs

@@ -272,23 +272,36 @@ impl ExecutionSpec {
         http_client: Arc<Http>,
     ) -> Arc<AppContext> {
         let blueprint = Blueprint::try_from(config).unwrap();
-        let http = if let Some(script) = blueprint.server.script.clone() {
+        let script = blueprint.server.script.clone();
+        let http = if let Some(script) = script.clone() {
             javascript::init_http(http_client, script)
         } else {
             http_client
         };
 
         let http2_only = http.clone();
 
+        let http_worker = if let Some(script) = script.clone() {
+            javascript::init_worker_io(script)
+        } else {
+            Arc::new(DefaultJsRuntime {})
+        };
+
+        let worker = if let Some(script) = script {
+            javascript::init_worker_io(script)
+        } else {
+            Arc::new(DefaultJsRuntime {})
+        };
+
         let runtime = TargetRuntime {
             http,
             http2_only,
             file: Arc::new(File::new(self.clone())),
             env: Arc::new(Env::init(env)),
             cache: Arc::new(InMemoryCache::new()),
             extensions: Arc::new(vec![]),
-            http_worker: Arc::new(DefaultJsRuntime {}),
-            worker: Arc::new(DefaultJsRuntime {}),
+            http_worker,
+            worker,
         };
 
         let endpoints = config

tests/core/runtime.rs

@@ -72,7 +72,7 @@ pub fn create_runtime(
         http_client.clone()
     };
 
-    let http2 = if let Some(script) = script {
+    let http2 = if let Some(script) = script.clone() {
         javascript::init_http(http_client.clone(), script)
     } else {
         http_client.clone()

tests/core/snapshots/js-directive.md_0.snap

@@ -0,0 +1,17 @@
+---
+source: tests/core/spec.rs
+expression: response
+---
+{
+  "status": 200,
+  "headers": {
+    "content-type": "application/json"
+  },
+  "body": {
+    "data": {
+      "hello": {
+        "name": "Leanne Graham"
+      }
+    }
+  }
+}

tests/core/snapshots/js-directive.md_client.snap

@@ -0,0 +1,28 @@
+---
+source: tests/core/spec.rs
+expression: formatted
+---
+scalar Date
+
+scalar Email
+
+scalar Empty
+
+scalar JSON
+
+scalar PhoneNumber
+
+type Query {
+  hello: User!
+}
+
+scalar Url
+
+type User {
+  id: Int!
+  name: String!
+}
+
+schema {
+  query: Query
+}

tests/core/snapshots/js-directive.md_merged.snap

@@ -0,0 +1,16 @@
+---
+source: tests/core/spec.rs
+expression: formatter
+---
+schema @server @upstream(baseURL: "https://jsonplaceholder.typicode.com") @link(src: "test.js", type: Script) {
+  query: Query
+}
+
+type Query {
+  hello: User! @http(path: "/users/1")
+}
+
+type User {
+  id: Int!
+  name: String! @js(name: "name")
+}