mirror of
https://github.com/gohugoio/hugo.git
synced 2024-11-07 20:30:36 -05:00
parent
6674189bc2
commit
36e0d005ed
2 changed files with 39 additions and 0 deletions
|
@ -15,6 +15,7 @@ package hugolib
|
|||
|
||||
import (
|
||||
"sort"
|
||||
"strings"
|
||||
)
|
||||
|
||||
var spc = newPageCache()
|
||||
|
@ -42,6 +43,9 @@ func (by PageBy) Sort(pages Pages) {
|
|||
|
||||
var DefaultPageSort = func(p1, p2 *Page) bool {
|
||||
if p1.Weight == p2.Weight {
|
||||
if p1.Date.Unix() == p2.Date.Unix() {
|
||||
return strings.Compare(p1.LinkTitle(), p2.LinkTitle()) == 1
|
||||
}
|
||||
return p1.Date.Unix() > p2.Date.Unix()
|
||||
}
|
||||
return p1.Weight < p2.Weight
|
||||
|
|
|
@ -5,10 +5,37 @@ import (
|
|||
"github.com/stretchr/testify/assert"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/spf13/hugo/source"
|
||||
)
|
||||
|
||||
func TesDefaultSort(t *testing.T) {
|
||||
|
||||
d1 := time.Now()
|
||||
d2 := d1.Add(1 * time.Hour)
|
||||
d3 := d1.Add(2 * time.Hour)
|
||||
|
||||
p := createSortTestPages(3)
|
||||
|
||||
// first by weight
|
||||
setSortVals([3]time.Time{d1, d2, d3}, [3]string{"a", "b", "c"}, [3]int{3, 2, 1}, p)
|
||||
p.Sort()
|
||||
assert.Equal(t, 1, p[0].Weight)
|
||||
|
||||
// next by date
|
||||
setSortVals([3]time.Time{d3, d1, d2}, [3]string{"a", "b", "c"}, [3]int{1, 1, 1}, p)
|
||||
p.Sort()
|
||||
assert.Equal(t, d1, p[0].Date)
|
||||
|
||||
// finally by title
|
||||
setSortVals([3]time.Time{d3, d3, d3}, [3]string{"b", "a", "c"}, [3]int{1, 1, 1}, p)
|
||||
p.Sort()
|
||||
assert.Equal(t, "a", p[0].Title)
|
||||
assert.Equal(t, "b", p[1].Title)
|
||||
assert.Equal(t, "c", p[2].Title)
|
||||
}
|
||||
|
||||
func TestPageSortReverse(t *testing.T) {
|
||||
p1 := createSortTestPages(10)
|
||||
assert.Equal(t, 0, p1[0].FuzzyWordCount)
|
||||
|
@ -28,6 +55,14 @@ func BenchmarkSortByWeightAndReverse(b *testing.B) {
|
|||
for i := 0; i < b.N; i++ {
|
||||
p = p.ByWeight().Reverse()
|
||||
}
|
||||
}
|
||||
|
||||
func setSortVals(dates [3]time.Time, titles [3]string, weights [3]int, pages Pages) {
|
||||
for i := range dates {
|
||||
pages[i].Date = dates[i]
|
||||
pages[i].Weight = weights[i]
|
||||
pages[i].Title = titles[i]
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in a new issue