创建使用指针接收器方法包装现有类型的接口

时间:2019-06-18 22:32:00

标签: pointers go interface

我需要测试一个使用Google Cloud Pubsub的应用程序,因此必须包装其类型test_derspubsub.Client进行测试。但是,尽管进行了几次尝试,但仍无法在它们周围进行编译。

我要包装的方法的定义是:

pubsub.Subscriber

这是当前代码。 func (s *Subscription) Receive( ctx context.Context, f func(context.Context, *Message)) error func (c *Client) Subscription(id string) *Subscription 界面(围绕Receiver的包装器)似乎可以正常工作,但是我怀疑可能需要对其进行更改才能修复Subscriber,因此我将两者都包括在内。

注意:我已经尝试了引用和取消引用指针的几种变体,因此请不要告诉我进行更改,除非您对为什么建议的配置是正确的说明有所解释,或者您已经亲自验证了它编译。

SubscriptionMaker

当前错误消息:

import (
    "context"

    "cloud.google.com/go/pubsub"
)

type Receiver interface {
    Receive(context.Context, func(ctx context.Context, msg *pubsub.Message)) (err error)
}

// Pubsub subscriptions implement Receiver
var _ Receiver = &pubsub.Subscription{}

type SubscriptionMaker interface {
    Subscription(name string) (s Receiver)
}

// Pubsub clients implement SubscriptionMaker
var _ SubscriptionMaker = pubsub.Client{}

2 个答案:

答案 0 :(得分:0)

首先,对于大多数用途而言,使用ptest包可能是一种测试pubsub的简便得多的方法。但是,当然,您的特定问题可以应用于任何库,并且以下方法可能对许多事情有用,而不仅仅是模拟pubsub


使用接口模拟这样的库的更广泛的目标是可行的。但是,当您要模拟的库返回无法模拟的具体类型时(这可能是由于未报告的字段),这很复杂。采取的方法比通常的方法要复杂得多,因为可能有更简单的方法来测试代码。

但是,如果您打算这样做,则必须采取的方法是不将整个程序包包装在接口中,而不仅是要模拟的特定方法。

您还需要包装您希望模拟的接口返回或接受的所有类型。通常,这意味着您还需要修改生产代码(而不仅仅是测试代码),因此有时可能会破坏现有代码库。

在模拟标准库的sql驱动程序之类的东西时,我通常会这样做,但是可以在此处应用相同的方法。本质上,您将需要为pubsub库创建一个包装程序包,甚至可以在生产代码中使用该程序包。同样,这对于现有代码库可能是相当麻烦的,但是为了说明起见。使用定义的接口:

package mypubsub

import "cloud.google.com/go/pubsub"

type Receiver interface {
    Recieve(context.Context, func(context.Context, *pubsub.Message) error)
}

type SubscriptionMaker interface {
    Subscription(string) Receiver
}

然后您可以包装默认实现,以用于生产代码:

// defaultClient wraps the default pubsub Client functionality.
type defaultClient struct {
    *pubsub.Client
}

func (d defaultImplementation) Subscription(name string) Receiver {
    return d.Client.Subscription()
}

自然,您需要扩展此程序包以包装您正在使用的大多数或全部pubsub程序包。这可能有点令人生畏。

但是,一旦这样做,就可以在代码中的任何地方使用mypubsub包,而不必直接依赖于pubsub包。现在,您可以轻松地在需要测试的任何地方换出模拟实现。

答案 1 :(得分:-1)

无法完成。

在接口上定义方法的类型签名时,它必须完全匹配。 func (c *Client) Subscription(id string) *Subscription返回一个*Subscription,一个*Subscription是有效的Receiver,但不算是符合接口方法Subscription(string) Receiver。 Go需要对功能签名进行精确匹配,而不是通常用于接口的鸭子键入样式。