如何在 Rust 中模拟特定方法而不是所有方法?

2024-03-10

我在确定目标结构的方法的单元测试时遇到了麻烦。

我有一个方法random_number根据结构体的属性返回一个随机值,还有另一种方法plus_one它获取第一个方法的结果并用它做一些事情:

pub struct RngTest {
    pub attr: u64,
}

impl RngTest {
    pub fn random_number(&self) -> u64 {
        let random = 42; // lets pretend it is random
        return random * self.attr;
    }

    pub fn plus_one(&self) -> u64 {
        return self.random_number() + 1;
    }
}

对第一种方法进行了单元测试,测试另一种方法的策略是什么?我想嘲笑self.random_number()单元测试的输出plus_one()在单元测试中拥有健全的代码。有一篇不错的文章比较了不同的模拟库 https://asomers.github.io/mock_shootout/并得出结论(很遗憾),他们中没有一个能够真正脱颖而出。

我在阅读这些库的说明时学到的唯一一件事是,我模拟方法的唯一方法是将它们移动到一个特征。我在这些库中没有看到任何测试与此类似的案例的示例(我查看了其中的 4 个或 5 个)。

将这些方法移至特征后(即使它们是),我如何模拟random_number对输出进行单元测试RngTest::plus_one?

pub trait SomeRng {
    fn random_number(&self) -> u64 {
        let random = 42; // lets pretend it is random
        return random * self.attr;
    }

    fn plus_one(&self) -> u64 {
        return self.random_number() + 1;
    }
}

impl SomeRng for RngTest {}

如何在 Rust 中模拟特定方法而不是所有方法?

正如您已经了解到的,您不能替换类型上的方法。您唯一能做的就是将方法移至特征,然后提供该特征的特定于生产和测试的实现。如何构造特征决定了您能够测试的粒度。

具有默认实现的特征

根据您的用例,您也许可以使用默认实现:

trait SomeRng {
    fn random_number(&self) -> u64;

    fn plus_one(&self) -> u64 {
        self.random_number() + 1
    }
}

struct RngTest(u64);
impl SomeRng for RngTest {
    fn random_number(&self) -> u64 {
        self.0
    }
}

#[test]
fn plus_one_works() {
    let rng = RngTest(41);
    assert_eq!(rng.plus_one(), 42);
}

Here, random_number是必需的方法,但是plus_one有一个默认实现。实施random_number给你plus_one默认情况下。您还可以选择实施plus_one如果你能做得更有效率的话。

真正的兰特箱子有什么作用?

真实的兰德箱 https://docs.rs/rand/0.6.5使用两个特征:

  • Rng https://docs.rs/rand/0.6.5/rand/trait.Rng.html

    pub trait Rng: RngCore { /* ... */ }
    

    自动实现的扩展特征RngCore提供采样值的高级通用方法和其他便捷方法。

  • RngCore https://docs.rs/rand/0.6.5/rand/trait.RngCore.html

    pub trait RngCore { /* ... */ }
    

    随机数生成器的核心。

这将实现的核心有趣部分与辅助方法分开。然后您可以控制核心并测试助手:

trait SomeRngCore {
    fn random_number(&self) -> u64;
}

trait SomeRng: SomeRngCore {
    fn plus_one(&self) -> u64 {
        self.random_number() + 1
    }
}

impl<R: SomeRngCore> SomeRng for R {}

struct RngTest(u64);
impl SomeRngCore for RngTest {
    fn random_number(&self) -> u64 {
        self.0
    }
}

#[test]
fn plus_one_works() {
    let rng = RngTest(41);
    assert_eq!(rng.plus_one(), 42);
}
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

如何在 Rust 中模拟特定方法而不是所有方法? 的相关文章

随机推荐