You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
sisyphus/database_test.go

74 lines
1.4 KiB
Go

package sisyphus_test
import (
"os"
"github.com/boltdb/bolt"
. "github.com/carlostrub/sisyphus"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
)
var _ = Describe("Database", func() {
Context("Bolt Database", func() {
AfterEach(func() {
err = os.Remove("test/Maildir/sisyphus.db")
Ω(err).ShouldNot(HaveOccurred())
})
It("Load databases for each maildir", func() {
dbs, err := LoadDatabases([]Maildir{"test/Maildir"})
Ω(err).ShouldNot(HaveOccurred())
dbTest := dbs["test/Maildir"]
var gN = 4
var jN = 4
var sN = 4
err = dbTest.View(func(tx *bolt.Tx) error {
b := tx.Bucket([]byte("Wordlists"))
good := b.Bucket([]byte("Good"))
gN = good.Stats().KeyN
junk := b.Bucket([]byte("Junk"))
jN = junk.Stats().KeyN
s := tx.Bucket([]byte("Statistics"))
sN = s.Stats().KeyN
return nil
})
Ω(err).ShouldNot(HaveOccurred())
Ω(gN).Should(Equal(0))
Ω(jN).Should(Equal(0))
Ω(sN).Should(Equal(0))
CloseDatabases(dbs)
})
It("Closes an open database", func() {
dbs, err := LoadDatabases([]Maildir{"test/Maildir"})
Ω(err).ShouldNot(HaveOccurred())
CloseDatabases(dbs)
dbTest := dbs["test/Maildir"]
n := 4
err = dbTest.View(func(tx *bolt.Tx) error {
b := tx.Bucket([]byte("Statistics"))
n = b.Stats().KeyN
return nil
})
Ω(err).Should(HaveOccurred())
Ω(n).Should(Equal(4))
})
})
})