From: Bjørn Erik Pedersen <bjorn.erik.pedersen@gmail.com>
Date: Sat, 6 Feb 2016 21:53:59 +0000 (+0100)
Subject: Fix buggy Menu.Limit()
X-Git-Tag: v0.16~364
X-Git-Url: http://git.maquefel.me/?a=commitdiff_plain;h=7c3dceeaedada1f4e7650864d8e23044c6129e46;p=brevno-suite%2Fhugo

Fix buggy Menu.Limit()
---

diff --git a/hugolib/menu.go b/hugolib/menu.go
index 4e3970fd..2e2670b3 100644
--- a/hugolib/menu.go
+++ b/hugolib/menu.go
@@ -157,7 +157,7 @@ func (m Menu) Sort() {
 }
 
 func (m Menu) Limit(n int) Menu {
-	if len(m) < n {
+	if len(m) > n {
 		return m[0:n]
 	}
 	return m
diff --git a/hugolib/menu_test.go b/hugolib/menu_test.go
index 33fa0a25..5687454c 100644
--- a/hugolib/menu_test.go
+++ b/hugolib/menu_test.go
@@ -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) {
 	viper.Reset()
 	defer viper.Reset()