使用 Go 进行 App Engine 服务代码单元测试教程

go get github.com/icub3d/appenginetesting

package myapp

import (
    "context"
    "net/http"
    "net/http/httptest"
    "testing"

    "github.com/icub3d/appenginetesting"
    "google.golang.org/appengine/datastore"
)

type MyEntity struct {
    Name string
}

func SaveEntity(ctx context.Context, name string) error {
    key := datastore.NewIncompleteKey(ctx, "MyEntity", nil)
    entity := MyEntity{Name: name}
    _, err := datastore.Put(ctx, key, &entity)
    return err
}

func MyHandler(w http.ResponseWriter, r *http.Request) {
    ctx := r.Context()
    name := r.FormValue("name")
    if err := SaveEntity(ctx, name); err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }
    w.WriteHeader(http.StatusOK)
}

func TestMyHandler(t *testing.T) {
    // 创建一个 App Engine 上下文
    ctx, closer := appenginetesting.NewTestContext(t)
    defer closer() // 确保在测试结束时关闭上下文

    // 创建一个 HTTP 请求
    req, err := http.NewRequest("POST", "/myhandler", nil)
    if err != nil {
        t.Fatalf("创建请求失败: %v", err)
    }
    req = req.WithContext(ctx)
    req.Form = map[string][]string{"name": {"testName"}}

    // 创建一个 HTTP 响应记录器
    recorder := httptest.NewRecorder()

    // 调用 handler
    MyHandler(recorder, req)

    // 检查响应状态码
    if recorder.Code != http.StatusOK {
        t.Errorf("期望状态码 %d, 实际 %d", http.StatusOK, recorder.Code)
    }

    // 验证数据是否已保存到 Datastore
    q := datastore.NewQuery("MyEntity")
    var entities []MyEntity
    _, err = q.GetAll(ctx, &entities)
    if err != nil {
        t.Fatalf("查询 Datastore 失败: %v", err)
    }

    if len(entities) != 1 {
        t.Errorf("期望 1 个实体, 实际 %d", len(entities))
    }

    if entities[0].Name != "testName" {
        t.Errorf("期望实体名称 'testName', 实际 '%s'", entities[0].Name)
    }
}