2
0
mirror of https://github.com/carlostrub/sisyphus synced 2024-11-13 13:10:40 +00:00
sisyphus/database_test.go
2017-05-13 22:34:54 +00:00

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