2

我正在努力理解 Go 中的模拟(正在寻找与 Go 中的 java 等效的 Mockito.spy 相关的东西)。

假设我在 Go 中有一个带有 5 个方法的接口。但是我要测试的那段代码只引用了两种方法。现在我如何在不实现所有方法的情况下模拟这种依赖关系,即我在源代码中的实际实现实现了 5 个接口方法,但是有没有办法避免在测试文件中实现 5 个方法的虚拟接口实现。以下是我目前正在做的方式,实现5个方法是可以管理的,但是如果接口有20个方法怎么办,模拟实现测试文件中的所有方法变得乏味。

例子:

handler.go 中的源代码:

type Client struct {}
type ClientStore interface {
  func(c *Client) methodOne() error {// some implementation}
  func(c *Client) methodTwo() error {// some implementation}
  func(c *Client) methodThree() error {// some implementation}
  func(c *Client) methodFour() error {// some implementation}
  func(c *Client) methodFive() error {// some implementation}
}

api.go 中的源代码:

 func processFeed(c Client) error {
     err := c.methodOne()
     if(err != null) {
      return err
    }
     err1 := c.methodTwo()
     if(err1 != null) {
      return err1
    }
 }

测试类代码:

import "testify/mock"

func TestFeed(t *testing.T){
   mockClient := &MockClient{}
   err := processFeed(mockClient)
   assert.NotNil(t , err)
 
}

type MockClient struct {
  mock.Mock
}

  func(c *MockClient ) methodOne() error {fmt.Printf("methodOne");nil}
  func(c *MockClient ) methodTwo() error {return errors.New("mocked error")}
  func(c *MockClient ) methodThree() error {fmt.Printf("methodThree");nil}
  func(c *MockClient ) methodFour() error {fmt.Printf("methodFour");nil}
  func(c *MockClient ) methodFive() error {fmt.Printf("methodFive");nil}


问题:

有没有办法只模拟我在上述情况下只需要 methodOne() 和 methodTwo() 方法,而不用担心测试中的剩余方法?如果存在,您能否建议任何其他替代方案?谢谢

4

1 回答 1

6

首先,如果您的界面有 5 个方法,而您只使用一个,那么您的界面太大了。使用更小的接口。

type BigInterface interface {
    Thing1()
    Thing2()
    ThingN()
}

type SmallInterface interface {
    Thing1()
}

func MyFunc(i SmallInterface) { ... }

另一种选择是通过嵌入完整接口来创建完整接口的完整实现。如果您尝试访问其他方法之一,这将导致恐慌,但如果您小心的话,它将适用于测试。(但请不要在生产代码中这样做!)

type BigInterface interface {
    Thing1()
    Thing2()
    ThingN()
}

type myImplementation struct {
    BigInterface
}

func (i *myImplementation) Thing1() { ... }

现在myImplementation通过BigInterface包含BigInterface. 如果您从未将该嵌入式实例设置为任何内容,那么调用这些方法将会恐慌,但您仍然可以定义Thing1为您的测试执行您想要的操作。

于 2019-07-05T18:07:25.677 回答