如何在Golang嘲笑Kafka的依赖性和作家？

Question

我有一个结构，

type Server struct {
    grpcServerPort  int
    grpcServer      *grpc.Server
    writer *kafka.Writer

}

该服务器有很多方法，包括 -

func NewServer(port int) *Server {
    server := &Server{
        grpcServerPort: port,
    }
    gs := grpc.NewServer()
    server.grpcServer = gs

    return server
}


func (s *Server) StartServer() {
    createKafkaTopic("brokker_url", "my_topic")

    s.writer = &kafka.Writer{
        Addr:        kafka.TCP("urls"),
        Topic:       "my_topic",
        Balancer:    &kafka.Hash{},
        MaxAttempts: 1,
        BatchSize:   1,
    }

    listener, err := net.Listen("tcp", fmt.Sprintf(":%d", s.grpcServerPort))
    if err != nil {
        fmt.Printf("failed to listen: %v", err)
    }

    go s.grpcServer.Serve(listener)
}

func (s *Server) produceEvents(key string, val string) error {

    msg := kafka.Message{
        Key:   []byte(key),
        Value: []byte(val),
    }
    err = s.writer.WriteMessages(context.Background(), msg)
    if err != nil {
        return err
    }
    return nil

}

现在肌方法，如果没有Kafka Brok在我的机器上启动并运行的话。我想知道如何模拟肌肉和 productevents ，以便我可以为其他服务器方法编写单元测试用例。

Answer 1

您可以定义接口，然后定义模拟结构以实现它，现在您可以对其进行测试。

type Server struct {
    grpcServerPort int
    grpcServer     *grpc.Server
    writer         KafkaWriterWrapper
}

type KafkaWriterWrapper interface {
    Write(msg string) error // Suppose kafka.Writer has func Write(msg string) error
}

type KafkaWriterWrapperImpl struct {
    *kafka.Writer
}

type MockKafkaWriterWrapperImpl struct {
}

func (m *MockKafkaWriterWrapperImpl) Write(msg string) error {
    // logic
    return nil
}

Answer 2

上周我刚刚遇到了你的案子。这就是我的解决方法：

type KafkaWriter interface {
    Method001fKafkaWriter()
    Method002fKafkaWriter()
}

type Server struct {
    grpcServerPort  int
    grpcServer      *grpc.Server
    writer KafkaWriter
}

现在，您可以用接口嘲笑Kafka依赖关系，

type mockKafkaWriter struct {
}

func (*mockKafkaWriter) Method001fKafkaWriter() {}
func (*mockKafkaWriter) Method002fKafkaWriter() {}

我希望这可以为您提供帮助！