在Golang中实现类似Python的任意值Mock

在 Go 语言中，实现类似 Python 中 mock.any 的任意值 Mock 有多种方法。其中一种方法是使用 cmp 库，该库提供了 Equal 方法，允许用户自定义比较行为。另一种方法是使用 cmp.FilterValues 和 cmp.Comparer 函数来构造支持特殊“等于任何值”值的选项。最后，还可以使用 cmpopts.IgnoreMapEntries 选项，该选项可以忽略具有特定值的映射条目。

我希望能够比较两个字典，同时忽略某些字段的值。在 python 中使用 mock.any 非常容易。

from unittest.mock import any
actual = {'username':'bob', 'lastmodified':'2012-01-01'}
expected = {'username':'bob', 'lastmodified': any}
assert actual == expected

这是我在 go 中的尝试，它利用 cmp ，它允许您实现 equal 方法（https://pkg.go.dev/github.com/google/go-cmp/cmp#equal），但它不起作用。我相信这是因为我的 matchall 结构不能“分配”给字符串。有什么办法可以解决这个问题吗？

package main

import (
    "fmt"

    "github.com/google/go-cmp/cmp"
)

type MatchAll string

func (m MatchAll) Equal(v string) bool {
    return true
}

func main() { 
    matchAll := MatchAll("bruh")
    actual := map[string]any{
        "key": matchAll,
    }

    expected := map[string]any{
        "key": "foobar",
    }

    fmt.Println(cmp.Equal(expected, actual))
}

正确答案

看看cmp代码，我相信这里的问题是，在比较地图元素时，用于检查equal方法是否存在的类型是地图的元素类型（这里是any），而不是具体的类型两个值之一的类型（此处为 string 或 matchall）。由于 any 类型未声明 equal 方法，因此它会退回到内置比较。

作为替代方案，您可以使用 cmp.FilterValues 和 cmp.Comparer 函数来构造支持特殊“等于任何值”值的选项。例如，这似乎按预期工作：

package main

import (
    "fmt"

    "github.com/google/go-cmp/cmp"
)

type matchalltype struct{}

var matchall = matchalltype{}

var matchalloption = cmp.filtervalues(
    // filter function: returns `true` only for values using this comparison.
    func(x, y any) bool {
        return x == matchall || y == matchall
    },
    // comparison function: called for pairs of values the filter accepts.
    cmp.comparer(func(x, y any) bool { return true }),
)

func main() {
    actual := map[string]any{
        "key": matchall,
    }

    expected := map[string]any{
        "key": "foobar",
    }

    fmt.println(cmp.equal(expected, actual, matchalloption))
}

always-true 比较函数也可能被 cmp.Ignore() 替换，尽管我不完全确定语义是否完全相同。至少在上面的例子中它确实有效。

还有 cmpopts.IgnoreMapEntries 选项，可以按如下方式使用，但语义略有不同：

var matchAllOption = cmpopts.IgnoreMapEntries(func(_, v any) bool { return v == matchAll })

不同之处在于，使用此选项，即使 actual 字典根本不包含具有匹配键的条目，比较也将是 true，因为内置映射比较测试是否“在 上递归调用 equal”所有未忽略的地图条目报告相同”，添加了强调。相比之下，早期的解决方案仍然要求另一个映射包含具有相同键的一些值。

好了，本文到此结束，带大家了解了《在Golang中实现类似Python的任意值Mock》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！