使用 Gin、GORM、Testify 和 PostgreSQL 在 Golang 中建立全面的整合測試設定涉及設定測試資料庫、為 CRUD 操作編寫測試以及使用 Testify 進行斷言。這是幫助您入門的逐步指南：
myapp/ |-- main.go |-- models/ | |-- models.go |-- handlers/ | |-- handlers.go |-- tests/ | |-- integration_test.go |-- go.mod |-- go.sum
使用 GORM 標籤定義模型以進行資料庫對應。
package models
import (
"time"
"gorm.io/gorm"
)
type User struct {
ID uint `gorm:"primaryKey"`
Name string `gorm:"not null"`
Email string `gorm:"unique;not null"`
CreatedAt time.Time
}
type Book struct {
ID uint `gorm:"primaryKey"`
Title string `gorm:"not null"`
Author string `gorm:"not null"`
PublishedDate time.Time `gorm:"not null"`
}
type BorrowLog struct {
ID uint `gorm:"primaryKey"`
UserID uint `gorm:"not null"`
BookID uint `gorm:"not null"`
BorrowedAt time.Time `gorm:"default:CURRENT_TIMESTAMP"`
ReturnedAt *time.Time
}
使用 Gin 定義 CRUD 操作的路由和處理程序。
package handlers
import (
"myapp/models"
"net/http"
"github.com/gin-gonic/gin"
"gorm.io/gorm"
)
type Handler struct {
DB *gorm.DB
}
func (h *Handler) CreateUser(c *gin.Context) {
var user models.User
if err := c.ShouldBindJSON(&user); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
if err := h.DB.Create(&user).Error; err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
return
}
c.JSON(http.StatusCreated, user)
}
func (h *Handler) GetUser(c *gin.Context) {
var user models.User
if err := h.DB.First(&user, c.Param("id")).Error; err != nil {
c.JSON(http.StatusNotFound, gin.H{"error": "User not found"})
return
}
c.JSON(http.StatusOK, user)
}
func (h *Handler) UpdateUser(c *gin.Context) {
var user models.User
if err := h.DB.First(&user, c.Param("id")).Error; err != nil {
c.JSON(http.StatusNotFound, gin.H{"error": "User not found"})
return
}
if err := c.ShouldBindJSON(&user); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
if err := h.DB.Save(&user).Error; err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
return
}
c.JSON(http.StatusOK, user)
}
func (h *Handler) DeleteUser(c *gin.Context) {
if err := h.DB.Delete(&models.User{}, c.Param("id")).Error; err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
return
}
c.JSON(http.StatusOK, gin.H{"message": "User deleted"})
}
設定資料庫連接和路由。
package main
import (
"myapp/handlers"
"myapp/models"
"github.com/gin-gonic/gin"
"gorm.io/driver/postgres"
"gorm.io/gorm"
"log"
"os"
)
func main() {
dsn := "host=localhost user=postgres password=yourpassword dbname=testdb port=5432 sslmode=disable"
db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
if err != nil {
log.Fatalf("failed to connect to database: %v", err)
}
// Auto migrate the models
db.AutoMigrate(&models.User{}, &models.Book{}, &models.BorrowLog{})
h := handlers.Handler{DB: db}
r := gin.Default()
r.POST("/users", h.CreateUser)
r.GET("/users/:id", h.GetUser)
r.PUT("/users/:id", h.UpdateUser)
r.DELETE("/users/:id", h.DeleteUser)
r.Run(":8080")
}
使用 Testify 設定和斷言測試結果。
對於資料庫，我們可以使用 Dockerized PostgreSQL 實例進行測試，該實例是隔離的，測試後可以快速拆除。以下是如何使用 testcontainers-go 在 Golang 中進行設定：
安裝 testcontainers-go:
go get github.com/testcontainers/testcontainers-go
以下是integration_test.go 文件，用於設定測試的 PostgreSQL 容器：
myapp/ |-- main.go |-- models/ | |-- models.go |-- handlers/ | |-- handlers.go |-- tests/ | |-- integration_test.go |-- go.mod |-- go.sum
使用以下命令執行測試：
myapp/ |-- main.go |-- models/ | |-- models.go |-- handlers/ | |-- handlers.go |-- tests/ | |-- integration_test.go |-- go.mod |-- go.sum
透過此設置，您可以測試使用者模型的 CRUD 操作，確保 API 按預期與 PostgreSQL 搭配使用。您可以類似地對 Book 和 BorrowLog 模型擴展測試。
以上是Golang 與 Gin、Gorm、Testify、PostgreSQL 的整合測試的詳細內容。更多資訊請關注PHP中文網其他相關文章！
