f4660639ca 2026-07-31 1: use smtp2tg::mail::MailServer;
f4660639ca 2026-07-31 2:
78b11b0319 2026-08-01 3: use config::FileFormat::Toml;
f4660639ca 2026-07-31 4: use stacked_errors::{
f4660639ca 2026-07-31 5: Result,
f4660639ca 2026-07-31 6: StackableErr,
78b11b0319 2026-08-01 7: ensure,
78b11b0319 2026-08-01 8: ensure_eq,
f4660639ca 2026-07-31 9: };
f4660639ca 2026-07-31 10:
f4660639ca 2026-07-31 11: use tgbot::types::ChatPeerId;
78b11b0319 2026-08-01 12:
78b11b0319 2026-08-01 13: #[test]
78b11b0319 2026-08-01 14: fn get_id_properly_resolves_addresses () -> Result<()> {
78b11b0319 2026-08-01 15: let server = MailServer::new(config::Config::builder()
f4660639ca 2026-07-31 16: .add_source(config::File::from_str(r#"
f4660639ca 2026-07-31 17: api_key = "test-api-key"
f4660639ca 2026-07-31 18: api_gateway = "https://api.telegram.org"
f4660639ca 2026-07-31 19: default = 0
f4660639ca 2026-07-31 20: fields = ["date", "from", "subject"]
f4660639ca 2026-07-31 21: domains = ["example.com"]
f4660639ca 2026-07-31 22:
f4660639ca 2026-07-31 23: [recipients]
f4660639ca 2026-07-31 24: "someone@example.com" = 1
f4660639ca 2026-07-31 25: "root" = -1
78b11b0319 2026-08-01 26: "#, Toml))
f4660639ca 2026-07-31 27: .build()
78b11b0319 2026-08-01 28: .stack()?)?;
f4660639ca 2026-07-31 29: let cases = [
f4660639ca 2026-07-31 30: ("someone@example.com", 1),
f4660639ca 2026-07-31 31: ("someone", 0),
f4660639ca 2026-07-31 32: ("root", -1),
f4660639ca 2026-07-31 33: ("unknown@example.com", 0),
78b11b0319 2026-08-01 34: ("SOMEONE@example.com", 1), // uppercase local part
78b11b0319 2026-08-01 35: ("someone@EXAMPLE.COM", 1), // uppercase domain
78b11b0319 2026-08-01 36: ("some.one@example.com", 0), // dot are not skipped
78b11b0319 2026-08-01 37: ("some-one-2", 0), // Hyphens
f4660639ca 2026-07-31 38: ];
f4660639ca 2026-07-31 39: for (email, id) in cases {
78b11b0319 2026-08-01 40: ensure_eq!(*server.get_id(email)?, ChatPeerId::from(id), format!("email [{email}] expected to return id [{id}]"));
78b11b0319 2026-08-01 41: }
78b11b0319 2026-08-01 42: let cases = [
78b11b0319 2026-08-01 43: "someone@otherdomain.net",
78b11b0319 2026-08-01 44: "@example.com", // empty local part
78b11b0319 2026-08-01 45: "some@one@example.com", // more than one '@'
78b11b0319 2026-08-01 46: "someone@example.com.evil",
78b11b0319 2026-08-01 47: "someone@example.org",
78b11b0319 2026-08-01 48: ];
78b11b0319 2026-08-01 49: for email in cases {
78b11b0319 2026-08-01 50: ensure!(server.get_id(email).is_err(), format!("this email should be rejected: {email}"));
78b11b0319 2026-08-01 51: }
78b11b0319 2026-08-01 52: Ok(())
78b11b0319 2026-08-01 53: }
78b11b0319 2026-08-01 54:
78b11b0319 2026-08-01 55: #[test]
78b11b0319 2026-08-01 56: fn wrong_server_config () -> Result<()> {
78b11b0319 2026-08-01 57: let configs = [
78b11b0319 2026-08-01 58: "domains = []",
78b11b0319 2026-08-01 59: "",
78b11b0319 2026-08-01 60: "[recipents]\na = 1",
78b11b0319 2026-08-01 61: r#"
78b11b0319 2026-08-01 62: api_key = "test-api-key"
78b11b0319 2026-08-01 63: api_gateway = "https://api.telegram.org"
78b11b0319 2026-08-01 64: default = 0
78b11b0319 2026-08-01 65: fields = ["date", "from", "subject"]
78b11b0319 2026-08-01 66: domains = ["example.com"]
78b11b0319 2026-08-01 67: # no recipients
78b11b0319 2026-08-01 68: "#,
78b11b0319 2026-08-01 69: r#"
78b11b0319 2026-08-01 70: ap_key = "test-api-key" # bad one
78b11b0319 2026-08-01 71: api_gateway = "https://api.telegram.org"
78b11b0319 2026-08-01 72: default = 0
78b11b0319 2026-08-01 73: fields = ["date", "from", "subject"]
78b11b0319 2026-08-01 74: domains = ["example.com"]
78b11b0319 2026-08-01 75:
78b11b0319 2026-08-01 76: [recipients]
78b11b0319 2026-08-01 77: "someone@example.com" = 1
78b11b0319 2026-08-01 78: "root" = -1"#,
78b11b0319 2026-08-01 79: r#"
78b11b0319 2026-08-01 80: api_key = "test-api-key"
78b11b0319 2026-08-01 81: api_gateway = "https://api.telegram.org"
78b11b0319 2026-08-01 82: default = 0
78b11b0319 2026-08-01 83: fields = ["date", "from", "subject"]
78b11b0319 2026-08-01 84: domains = [] # empty
78b11b0319 2026-08-01 85:
78b11b0319 2026-08-01 86: [recipients]
78b11b0319 2026-08-01 87: "someone@example.com" = 1
78b11b0319 2026-08-01 88: "root" = -1"#,
78b11b0319 2026-08-01 89: ];
78b11b0319 2026-08-01 90: for config in configs {
78b11b0319 2026-08-01 91: let settings = config::Config::builder()
78b11b0319 2026-08-01 92: .add_source(config::File::from_str(config, Toml))
78b11b0319 2026-08-01 93: .build()
78b11b0319 2026-08-01 94: .stack()?;
78b11b0319 2026-08-01 95: ensure!(MailServer::new(settings).is_err(), format!("this config shouldn't be valid:\n{config}"));
f4660639ca 2026-07-31 96: }
78b11b0319 2026-08-01 97:
f4660639ca 2026-07-31 98: Ok(())
f4660639ca 2026-07-31 99: }