使用stretchr/testify套件在测试上运行http.ListenAndServe() 停止继续测试

本篇文章向大家介绍《使用stretchr/testify套件在测试上运行http.ListenAndServe() 停止继续测试》，主要包括，具有一定的参考价值，需要的朋友可以参考一下。

我正在尝试为我使用 gorilla/mux、gorm.io 和 golang-migrate/v4 制作的 rest api 应用程序创建集成测试

对于我正在使用 testify 的测试。

我的 integration_test.go 中的 setupsuite() 是这样的：

func (s *receiptservicetestsuite) setupsuite() {
    s.require().noerror(godotenv.load("test.env"))
    s.require().noerror(database.connect())
    s.db = database.db
    s.m = database.m

    router.handlerequests()
}

我的 router.handlerequests() 是这样的：

func HandleRequests() {
    router := mux.NewRouter()
    router.Use(middleware)
    // lots of router.HandleFunc()
    
    http.ListenAndServe(":8080", router)
}

问题是：如果我从 setupsuite() 中删除 router.handlerequests()，我的所有数据库测试都会正常运行，但如果我尝试 http.listenandserve() 测试工作流程会停止并且不会发生任何事情。

我相信我应该将 goroutine 与 router.handlerequests() 一起使用，这样它就可以与测试并行运行，我只是不知道该怎么做。

有关更多信息，这里是项目存储库，我不知道它是否相关，但我正在使用 docker-compose 运行两个 postgres 实例，一个用于运行项目，另一个用于测试。

正确答案

我想出了如何使用 go 例程来做到这一点。

我读了这篇 example，它教如何使用通道处理 os.signals ，我这样实现了它：

首先，我在路由器上进行了以下更改：

func handlerequests() {
    router := mux.newrouter()
    router.use(middleware)
    // lots of router.handlefunc()
    
    go http.listenandserve(":8080", router)

    quit := make(chan os.signal, 1)
    signal.notify(quit, syscall.sigint, syscall.sigterm)

    <-quit

}

现在 http.listenandserve() 正在新线程中运行。我相信，默认情况下，它会在程序执行结束时停止。

在 http.listenandserve() 之后，我创建了一个 os.signal 通道，它将侦听来自系统的传入信号（sigint 和 sigterm），然后该函数将停止在 <-quit 处，它将等待一个信号继续。

然后我修复了 setupsuite() 以在 goroutine 中运行 router.handlerequests()：

func (s *receiptservicetestsuite) setupsuite() {
    s.require().noerror(godotenv.load("test.env"))
    s.require().noerror(database.connect())
    s.db = database.db
    s.m = database.m

    go router.handlerequests()
}

在我的 teardownsuite() 中，我向当前进程发送一个 sigterm 信号，该信号将由我之前在 router.handlerequests() 创建的 quit 通道侦听，并且该函数将继续终止程序。

func (s *ReceiptServiceTestSuite) TearDownSuite() {
    // some database code 
    p, _ := os.FindProcess(os.Getpid())
    p.Signal(syscall.SIGINT)
}

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。