shithub: hugo

Download patch

ref: de76d4a84ef38ed92887ddd78da639000d6e6371
parent: 0c2297075afa28b22485d7c15410216f82a759b3
author: bep <[email protected]>
date: Mon Jan 26 20:27:07 EST 2015

Improve pagination testing

* Add missing pagination test cases
* Remove some unreachable code paths
* Fix some corner cases

--- a/hugolib/pagination.go
+++ b/hugolib/pagination.go
@@ -88,19 +88,11 @@
 
 // First returns the pager for the first page.
 func (p *pager) First() *pager {
-	if p.TotalPages() == 0 {
-		return nil
-	}
-
 	return p.pagers[0]
 }
 
 // Last returns the pager for the last page.
 func (p *pager) Last() *pager {
-	if p.TotalPages() == 0 {
-		return nil
-	}
-
 	return p.pagers[len(p.pagers)-1]
 }
 
@@ -138,10 +130,6 @@
 // If it's not, one will be created with all pages in Data["Pages"].
 func (n *Node) Paginator() (*pager, error) {
 
-	if n.IsPage() {
-		return nil, errors.New("Paginators isn't supported for content pages.")
-	}
-
 	var initError error
 
 	n.paginatorInit.Do(func() {
@@ -170,15 +158,18 @@
 	return n.paginator, nil
 }
 
+func (p *Page) Paginator() (*pager, error) {
+	return nil, errors.New("Paginators not supported for content pages.")
+}
+func (p *Page) Paginate(seq interface{}) (*pager, error) {
+	return nil, errors.New("Paginators not supported for content pages.")
+}
+
 // Paginate gets this Node's paginator if it's already created.
 // If it's not, one will be created with the qiven sequence.
 // Note that repeated calls will return the same result, even if the sequence is different.
 func (n *Node) Paginate(seq interface{}) (*pager, error) {
 
-	if n.IsPage() {
-		return nil, errors.New("Paginators isn't supported for content pages.")
-	}
-
 	var initError error
 
 	n.paginatorInit.Do(func() {
@@ -227,16 +218,16 @@
 	}
 
 	urlFactory := newPaginationUrlFactory(section)
-	paginator := newPaginator(pages, paginateSize, urlFactory)
+	paginator, _ := newPaginator(pages, paginateSize, urlFactory)
 	pagers := paginator.Pagers()
 
 	return pagers, nil
 }
 
-func newPaginator(pages Pages, size int, urlFactory paginationUrlFactory) *paginator {
+func newPaginator(pages Pages, size int, urlFactory paginationUrlFactory) (*paginator, error) {
 
 	if size <= 0 {
-		panic("Paginator size must be positive")
+		return nil, errors.New("Paginator size must be positive")
 	}
 
 	split := splitPages(pages, size)
@@ -250,7 +241,7 @@
 
 	p.pagers = pagers
 
-	return p
+	return p, nil
 }
 
 func newPaginationUrlFactory(pathElements ...string) paginationUrlFactory {
--- a/hugolib/pagination_test.go
+++ b/hugolib/pagination_test.go
@@ -24,7 +24,7 @@
 
 }
 
-func TestPaginator(t *testing.T) {
+func TestPager(t *testing.T) {
 
 	pages := createTestPages(21)
 	urlFactory := func(page int) string {
@@ -31,7 +31,10 @@
 		return fmt.Sprintf("page/%d/", page)
 	}
 
-	paginator := newPaginator(pages, 5, urlFactory)
+	_, err := newPaginator(pages, -1, urlFactory)
+	assert.NotNil(t, err)
+
+	paginator, _ := newPaginator(pages, 5, urlFactory)
 	paginatorPages := paginator.Pagers()
 
 	assert.Equal(t, 5, len(paginatorPages))
@@ -43,7 +46,9 @@
 	assert.Equal(t, "page/1/", first.Url())
 	assert.Equal(t, first, first.First())
 	assert.Equal(t, true, first.HasNext())
+	assert.Equal(t, paginatorPages[1], first.Next())
 	assert.Equal(t, false, first.HasPrev())
+	assert.Nil(t, first.Prev())
 	assert.Equal(t, 5, first.NumberOfElements())
 	assert.Equal(t, 1, first.PageNumber())
 
@@ -50,15 +55,31 @@
 	third := paginatorPages[2]
 	assert.Equal(t, true, third.HasNext())
 	assert.Equal(t, true, third.HasPrev())
+	assert.Equal(t, paginatorPages[1], third.Prev())
 
 	last := paginatorPages[4]
 	assert.Equal(t, "page/5/", last.Url())
 	assert.Equal(t, last, last.Last())
 	assert.Equal(t, false, last.HasNext())
+	assert.Nil(t, last.Next())
 	assert.Equal(t, true, last.HasPrev())
 	assert.Equal(t, 1, last.NumberOfElements())
 	assert.Equal(t, 5, last.PageNumber())
+}
 
+func TestPagerNoPages(t *testing.T) {
+	pages := createTestPages(0)
+	urlFactory := func(page int) string {
+		return fmt.Sprintf("page/%d/", page)
+	}
+
+	paginator, _ := newPaginator(pages, 5, urlFactory)
+	paginatorPages := paginator.Pagers()
+
+	assert.Equal(t, 0, len(paginatorPages))
+	assert.Equal(t, 0, paginator.TotalNumberOfElements())
+	assert.Equal(t, 5, paginator.PageSize())
+	assert.Equal(t, 0, paginator.TotalPages())
 }
 
 func TestPaginationUrlFactory(t *testing.T) {
@@ -70,6 +91,88 @@
 	assert.Equal(t, "/foo/bar/", fooBar(1))
 	assert.Equal(t, "/%D0%BD%D0%BE%D0%B2%D0%BE%D1%81%D1%82%D0%B8-%D0%BF%D1%80%D0%BE%D0%B5%D0%BA%D1%82%D0%B0/zoo/4/", unicode(4))
 	assert.Equal(t, "/foo/bar/zoo/12345/", fooBar(12345))
+
+}
+
+func TestPaginator(t *testing.T) {
+	viper.Set("paginate", 5)
+	pages := createTestPages(12)
+	s := &Site{}
+	n1 := s.newHomeNode()
+	n2 := s.newHomeNode()
+	n1.Data["Pages"] = pages
+
+	paginator1, err := n1.Paginator()
+
+	assert.Nil(t, err)
+	assert.NotNil(t, paginator1)
+	assert.Equal(t, 3, paginator1.TotalPages())
+	assert.Equal(t, 12, paginator1.TotalNumberOfElements())
+
+	n2.paginator = paginator1.Next()
+	paginator2, err := n2.Paginator()
+	assert.Nil(t, err)
+	assert.Equal(t, paginator2, paginator1.Next())
+
+	n1.Data["Pages"] = createTestPages(1)
+	samePaginator, _ := n1.Paginator()
+	assert.Equal(t, paginator1, samePaginator)
+
+	p, _ := NewPage("test")
+	_, err = p.Paginator()
+	assert.NotNil(t, err)
+}
+
+func TestPaginatorWithNegativePaginate(t *testing.T) {
+	viper.Set("paginate", -1)
+	s := &Site{}
+	_, err := s.newHomeNode().Paginator()
+	assert.NotNil(t, err)
+}
+
+func TestPaginate(t *testing.T) {
+	viper.Set("paginate", 5)
+	pages := createTestPages(6)
+	s := &Site{}
+	n1 := s.newHomeNode()
+	n2 := s.newHomeNode()
+
+	paginator1, err := n1.Paginate(pages)
+
+	assert.Nil(t, err)
+	assert.NotNil(t, paginator1)
+	assert.Equal(t, 2, paginator1.TotalPages())
+	assert.Equal(t, 6, paginator1.TotalNumberOfElements())
+
+	n2.paginator = paginator1.Next()
+	paginator2, err := n2.Paginate(pages)
+	assert.Nil(t, err)
+	assert.Equal(t, paginator2, paginator1.Next())
+
+	samePaginator, err := n1.Paginate(createTestPages(2))
+	assert.Equal(t, paginator1, samePaginator)
+
+	p, _ := NewPage("test")
+	_, err = p.Paginate(pages)
+	assert.NotNil(t, err)
+}
+
+func TestPaginateWithNegativePaginate(t *testing.T) {
+	viper.Set("paginate", -1)
+	s := &Site{}
+	_, err := s.newHomeNode().Paginate(createTestPages(2))
+	assert.NotNil(t, err)
+}
+
+func TestPaginatePages(t *testing.T) {
+	viper.Set("paginate", 11)
+	for i, seq := range []interface{}{createTestPages(11), WeightedPages{}, PageGroup{}, &Pages{}} {
+		v, err := paginatePages(seq, "t")
+		assert.NotNil(t, v, "Val %d", i)
+		assert.Nil(t, err, "Err %d", i)
+	}
+	_, err := paginatePages(Site{}, "t")
+	assert.NotNil(t, err)
 
 }