add tests

src/lib.rs

@@ -69,10 +69,12 @@ use log::{Level, Log, Metadata, Record};
 mod errors;
 mod facility;
 mod format;
+#[cfg(test)]
+mod tests;
+
 pub use errors::*;
 pub use facility::Facility;
 pub use format::Severity;
-
 pub use format::{Formatter3164, Formatter5424, LogFormat};
 
 pub type Priority = u8;

src/tests.rs

@@ -0,0 +1,113 @@
+use io::Read;
+use std::collections::BTreeMap;
+
+#[test]
+fn test_unix_socket() {
+    use super::*;
+    // create a unix socket listener on a random path
+    let path = std::env::temp_dir().join(format!("syslog-test-{}", std::process::id()));
+    let formatter = Formatter3164 {
+        facility: Facility::LOG_USER,
+        hostname: None,
+        process: "myprogram".into(),
+        pid: 0,
+    };
+
+    let err = unix_custom(formatter.clone(), &path)
+        .map(|_| ())
+        .unwrap_err();
+    println!("expected initialization error: {err:?}");
+
+    let listener = std::os::unix::net::UnixListener::bind(&path).unwrap();
+
+    let str = Arc::new(Mutex::new(String::new()));
+    let s = str.clone();
+
+    std::thread::spawn(move || {
+        let mut stream = listener.accept().unwrap().0;
+        loop {
+            let mut str = s.lock().unwrap();
+            stream.read_to_string(&mut str).unwrap();
+        }
+    });
+
+    let mut writer: Logger<LoggerBackend, Formatter3164> = unix_custom(formatter, &path).unwrap();
+    writer.emerg("a1").unwrap();
+    writer.alert("a2").unwrap();
+    writer.crit("a3").unwrap();
+    writer.err("a4").unwrap();
+    writer.warning("a5").unwrap();
+    writer.notice("a6").unwrap();
+    writer.info("a7").unwrap();
+    writer.debug("a8").unwrap();
+    drop(writer);
+
+    let s = str.lock().unwrap();
+    println!("messages:\n{}", *s);
+
+    assert!(s.contains("a1"));
+    assert!(s.contains("a2"));
+    assert!(s.contains("a3"));
+    assert!(s.contains("a4"));
+    assert!(s.contains("a5"));
+    assert!(s.contains("a6"));
+    assert!(s.contains("a7"));
+    assert!(s.contains("a8"));
+    assert!(s.contains("myprogram"));
+}
+
+#[test]
+fn test_tcp() {
+    use super::*;
+    let formatter = Formatter5424 {
+        facility: Facility::LOG_USER,
+        hostname: None,
+        process: "myprogram".into(),
+        pid: 0,
+    };
+
+    let listener = std::net::TcpListener::bind("127.0.0.1:0").unwrap();
+    let local_address = listener.local_addr().unwrap();
+
+    let str = Arc::new(Mutex::new(String::new()));
+    let s = str.clone();
+
+    std::thread::spawn(move || {
+        let mut stream = listener.accept().unwrap().0;
+        while let Ok(mut str) = s.lock() {
+            stream.read_to_string(&mut str).unwrap();
+        }
+    });
+
+    let mut writer: Logger<LoggerBackend, Formatter5424> = tcp(formatter, local_address).unwrap();
+
+    writer.emerg((1, BTreeMap::new(), "a1")).unwrap();
+    let mut data = BTreeMap::new();
+    let mut inner = BTreeMap::new();
+    inner.insert("key".to_string(), "value".to_string());
+    data.insert("param".to_string(), inner);
+
+    writer.alert((1, data.clone(), "a2")).unwrap();
+    writer.crit((1, data.clone(), "a3")).unwrap();
+    writer.err((1, data.clone(), "a4")).unwrap();
+    writer.warning((1, data.clone(), "a5")).unwrap();
+    writer.notice((1, data.clone(), "a6")).unwrap();
+    writer.info((1, data.clone(), "a7")).unwrap();
+    writer.debug((1, data.clone(), "a8")).unwrap();
+
+    drop(writer);
+
+    let s = str.lock().unwrap();
+    println!("messages:\n{}", *s);
+
+    assert!(s.contains("a1"));
+    assert!(s.contains("a2"));
+    assert!(s.contains("a3"));
+    assert!(s.contains("a4"));
+    assert!(s.contains("a5"));
+    assert!(s.contains("a6"));
+    assert!(s.contains("a7"));
+    assert!(s.contains("a8"));
+    assert!(s.contains("myprogram"));
+    assert!(s.contains("[param key=\"value\"]"));
+}