func init() {
var err error
// Create new bold db
db, err := bolt.Open(dbName, 0600, nil)
if err != nil {
log.Fatal(err)
}
// Create new bucket bucketName
err = db.Update(func(tx *bolt.Tx) error {
_, err = tx.CreateBucketIfNotExists([]byte(bucketName))
if err != nil {
return fmt.Errorf("create bucket: %s", err)
}
return nil
})
if err != nil {
log.Fatal(err)
}
// Tell task package to use Bolt store
task.RegisterDB(&BoltStore{db})
}
func TestRegisterDB(t *testing.T) {
// nothing to test here
task.RegisterDB(&testdb.TestDB{})
}
func init() {
task.RegisterDB(&TestDB{})
}