54 lines
1.0 KiB
Go
54 lines
1.0 KiB
Go
package main
|
|
|
|
import (
|
|
"path/filepath"
|
|
|
|
. "gopkg.in/check.v1"
|
|
)
|
|
|
|
type AlbumDatabaseSuite struct {
|
|
db *AlbumDatabase
|
|
}
|
|
|
|
var _ = Suite(&AlbumDatabaseSuite{})
|
|
|
|
func (s *AlbumDatabaseSuite) SetUpSuite(c *C) {
|
|
var err error
|
|
s.db, err = OpenAlbumDatabase(filepath.Join(c.MkDir(), "satdb.bar.satellite/db"))
|
|
c.Assert(err, IsNil)
|
|
|
|
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
|
|
data := []AlbumID{}
|
|
sorted, err := s.db.ByPurchaseDate()
|
|
c.Assert(err, IsNil)
|
|
c.Assert(len(sorted), Equals, len(data))
|
|
for i, a := range sorted {
|
|
c.Check(a, Equals, data[i])
|
|
}
|
|
}
|