gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-11-29 08:23:17 25 4
gpt4 key购买 nike

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

我有一个方法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;
}
}

对第一种方法进行了单元测试,测试另一种方法的策略是什么?我想为 plus_one() 的单元测试模拟 self.random_number() 输出,以便在单元测试中拥有合理的代码。有一篇很好的帖子比较了different mocking libraries并得出结论(可悲的是)他们中没有一个真正能从其他人中脱颖而出。

我在阅读这些库的说明时学到的唯一一件事是,我可以模拟方法的唯一方法是将它们移动到特征。我没有在这些库中看到任何示例(我查看了其中的 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 {}

最佳答案

How to mock specific methods but not all of them in 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);
}

这里,random_number 是一个必需的方法,但是plus_one 有一个默认实现。默认情况下,实现 random_number 会为您提供 plus_one。如果可以更高效地实现 plus_one,您也可以选择实现它。

real rand crate 有什么作用?

真实rand crate使用两个特征:

  • Rng

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

    An automatically-implemented extension trait on RngCore providing high-level generic methods for sampling values and other convenience methods.

  • RngCore

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

    The core of a random number generator.

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

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);
}

关于unit-testing - 如何在 Rust 中模拟特定方法而不是所有方法?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55152927/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com