Adds an indexer
This commit is contained in:
62
bleve_indexer_test.go
Normal file
62
bleve_indexer_test.go
Normal file
@@ -0,0 +1,62 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"log"
|
||||
"time"
|
||||
|
||||
. "gopkg.in/check.v1"
|
||||
)
|
||||
|
||||
type BleveIndexerSuite struct {
|
||||
i Indexer
|
||||
}
|
||||
|
||||
var _ = Suite(&BleveIndexerSuite{})
|
||||
|
||||
func (s *BleveIndexerSuite) SetUpSuite(c *C) {
|
||||
var err error
|
||||
start := time.Now()
|
||||
s.i, err = NewBleveIndexer("satbd-test.bar.satellite")
|
||||
c.Assert(err, IsNil)
|
||||
for _, a := range albumsDataTest {
|
||||
c.Assert(s.i.Index(&a), IsNil)
|
||||
}
|
||||
log.Printf("Indexing took %s", time.Since(start))
|
||||
}
|
||||
|
||||
func (s *BleveIndexerSuite) TestCanRetrieveAlbum(c *C) {
|
||||
start := time.Now()
|
||||
for _, a := range albumsDataTest {
|
||||
fromIndex, err := s.i.Get(a.ID)
|
||||
if c.Check(err, IsNil) == true {
|
||||
c.Check(*fromIndex, DeepEquals, a)
|
||||
}
|
||||
}
|
||||
log.Printf("%s: %s", c.TestName(), time.Since(start))
|
||||
}
|
||||
|
||||
func (s *BleveIndexerSuite) TestCanDeleteAlbum(c *C) {
|
||||
c.Check(s.i.Delete(albumsDataTest[0].ID), IsNil)
|
||||
c.Check(s.i.Index(&(albumsDataTest[0])), IsNil)
|
||||
c.Check(s.i.Delete(0), ErrorMatches, "No album 0 in the index")
|
||||
}
|
||||
|
||||
func (s *BleveIndexerSuite) TestCanSearch(c *C) {
|
||||
data := map[string][]AlbumID{
|
||||
"Black Crow Hermione": []AlbumID{albumsDataTest[10].ID},
|
||||
"Plantu": []AlbumID{albumsDataTest[2].ID},
|
||||
"FOOOOOOOBAAAAAR": []AlbumID{},
|
||||
}
|
||||
start := time.Now()
|
||||
for q, expected := range data {
|
||||
res, err := s.i.Search(q)
|
||||
if c.Check(err, IsNil) == true {
|
||||
if c.Check(len(res), Equals, len(expected)) == true {
|
||||
for idx, resAlbum := range res {
|
||||
c.Check(resAlbum.ID, Equals, expected[idx])
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
log.Printf("%s: %s", c.TestName(), time.Since(start))
|
||||
}
|
||||
Reference in New Issue
Block a user