mirror of
https://github.com/gohugoio/hugo.git
synced 2024-07-07 09:36:03 +00:00
58 lines
1 KiB
Go
58 lines
1 KiB
Go
package hugolib
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/stretchr/testify/assert"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/spf13/hugo/source"
|
|
)
|
|
|
|
func TestPageSortReverse(t *testing.T) {
|
|
p := createSortTestPages(10)
|
|
assert.Equal(t, 0, p[0].FuzzyWordCount)
|
|
assert.Equal(t, 9, p[9].FuzzyWordCount)
|
|
p = p.Reverse()
|
|
assert.Equal(t, 9, p[0].FuzzyWordCount)
|
|
assert.Equal(t, 1, p[9].FuzzyWordCount)
|
|
}
|
|
|
|
func BenchmarkSortByWeightAndReverse(b *testing.B) {
|
|
|
|
p := createSortTestPages(300)
|
|
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
p = p.ByWeight().Reverse()
|
|
}
|
|
|
|
}
|
|
|
|
func createSortTestPages(num int) Pages {
|
|
pages := make(Pages, num)
|
|
|
|
for i := 0; i < num; i++ {
|
|
pages[i] = &Page{
|
|
Node: Node{
|
|
URLPath: URLPath{
|
|
Section: "z",
|
|
URL: fmt.Sprintf("http://base/x/y/p%d.html", i),
|
|
},
|
|
Site: &SiteInfo{
|
|
BaseURL: "http://base/",
|
|
},
|
|
},
|
|
Source: Source{File: *source.NewFile(filepath.FromSlash(fmt.Sprintf("/x/y/p%d.md", i)))},
|
|
}
|
|
w := 5
|
|
if i%2 == 0 {
|
|
w = 10
|
|
}
|
|
pages[i].FuzzyWordCount = i
|
|
pages[i].Weight = w
|
|
}
|
|
|
|
return pages
|
|
}
|