56 lines
1.7 KiB
Go
56 lines
1.7 KiB
Go
|
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
|
||
|
|
// 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
|