use serde::{Deserialize, Serialize};
#[derive(Debug, Clone, Default, PartialEq, Eq, Deserialize, Serialize)]
#[serde(rename_all = "kebab-case", deny_unknown_fields)]
pub struct WasmLimitsConfig {
#[serde(default, skip_serializing_if = "Option::is_none")]
pub timeout_secs: Option<u64>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub max_memory: Option<u64>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub max_concurrent_calls: Option<usize>,
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn defaults_to_all_none() {
let cfg = WasmLimitsConfig::default();
assert_eq!(cfg.timeout_secs, None);
assert_eq!(cfg.max_memory, None);
assert_eq!(cfg.max_concurrent_calls, None);
}
#[test]
fn deserialises_full_block() {
let toml = toml::toml! {
timeout-secs = 600
max-memory = 4294967296i64
max-concurrent-calls = 2
};
let cfg: WasmLimitsConfig = toml.try_into().expect("deserialize");
assert_eq!(cfg.timeout_secs, Some(600));
assert_eq!(cfg.max_memory, Some(4_294_967_296));
assert_eq!(cfg.max_concurrent_calls, Some(2));
}
#[test]
fn deserialises_partial_block() {
let toml = toml::toml! {
timeout-secs = 120
};
let cfg: WasmLimitsConfig = toml.try_into().expect("deserialize");
assert_eq!(cfg.timeout_secs, Some(120));
assert_eq!(cfg.max_memory, None);
assert_eq!(cfg.max_concurrent_calls, None);
}
#[test]
fn rejects_unknown_field() {
let toml = toml::toml! {
timeout-secs = 10
fuel = 1000
};
let result: Result<WasmLimitsConfig, _> = toml.try_into();
assert!(result.is_err(), "deny_unknown_fields must reject `fuel`");
}
#[test]
fn serde_round_trip_preserves_set_fields() {
let original = WasmLimitsConfig {
timeout_secs: Some(45),
max_memory: Some(64 * 1024 * 1024),
max_concurrent_calls: None,
};
let serialized = toml::to_string(&original).expect("serialize");
let back: WasmLimitsConfig = toml::from_str(&serialized).expect("deserialize");
assert_eq!(original, back);
}
#[test]
fn skip_serializing_none_fields() {
let cfg = WasmLimitsConfig {
timeout_secs: Some(30),
max_memory: None,
max_concurrent_calls: None,
};
let s = toml::to_string(&cfg).expect("serialize");
assert!(s.contains("timeout-secs"));
assert!(!s.contains("max-memory"));
assert!(!s.contains("max-concurrent-calls"));
}
}