unit-testing - 是否有可能改变整个 panic 信息？

NUnit 是一个 C# 单元测试框架，允许您编写如下代码:

Assert.That(someInt, Is.EqualTo(42));
Assert.That(someList, Has.Member(someMember));

我喜欢这种代码，因为它看起来像英文，很容易阅读。

我正在使用 Rust，看看我是否可以创建一个具有相同感觉的库:

use std::fmt::Debug;

struct Is;

enum Verb<T> {
    EqualTo(T),
}

impl Is {
    fn equal_to<T>(&self, obj: T) -> Verb<T> {
        Verb::EqualTo(obj)
    }
}

#[allow(non_upper_case_globals)]
const is: Is = Is{};

fn assert_that<T: Eq + Debug>(obj: T, verb: Verb<T>) {
    match verb {
        Verb::EqualTo(rhs)    => assert_eq!(obj, rhs),
    }
}

fn main() {
    assert_that(42, is.equal_to(42));
    assert_that(42, is.equal_to(0));
}

这很好，但一方面:当代码在 assert_that(42, is.equal_to(0)) 处发生 panic 时， panic 给出的行是 assert_eq!(obj, rhs) 行(即在库中而不是用户代码中)。我知道这种行为很正常，但我会有更有用的信息。

如何在 panic 中指示正确的行号？

最佳答案

没有直接的方法来调整 panic! 打印的行号。

a proto-RFC 添加一个属性，允许某些方法从回溯中“隐藏”。这样的属性也可能会影响行号，但尚不清楚。

How to write a panic! like macro in Rust? 描述了如何编写自己的 panic! 宏，但它选择拆除整个进程，而不仅仅是当前线程。

重要的是您只想控制消息，这可以通过 panic::set_hook 实现。您可以通过线程局部变量将侧信道信息从测试传递到 panic 处理程序。

use std::cell::Cell;

thread_local! {
    static ASSERT_LOCATION: Cell<Option<(&'static str, u32)>> = Cell::new(None)
}

fn report_my_error(info: &std::panic::PanicInfo) {
    match info.location() {
        Some(location) => {
            let file = location.file();
            let line = location.line();
            println!("The panic actually happened at: {}, {}", file, line);
        }
        None => println!("I don't know where the panic actually happened"),
    }

    ASSERT_LOCATION.with(|location| if let Some((file, line)) = location.get() {
        println!(
            "But I'm going to tell you it happened at {}, {}",
            file,
            line
        );
    });

    if let Some(msg) = info.payload().downcast_ref::<&str>() {
        println!("The error message was: {}", msg);
    }
}

#[test]
fn alpha() {
    std::panic::set_hook(Box::new(report_my_error));

    ASSERT_LOCATION.with(|location| {
        location.set(Some((file!(), line!())));
    });

    panic!("This was only a test")
}

您需要确保在每个测试中都设置了紧急处理程序，然后设置了位置信息。您可能还希望更新紧急处理程序以将位置信息设置回 None 以避免位置信息在线程之间泄漏。

您可能希望编写自己的宏，用户可以在测试中使用该宏来隐式设置行号。与此类似的语法可以为这个设置代码提供一个存在的地方:

assert_that!(42, is.equal_to(0));

可以扩展为:

assert_that(file!(), line!(), 42, is.equal_to(0));

我可能会在 assert_that 内设置 panic 处理程序。

关于unit-testing - 是否有可能改变整个 panic 信息？，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/45053381/