Spaces:
Runtime error
Runtime error
File size: 2,670 Bytes
9552aa0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 | use crate::ferron_common::ErrorLogger;
use pyo3::prelude::*;
#[pyclass]
pub struct WsgiErrorStream {
error_logger: ErrorLogger,
}
impl WsgiErrorStream {
pub fn new(error_logger: ErrorLogger) -> Self {
Self { error_logger }
}
}
#[pymethods]
impl WsgiErrorStream {
fn write(&self, data: &str) -> PyResult<usize> {
futures_lite::future::block_on(
self
.error_logger
.log(&format!("There was a WSGI error: {}", data)),
);
Ok(data.len())
}
fn writelines(&self, lines: Vec<String>) -> PyResult<()> {
for line in lines {
// Each `log_blocking` call prints a separate line
futures_lite::future::block_on(
self
.error_logger
.log(&format!("There was a WSGI error: {}", line)),
);
}
Ok(())
}
fn flush(&self) -> PyResult<()> {
// This is a no-op function
Ok(())
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::ferron_common::LogMessage;
use async_channel::{bounded, Receiver};
#[test]
fn test_write_logs_error_message() {
let (tx, rx): (async_channel::Sender<LogMessage>, Receiver<LogMessage>) = bounded(100);
let error_logger = ErrorLogger::new(tx);
let wsgi_error_stream = WsgiErrorStream::new(error_logger);
let data = "Some error occurred";
let result = wsgi_error_stream.write(data).unwrap();
assert_eq!(result, data.len());
// Check if the log message was sent
let log_message = rx.recv_blocking().unwrap();
let (message, is_error) = log_message.get_message();
assert_eq!(message, format!("There was a WSGI error: {}", data));
assert!(is_error);
}
#[test]
fn test_writelines_logs_multiple_error_messages() {
let (tx, rx): (async_channel::Sender<LogMessage>, Receiver<LogMessage>) = bounded(100);
let error_logger = ErrorLogger::new(tx);
let wsgi_error_stream = WsgiErrorStream::new(error_logger);
let lines = vec!["Error 1".to_string(), "Error 2".to_string()];
let result = wsgi_error_stream.writelines(lines.clone());
assert!(result.is_ok());
// Check if the log messages were sent
for line in lines {
let log_message = rx.recv_blocking().unwrap();
let (message, is_error) = log_message.get_message();
assert_eq!(message, format!("There was a WSGI error: {}", line));
assert!(is_error);
}
}
#[test]
fn test_flush_no_op() {
let (tx, _rx): (async_channel::Sender<LogMessage>, Receiver<LogMessage>) = bounded(100);
let error_logger = ErrorLogger::new(tx);
let wsgi_error_stream = WsgiErrorStream::new(error_logger);
let result = wsgi_error_stream.flush();
assert!(result.is_ok());
}
}
|