vendor/eyre/tests/test_pyo3.rs


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

#![cfg(feature = "pyo3")]

use pyo3::prelude::*;

use eyre::{bail, Result, WrapErr};

fn f() -> Result<()> {
    use std::io;
    bail!(io::Error::new(io::ErrorKind::PermissionDenied, "oh no!"));
}

fn g() -> Result<()> {
    f().wrap_err("f failed")
}

fn h() -> Result<()> {
    g().wrap_err("g failed")
}

#[test]
fn test_pyo3_exception_contents() {
    use pyo3::types::IntoPyDict;

    let err = h().unwrap_err();
    let expected_contents = format!("{:?}", err);
    let pyerr = PyErr::from(err);

    Python::with_gil(|py| {
        let locals = [("err", pyerr)].into_py_dict(py);
        let pyerr = py.run("raise err", None, Some(locals)).unwrap_err();
        assert_eq!(pyerr.pvalue(py).to_string(), expected_contents);
    })
}