无法在 func 参数中使用 bytes.Buffer 作为 io.WriterAt 的类型

在 Go 语言中，无法在函数参数中使用 bytes.Buffer 作为 io.WriterAt 类型，因为 bytes.Buffer 仅实现了 io.Writer 接口，而 io.WriterAt 要求实现 writeat 方法。为了解决此问题，可以创建一个自定义类型来实现 io.WriterAt，以便在测试中使用。

今天在 go 上苦苦挣扎..我不得不问的第二个问题。

我有 2 个测试来测试写入函数 write()，该函数采用 writer io.writerat 和 content 接口{}。

我正在使用为函数 testwritesuccessful 和 testwritefail 编写的 (2) 个测试。

测试时我收到的两个函数的错误是：

cannot use &b (type *bytes.buffer) as type io.writerat in argument to write:

问题

什么实现了 writerat，我可以在这些测试中替换 bytes.buffer 以使测试正常运行？

我尝试过的

1. 将 b 的类型更改为 os.file - b.len() > 0 会失败。
2. 广泛搜索如何测试 io.writerat 和实现它的函数，我唯一发现的是 bytes.buffer 没有实现 io.writerat，只有 io.writer，但没有更多信息。
3. 检查了 godocs，我看到 io.writerat 调用 writeat 并获取字节切片和偏移量： writeat(p []byte, off int64) (n int, err error)，我看到了该字节。 buffer可以用来调用write，但是没有关于writeat的信息

代码

写入函数：

func write(writer io.writerat, content interface{}) error {
    data, err := json.marshal(content)
    if err != nil {
        return err
    }
    writer.writeat(data, 0)
    return nil
}

write 函数测试（测试导入 testify）：

func testwritesuccessful(t *testing.t) {
    var b bytes.buffer
    err := write(&b, examplesystemconfig)
    assert.nil(t, err)
    assert.true(t, b.len() > 0)
}

func testwritefail(t *testing.t) {
    var b bytes.buffer
    err := write(&b, make(chan int)) // write will return unsupportedtypeerror
    assert.notnil(t, err)
}

预期结果

两个测试都成功。

实际结果

internal/platform/store/store_test.go:33:15: cannot use &b (type *bytes.Buffer) as type io.WriterAt in argument to Write:
        *bytes.Buffer does not implement io.WriterAt (missing WriteAt method)
internal/platform/store/store_test.go:40:15: cannot use &b (type *bytes.Buffer) as type io.WriterAt in argument to Write:
        *bytes.Buffer does not implement io.WriterAt (missing WriteAt method)

解决方案

出于测试目的，很容易制作自己的 writerat — 它是任何使用正确签名实现 writeat 函数的类型。如果您只想知道写入的字节数，那么它可以很简单

type testwriter struct {
    byteswritten int
}

func (tw *testwriter) writeat(b []byte, _ int64) (n int, err error) {
    tw.byteswritten += len(b)
    return len(b), nil
}

然后你可以测试

func TestWriteSuccessful(t *testing.T) {
    var tw TestWriter
    err := Write(&tw, exampleSystemConfig)
    assert.Nil(t, err)
    assert.True(t, tw.BytesWritten > 0)
}

func TestWriteFail(t *testing.T) {
    var tw bytes.Buffer
    err := Write(&tw, make(chan int)) // Write will return UnsupportedTypeError
    assert.NotNil(t, err)
}

终于介绍完啦！小伙伴们，这篇关于《无法在 func 参数中使用 bytes.Buffer 作为 io.WriterAt 的类型》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！