如何对Google App Engine Go HTTP处理程序进行单元测试?

时间:2013-10-16 15:21:19

标签: google-app-engine unit-testing go

本地设备测试距离Google App Engine Go SDK版本1.8.6 supportedappengine/aetest包允许我使用。

创建Context到单元测试

如何在net/http/httptest中使用它来测试我的HTTP处理程序?

2 个答案:

答案 0 :(得分:16)

请参阅goroot / src / pkg / appengine / aetest / context.go的顶部(更新的源尚未发布在https://code.google.com/p/appengine-go)。乍一看,新的测试应用程序看起来是appenginetesting稍微更强大/不同的版本,因此您可以进行相同类型的测试,请参阅here了解sampleHandler(w)的一种方法http.ResponseWriter,r * http.Request)被调用。

或者,您可以将http.Handler的ContextHandler设置如下:

type ContextHandler struct {
    Real func(*appengine.Context, http.ResponseWriter, *http.Request)
}

func (f ContextHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    c := appengine.NewContext(r)
    f.Real(c, w, r)
}

func myNewHandler(c appengine.Context, w http.ResponseWriter, r *http.Request) {
// do something
}

然后你可以在init()中执行此操作以支持生产:

http.Handle("/myNewHandler", ContextHandler{myNewHandler})

这使得测试功能变得简单:

func TestMyNewHandler(t *testing.T) {
    c := aetest.NewContext()
    r, _ := http.NewRequest("GET", "/tasks/findOverdueSchedules", nil)
    w := httptest.NewRecorder()
    myNewHandler(c, w, r)
    if 200 != w.Code {
        t.Fail()
    }
}

这是来自appengine / aetest中的context.go的内容:

  

/ *   包aetest提供了一个appengine.Context用于测试。

     

示例测试文件:       包foo_test

import (
    "testing"

    "appengine/memcache"
    "appengine/aetest"
)

func TestFoo(t *testing.T) {
    c, err := aetest.NewContext(nil)
    if err != nil {
        t.Fatal(err)
    }
    defer c.Close()

    it := &memcache.Item{
        Key:   "some-key",
        Value: []byte("some-value"),
    }
    err = memcache.Set(c, it)
    if err != nil {
        t.Fatalf("Set err: %v", err)
    }
    it, err = memcache.Get(c, "some-key")
    if err != nil {
        t.Fatalf("Get err: %v; want no error", err)
    }
    if g, w := string(it.Value), "some-value" ; g != w {
        t.Errorf("retrieved Item.Value = %q, want %q", g, w)
    }
}
  

环境变量APPENGINE_API_SERVER指定的位置   要使用的api_server.py可执行文件。如果未设置,则咨询系统PATH。   * /

答案 1 :(得分:1)

如果您不反对使用Martini,依赖注入是解决问题的好方法。以下是设置测试的方法(使用Ginkgo):

var _ = Describe("Items", func() {

    var (
        m *martini.ClassicMartini
    )

    BeforeEach(func() {
        m = martini.Classic()

        // injects app engine context into requests
        m.Use(func(c martini.Context, req *http.Request) {
             con, _ := aetest.NewContext(nil)
             c.MapTo(con, (*appengine.Context)(nil))
        })

        m.Get("/items", func(c martini.Context){
             // code you want to test
        })
    })

    It("should get items", func() {
        recorder := httptest.NewRecorder()
        r, _ := http.NewRequest("GET", "/items", nil)
        m.ServeHTTP(recorder, r) // martini server used
        Expect(recorder.Code).To(Equal(200))
    })
})

在制作过程中会注入实际情况:

m.Use(func(c martini.Context, req *http.Request) {
    c.MapTo(appengine.NewContext(req), (*appengine.Context)(nil))
})