Функция тестирования, которая возвращает интерфейс

Question

Мне сложно написать тест для приведенного ниже случая.

Я могу написать тест для "помощника" с фиктивным объектом, который реализует только те функции, которые используются мной.

Как мне написать тестовый код для функции 'new', используя фиктивный объект без фиктивных функций C (), D ()?

Возможно, что другой пакет написан плохо, что он не должен возвращатьсяИнтерфейс, а точнее фактическая структура?

package main

import (
    "fmt"
)

func main() {
    New()
}

func New() {
    new(NewFromEnvironment)
}

type newTopology func()(Interface,error)

// new is non-exposed simply used for testing purpose.
func new(newTopology newTopology) {
  t,_ :=  newTopology()
  helper(t)
}

// I need to call only A and B
type topologyInterface interface {
    A() string
    B() string
}

func helper(topology topologyInterface) {
    s1 := topology.A()
    s2 := topology.B()
    fmt.Println(s1 + "," + s2)
}

// Below are from other package named "topology".
// I have no control to the code below.

type Interface interface {
    A() string
    B() string
    C() string
    D() string
    //... more
}

func NewFromEnvironment() (Interface, error) {
    return P{}, nil
}

type P struct{}

func (p P) A() string {
    return "A"
}

func (p P) B() string {
    return "B"
}

func (p P) C() string {
    return "C"
}

func (p P) D() string {
    return "D"
}

// more...

Answer 1

Вы можете попробовать создать структуру MockP, которая встраивает P. Затем MockP наследует все методы P, но вы можете затенять A() и B() с помощью своих собственных ложных реализаций. Вот пример:

package main

import (
    "fmt"
)

type Interface interface {
    A()
    B()
}

type P struct {
}

func (p P) A() {
    fmt.Println("A")
}


func (p P) B() {
    fmt.Println("B")
}

type MockP struct {
    P
}

// Shadow P's B() method
func (p MockP) B() {
    fmt.Println("Mock B")
}

func main() {
    var p Interface = MockP{}
    p.A()
    p.B()
}

Выход:

A
Mock B