// runTest initializes the environment for the tests and allows for
// the proper return code if the test fails or succeeds.
func runTest(m *testing.M) int {
// Create stub server for Sponged.
server := setup()
cfg.SetString("SPONGED_URL", server)
mongoURI := cfg.MustURL("MONGO_URI")
// Initialize MongoDB using the `tests.TestSession` as the name of the
// master session.
if err := db.RegMasterSession(tests.Context, tests.TestSession, mongoURI.String(), 0); err != nil {
fmt.Println("Can't register master session: " + err.Error())
return 1
}
a = routes.API()
// Snatch the mongo session so we can create some test data.
db, err := db.NewMGO(tests.Context, tests.TestSession)
if err != nil {
fmt.Println("Unable to get Mongo session")
return 1
}
defer db.CloseMGO(tests.Context)
if err = db.NewCayley(tests.Context, tests.TestSession); err != nil {
fmt.Println("Unable to get Cayley support")
}
store, err := db.GraphHandle(tests.Context)
if err != nil {
fmt.Println("Unable to get Cayley handle")
return 1
}
defer store.Close()
if err := tstdata.Generate(db); err != nil {
fmt.Println("Could not generate test data.")
return 1
}
defer tstdata.Drop(db)
if err := loadItems("context", db, store); err != nil {
fmt.Println("Could not import items")
return 1
}
defer unloadItems("context", db, store)
return m.Run()
}
func main() {
log.User("startup", "Init", "Revision : %q", GitRevision)
log.User("startup", "Init", "Version : %q", GitVersion)
log.User("startup", "Init", "Build Date : %q", BuildDate)
log.User("startup", "Init", "Int Version : %q", IntVersion)
log.User("startup", "Init", "Go Version : %q", runtime.Version())
log.User("startup", "Init", "Go Compiler : %q", runtime.Compiler)
log.User("startup", "Init", "Go ARCH : %q", runtime.GOARCH)
log.User("startup", "Init", "Go OS : %q", runtime.GOOS)
handlers.Version.GitRevision = GitRevision
handlers.Version.GitVersion = GitVersion
handlers.Version.BuildDate = BuildDate
handlers.Version.IntVersion = IntVersion
// These are the absolute read and write timeouts.
const (
// ReadTimeout covers the time from when the connection is accepted to when the
// request body is fully read.
readTimeout = 10 * time.Second
// WriteTimeout normally covers the time from the end of the request header read
// to the end of the response write.
writeTimeout = 30 * time.Second
)
host := cfg.MustString(cfgHost)
log.User("startup", "Init", "Binding web service to %s", host)
if err := web.Run(host, routes.API(), readTimeout, writeTimeout); err != nil {
log.Error("shutdown", "Init", err, "App Shutdown")
os.Exit(1)
}
log.User("shutdown", "Init", "App Shutdown")
}
