diff --git a/stdlib/src/database.rs b/stdlib/src/database.rs index 37b12bd7..b27b2f6a 100644 --- a/stdlib/src/database.rs +++ b/stdlib/src/database.rs @@ -283,3 +283,27 @@ impl StdlibRegistry { ); } } + +#[cfg(test)] +mod tests { + use super::*; + use std::collections::HashMap; + + #[test] + fn test_register_database() { + let mut registry = StdlibRegistry { + modules: HashMap::new(), + }; + registry.register_database(); + + assert!(registry.has_module("std.database")); + let module = registry.get_module("std.database").unwrap(); + assert_eq!(module.name, "std.database"); + assert_eq!(module.version, "1.0.0"); + assert!(module.exports.contains_key("connect")); + assert!(module.exports.contains_key("query")); + assert!(module.exports.contains_key("execute")); + assert!(module.exports.contains_key("close")); + assert_eq!(module.exports.len(), 4); + } +}