Fix buggy Menu.Limit()

1 files changed, 17 insertions, 0 deletions

diff --git a/hugolib/menu_test.go b/hugolib/menu_test.go
index 33fa0a25f..5687454c2 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()