func (t *ModelTestSuite) SetUpTest(c *C) {
memGraph, _ := cayley.NewMemoryGraph()
memNg, _ := graph.NewGraph(memGraph)
t.nodeGraph, t.tmpDir = testutils.TestInit()
t.server = httptest.NewServer(api.NewApi(memNg))
t.client = apiclient.ApiClient{t.server.URL, api.JsonEncoding}
}
func NewManager(client apiclient.OlympusClient, handle *cayley.Handle) *Manager {
manager := new(Manager)
if ng, err := graph.NewGraph(handle); err != nil {
panic(err)
} else {
manager.graph = ng
}
manager.api = client
return manager
}
func initDb() (*graph.NodeGraph, error) {
var handle *cayley.Handle
var err error
if !debug {
dbPath := filepath.Join(env.EnvPath(env.DbPath), "db.dat")
if !env.Exists(dbPath) {
if err = cgraph.InitQuadStore("bolt", dbPath, nil); err != nil {
return nil, err
}
}
if handle, err = cayley.NewGraph("bolt", dbPath, nil); err != nil {
return nil, err
}
} else {
if handle, err = cayley.NewMemoryGraph(); err != nil {
return nil, err
}
}
return graph.NewGraph(handle)
}
func TestInit() (*graph.NodeGraph, string) {
if dir, err := ioutil.TempDir(os.TempDir(), ".olympus"); err != nil {
panic(err)
} else {
os.Setenv("OLYMPUS_HOME", dir)
if err = env.InitializeEnvironment(); err != nil {
panic(err)
}
dbPath := filepath.Join(env.EnvPath(env.DbPath), "db.dat")
if !env.Exists(dbPath) {
cgraph.InitQuadStore("bolt", dbPath, nil)
if handle, err := cayley.NewGraph("bolt", dbPath, nil); err != nil {
panic(err)
} else if ng, err := graph.NewGraph(handle); err != nil {
panic(err)
} else {
return ng, dir
}
} else {
return nil, ""
}
}
}