Function test_logging

crates/chat-cli/src/logging.rs:275–312

()

    #[test]
    fn test_logging() {
        // Create a temp path for where we write logs to.
        let tempdir = tempfile::TempDir::new().unwrap();
        let log_path = tempdir.path().join("test.log");

        // Assert that initialize logging simply doesn't panic.
        let _guard = initialize_logging(LogArgs {
            log_level: Some("trace".to_owned()),
            log_to_stdout: true,
            log_file_path: Some(&log_path),
            delete_old_log_file: true,
        })
        .unwrap();

        // Test that get log level functions as expected.
        assert_eq!(get_log_level(), "trace");

        // Write some log messages out to file. (and stderr)
        trace!("abc");
        debug!("def");
        info!("ghi");
        warn!("jkl");
        error!("mno");

        // Test that set log level functions as expected.
        // This also restores the default log level.
        set_log_level(DEFAULT_FILTER.to_string()).unwrap();
        assert_eq!(get_log_level(), DEFAULT_FILTER.to_string());

        // Sleep in order to ensure logs get written to file, then assert on the contents
        std::thread::sleep(Duration::from_millis(100));
        let logs = read_to_string(&log_path).unwrap();
        for i in [
            "TRACE", "DEBUG", "INFO", "WARN", "ERROR", "abc", "def", "ghi", "jkl", "mno",
        ] {
            assert!(logs.contains(i));
        }
    }
}