mirror of
https://github.com/gohugoio/hugo.git
synced 2024-11-07 20:30:36 -05:00
653e6856ea
This is obviously much faster for lager data sets: ```bash name old time/op new time/op delta SearchPage/ByWeight-100-4 267ns ± 4% 272ns ± 5% ~ (p=0.457 n=4+4) SearchPage/ByWeight-5000-4 10.8µs ± 3% 1.2µs ± 2% -88.99% (p=0.029 n=4+4) SearchPage/ByWeight-10000-4 21.1µs ± 1% 1.4µs ±11% -93.28% (p=0.029 n=4+4) ``` See #4500
292 lines
7.7 KiB
Go
292 lines
7.7 KiB
Go
// Copyright 2019 The Hugo Authors. All rights reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package page
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/gohugoio/hugo/htesting/hqt"
|
|
"github.com/gohugoio/hugo/source"
|
|
|
|
"github.com/gohugoio/hugo/resources/resource"
|
|
|
|
qt "github.com/frankban/quicktest"
|
|
)
|
|
|
|
var eq = qt.CmpEquals(hqt.DeepAllowUnexported(
|
|
&testPage{},
|
|
&source.FileInfo{},
|
|
))
|
|
|
|
func TestDefaultSort(t *testing.T) {
|
|
t.Parallel()
|
|
c := qt.New(t)
|
|
d1 := time.Now()
|
|
d2 := d1.Add(-1 * time.Hour)
|
|
d3 := d1.Add(-2 * time.Hour)
|
|
d4 := d1.Add(-3 * time.Hour)
|
|
|
|
p := createSortTestPages(4)
|
|
|
|
// first by weight
|
|
setSortVals([4]time.Time{d1, d2, d3, d4}, [4]string{"b", "a", "c", "d"}, [4]int{4, 3, 2, 1}, p)
|
|
SortByDefault(p)
|
|
|
|
c.Assert(p[0].Weight(), qt.Equals, 1)
|
|
|
|
// Consider zero weight, issue #2673
|
|
setSortVals([4]time.Time{d1, d2, d3, d4}, [4]string{"b", "a", "d", "c"}, [4]int{0, 0, 0, 1}, p)
|
|
SortByDefault(p)
|
|
|
|
c.Assert(p[0].Weight(), qt.Equals, 1)
|
|
|
|
// next by date
|
|
setSortVals([4]time.Time{d3, d4, d1, d2}, [4]string{"a", "b", "c", "d"}, [4]int{1, 1, 1, 1}, p)
|
|
SortByDefault(p)
|
|
c.Assert(p[0].Date(), qt.Equals, d1)
|
|
|
|
// finally by link title
|
|
setSortVals([4]time.Time{d3, d3, d3, d3}, [4]string{"b", "c", "a", "d"}, [4]int{1, 1, 1, 1}, p)
|
|
SortByDefault(p)
|
|
c.Assert(p[0].LinkTitle(), qt.Equals, "al")
|
|
c.Assert(p[1].LinkTitle(), qt.Equals, "bl")
|
|
c.Assert(p[2].LinkTitle(), qt.Equals, "cl")
|
|
}
|
|
|
|
// https://github.com/gohugoio/hugo/issues/4953
|
|
func TestSortByLinkTitle(t *testing.T) {
|
|
t.Parallel()
|
|
c := qt.New(t)
|
|
pages := createSortTestPages(6)
|
|
|
|
for i, p := range pages {
|
|
pp := p.(*testPage)
|
|
if i < 5 {
|
|
pp.title = fmt.Sprintf("title%d", i)
|
|
}
|
|
|
|
if i > 2 {
|
|
pp.linkTitle = fmt.Sprintf("linkTitle%d", i)
|
|
}
|
|
|
|
}
|
|
|
|
pages.shuffle()
|
|
|
|
bylt := pages.ByLinkTitle()
|
|
|
|
for i, p := range bylt {
|
|
if i < 3 {
|
|
c.Assert(p.LinkTitle(), qt.Equals, fmt.Sprintf("linkTitle%d", i+3))
|
|
} else {
|
|
c.Assert(p.LinkTitle(), qt.Equals, fmt.Sprintf("title%d", i-3))
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestSortByN(t *testing.T) {
|
|
t.Parallel()
|
|
d1 := time.Now()
|
|
d2 := d1.Add(-2 * time.Hour)
|
|
d3 := d1.Add(-10 * time.Hour)
|
|
d4 := d1.Add(-20 * time.Hour)
|
|
|
|
p := createSortTestPages(4)
|
|
|
|
for i, this := range []struct {
|
|
sortFunc func(p Pages) Pages
|
|
assertFunc func(p Pages) bool
|
|
}{
|
|
{(Pages).ByWeight, func(p Pages) bool { return p[0].Weight() == 1 }},
|
|
{(Pages).ByTitle, func(p Pages) bool { return p[0].Title() == "ab" }},
|
|
{(Pages).ByLinkTitle, func(p Pages) bool { return p[0].LinkTitle() == "abl" }},
|
|
{(Pages).ByDate, func(p Pages) bool { return p[0].Date() == d4 }},
|
|
{(Pages).ByPublishDate, func(p Pages) bool { return p[0].PublishDate() == d4 }},
|
|
{(Pages).ByExpiryDate, func(p Pages) bool { return p[0].ExpiryDate() == d4 }},
|
|
{(Pages).ByLastmod, func(p Pages) bool { return p[1].Lastmod() == d3 }},
|
|
{(Pages).ByLength, func(p Pages) bool { return p[0].(resource.LengthProvider).Len() == len(p[0].(*testPage).content) }},
|
|
} {
|
|
setSortVals([4]time.Time{d1, d2, d3, d4}, [4]string{"b", "ab", "cde", "fg"}, [4]int{0, 3, 2, 1}, p)
|
|
|
|
sorted := this.sortFunc(p)
|
|
if !this.assertFunc(sorted) {
|
|
t.Errorf("[%d] sort error", i)
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
func TestLimit(t *testing.T) {
|
|
t.Parallel()
|
|
c := qt.New(t)
|
|
p := createSortTestPages(10)
|
|
firstFive := p.Limit(5)
|
|
c.Assert(len(firstFive), qt.Equals, 5)
|
|
for i := 0; i < 5; i++ {
|
|
c.Assert(firstFive[i], qt.Equals, p[i])
|
|
}
|
|
c.Assert(p.Limit(10), eq, p)
|
|
c.Assert(p.Limit(11), eq, p)
|
|
}
|
|
|
|
func TestPageSortReverse(t *testing.T) {
|
|
t.Parallel()
|
|
c := qt.New(t)
|
|
p1 := createSortTestPages(10)
|
|
c.Assert(p1[0].(*testPage).fuzzyWordCount, qt.Equals, 0)
|
|
c.Assert(p1[9].(*testPage).fuzzyWordCount, qt.Equals, 9)
|
|
p2 := p1.Reverse()
|
|
c.Assert(p2[0].(*testPage).fuzzyWordCount, qt.Equals, 9)
|
|
c.Assert(p2[9].(*testPage).fuzzyWordCount, qt.Equals, 0)
|
|
// cached
|
|
c.Assert(pagesEqual(p2, p1.Reverse()), qt.Equals, true)
|
|
}
|
|
|
|
func TestPageSortByParam(t *testing.T) {
|
|
t.Parallel()
|
|
c := qt.New(t)
|
|
var k interface{} = "arbitrarily.nested"
|
|
|
|
unsorted := createSortTestPages(10)
|
|
delete(unsorted[9].Params(), "arbitrarily")
|
|
|
|
firstSetValue, _ := unsorted[0].Param(k)
|
|
secondSetValue, _ := unsorted[1].Param(k)
|
|
lastSetValue, _ := unsorted[8].Param(k)
|
|
unsetValue, _ := unsorted[9].Param(k)
|
|
|
|
c.Assert(firstSetValue, qt.Equals, "xyz100")
|
|
c.Assert(secondSetValue, qt.Equals, "xyz99")
|
|
c.Assert(lastSetValue, qt.Equals, "xyz92")
|
|
c.Assert(unsetValue, qt.Equals, nil)
|
|
|
|
sorted := unsorted.ByParam("arbitrarily.nested")
|
|
firstSetSortedValue, _ := sorted[0].Param(k)
|
|
secondSetSortedValue, _ := sorted[1].Param(k)
|
|
lastSetSortedValue, _ := sorted[8].Param(k)
|
|
unsetSortedValue, _ := sorted[9].Param(k)
|
|
|
|
c.Assert(firstSetSortedValue, qt.Equals, firstSetValue)
|
|
c.Assert(lastSetSortedValue, qt.Equals, secondSetValue)
|
|
c.Assert(secondSetSortedValue, qt.Equals, lastSetValue)
|
|
c.Assert(unsetSortedValue, qt.Equals, unsetValue)
|
|
}
|
|
|
|
func TestPageSortByParamNumeric(t *testing.T) {
|
|
t.Parallel()
|
|
c := qt.New(t)
|
|
|
|
var k interface{} = "arbitrarily.nested"
|
|
|
|
n := 10
|
|
unsorted := createSortTestPages(n)
|
|
for i := 0; i < n; i++ {
|
|
v := 100 - i
|
|
if i%2 == 0 {
|
|
v = 100.0 - i
|
|
}
|
|
|
|
unsorted[i].(*testPage).params = map[string]interface{}{
|
|
"arbitrarily": map[string]interface{}{
|
|
"nested": v,
|
|
},
|
|
}
|
|
}
|
|
delete(unsorted[9].Params(), "arbitrarily")
|
|
|
|
firstSetValue, _ := unsorted[0].Param(k)
|
|
secondSetValue, _ := unsorted[1].Param(k)
|
|
lastSetValue, _ := unsorted[8].Param(k)
|
|
unsetValue, _ := unsorted[9].Param(k)
|
|
|
|
c.Assert(firstSetValue, qt.Equals, 100)
|
|
c.Assert(secondSetValue, qt.Equals, 99)
|
|
c.Assert(lastSetValue, qt.Equals, 92)
|
|
c.Assert(unsetValue, qt.Equals, nil)
|
|
|
|
sorted := unsorted.ByParam("arbitrarily.nested")
|
|
firstSetSortedValue, _ := sorted[0].Param(k)
|
|
secondSetSortedValue, _ := sorted[1].Param(k)
|
|
lastSetSortedValue, _ := sorted[8].Param(k)
|
|
unsetSortedValue, _ := sorted[9].Param(k)
|
|
|
|
c.Assert(firstSetSortedValue, qt.Equals, 92)
|
|
c.Assert(secondSetSortedValue, qt.Equals, 93)
|
|
c.Assert(lastSetSortedValue, qt.Equals, 100)
|
|
c.Assert(unsetSortedValue, qt.Equals, unsetValue)
|
|
}
|
|
|
|
func BenchmarkSortByWeightAndReverse(b *testing.B) {
|
|
p := createSortTestPages(300)
|
|
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
p = p.ByWeight().Reverse()
|
|
}
|
|
}
|
|
|
|
func setSortVals(dates [4]time.Time, titles [4]string, weights [4]int, pages Pages) {
|
|
for i := range dates {
|
|
this := pages[i].(*testPage)
|
|
other := pages[len(dates)-1-i].(*testPage)
|
|
|
|
this.date = dates[i]
|
|
this.lastMod = dates[i]
|
|
this.weight = weights[i]
|
|
this.title = titles[i]
|
|
// make sure we compare apples and ... apples ...
|
|
other.linkTitle = this.Title() + "l"
|
|
other.pubDate = dates[i]
|
|
other.expiryDate = dates[i]
|
|
other.content = titles[i] + "_content"
|
|
}
|
|
lastLastMod := pages[2].Lastmod()
|
|
pages[2].(*testPage).lastMod = pages[1].Lastmod()
|
|
pages[1].(*testPage).lastMod = lastLastMod
|
|
|
|
for _, p := range pages {
|
|
p.(*testPage).content = ""
|
|
}
|
|
|
|
}
|
|
|
|
func createSortTestPages(num int) Pages {
|
|
pages := make(Pages, num)
|
|
|
|
for i := 0; i < num; i++ {
|
|
p := newTestPage()
|
|
p.path = fmt.Sprintf("/x/y/p%d.md", i)
|
|
p.title = fmt.Sprintf("Title %d", i%(num+1/2))
|
|
p.params = map[string]interface{}{
|
|
"arbitrarily": map[string]interface{}{
|
|
"nested": ("xyz" + fmt.Sprintf("%v", 100-i)),
|
|
},
|
|
}
|
|
|
|
w := 5
|
|
|
|
if i%2 == 0 {
|
|
w = 10
|
|
}
|
|
p.fuzzyWordCount = i
|
|
p.weight = w
|
|
p.description = "initial"
|
|
|
|
pages[i] = p
|
|
}
|
|
|
|
return pages
|
|
}
|