单元测试是测试单个功能或组件的隔离性测试,使用 testing 包编写;集成测试是测试多个组件或系统交互的端到端测试,可以使用 httptest 包进行。1. 单元测试:编写单元测试来验证单个功能或组件的正确性。2. 集成测试:编写集成测试来验证多个组件或系统交互的正确性。3. 实战案例:展示单元测试和集成测试在验证 getuser 函数中的应用。
Go 框架的单元测试与集成测试
单元测试
单元测试是测试单个功能或组件的隔离性测试。在 Go 中,我们可以使用 testing 包编写单元测试。
立即学习“go语言免费学习笔记(深入)”;
package mypackage
import (
"testing"
)
func TestAdd(t *testing.T) {
sum := Add(1, 2)
if sum != 3 {
t.Errorf("Add(1, 2) = %d; want 3", sum)
}
}
集成测试
集成测试是测试多个组件或系统交互的端到端测试。它们比单元测试复杂,但可以确保整个系统的正确性。在 Go 中,我们可以使用 httptest 包进行集成测试。
package mypackage
import (
"net/http"
"net/http/httptest"
"testing"
)
func TestHelloWorldHandler(t *testing.T) {
req, err := http.NewRequest(http.MethodGet, "/", nil)
if err != nil {
t.Fatalf("http.NewRequest: %v", err)
}
rr := httptest.NewRecorder()
HelloWorldHandler(rr, req)
if rr.Code != http.StatusOK {
t.Errorf("HelloWorldHandler returned status %d; want %d", rr.Code, http.StatusOK)
}
if rr.Body.String() != "Hello, World!" {
t.Errorf("HelloWorldHandler returned body %q; want %q", rr.Body.String(), "Hello, World!")
}
}
实战案例
假设我们有一个 GetUser 函数,可以从数据库中获取用户。我们可以编写单元测试和集成测试来验证其正确性。
单元测试
package mypackage
import (
"database/sql"
"testing"
)
func TestGetUser(t *testing.T) {
db, err := sql.Open("postgres", "host=localhost user=root password=root")
if err != nil {
t.Fatalf("sql.Open: %v", err)
}
user, err := GetUser(db, 1)
if err != nil {
t.Fatalf("GetUser: %v", err)
}
if user.ID != 1 || user.Name != "John Doe" {
t.Errorf("GetUser returned unexpected user: %v", user)
}
}
集成测试
package mypackage
import (
"net/http"
"net/http/httptest"
"testing"
)
func TestGetUserHandler(t *testing.T) {
db, err := sql.Open("postgres", "host=localhost user=root password=root")
if err != nil {
t.Fatalf("sql.Open: %v", err)
}
req, err := http.NewRequest(http.MethodGet, "/users/1", nil)
if err != nil {
t.Fatalf("http.NewRequest: %v", err)
}
rr := httptest.NewRecorder()
GetUserHandler(db, rr, req)
if rr.Code != http.StatusOK {
t.Errorf("GetUserHandler returned status %d; want %d", rr.Code, http.StatusOK)
}
if rr.Body.String() != "{"id":1,"name":"John Doe"}" {
t.Errorf("GetUserHandler returned body %q; want %q", rr.Body.String(), "{"id":1,"name":"John Doe"}")
}
}