-1

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

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

我SetupSuite()的我integration_Test.go是这样的：

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)
}

问题是：如果我router.HandleRequests()从 中删除SetupSuite()，我所有的数据库测试都会正常运行，但是如果我尝试http.ListenAndServe()停止测试工作流并且什​​么也没有发生。

我相信我应该使用 goroutinesrouter.HandleRequests()以便它可以与测试并行运行，我只是不知道该怎么做。

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

4

1 回答 1

1

我想出了如何使用 goroutines 来做到这一点。

我读了这个例子，它教如何处理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.Signalchanel，它将监听来自系统的传入信号（SIGINT 和 SIGTERM），然后该函数将停止在<-quit，它将等待一个信号继续。

然后我修复了我在 goroutineSetupSuite()中运行：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 信号时，quit我之前创建的通道将监听该信号，router.HandleRequests()并且该函数将继续终止程序。

func (s *ReceiptServiceTestSuite) TearDownSuite() {
    // some database code 
    p, _ := os.FindProcess(os.Getpid())
    p.Signal(syscall.SIGINT)
}
于 2022-01-28T18:28:08.180 回答