github/hugo
1
0
Fork 0
mirror of https://github.com/gohugoio/hugo.git synced 2025-03-25 01:31:27 +00:00
Code Issues Projects Releases Packages Wiki Activity

Fix buggy Menu.Limit()

Browse source
This commit is contained in:
Bjørn Erik Pedersen 2016-02-06 22:53:59 +01:00
parent a95426d40c
commit 7c3dceeaed
2 changed files with 18 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
hugolib/menu.go
View file

@ -157,7 +157,7 @@ func (m Menu) Sort() {
} }
func (m Menu) Limit(n int) Menu { func (m Menu) Limit(n int) Menu {
if len(m) < n { if len(m) > n {
return m[0:n] return m[0:n]
} }
return m return m

17
hugolib/menu_test.go
View file

@ -480,6 +480,23 @@ func TestTaxonomyNodeMenu(t *testing.T) {
} }
} }
func TestMenuLimit(t *testing.T) {
viper.Reset()
defer viper.Reset()
s := setupMenuTests(t, MENU_PAGE_SOURCES)
m := *s.Menus["main"]
// main menu has 4 entries
firstTwo := m.Limit(2)
assert.Equal(t, 2, len(firstTwo))
for i := 0; i < 2; i++ {
assert.Equal(t, m[i], firstTwo[i])
}
assert.Equal(t, m, m.Limit(4))
assert.Equal(t, m, m.Limit(5))
}
func TestHomeNodeMenu(t *testing.T) { func TestHomeNodeMenu(t *testing.T) {
viper.Reset() viper.Reset()
defer viper.Reset() defer viper.Reset()