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{}) }