测试私有字段的状态设置

大家好，我们又见面了啊~本文《测试私有字段的状态设置》的内容中将会涉及到等等。如果你正在学习Golang相关知识，欢迎关注我，以后会给大家带来更多Golang相关文章，希望我们能一起进步！下面就开始本文的正式内容~

• 库包含一个导出方法 a 的结构类型，该方法是根据未导出的字段数据 a 计算的。
• 由于 a 未导出，因此直接构造类型将产生空字段数据。
• 设置 a 的唯一位置是当库以使用导出函数/字段不易复制的方式实例化类型时（例如，通过网络 api 调用填充数据）

type s struct {
  a []byte
}
func (s *s) a() ... {
  // produces a meaningful result using the blob in field a
}
...
type i interface {
  dolibrarything() (*s)
  ...
}
struct client type {...}
func (c *client) dolibrarything() (*s) {
  ...
}
...

我想要编写测试的程序使用 dolibrarything，特别是依赖于并操纵 s.a() 的结果

func testme(client i) {
  // do a bunch of things with the client interface
  s := client.dolibrarything()
  data := s.a()
  // do more things with client interface and data
}

我已经模拟了返回此类型的库 client，但无法返回适当的实例，以便调用 a 将生成可在测试的其余部分中使用的有用数据。正在测试的函数对客户端/库进行了许多其他调用。

有没有办法用 a 构造有用数据的类型，或者以其他方式替换可以从 a 返回有用数据的模拟 a？ 让客户端模拟返回导出 a 的不同结构硬编码行为不起作用，因为从模拟返回时它不是 *s。

考虑的替代方案：

• 在此库调用点拆分函数，测试调用前后的行为，并在测试函数后一部分时将 a 的结果提供给测试作为测试输入。在这里拆分函数只会对测试真正有意义，并且会涉及将状态从函数实现的第一部分传递到函数实现的第二部分的不必要的复杂性。

func testMe(client I) {
  kludge, data := testMe1(client)
  return testMe2(kludge, data)
}
func testMe1(client I) {
  // do a bunch of things with the client interface
  s := client.DoLibraryThing()
  data := s.A()
  return Kludge{ /* pack up all the state */ }, data
}
func testMe2(k Kludge, d data) {
  // unpack all the state
  // do more things with client interface and data
}

^ testme1和testme2现在可以独立测试，并且在测试testme2时可以传入预定的测试数据。

正确答案

这是一个设置结构体字段值的函数，包括未导出的字段。我以此为依据https://stackoverflow.com/a/43918797/3201891

除了测试之外，我不建议在任何地方使用此函数。并且仅当没有其他方法来设置测试数据时。例如，在依赖项是无法编辑的第三方包的情况下。请注意，如果外部包发生更改，测试可能会中断。

func setfieldvalue(target any, fieldname string, value any) {
    rv := reflect.valueof(target)
    for rv.kind() == reflect.ptr && !rv.isnil() {
        rv = rv.elem()
    }
    if !rv.canaddr() {
        panic("target must be addressable")
    }
    if rv.kind() != reflect.struct {
        panic(fmt.sprintf(
            "unable to set the '%s' field value of the type %t, target must be a struct",
            fieldname,
            target,
        ))
    }
    rf := rv.fieldbyname(fieldname)

    reflect.newat(rf.type(), unsafe.pointer(rf.unsafeaddr())).elem().set(reflect.valueof(value))
}

这里是使用示例 https://go.dev/play/p/i9w0RPcmKHc

我认为你需要变得不安全。

为此，您需要能够读取库的源代码。

// lib

type useful struct {
    exported   bool
    unexported int
}

var s useful = useful{unexported: 3}

未导出的字段位于布尔值之后，在我的计算机中使用 8 位。

你要做的就是获取一个指向 s 的指针，将其移动 8 位，然后读取此时的值。

// Your side
func main() {
    fmt.Println(unsafe.Offsetof(s.unexported)) // we get 8, it's try and error in your case
    point := (*int)(unsafe.Pointer(uintptr(unsafe.Pointer(&s)) + 8))
    fmt.Println(*point)
}

来自一位疲惫的旅行者的一点警告。该包被称为不安全是有非常充分的理由的。不要在任何需要花费超过 crtl+c 来修复的问题上尝试此操作。您获得的值可能因计算机而异，特别是如果您使用的机器是用于特定目的的。我不会对 unsaffe 释放的任何黑洞或时间异常负责。

You can find my playground test here.

好了，本文到此结束，带大家了解了《测试私有字段的状态设置》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！