57 lines
1.2 KiB
Go
57 lines
1.2 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
. "gopkg.in/check.v1"
|
|
)
|
|
|
|
type AlbumDatabaseSuite struct {
|
|
db *AlbumDatabase
|
|
}
|
|
|
|
var _ = Suite(&AlbumDatabaseSuite{})
|
|
|
|
func (s *AlbumDatabaseSuite) SetUpSuite(c *C) {
|
|
s.db = OpenAlbumDatabase(filepath.Join(c.MkDir(), "satdb.bar.satellite/db"))
|
|
for _, a := range albumsDataTest {
|
|
c.Assert(s.db.AddOrUpdate(&a), IsNil)
|
|
}
|
|
}
|
|
|
|
func (s *AlbumDatabaseSuite) TestCanDelete(c *C) {
|
|
defer func() {
|
|
//we can always add it twice
|
|
c.Assert(s.db.AddOrUpdate(&(albumsDataTest[0])), IsNil)
|
|
}()
|
|
|
|
c.Assert(s.db.Delete(albumsDataTest[0].ID), IsNil)
|
|
}
|
|
|
|
func (s *AlbumDatabaseSuite) TestCanGet(c *C) {
|
|
|
|
for _, a := range albumsDataTest {
|
|
fromDb, err := s.db.Get(a.ID)
|
|
if c.Check(err, IsNil) == true {
|
|
c.Check(*fromDb, DeepEquals, a)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (s *AlbumDatabaseSuite) TestCanSort(c *C) {
|
|
// here
|
|
start := time.Now()
|
|
data := []AlbumID{160366, 58595, 15875, 9935, 84448, 46005, 19762, 164, 52100, 8179, 44989, 32043, 22737, 754}
|
|
sorted, err := s.db.ByPurchaseDate()
|
|
|
|
log.Printf("sorting took %s", time.Since(start))
|
|
|
|
c.Assert(err, IsNil)
|
|
c.Assert(len(sorted), Equals, len(data))
|
|
for i, a := range sorted {
|
|
c.Check(a, Equals, data[i], Commentf("expected %d", a))
|
|
}
|
|
}
|