use std::collections::HashMap; use std::time::Duration; use crate::config::*; // Environment variables are set for the entire process and // tests are run whithin the same process. // => We cannot test ConfigOpts::from_env(), // because tests modify each other's environment. // This is why we only test ConfigOpts::from_iter(iter). fn minimal_valid_options() -> HashMap { let mut opts = HashMap::new(); opts.insert("DIPLONAT_PRIVATE_IP".to_string(), "172.123.43.555".to_string()); opts.insert("DIPLONAT_CONSUL_NODE_NAME".to_string(), "consul_node".to_string()); opts } fn all_valid_options() -> HashMap { let mut opts = minimal_valid_options(); opts.insert("DIPLONAT_EXPIRATION_TIME".to_string(), "30".to_string()); opts.insert("DIPLONAT_REFRESH_TIME".to_string(), "10".to_string()); opts.insert("DIPLONAT_CONSUL_URL".to_string(), "http://127.0.0.1:9999".to_string()); opts.insert("DIPLONAT_ACME_ENABLE".to_string(), "true".to_string()); opts.insert("DIPLONAT_ACME_EMAIL".to_string(), "bozo@bozo.net".to_string()); opts } #[test] #[should_panic] fn err_empty_env() { std::env::remove_var("DIPLONAT_PRIVATE_IP"); std::env::remove_var("DIPLONAT_CONSUL_NODE_NAME"); ConfigOpts::from_env().unwrap(); } #[test] fn ok_from_iter_minimal_valid_options() { let opts = minimal_valid_options(); let rt_config = ConfigOpts::from_iter(opts.clone()).unwrap(); assert!(rt_config.acme.is_none()); assert_eq!( &rt_config.consul.node_name, opts.get(&"DIPLONAT_CONSUL_NODE_NAME".to_string()).unwrap() ); assert_eq!( rt_config.consul.url, CONSUL_URL.to_string() ); assert_eq!( rt_config.firewall.refresh_time, Duration::from_secs(REFRESH_TIME.into()) ); assert_eq!( &rt_config.igd.private_ip, opts.get(&"DIPLONAT_PRIVATE_IP".to_string()).unwrap() ); assert_eq!( rt_config.igd.expiration_time, Duration::from_secs(EXPIRATION_TIME.into()) ); assert_eq!( rt_config.igd.refresh_time, Duration::from_secs(REFRESH_TIME.into()) ); } #[test] #[should_panic] fn err_from_iter_invalid_refresh_time() { let mut opts = minimal_valid_options(); opts.insert("DIPLONAT_EXPIRATION_TIME".to_string(), "60".to_string()); opts.insert("DIPLONAT_REFRESH_TIME".to_string(), "60".to_string()); ConfigOpts::from_iter(opts).unwrap(); } #[test] fn ok_from_iter_all_valid_options() { let opts = all_valid_options(); let rt_config = ConfigOpts::from_iter(opts.clone()).unwrap(); let expiration_time = Duration::from_secs( opts.get(&"DIPLONAT_EXPIRATION_TIME".to_string()).unwrap() .parse::().unwrap() .into()); let refresh_time = Duration::from_secs( opts.get(&"DIPLONAT_REFRESH_TIME".to_string()).unwrap() .parse::().unwrap() .into()); assert!(rt_config.acme.is_some()); assert_eq!( &rt_config.acme.unwrap().email, opts.get(&"DIPLONAT_ACME_EMAIL".to_string()).unwrap()); assert_eq!( &rt_config.consul.node_name, opts.get(&"DIPLONAT_CONSUL_NODE_NAME".to_string()).unwrap() ); assert_eq!( &rt_config.consul.url, opts.get(&"DIPLONAT_CONSUL_URL".to_string()).unwrap() ); assert_eq!( rt_config.firewall.refresh_time, refresh_time ); assert_eq!( &rt_config.igd.private_ip, opts.get(&"DIPLONAT_PRIVATE_IP".to_string()).unwrap() ); assert_eq!( rt_config.igd.expiration_time, expiration_time ); assert_eq!( rt_config.igd.refresh_time, refresh_time ); }