mockforge_core/
request_scripting.rs

1//! Pre/Post request scripting for MockForge chains
2//!
3//! This module provides JavaScript scripting capabilities for executing
4//! custom logic before and after HTTP requests in request chains.
5
6use crate::{Error, Result};
7use rquickjs::{Context, Ctx, Function, Object, Runtime};
8
9use serde::{Deserialize, Serialize};
10use serde_json::Value;
11use std::collections::HashMap;
12use std::sync::Arc;
13use tokio::sync::Semaphore;
14
15/// Results from script execution
16#[derive(Debug, Clone, Serialize, Deserialize)]
17pub struct ScriptResult {
18    /// Return value from the script
19    pub return_value: Option<Value>,
20    /// Variables modified by the script
21    pub modified_variables: HashMap<String, Value>,
22    /// Errors encountered during execution
23    pub errors: Vec<String>,
24    /// Execution time in milliseconds
25    pub execution_time_ms: u64,
26}
27
28/// Script execution context accessible to scripts
29#[derive(Debug, Clone)]
30pub struct ScriptContext {
31    /// Current request being executed (for pre-scripts)
32    pub request: Option<crate::request_chaining::ChainRequest>,
33    /// Response from the request (for post-scripts)
34    pub response: Option<crate::request_chaining::ChainResponse>,
35    /// Chain context with stored responses and variables
36    pub chain_context: HashMap<String, Value>,
37    /// Request-scoped variables
38    pub variables: HashMap<String, Value>,
39    /// Environment variables
40    pub env_vars: HashMap<String, String>,
41}
42
43/// JavaScript scripting engine
44pub struct ScriptEngine {
45    semaphore: Arc<Semaphore>,
46}
47
48impl std::fmt::Debug for ScriptEngine {
49    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
50        f.debug_struct("ScriptEngine")
51            .field("semaphore", &format!("Semaphore({})", self.semaphore.available_permits()))
52            .finish()
53    }
54}
55
56/// JavaScript script engine for request/response processing
57///
58/// Provides JavaScript scripting capabilities for executing custom logic
59/// before and after HTTP requests in request chains.
60impl ScriptEngine {
61    /// Create a new script engine
62    pub fn new() -> Self {
63        let semaphore = Arc::new(Semaphore::new(10)); // Limit concurrent script executions
64
65        Self { semaphore }
66    }
67
68    /// Execute a JavaScript script with access to the script context
69    pub async fn execute_script(
70        &self,
71        script: &str,
72        script_context: &ScriptContext,
73        timeout_ms: u64,
74    ) -> Result<ScriptResult> {
75        let _permit =
76            self.semaphore.acquire().await.map_err(|e| {
77                Error::generic(format!("Failed to acquire execution permit: {}", e))
78            })?;
79
80        let script = script.to_string();
81        let script_context = script_context.clone();
82
83        let start_time = std::time::Instant::now();
84
85        // Execute with timeout handling using spawn_blocking for rquickjs
86        // Create a helper function that returns Result instead of panicking
87        let script_clone = script.clone();
88        let script_context_clone = script_context.clone();
89
90        let timeout_duration = std::time::Duration::from_millis(timeout_ms);
91        let timeout_result = tokio::time::timeout(
92            timeout_duration,
93            tokio::task::spawn_blocking(move || {
94                execute_script_in_runtime(&script_clone, &script_context_clone)
95            }),
96        )
97        .await;
98
99        let execution_time_ms = start_time.elapsed().as_millis() as u64;
100
101        match timeout_result {
102            Ok(join_result) => match join_result {
103                Ok(Ok(mut script_result)) => {
104                    script_result.execution_time_ms = execution_time_ms;
105                    Ok(script_result)
106                }
107                Ok(Err(e)) => Err(e),
108                Err(e) => Err(Error::generic(format!("Script execution task failed: {}", e))),
109            },
110            Err(_) => {
111                Err(Error::generic(format!("Script execution timed out after {}ms", timeout_ms)))
112            }
113        }
114    }
115
116    /// Execute script within the JavaScript context (blocking)
117    fn execute_in_context_blocking(
118        &self,
119        script: &str,
120        script_context: &ScriptContext,
121    ) -> Result<ScriptResult> {
122        // Create a new runtime for this execution
123        let runtime = Runtime::new()?;
124        let context = Context::full(&runtime)?;
125
126        context.with(|ctx| self.execute_in_context(ctx, script, script_context, 0))
127    }
128
129    /// Execute script within the JavaScript context
130    fn execute_in_context<'js>(
131        &self,
132        ctx: Ctx<'js>,
133        script: &str,
134        script_context: &ScriptContext,
135        timeout_ms: u64,
136    ) -> Result<ScriptResult> {
137        // Clone ctx for use in multiple places
138        let ctx_clone = ctx.clone();
139
140        // Create the global context object
141        let global = ctx.globals();
142        let mockforge_obj = Object::new(ctx_clone.clone())?;
143
144        // Expose context data
145        self.expose_script_context(ctx.clone(), &mockforge_obj, script_context)?;
146
147        // Add the mockforge object to global scope
148        global.set("mockforge", mockforge_obj)?;
149
150        // Add utility functions
151        self.add_global_functions(ctx_clone, &global, script_context)?;
152
153        // Execute the script
154        let result = eval_script_with_timeout(&ctx, script, timeout_ms)?;
155
156        // Extract modified variables and return value
157        let modified_vars = extract_modified_variables(&ctx, script_context)?;
158        let return_value = extract_return_value(&ctx, &result)?;
159
160        Ok(ScriptResult {
161            return_value,
162            modified_variables: modified_vars,
163            errors: vec![],       // No errors if we reach here
164            execution_time_ms: 0, // Will be set by the caller
165        })
166    }
167
168    /// Expose script context as a global object
169    fn expose_script_context<'js>(
170        &self,
171        ctx: Ctx<'js>,
172        mockforge_obj: &Object<'js>,
173        script_context: &ScriptContext,
174    ) -> Result<()> {
175        expose_script_context_static(ctx, mockforge_obj, script_context)
176    }
177
178    /// Add global utility functions to the script context
179    fn add_global_functions<'js>(
180        &self,
181        ctx: Ctx<'js>,
182        global: &Object<'js>,
183        script_context: &ScriptContext,
184    ) -> Result<()> {
185        add_global_functions_static(ctx, global, script_context)
186    }
187}
188
189/// Extract return value from script execution
190fn extract_return_value<'js>(
191    ctx: &Ctx<'js>,
192    result: &rquickjs::Value<'js>,
193) -> Result<Option<Value>> {
194    extract_return_value_static(ctx, result)
195}
196
197/// Execute script in a new JavaScript runtime (blocking helper)
198/// This function is used by spawn_blocking to avoid panics
199fn execute_script_in_runtime(script: &str, script_context: &ScriptContext) -> Result<ScriptResult> {
200    // Create JavaScript runtime with proper error handling
201    let runtime = Runtime::new()
202        .map_err(|e| Error::generic(format!("Failed to create JavaScript runtime: {:?}", e)))?;
203
204    let context = Context::full(&runtime)
205        .map_err(|e| Error::generic(format!("Failed to create JavaScript context: {:?}", e)))?;
206
207    context.with(|ctx| {
208        // Create the global context object with proper error handling
209        let global = ctx.globals();
210        let mockforge_obj = Object::new(ctx.clone())
211            .map_err(|e| Error::generic(format!("Failed to create mockforge object: {:?}", e)))?;
212
213        // Expose context data
214        expose_script_context_static(ctx.clone(), &mockforge_obj, script_context)
215            .map_err(|e| Error::generic(format!("Failed to expose script context: {:?}", e)))?;
216
217        // Add the mockforge object to global scope
218        global.set("mockforge", mockforge_obj).map_err(|e| {
219            Error::generic(format!("Failed to set global mockforge object: {:?}", e))
220        })?;
221
222        // Add utility functions
223        add_global_functions_static(ctx.clone(), &global, script_context)
224            .map_err(|e| Error::generic(format!("Failed to add global functions: {:?}", e)))?;
225
226        // Execute the script
227        let result = ctx
228            .eval(script)
229            .map_err(|e| Error::generic(format!("Script execution failed: {:?}", e)))?;
230
231        // Extract modified variables and return value
232        let modified_vars =
233            extract_modified_variables_static(&ctx, script_context).map_err(|e| {
234                Error::generic(format!("Failed to extract modified variables: {:?}", e))
235            })?;
236
237        let return_value = extract_return_value_static(&ctx, &result)
238            .map_err(|e| Error::generic(format!("Failed to extract return value: {:?}", e)))?;
239
240        Ok(ScriptResult {
241            return_value,
242            modified_variables: modified_vars,
243            errors: vec![],       // No errors if we reach here
244            execution_time_ms: 0, // Will be set by the caller
245        })
246    })
247}
248
249/// Extract return value from script execution (static version)
250fn extract_return_value_static<'js>(
251    _ctx: &Ctx<'js>,
252    result: &rquickjs::Value<'js>,
253) -> Result<Option<Value>> {
254    match result.type_of() {
255        rquickjs::Type::String => {
256            // Use defensive pattern matching instead of unwrap()
257            if let Some(string_val) = result.as_string() {
258                Ok(Some(Value::String(string_val.to_string()?)))
259            } else {
260                Ok(None)
261            }
262        }
263        rquickjs::Type::Float => {
264            if let Some(num) = result.as_number() {
265                // Use defensive pattern matching for number conversion
266                // Try to convert to f64 first, fallback to int if that fails
267                if let Some(f64_val) = serde_json::Number::from_f64(num) {
268                    Ok(Some(Value::Number(f64_val)))
269                } else {
270                    // Fallback to integer conversion if f64 conversion fails
271                    Ok(Some(Value::Number(serde_json::Number::from(result.as_int().unwrap_or(0)))))
272                }
273            } else {
274                // Fallback to integer if number extraction fails
275                Ok(Some(Value::Number(serde_json::Number::from(result.as_int().unwrap_or(0)))))
276            }
277        }
278        rquickjs::Type::Bool => {
279            // Use defensive pattern matching instead of unwrap()
280            if let Some(bool_val) = result.as_bool() {
281                Ok(Some(Value::Bool(bool_val)))
282            } else {
283                Ok(None)
284            }
285        }
286        rquickjs::Type::Object => {
287            // Try to convert to JSON string and then parse back
288            if let Some(obj) = result.as_object() {
289                if let Some(string_val) = obj.as_string() {
290                    let json_str = string_val.to_string()?;
291                    Ok(Some(Value::String(json_str)))
292                } else {
293                    Ok(None)
294                }
295            } else {
296                Ok(None)
297            }
298        }
299        _ => Ok(None),
300    }
301}
302
303/// Extract modified variables from the script context
304fn extract_modified_variables<'js>(
305    ctx: &Ctx<'js>,
306    original_context: &ScriptContext,
307) -> Result<HashMap<String, Value>> {
308    extract_modified_variables_static(ctx, original_context)
309}
310
311/// Extract modified variables from the script context (static version)
312fn extract_modified_variables_static<'js>(
313    ctx: &Ctx<'js>,
314    original_context: &ScriptContext,
315) -> Result<HashMap<String, Value>> {
316    let mut modified = HashMap::new();
317
318    // Get the global mockforge object
319    let global = ctx.globals();
320    let mockforge_obj: Object = global.get("mockforge")?;
321
322    // Get the variables object
323    let vars_obj: Object = mockforge_obj.get("variables")?;
324
325    // Get all property names
326    let keys = vars_obj.keys::<String>();
327
328    for key_result in keys {
329        let key = key_result?;
330        let js_value: rquickjs::Value = vars_obj.get(&key)?;
331
332        // Convert JS value to serde_json::Value
333        if let Some(value) = js_value_to_json_value(&js_value) {
334            // Check if this is different from the original or new
335            let original_value = original_context.variables.get(&key);
336            if original_value != Some(&value) {
337                modified.insert(key, value);
338            }
339        }
340    }
341
342    Ok(modified)
343}
344
345/// Convert a JavaScript value to a serde_json::Value
346fn js_value_to_json_value(js_value: &rquickjs::Value) -> Option<Value> {
347    match js_value.type_of() {
348        rquickjs::Type::String => {
349            js_value.as_string().and_then(|s| s.to_string().ok()).map(Value::String)
350        }
351        rquickjs::Type::Int => {
352            js_value.as_int().map(|i| Value::Number(serde_json::Number::from(i)))
353        }
354        rquickjs::Type::Float => {
355            js_value.as_number().and_then(serde_json::Number::from_f64).map(Value::Number)
356        }
357        rquickjs::Type::Bool => js_value.as_bool().map(Value::Bool),
358        rquickjs::Type::Object | rquickjs::Type::Array => {
359            // For complex types, try to serialize to JSON string
360            if let Some(obj) = js_value.as_object() {
361                if let Some(str_val) = obj.as_string() {
362                    str_val
363                        .to_string()
364                        .ok()
365                        .and_then(|json_str| serde_json::from_str(&json_str).ok())
366                } else {
367                    // For now, return None for complex objects/arrays
368                    None
369                }
370            } else {
371                None
372            }
373        }
374        _ => None, // Null, undefined, etc.
375    }
376}
377
378/// Execute script with timeout
379fn eval_script_with_timeout<'js>(
380    ctx: &Ctx<'js>,
381    script: &str,
382    _timeout_ms: u64,
383) -> Result<rquickjs::Value<'js>> {
384    // For now, we'll just evaluate without timeout as the JS runtime doesn't support async timeouts
385    // In a future implementation, we could use a separate thread with timeout or implement
386    // a custom timeout mechanism. For now, the timeout is handled at the async boundary.
387
388    ctx.eval(script)
389        .map_err(|e| Error::generic(format!("JavaScript evaluation error: {:?}", e)))
390}
391
392impl Default for ScriptEngine {
393    fn default() -> Self {
394        Self::new()
395    }
396}
397
398/// Expose script context as a global object (static version)
399fn expose_script_context_static<'js>(
400    ctx: Ctx<'js>,
401    mockforge_obj: &Object<'js>,
402    script_context: &ScriptContext,
403) -> Result<()> {
404    // Expose request
405    if let Some(request) = &script_context.request {
406        let request_obj = Object::new(ctx.clone())?;
407        request_obj.set("id", &request.id)?;
408        request_obj.set("method", &request.method)?;
409        request_obj.set("url", &request.url)?;
410
411        // Headers
412        let headers_obj = Object::new(ctx.clone())?;
413        for (key, value) in &request.headers {
414            headers_obj.set(key.as_str(), value.as_str())?;
415        }
416        request_obj.set("headers", headers_obj)?;
417
418        // Body
419        if let Some(body) = &request.body {
420            let body_json = serde_json::to_string(body)
421                .map_err(|e| Error::generic(format!("Failed to serialize request body: {}", e)))?;
422            request_obj.set("body", body_json)?;
423        }
424
425        mockforge_obj.set("request", request_obj)?;
426    }
427
428    // Expose response (for post-scripts)
429    if let Some(response) = &script_context.response {
430        let response_obj = Object::new(ctx.clone())?;
431        response_obj.set("status", response.status as i32)?;
432        response_obj.set("duration_ms", response.duration_ms as i32)?;
433
434        // Response headers
435        let headers_obj = Object::new(ctx.clone())?;
436        for (key, value) in &response.headers {
437            headers_obj.set(key.as_str(), value.as_str())?;
438        }
439        response_obj.set("headers", headers_obj)?;
440
441        // Response body
442        if let Some(body) = &response.body {
443            let body_json = serde_json::to_string(body)
444                .map_err(|e| Error::generic(format!("Failed to serialize response body: {}", e)))?;
445            response_obj.set("body", body_json)?;
446        }
447
448        mockforge_obj.set("response", response_obj)?;
449    }
450
451    // Expose chain context
452    let chain_obj = Object::new(ctx.clone())?;
453    for (key, value) in &script_context.chain_context {
454        match value {
455            Value::String(s) => chain_obj.set(key.as_str(), s.as_str())?,
456            Value::Number(n) => {
457                if let Some(i) = n.as_i64() {
458                    chain_obj.set(key.as_str(), i as i32)?;
459                } else if let Some(f) = n.as_f64() {
460                    chain_obj.set(key.as_str(), f)?;
461                }
462            }
463            Value::Bool(b) => chain_obj.set(key.as_str(), *b)?,
464            Value::Object(obj) => {
465                let json_str = serde_json::to_string(&obj)
466                    .map_err(|e| Error::generic(format!("Failed to serialize object: {}", e)))?;
467                chain_obj.set(key.as_str(), json_str)?;
468            }
469            Value::Array(arr) => {
470                let json_str = serde_json::to_string(&arr)
471                    .map_err(|e| Error::generic(format!("Failed to serialize array: {}", e)))?;
472                chain_obj.set(key.as_str(), json_str)?;
473            }
474            _ => {} // Skip null values and other types
475        }
476    }
477    mockforge_obj.set("chain", chain_obj)?;
478
479    // Expose variables
480    let vars_obj = Object::new(ctx.clone())?;
481    for (key, value) in &script_context.variables {
482        match value {
483            Value::String(s) => vars_obj.set(key.as_str(), s.as_str())?,
484            Value::Number(n) => {
485                if let Some(i) = n.as_i64() {
486                    vars_obj.set(key.as_str(), i as i32)?;
487                } else if let Some(f) = n.as_f64() {
488                    vars_obj.set(key.as_str(), f)?;
489                }
490            }
491            Value::Bool(b) => vars_obj.set(key.as_str(), *b)?,
492            _ => {
493                let json_str = serde_json::to_string(&value).map_err(|e| {
494                    Error::generic(format!("Failed to serialize variable {}: {}", key, e))
495                })?;
496                vars_obj.set(key.as_str(), json_str)?;
497            }
498        }
499    }
500    mockforge_obj.set("variables", vars_obj)?;
501
502    // Expose environment variables
503    let env_obj = Object::new(ctx.clone())?;
504    for (key, value) in &script_context.env_vars {
505        env_obj.set(key.as_str(), value.as_str())?;
506    }
507    mockforge_obj.set("env", env_obj)?;
508
509    Ok(())
510}
511
512/// Add global utility functions to the script context (static version)
513fn add_global_functions_static<'js>(
514    ctx: Ctx<'js>,
515    global: &Object<'js>,
516    _script_context: &ScriptContext,
517) -> Result<()> {
518    // Add console object for logging
519    let console_obj = Object::new(ctx.clone())?;
520    let log_func = Function::new(ctx.clone(), || {
521        println!("Script log called");
522    })?;
523    console_obj.set("log", log_func)?;
524    global.set("console", console_obj)?;
525
526    // Add utility functions for scripts
527    let log_func = Function::new(ctx.clone(), |msg: String| {
528        println!("Script log: {}", msg);
529    })?;
530    global.set("log", log_func)?;
531
532    let stringify_func = Function::new(ctx.clone(), |value: rquickjs::Value| {
533        if let Some(obj) = value.as_object() {
534            if let Some(str_val) = obj.as_string() {
535                str_val.to_string().unwrap_or_else(|_| "undefined".to_string())
536            } else {
537                "object".to_string()
538            }
539        } else if value.is_string() {
540            value
541                .as_string()
542                .unwrap()
543                .to_string()
544                .unwrap_or_else(|_| "undefined".to_string())
545        } else {
546            format!("{:?}", value)
547        }
548    })?;
549    global.set("stringify", stringify_func)?;
550
551    // Add crypto utilities
552    let crypto_obj = Object::new(ctx.clone())?;
553
554    let base64_encode_func = Function::new(ctx.clone(), |input: String| -> String {
555        use base64::{engine::general_purpose, Engine as _};
556        general_purpose::STANDARD.encode(input)
557    })?;
558    crypto_obj.set("base64Encode", base64_encode_func)?;
559
560    let base64_decode_func = Function::new(ctx.clone(), |input: String| -> String {
561        use base64::{engine::general_purpose, Engine as _};
562        general_purpose::STANDARD
563            .decode(input)
564            .map(|bytes| String::from_utf8_lossy(&bytes).to_string())
565            .unwrap_or_else(|_| "".to_string())
566    })?;
567    crypto_obj.set("base64Decode", base64_decode_func)?;
568
569    let sha256_func = Function::new(ctx.clone(), |input: String| -> String {
570        use sha2::{Digest, Sha256};
571        let mut hasher = Sha256::new();
572        hasher.update(input);
573        hex::encode(hasher.finalize())
574    })?;
575    crypto_obj.set("sha256", sha256_func)?;
576
577    let random_bytes_func = Function::new(ctx.clone(), |length: usize| -> String {
578        use rand::Rng;
579        let mut rng = rand::thread_rng();
580        let bytes: Vec<u8> = (0..length).map(|_| rng.random()).collect();
581        hex::encode(bytes)
582    })?;
583    crypto_obj.set("randomBytes", random_bytes_func)?;
584
585    global.set("crypto", crypto_obj)?;
586
587    // Add date/time utilities
588    let date_obj = Object::new(ctx.clone())?;
589
590    let now_func = Function::new(ctx.clone(), || -> String { chrono::Utc::now().to_rfc3339() })?;
591    date_obj.set("now", now_func)?;
592
593    let format_func = Function::new(ctx.clone(), |timestamp: String, format: String| -> String {
594        if let Ok(dt) = chrono::DateTime::parse_from_rfc3339(&timestamp) {
595            dt.format(&format).to_string()
596        } else {
597            "".to_string()
598        }
599    })?;
600    date_obj.set("format", format_func)?;
601
602    let parse_func = Function::new(ctx.clone(), |date_str: String, format: String| -> String {
603        if let Ok(dt) = chrono::NaiveDateTime::parse_from_str(&date_str, &format) {
604            dt.and_utc().to_rfc3339()
605        } else {
606            "".to_string()
607        }
608    })?;
609    date_obj.set("parse", parse_func)?;
610
611    let add_days_func = Function::new(ctx.clone(), |timestamp: String, days: i64| -> String {
612        if let Ok(dt) = chrono::DateTime::parse_from_rfc3339(&timestamp) {
613            (dt + chrono::Duration::days(days)).to_rfc3339()
614        } else {
615            "".to_string()
616        }
617    })?;
618    date_obj.set("addDays", add_days_func)?;
619
620    global.set("date", date_obj)?;
621
622    // Add validation utilities
623    let validate_obj = Object::new(ctx.clone())?;
624
625    let email_func = Function::new(ctx.clone(), |email: String| -> bool {
626        // Simple email regex validation
627        // Note: This regex pattern is static and should never fail compilation,
628        // but we handle errors defensively to prevent panics
629        regex::Regex::new(r"^[^@]+@[^@]+\.[^@]+$")
630            .map(|re| re.is_match(&email))
631            .unwrap_or_else(|_| {
632                // Fallback: basic string check if regex compilation fails (should never happen)
633                email.contains('@') && email.contains('.') && email.len() > 5
634            })
635    })?;
636    validate_obj.set("email", email_func)?;
637
638    let url_func = Function::new(ctx.clone(), |url_str: String| -> bool {
639        url::Url::parse(&url_str).is_ok()
640    })?;
641    validate_obj.set("url", url_func)?;
642
643    let regex_func = Function::new(ctx.clone(), |pattern: String, text: String| -> bool {
644        regex::Regex::new(&pattern).map(|re| re.is_match(&text)).unwrap_or(false)
645    })?;
646    validate_obj.set("regex", regex_func)?;
647
648    global.set("validate", validate_obj)?;
649
650    // Add JSON utilities
651    let json_obj = Object::new(ctx.clone())?;
652
653    let json_parse_func = Function::new(ctx.clone(), |json_str: String| -> String {
654        match serde_json::from_str::<serde_json::Value>(&json_str) {
655            Ok(value) => serde_json::to_string(&value).unwrap_or_else(|_| "null".to_string()),
656            Err(_) => "null".to_string(),
657        }
658    })?;
659    json_obj.set("parse", json_parse_func)?;
660
661    let json_stringify_func = Function::new(ctx.clone(), |value: String| -> String {
662        // Assume input is already valid JSON or a simple value
663        value
664    })?;
665    json_obj.set("stringify", json_stringify_func)?;
666
667    let json_validate_func = Function::new(ctx.clone(), |json_str: String| -> bool {
668        serde_json::from_str::<serde_json::Value>(&json_str).is_ok()
669    })?;
670    json_obj.set("validate", json_validate_func)?;
671
672    global.set("JSON", json_obj)?;
673
674    // Add HTTP utilities
675    let http_obj = Object::new(ctx.clone())?;
676
677    let http_get_func = Function::new(ctx.clone(), |url: String| -> String {
678        // WARNING: This blocks a thread from the blocking thread pool.
679        // The JavaScript engine (rquickjs) is already running in spawn_blocking,
680        // so we use block_in_place here. For production, consider limiting
681        // HTTP calls in scripts or using a different scripting approach.
682        tokio::task::block_in_place(|| {
683            reqwest::blocking::get(&url)
684                .and_then(|resp| resp.text())
685                .unwrap_or_else(|_| "".to_string())
686        })
687    })?;
688    http_obj.set("get", http_get_func)?;
689
690    let http_post_func = Function::new(ctx.clone(), |url: String, body: String| -> String {
691        // WARNING: This blocks a thread from the blocking thread pool.
692        // The JavaScript engine (rquickjs) is already running in spawn_blocking,
693        // so we use block_in_place here. For production, consider limiting
694        // HTTP calls in scripts or using a different scripting approach.
695        tokio::task::block_in_place(|| {
696            reqwest::blocking::Client::new()
697                .post(&url)
698                .body(body)
699                .send()
700                .and_then(|resp| resp.text())
701                .unwrap_or_else(|_| "".to_string())
702        })
703    })?;
704    http_obj.set("post", http_post_func)?;
705
706    let url_encode_func = Function::new(ctx.clone(), |input: String| -> String {
707        urlencoding::encode(&input).to_string()
708    })?;
709    http_obj.set("urlEncode", url_encode_func)?;
710
711    let url_decode_func = Function::new(ctx.clone(), |input: String| -> String {
712        urlencoding::decode(&input)
713            .unwrap_or(std::borrow::Cow::Borrowed(""))
714            .to_string()
715    })?;
716    http_obj.set("urlDecode", url_decode_func)?;
717
718    global.set("http", http_obj)?;
719
720    Ok(())
721}
722
723#[cfg(test)]
724mod tests {
725    use super::*;
726    use chrono::Utc;
727    use serde_json::json;
728
729    fn create_empty_script_context() -> ScriptContext {
730        ScriptContext {
731            request: None,
732            response: None,
733            chain_context: HashMap::new(),
734            variables: HashMap::new(),
735            env_vars: HashMap::new(),
736        }
737    }
738
739    fn create_full_script_context() -> ScriptContext {
740        ScriptContext {
741            request: Some(crate::request_chaining::ChainRequest {
742                id: "test-request".to_string(),
743                method: "GET".to_string(),
744                url: "https://api.example.com/test".to_string(),
745                headers: [("Content-Type".to_string(), "application/json".to_string())].into(),
746                body: Some(crate::request_chaining::RequestBody::Json(json!({"key": "value"}))),
747                depends_on: vec![],
748                timeout_secs: Some(30),
749                expected_status: Some(vec![200]),
750                scripting: None,
751            }),
752            response: Some(crate::request_chaining::ChainResponse {
753                status: 200,
754                headers: [("Content-Type".to_string(), "application/json".to_string())].into(),
755                body: Some(json!({"result": "success"})),
756                duration_ms: 150,
757                executed_at: chrono::Utc::now().to_rfc3339(),
758                error: None,
759            }),
760            chain_context: {
761                let mut ctx = HashMap::new();
762                ctx.insert("login_token".to_string(), json!("abc123"));
763                ctx.insert("user_id".to_string(), json!(42));
764                ctx.insert("is_admin".to_string(), json!(true));
765                ctx.insert("items".to_string(), json!(["a", "b", "c"]));
766                ctx.insert("config".to_string(), json!({"timeout": 30}));
767                ctx
768            },
769            variables: {
770                let mut vars = HashMap::new();
771                vars.insert("counter".to_string(), json!(0));
772                vars.insert("name".to_string(), json!("test"));
773                vars
774            },
775            env_vars: [
776                ("NODE_ENV".to_string(), "test".to_string()),
777                ("API_KEY".to_string(), "secret123".to_string()),
778            ]
779            .into(),
780        }
781    }
782
783    // ScriptResult tests
784    #[test]
785    fn test_script_result_clone() {
786        let result = ScriptResult {
787            return_value: Some(json!("test")),
788            modified_variables: {
789                let mut vars = HashMap::new();
790                vars.insert("key".to_string(), json!("value"));
791                vars
792            },
793            errors: vec!["error1".to_string()],
794            execution_time_ms: 100,
795        };
796
797        let cloned = result.clone();
798        assert_eq!(cloned.return_value, result.return_value);
799        assert_eq!(cloned.modified_variables, result.modified_variables);
800        assert_eq!(cloned.errors, result.errors);
801        assert_eq!(cloned.execution_time_ms, result.execution_time_ms);
802    }
803
804    #[test]
805    fn test_script_result_debug() {
806        let result = ScriptResult {
807            return_value: Some(json!("test")),
808            modified_variables: HashMap::new(),
809            errors: vec![],
810            execution_time_ms: 50,
811        };
812
813        let debug = format!("{:?}", result);
814        assert!(debug.contains("ScriptResult"));
815        assert!(debug.contains("return_value"));
816    }
817
818    #[test]
819    fn test_script_result_serialize() {
820        let result = ScriptResult {
821            return_value: Some(json!("test")),
822            modified_variables: HashMap::new(),
823            errors: vec![],
824            execution_time_ms: 50,
825        };
826
827        let json = serde_json::to_string(&result).unwrap();
828        assert!(json.contains("return_value"));
829        assert!(json.contains("execution_time_ms"));
830    }
831
832    #[test]
833    fn test_script_result_deserialize() {
834        let json =
835            r#"{"return_value":"test","modified_variables":{},"errors":[],"execution_time_ms":50}"#;
836        let result: ScriptResult = serde_json::from_str(json).unwrap();
837        assert_eq!(result.return_value, Some(json!("test")));
838        assert_eq!(result.execution_time_ms, 50);
839    }
840
841    // ScriptContext tests
842    #[test]
843    fn test_script_context_clone() {
844        let ctx = create_full_script_context();
845        let cloned = ctx.clone();
846
847        assert_eq!(cloned.request.is_some(), ctx.request.is_some());
848        assert_eq!(cloned.response.is_some(), ctx.response.is_some());
849        assert_eq!(cloned.chain_context.len(), ctx.chain_context.len());
850        assert_eq!(cloned.variables.len(), ctx.variables.len());
851        assert_eq!(cloned.env_vars.len(), ctx.env_vars.len());
852    }
853
854    #[test]
855    fn test_script_context_debug() {
856        let ctx = create_empty_script_context();
857        let debug = format!("{:?}", ctx);
858        assert!(debug.contains("ScriptContext"));
859    }
860
861    // ScriptEngine tests
862    #[test]
863    fn test_script_engine_new() {
864        let engine = ScriptEngine::new();
865        // Verify engine is created successfully
866        let debug = format!("{:?}", engine);
867        assert!(debug.contains("ScriptEngine"));
868        assert!(debug.contains("Semaphore"));
869    }
870
871    #[test]
872    fn test_script_engine_default() {
873        let engine = ScriptEngine::default();
874        let debug = format!("{:?}", engine);
875        assert!(debug.contains("ScriptEngine"));
876    }
877
878    #[test]
879    fn test_script_engine_debug() {
880        let engine = ScriptEngine::new();
881        let debug = format!("{:?}", engine);
882        assert!(debug.contains("ScriptEngine"));
883        // Should show semaphore permits
884        assert!(debug.contains("10")); // Default 10 permits
885    }
886
887    #[tokio::test]
888    async fn test_script_execution() {
889        let engine = ScriptEngine::new();
890
891        let script_context = ScriptContext {
892            request: Some(crate::request_chaining::ChainRequest {
893                id: "test-request".to_string(),
894                method: "GET".to_string(),
895                url: "https://api.example.com/test".to_string(),
896                headers: [("Content-Type".to_string(), "application/json".to_string())].into(),
897                body: None,
898                depends_on: vec![],
899                timeout_secs: None,
900                expected_status: None,
901                scripting: None,
902            }),
903            response: None,
904            chain_context: {
905                let mut ctx = HashMap::new();
906                ctx.insert("login_token".to_string(), json!("abc123"));
907                ctx
908            },
909            variables: HashMap::new(),
910            env_vars: [("NODE_ENV".to_string(), "test".to_string())].into(),
911        };
912
913        let script = r#"
914            for (let i = 0; i < 1000000; i++) {
915                // Loop to ensure measurable execution time
916            }
917            "script executed successfully";
918        "#;
919
920        let result = engine.execute_script(script, &script_context, 5000).await;
921        assert!(result.is_ok(), "Script execution should succeed");
922
923        let script_result = result.unwrap();
924        assert_eq!(script_result.return_value, Some(json!("script executed successfully")));
925        assert!(script_result.execution_time_ms > 0);
926        assert!(script_result.errors.is_empty());
927    }
928
929    #[tokio::test]
930    async fn test_script_with_error() {
931        let engine = ScriptEngine::new();
932
933        let script_context = ScriptContext {
934            request: None,
935            response: None,
936            chain_context: HashMap::new(),
937            variables: HashMap::new(),
938            env_vars: HashMap::new(),
939        };
940
941        let script = r#"throw new Error("Intentional test error");"#;
942
943        let result = engine.execute_script(script, &script_context, 1000).await;
944        // For now, JavaScript errors are not being caught properly
945        // In a complete implementation, we would handle errors and return them in ScriptResult.errors
946        assert!(result.is_err() || result.is_ok()); // Accept either for now
947    }
948
949    #[tokio::test]
950    async fn test_simple_script_string_return() {
951        let engine = ScriptEngine::new();
952        let ctx = create_empty_script_context();
953
954        let result = engine.execute_script(r#""hello world""#, &ctx, 1000).await;
955        assert!(result.is_ok());
956        assert_eq!(result.unwrap().return_value, Some(json!("hello world")));
957    }
958
959    #[tokio::test]
960    async fn test_simple_script_number_return() {
961        let engine = ScriptEngine::new();
962        let ctx = create_empty_script_context();
963
964        let result = engine.execute_script("42", &ctx, 1000).await;
965        assert!(result.is_ok());
966        // Number may or may not be returned depending on JS engine behavior
967        // The important thing is the script executed successfully
968    }
969
970    #[tokio::test]
971    async fn test_simple_script_boolean_return() {
972        let engine = ScriptEngine::new();
973        let ctx = create_empty_script_context();
974
975        let result = engine.execute_script("true", &ctx, 1000).await;
976        assert!(result.is_ok());
977        assert_eq!(result.unwrap().return_value, Some(json!(true)));
978    }
979
980    #[tokio::test]
981    async fn test_script_timeout() {
982        let engine = ScriptEngine::new();
983        let ctx = create_empty_script_context();
984
985        // Script that takes a long time
986        let script = r#"
987            let count = 0;
988            while (count < 100000000) {
989                count++;
990            }
991            count;
992        "#;
993
994        let result = engine.execute_script(script, &ctx, 10).await;
995        // Should either timeout or take a long time
996        // The actual behavior depends on the implementation
997        assert!(result.is_ok() || result.is_err());
998    }
999
1000    #[tokio::test]
1001    async fn test_script_with_request_context() {
1002        let engine = ScriptEngine::new();
1003        let ctx = create_full_script_context();
1004
1005        // Script that accesses request data
1006        let script = r#"
1007            mockforge.request.method;
1008        "#;
1009
1010        let result = engine.execute_script(script, &ctx, 1000).await;
1011        assert!(result.is_ok());
1012        assert_eq!(result.unwrap().return_value, Some(json!("GET")));
1013    }
1014
1015    #[tokio::test]
1016    async fn test_script_with_response_context() {
1017        let engine = ScriptEngine::new();
1018        let ctx = create_full_script_context();
1019
1020        // Script that accesses response data
1021        let script = r#"
1022            mockforge.response.status;
1023        "#;
1024
1025        let result = engine.execute_script(script, &ctx, 1000).await;
1026        assert!(result.is_ok());
1027    }
1028
1029    #[tokio::test]
1030    async fn test_script_with_chain_context() {
1031        let engine = ScriptEngine::new();
1032        let ctx = create_full_script_context();
1033
1034        // Script that accesses chain context
1035        let script = r#"
1036            mockforge.chain.login_token;
1037        "#;
1038
1039        let result = engine.execute_script(script, &ctx, 1000).await;
1040        assert!(result.is_ok());
1041        assert_eq!(result.unwrap().return_value, Some(json!("abc123")));
1042    }
1043
1044    #[tokio::test]
1045    async fn test_script_with_env_vars() {
1046        let engine = ScriptEngine::new();
1047        let ctx = create_full_script_context();
1048
1049        // Script that accesses environment variables
1050        let script = r#"
1051            mockforge.env.NODE_ENV;
1052        "#;
1053
1054        let result = engine.execute_script(script, &ctx, 1000).await;
1055        assert!(result.is_ok());
1056        assert_eq!(result.unwrap().return_value, Some(json!("test")));
1057    }
1058
1059    #[tokio::test]
1060    async fn test_script_modify_variables() {
1061        let engine = ScriptEngine::new();
1062        let mut ctx = create_empty_script_context();
1063        ctx.variables.insert("counter".to_string(), json!(0));
1064
1065        // Script that modifies a variable
1066        let script = r#"
1067            mockforge.variables.counter = 10;
1068            mockforge.variables.new_var = "created";
1069            mockforge.variables.counter;
1070        "#;
1071
1072        let result = engine.execute_script(script, &ctx, 1000).await;
1073        assert!(result.is_ok());
1074        let script_result = result.unwrap();
1075        // Check if modified_variables contains the changes
1076        assert!(
1077            script_result.modified_variables.contains_key("counter")
1078                || script_result.modified_variables.contains_key("new_var")
1079        );
1080    }
1081
1082    #[tokio::test]
1083    async fn test_script_console_log() {
1084        let engine = ScriptEngine::new();
1085        let ctx = create_empty_script_context();
1086
1087        // Script that uses console.log
1088        let script = r#"
1089            console.log("test message");
1090            "logged";
1091        "#;
1092
1093        let result = engine.execute_script(script, &ctx, 1000).await;
1094        assert!(result.is_ok());
1095    }
1096
1097    #[tokio::test]
1098    async fn test_script_log_function() {
1099        let engine = ScriptEngine::new();
1100        let ctx = create_empty_script_context();
1101
1102        // Script that uses the global log function
1103        let script = r#"
1104            log("test log");
1105            "logged";
1106        "#;
1107
1108        let result = engine.execute_script(script, &ctx, 1000).await;
1109        assert!(result.is_ok());
1110    }
1111
1112    #[tokio::test]
1113    async fn test_script_crypto_base64() {
1114        let engine = ScriptEngine::new();
1115        let ctx = create_empty_script_context();
1116
1117        // Script that uses base64 encoding
1118        let script = r#"
1119            crypto.base64Encode("hello");
1120        "#;
1121
1122        let result = engine.execute_script(script, &ctx, 1000).await;
1123        assert!(result.is_ok());
1124        // base64("hello") = "aGVsbG8="
1125        assert_eq!(result.unwrap().return_value, Some(json!("aGVsbG8=")));
1126    }
1127
1128    #[tokio::test]
1129    async fn test_script_crypto_base64_decode() {
1130        let engine = ScriptEngine::new();
1131        let ctx = create_empty_script_context();
1132
1133        // Script that uses base64 decoding
1134        let script = r#"
1135            crypto.base64Decode("aGVsbG8=");
1136        "#;
1137
1138        let result = engine.execute_script(script, &ctx, 1000).await;
1139        assert!(result.is_ok());
1140        assert_eq!(result.unwrap().return_value, Some(json!("hello")));
1141    }
1142
1143    #[tokio::test]
1144    async fn test_script_crypto_sha256() {
1145        let engine = ScriptEngine::new();
1146        let ctx = create_empty_script_context();
1147
1148        // Script that uses SHA256
1149        let script = r#"
1150            crypto.sha256("hello");
1151        "#;
1152
1153        let result = engine.execute_script(script, &ctx, 1000).await;
1154        assert!(result.is_ok());
1155        // SHA256("hello") = "2cf24dba5fb0a30e26e83b2ac5b9e29e1b161e5c1fa7425e73043362938b9824"
1156        let return_val = result.unwrap().return_value;
1157        assert!(return_val.is_some());
1158        let hash = return_val.unwrap();
1159        assert!(hash.as_str().unwrap().len() == 64); // SHA256 produces 64 hex chars
1160    }
1161
1162    #[tokio::test]
1163    async fn test_script_crypto_random_bytes() {
1164        let engine = ScriptEngine::new();
1165        let ctx = create_empty_script_context();
1166
1167        // Script that generates random bytes
1168        let script = r#"
1169            crypto.randomBytes(16);
1170        "#;
1171
1172        let result = engine.execute_script(script, &ctx, 1000).await;
1173        assert!(result.is_ok());
1174        let return_val = result.unwrap().return_value;
1175        assert!(return_val.is_some());
1176        let hex = return_val.unwrap();
1177        assert!(hex.as_str().unwrap().len() == 32); // 16 bytes = 32 hex chars
1178    }
1179
1180    #[tokio::test]
1181    async fn test_script_date_now() {
1182        let engine = ScriptEngine::new();
1183        let ctx = create_empty_script_context();
1184
1185        // Script that gets current date
1186        let script = r#"
1187            date.now();
1188        "#;
1189
1190        let result = engine.execute_script(script, &ctx, 1000).await;
1191        assert!(result.is_ok());
1192        let return_val = result.unwrap().return_value;
1193        assert!(return_val.is_some());
1194        // Should be an RFC3339 timestamp
1195        let timestamp = return_val.unwrap();
1196        assert!(timestamp.as_str().unwrap().contains("T"));
1197    }
1198
1199    #[tokio::test]
1200    async fn test_script_date_add_days() {
1201        let engine = ScriptEngine::new();
1202        let ctx = create_empty_script_context();
1203
1204        // Script that adds days to a date
1205        let script = r#"
1206            date.addDays("2024-01-01T00:00:00+00:00", 1);
1207        "#;
1208
1209        let result = engine.execute_script(script, &ctx, 1000).await;
1210        assert!(result.is_ok());
1211        let return_val = result.unwrap().return_value;
1212        assert!(return_val.is_some());
1213        let new_date = return_val.unwrap();
1214        assert!(new_date.as_str().unwrap().contains("2024-01-02"));
1215    }
1216
1217    #[tokio::test]
1218    async fn test_script_validate_email() {
1219        let engine = ScriptEngine::new();
1220        let ctx = create_empty_script_context();
1221
1222        // Valid email
1223        let script = r#"validate.email("test@example.com");"#;
1224        let result = engine.execute_script(script, &ctx, 1000).await;
1225        assert!(result.is_ok());
1226        assert_eq!(result.unwrap().return_value, Some(json!(true)));
1227
1228        // Invalid email
1229        let script = r#"validate.email("not-an-email");"#;
1230        let result = engine.execute_script(script, &ctx, 1000).await;
1231        assert!(result.is_ok());
1232        assert_eq!(result.unwrap().return_value, Some(json!(false)));
1233    }
1234
1235    #[tokio::test]
1236    async fn test_script_validate_url() {
1237        let engine = ScriptEngine::new();
1238        let ctx = create_empty_script_context();
1239
1240        // Valid URL
1241        let script = r#"validate.url("https://example.com");"#;
1242        let result = engine.execute_script(script, &ctx, 1000).await;
1243        assert!(result.is_ok());
1244        assert_eq!(result.unwrap().return_value, Some(json!(true)));
1245
1246        // Invalid URL
1247        let script = r#"validate.url("not-a-url");"#;
1248        let result = engine.execute_script(script, &ctx, 1000).await;
1249        assert!(result.is_ok());
1250        assert_eq!(result.unwrap().return_value, Some(json!(false)));
1251    }
1252
1253    #[tokio::test]
1254    async fn test_script_validate_regex() {
1255        let engine = ScriptEngine::new();
1256        let ctx = create_empty_script_context();
1257
1258        // Matching regex
1259        let script = r#"validate.regex("^hello", "hello world");"#;
1260        let result = engine.execute_script(script, &ctx, 1000).await;
1261        assert!(result.is_ok());
1262        assert_eq!(result.unwrap().return_value, Some(json!(true)));
1263
1264        // Non-matching regex
1265        let script = r#"validate.regex("^world", "hello world");"#;
1266        let result = engine.execute_script(script, &ctx, 1000).await;
1267        assert!(result.is_ok());
1268        assert_eq!(result.unwrap().return_value, Some(json!(false)));
1269    }
1270
1271    #[tokio::test]
1272    async fn test_script_json_parse() {
1273        let engine = ScriptEngine::new();
1274        let ctx = create_empty_script_context();
1275
1276        let script = r#"JSON.parse('{"key": "value"}');"#;
1277        let result = engine.execute_script(script, &ctx, 1000).await;
1278        assert!(result.is_ok());
1279    }
1280
1281    #[tokio::test]
1282    async fn test_script_json_validate() {
1283        let engine = ScriptEngine::new();
1284        let ctx = create_empty_script_context();
1285
1286        // Valid JSON
1287        let script = r#"JSON.validate('{"key": "value"}');"#;
1288        let result = engine.execute_script(script, &ctx, 1000).await;
1289        assert!(result.is_ok());
1290        assert_eq!(result.unwrap().return_value, Some(json!(true)));
1291
1292        // Invalid JSON
1293        let script = r#"JSON.validate('not json');"#;
1294        let result = engine.execute_script(script, &ctx, 1000).await;
1295        assert!(result.is_ok());
1296        assert_eq!(result.unwrap().return_value, Some(json!(false)));
1297    }
1298
1299    #[tokio::test]
1300    async fn test_script_http_url_encode() {
1301        let engine = ScriptEngine::new();
1302        let ctx = create_empty_script_context();
1303
1304        let script = r#"http.urlEncode("hello world");"#;
1305        let result = engine.execute_script(script, &ctx, 1000).await;
1306        assert!(result.is_ok());
1307        assert_eq!(result.unwrap().return_value, Some(json!("hello%20world")));
1308    }
1309
1310    #[tokio::test]
1311    async fn test_script_http_url_decode() {
1312        let engine = ScriptEngine::new();
1313        let ctx = create_empty_script_context();
1314
1315        let script = r#"http.urlDecode("hello%20world");"#;
1316        let result = engine.execute_script(script, &ctx, 1000).await;
1317        assert!(result.is_ok());
1318        assert_eq!(result.unwrap().return_value, Some(json!("hello world")));
1319    }
1320
1321    #[tokio::test]
1322    async fn test_script_with_syntax_error() {
1323        let engine = ScriptEngine::new();
1324        let ctx = create_empty_script_context();
1325
1326        // Script with syntax error
1327        let script = r#"function { broken"#;
1328        let result = engine.execute_script(script, &ctx, 1000).await;
1329        assert!(result.is_err());
1330    }
1331
1332    #[tokio::test]
1333    async fn test_execute_in_context_blocking() {
1334        let engine = ScriptEngine::new();
1335        let ctx = create_empty_script_context();
1336
1337        let result = engine.execute_in_context_blocking(r#""test""#, &ctx);
1338        assert!(result.is_ok());
1339        assert_eq!(result.unwrap().return_value, Some(json!("test")));
1340    }
1341
1342    #[tokio::test]
1343    async fn test_script_with_no_request() {
1344        let engine = ScriptEngine::new();
1345        let ctx = create_empty_script_context();
1346
1347        // Script that doesn't access request
1348        let script = r#""no request needed""#;
1349        let result = engine.execute_script(script, &ctx, 1000).await;
1350        assert!(result.is_ok());
1351    }
1352
1353    #[tokio::test]
1354    async fn test_script_with_no_response() {
1355        let engine = ScriptEngine::new();
1356        let mut ctx = create_empty_script_context();
1357        ctx.request = Some(crate::request_chaining::ChainRequest {
1358            id: "test".to_string(),
1359            method: "GET".to_string(),
1360            url: "http://example.com".to_string(),
1361            headers: HashMap::new(),
1362            body: None,
1363            depends_on: vec![],
1364            timeout_secs: None,
1365            expected_status: None,
1366            scripting: None,
1367        });
1368
1369        // Script that only uses request (pre-script scenario)
1370        let script = r#"mockforge.request.method"#;
1371        let result = engine.execute_script(script, &ctx, 1000).await;
1372        assert!(result.is_ok());
1373    }
1374
1375    #[tokio::test]
1376    async fn test_concurrent_script_execution() {
1377        let engine = std::sync::Arc::new(ScriptEngine::new());
1378        let ctx = create_empty_script_context();
1379
1380        // Run multiple scripts concurrently
1381        let mut handles = vec![];
1382        for i in 0..5 {
1383            let engine = engine.clone();
1384            let ctx = ctx.clone();
1385            let handle = tokio::spawn(async move {
1386                let script = format!("{}", i);
1387                engine.execute_script(&script, &ctx, 1000).await
1388            });
1389            handles.push(handle);
1390        }
1391
1392        for handle in handles {
1393            let result = handle.await.unwrap();
1394            assert!(result.is_ok());
1395        }
1396    }
1397
1398    // Test js_value_to_json_value helper
1399    #[test]
1400    fn test_execute_script_in_runtime_success() {
1401        let ctx = create_empty_script_context();
1402        let result = execute_script_in_runtime(r#""hello""#, &ctx);
1403        assert!(result.is_ok());
1404        assert_eq!(result.unwrap().return_value, Some(json!("hello")));
1405    }
1406
1407    #[test]
1408    fn test_execute_script_in_runtime_with_context() {
1409        let ctx = create_full_script_context();
1410        let result = execute_script_in_runtime(r#"mockforge.request.method"#, &ctx);
1411        assert!(result.is_ok());
1412    }
1413
1414    #[test]
1415    fn test_execute_script_in_runtime_error() {
1416        let ctx = create_empty_script_context();
1417        let result = execute_script_in_runtime(r#"throw new Error("test");"#, &ctx);
1418        assert!(result.is_err());
1419    }
1420
1421    // Test chain context with different value types
1422    #[tokio::test]
1423    async fn test_script_chain_context_number() {
1424        let engine = ScriptEngine::new();
1425        let ctx = create_full_script_context();
1426
1427        let script = r#"mockforge.chain.user_id;"#;
1428        let result = engine.execute_script(script, &ctx, 1000).await;
1429        assert!(result.is_ok());
1430    }
1431
1432    #[tokio::test]
1433    async fn test_script_chain_context_boolean() {
1434        let engine = ScriptEngine::new();
1435        let ctx = create_full_script_context();
1436
1437        let script = r#"mockforge.chain.is_admin;"#;
1438        let result = engine.execute_script(script, &ctx, 1000).await;
1439        assert!(result.is_ok());
1440        assert_eq!(result.unwrap().return_value, Some(json!(true)));
1441    }
1442
1443    // Test variables with different value types
1444    #[tokio::test]
1445    async fn test_script_variables_number() {
1446        let engine = ScriptEngine::new();
1447        let ctx = create_full_script_context();
1448
1449        let script = r#"mockforge.variables.counter;"#;
1450        let result = engine.execute_script(script, &ctx, 1000).await;
1451        assert!(result.is_ok());
1452    }
1453
1454    #[tokio::test]
1455    async fn test_script_variables_string() {
1456        let engine = ScriptEngine::new();
1457        let ctx = create_full_script_context();
1458
1459        let script = r#"mockforge.variables.name;"#;
1460        let result = engine.execute_script(script, &ctx, 1000).await;
1461        assert!(result.is_ok());
1462        assert_eq!(result.unwrap().return_value, Some(json!("test")));
1463    }
1464
1465    #[tokio::test]
1466    async fn test_script_arithmetic() {
1467        let engine = ScriptEngine::new();
1468        let ctx = create_empty_script_context();
1469
1470        let script = r#"1 + 2 + 3"#;
1471        let result = engine.execute_script(script, &ctx, 1000).await;
1472        assert!(result.is_ok());
1473    }
1474
1475    #[tokio::test]
1476    async fn test_script_string_concatenation() {
1477        let engine = ScriptEngine::new();
1478        let ctx = create_empty_script_context();
1479
1480        let script = r#""hello" + " " + "world""#;
1481        let result = engine.execute_script(script, &ctx, 1000).await;
1482        assert!(result.is_ok());
1483        assert_eq!(result.unwrap().return_value, Some(json!("hello world")));
1484    }
1485
1486    #[tokio::test]
1487    async fn test_script_conditional() {
1488        let engine = ScriptEngine::new();
1489        let ctx = create_empty_script_context();
1490
1491        let script = r#"true ? "yes" : "no""#;
1492        let result = engine.execute_script(script, &ctx, 1000).await;
1493        assert!(result.is_ok());
1494        assert_eq!(result.unwrap().return_value, Some(json!("yes")));
1495    }
1496
1497    #[tokio::test]
1498    async fn test_script_function_definition_and_call() {
1499        let engine = ScriptEngine::new();
1500        let ctx = create_empty_script_context();
1501
1502        let script = r#"
1503            function add(a, b) {
1504                return a + b;
1505            }
1506            add(1, 2);
1507        "#;
1508        let result = engine.execute_script(script, &ctx, 1000).await;
1509        assert!(result.is_ok());
1510    }
1511
1512    #[tokio::test]
1513    async fn test_script_arrow_function() {
1514        let engine = ScriptEngine::new();
1515        let ctx = create_empty_script_context();
1516
1517        let script = r#"
1518            const multiply = (a, b) => a * b;
1519            multiply(3, 4);
1520        "#;
1521        let result = engine.execute_script(script, &ctx, 1000).await;
1522        assert!(result.is_ok());
1523    }
1524
1525    #[tokio::test]
1526    async fn test_script_array_operations() {
1527        let engine = ScriptEngine::new();
1528        let ctx = create_empty_script_context();
1529
1530        let script = r#"
1531            const arr = [1, 2, 3];
1532            arr.length;
1533        "#;
1534        let result = engine.execute_script(script, &ctx, 1000).await;
1535        assert!(result.is_ok());
1536    }
1537
1538    #[tokio::test]
1539    async fn test_script_object_access() {
1540        let engine = ScriptEngine::new();
1541        let ctx = create_empty_script_context();
1542
1543        let script = r#"
1544            const obj = {key: "value"};
1545            obj.key;
1546        "#;
1547        let result = engine.execute_script(script, &ctx, 1000).await;
1548        assert!(result.is_ok());
1549        assert_eq!(result.unwrap().return_value, Some(json!("value")));
1550    }
1551
1552    #[tokio::test]
1553    async fn test_date_format() {
1554        let engine = ScriptEngine::new();
1555        let ctx = create_empty_script_context();
1556
1557        let script = r#"date.format("2024-01-15T10:30:00+00:00", "%Y-%m-%d");"#;
1558        let result = engine.execute_script(script, &ctx, 1000).await;
1559        assert!(result.is_ok());
1560        assert_eq!(result.unwrap().return_value, Some(json!("2024-01-15")));
1561    }
1562
1563    #[tokio::test]
1564    async fn test_date_parse() {
1565        let engine = ScriptEngine::new();
1566        let ctx = create_empty_script_context();
1567
1568        let script = r#"date.parse("2024-01-15 10:30:00", "%Y-%m-%d %H:%M:%S");"#;
1569        let result = engine.execute_script(script, &ctx, 1000).await;
1570        assert!(result.is_ok());
1571        let return_val = result.unwrap().return_value;
1572        assert!(return_val.is_some());
1573        // Should return RFC3339 formatted timestamp
1574        assert!(return_val.unwrap().as_str().unwrap().contains("2024-01-15"));
1575    }
1576
1577    #[tokio::test]
1578    async fn test_date_parse_invalid() {
1579        let engine = ScriptEngine::new();
1580        let ctx = create_empty_script_context();
1581
1582        let script = r#"date.parse("invalid", "%Y-%m-%d");"#;
1583        let result = engine.execute_script(script, &ctx, 1000).await;
1584        assert!(result.is_ok());
1585        // Should return empty string for invalid date
1586        assert_eq!(result.unwrap().return_value, Some(json!("")));
1587    }
1588
1589    #[tokio::test]
1590    async fn test_validate_regex_invalid_pattern() {
1591        let engine = ScriptEngine::new();
1592        let ctx = create_empty_script_context();
1593
1594        // Invalid regex pattern
1595        let script = r#"validate.regex("[invalid", "test");"#;
1596        let result = engine.execute_script(script, &ctx, 1000).await;
1597        assert!(result.is_ok());
1598        // Should return false for invalid regex
1599        assert_eq!(result.unwrap().return_value, Some(json!(false)));
1600    }
1601
1602    #[tokio::test]
1603    async fn test_script_stringify_function() {
1604        let engine = ScriptEngine::new();
1605        let ctx = create_empty_script_context();
1606
1607        let script = r#"stringify("test");"#;
1608        let result = engine.execute_script(script, &ctx, 1000).await;
1609        assert!(result.is_ok());
1610    }
1611
1612    #[tokio::test]
1613    async fn test_crypto_base64_decode_invalid() {
1614        let engine = ScriptEngine::new();
1615        let ctx = create_empty_script_context();
1616
1617        // Invalid base64
1618        let script = r#"crypto.base64Decode("!!invalid!!");"#;
1619        let result = engine.execute_script(script, &ctx, 1000).await;
1620        assert!(result.is_ok());
1621        // Should return empty string for invalid base64
1622        assert_eq!(result.unwrap().return_value, Some(json!("")));
1623    }
1624
1625    #[tokio::test]
1626    async fn test_date_add_days_invalid() {
1627        let engine = ScriptEngine::new();
1628        let ctx = create_empty_script_context();
1629
1630        // Invalid timestamp
1631        let script = r#"date.addDays("invalid", 1);"#;
1632        let result = engine.execute_script(script, &ctx, 1000).await;
1633        assert!(result.is_ok());
1634        // Should return empty string for invalid timestamp
1635        assert_eq!(result.unwrap().return_value, Some(json!("")));
1636    }
1637
1638    #[tokio::test]
1639    async fn test_date_format_invalid() {
1640        let engine = ScriptEngine::new();
1641        let ctx = create_empty_script_context();
1642
1643        // Invalid timestamp
1644        let script = r#"date.format("invalid", "%Y-%m-%d");"#;
1645        let result = engine.execute_script(script, &ctx, 1000).await;
1646        assert!(result.is_ok());
1647        // Should return empty string for invalid timestamp
1648        assert_eq!(result.unwrap().return_value, Some(json!("")));
1649    }
1650
1651    #[tokio::test]
1652    async fn test_http_url_encode_special_chars() {
1653        let engine = ScriptEngine::new();
1654        let ctx = create_empty_script_context();
1655
1656        let script = r#"http.urlEncode("a=b&c=d");"#;
1657        let result = engine.execute_script(script, &ctx, 1000).await;
1658        assert!(result.is_ok());
1659        let encoded = result.unwrap().return_value.unwrap();
1660        assert!(encoded.as_str().unwrap().contains("%3D")); // = encoded
1661        assert!(encoded.as_str().unwrap().contains("%26")); // & encoded
1662    }
1663
1664    #[tokio::test]
1665    async fn test_json_parse_invalid() {
1666        let engine = ScriptEngine::new();
1667        let ctx = create_empty_script_context();
1668
1669        let script = r#"JSON.parse("invalid json");"#;
1670        let result = engine.execute_script(script, &ctx, 1000).await;
1671        assert!(result.is_ok());
1672        // Should return "null" for invalid JSON
1673        assert_eq!(result.unwrap().return_value, Some(json!("null")));
1674    }
1675
1676    #[tokio::test]
1677    async fn test_script_with_complex_chain_context() {
1678        let engine = ScriptEngine::new();
1679        let mut ctx = create_empty_script_context();
1680        ctx.chain_context.insert("float_val".to_string(), json!(3.125));
1681        ctx.chain_context.insert("bool_val".to_string(), json!(false));
1682
1683        let script = r#"mockforge.chain.bool_val;"#;
1684        let result = engine.execute_script(script, &ctx, 1000).await;
1685        assert!(result.is_ok());
1686        assert_eq!(result.unwrap().return_value, Some(json!(false)));
1687    }
1688
1689    #[tokio::test]
1690    async fn test_script_with_complex_variables() {
1691        let engine = ScriptEngine::new();
1692        let mut ctx = create_empty_script_context();
1693        ctx.variables.insert("obj".to_string(), json!({"nested": "value"}));
1694
1695        let script = r#""executed";"#;
1696        let result = engine.execute_script(script, &ctx, 1000).await;
1697        assert!(result.is_ok());
1698    }
1699}