package aetest import ( "io" "net/http" "time" "golang.org/x/net/context" "google.golang.org/appengine" ) // Instance represents a running instance of the development API Server. type Instance interface { // Close kills the child api_server.py process, releasing its resources. io.Closer // NewRequest returns an *http.Request associated with this instance. NewRequest(method, urlStr string, body io.Reader) (*http.Request, error) } // Options is used to specify options when creating an Instance. type Options struct { // AppID specifies the App ID to use during tests. // By default, "testapp". AppID string // StronglyConsistentDatastore is whether the local datastore should be // strongly consistent. This will diverge from production behaviour. StronglyConsistentDatastore bool // SuppressDevAppServerLog is whether the dev_appserver running in tests // should output logs. SuppressDevAppServerLog bool // StartupTimeout is a duration to wait for instance startup. // By default, 15 seconds. StartupTimeout time.Duration } // NewContext starts an instance of the development API server, and returns // a context that will route all API calls to that server, as well as a // closure that must be called when the Context is no longer required. func NewContext() (context.Context, func(), error) { inst, err := NewInstance(nil) if err != nil { return nil, nil, err } req, err := inst.NewRequest("GET", "/", nil) if err != nil { inst.Close() return nil, nil, err } ctx := appengine.NewContext(req) return ctx, func() { inst.Close() }, nil } // PrepareDevAppserver is a hook which, if set, will be called before the // dev_appserver.py is started, each time it is started. If aetest.NewContext // is invoked from the goapp test tool, this hook is unnecessary. var PrepareDevAppserver func() error