2

我正在尝试模拟 Stripe 进行一些测试。

//testify mock
type Backend struct {
    mock.Mock
}

func (s Backend) Call(method, path, key string, params stripe.ParamsContainer, v interface{}) error {
    args := s.Called(params)
    return args.Error(0)
}

func (s Backend) CallRaw(method, path, key string, body *form.Values, params *stripe.Params, v interface{}) error {
    args := s.Called(params)
    return args.Error(0)
}

func (s Backend) CallMultipart(method, path, key, boundary string, body *bytes.Buffer, params *stripe.Params, v interface{}) error {
    args := s.Called(params)
    return args.Error(0)
}

func (s Backend) SetMaxNetworkRetries(maxNetworkRetries int) {
    s.Called(maxNetworkRetries)
}

然后在测试初始化​​：

//stripe
backend := new(mock.Backend)
backend.On("CallRaw", testify.Anything, testify.Anything, testify.Anything, testify.Anything, testify.Anything).
    Return(nil)
backend.On("Call",
    testify.
        MatchedBy(func(req stripe.ParamsContainer) bool {
            customerParams, ok := req.(*stripe.CustomerParams)
            if ok {
                return *customerParams.Email == "failure@email.com"
            }
            return false
        })).
    Return(fmt.Errorf("downstream stripe error"))
backend.On("Call",
    testify.
        MatchedBy(func(req stripe.ParamsContainer) bool {
            customerParams, ok := req.(*stripe.CustomerParams)
            if ok {
                return *customerParams.Email == "success@email.com"
            }
            return false
        })).
    Return(nil)
sc := &client.API{}
sc.Init("", &stripe.Backends{
    API:     backend,
    Connect: backend,
    Uploads: backend,
})

这行得通-但我无法弄清楚如何模拟以实际获得客户？我不想嘲笑client.API。API 代码：https ://github.com/stripe/stripe-go/blob/9c5fd87e31fd4a072b4d92571d67437e329dc9db/customer/client.go#L23

还有其他人这样做吗？:)

谢谢

4

2 回答 2

1

Here the solution for stripe-go v72 (partially inspired by @kozmo answer)

import (
    "bytes"

    "github.com/stretchr/testify/mock"
    "github.com/stripe/stripe-go/v72"
    "github.com/stripe/stripe-go/v72/form"
)

type stripeBackendMock struct {
    mock.Mock
}

func (s *stripeBackendMock) Call(method, path, key string, params stripe.ParamsContainer, v stripe.LastResponseSetter) error {
    args := s.Called(method, path, key, params, v)
    return args.Error(0)
}

func (s *stripeBackendMock) CallStreaming(method, path, key string, params stripe.ParamsContainer, v stripe.StreamingLastResponseSetter) error {
    args := s.Called(method, path, key, params, v)
    return args.Error(0)
}

func (s *stripeBackendMock) CallRaw(method, path, key string, body *form.Values, params *stripe.Params, v stripe.LastResponseSetter) error {
    args := s.Called(method, path, key, params, v)
    return args.Error(0)
}

func (s *stripeBackendMock) CallMultipart(method, path, key, boundary string, body *bytes.Buffer, params *stripe.Params, v stripe.LastResponseSetter) error {
    args := s.Called(method, path, key, boundary, body, params, v)
    return args.Error(0)
}

func (s *stripeBackendMock) SetMaxNetworkRetries(maxNetworkRetries int64) {
    s.Called(maxNetworkRetries)
}

Then in your tests

stripeBackendMock := new(stripeBackendMock)    
stripeTestBackends := &stripe.Backends{
    API: stripeBackendMock,
    Connect: stripeBackendMock,
    Uploads: stripeBackendMock,
}

stripe = client.New("sk_test", stripeTestBackends)
// ...

stripeBackendMock.On("Call", mock.Anything, mock.Anything, mock.Anything, mock.Anything, mock.Anything).Run(func(args mock.Arguments) {
    stripeCustomer := args.Get(4).(*stripe.Customer)
    *stripeCustomer = stripe.Customer{ /* mock here */ }
}).Return(nil).Once()
// check stripe.Customers.New(...)
于 2022-02-10T22:43:57.830 回答
0

我认为你不正确地实现了接口。

我的实现github.com/stripe/stripe-go/v71 v71.27.0：

import (
    "bytes"
    "github.com/stretchr/testify/mock"
    "github.com/stripe/stripe-go/v71"
    "github.com/stripe/stripe-go/v71/form"
)

type Backend struct {
    mock.Mock
}

func (s *Backend) Call(method, path, key string, params stripe.ParamsContainer, v stripe.LastResponseSetter) error {
    args := s.Called(method, path, key, params, v)
    return args.Error(0)
}

func (s *Backend) CallRaw(method, path, key string, body *form.Values, params *stripe.Params, v stripe.LastResponseSetter) error {
    args := s.Called(method, path, key, body, params, v)
    return args.Error(0)
}

func (s *Backend) CallMultipart(method, path, key, boundary string, body *bytes.Buffer, params *stripe.Params, v stripe.LastResponseSetter) error {
    args := s.Called(method, path, key, boundary, body, params, v)
    return args.Error(0)
}

func (s *Backend) SetMaxNetworkRetries(maxNetworkRetries int64) {
    s.Called(maxNetworkRetries)
}

要实现mock，您必须使用指针作为目标func (s *Backend) SetMaxNetworkRetries(maxNetworkRetries int64){...}，而不是将目标 ( func (s Backend) SetMaxNetworkRetries(maxNetworkRetries int64){...}) 复制到方法。

Аlso 你必须严格遵循方法签名。

于 2020-06-23T14:47:07.127 回答