313 lines
		
	
	
		
			8.9 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			313 lines
		
	
	
		
			8.9 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| // Copyright 2022 The Gitea Authors.
 | |
| // Copyright 2015 https://github.com/unknwon. Licensed under the Apache License, Version 2.0
 | |
| // SPDX-License-Identifier: Apache-2.0
 | |
| 
 | |
| package paginator
 | |
| 
 | |
| import (
 | |
| 	"testing"
 | |
| 
 | |
| 	"github.com/stretchr/testify/assert"
 | |
| )
 | |
| 
 | |
| func TestPaginator(t *testing.T) {
 | |
| 	t.Run("Basic logics", func(t *testing.T) {
 | |
| 		p := New(0, -1, -1, 0)
 | |
| 		assert.Equal(t, 1, p.PagingNum())
 | |
| 		assert.True(t, p.IsFirst())
 | |
| 		assert.False(t, p.HasPrevious())
 | |
| 		assert.Equal(t, 1, p.Previous())
 | |
| 		assert.False(t, p.HasNext())
 | |
| 		assert.Equal(t, 1, p.Next())
 | |
| 		assert.True(t, p.IsLast())
 | |
| 		assert.Equal(t, 0, p.Total())
 | |
| 
 | |
| 		p = New(1, 10, 2, 0)
 | |
| 		assert.Equal(t, 10, p.PagingNum())
 | |
| 		assert.True(t, p.IsFirst())
 | |
| 		assert.False(t, p.HasPrevious())
 | |
| 		assert.False(t, p.HasNext())
 | |
| 		assert.True(t, p.IsLast())
 | |
| 
 | |
| 		p = New(10, 10, 1, 0)
 | |
| 		assert.Equal(t, 10, p.PagingNum())
 | |
| 		assert.True(t, p.IsFirst())
 | |
| 		assert.False(t, p.HasPrevious())
 | |
| 		assert.False(t, p.HasNext())
 | |
| 		assert.True(t, p.IsLast())
 | |
| 
 | |
| 		p = New(11, 10, 1, 0)
 | |
| 		assert.Equal(t, 10, p.PagingNum())
 | |
| 		assert.True(t, p.IsFirst())
 | |
| 		assert.False(t, p.HasPrevious())
 | |
| 		assert.True(t, p.HasNext())
 | |
| 		assert.Equal(t, 2, p.Next())
 | |
| 		assert.False(t, p.IsLast())
 | |
| 
 | |
| 		p = New(11, 10, 2, 0)
 | |
| 		assert.Equal(t, 10, p.PagingNum())
 | |
| 		assert.False(t, p.IsFirst())
 | |
| 		assert.True(t, p.HasPrevious())
 | |
| 		assert.Equal(t, 1, p.Previous())
 | |
| 		assert.False(t, p.HasNext())
 | |
| 		assert.True(t, p.IsLast())
 | |
| 
 | |
| 		p = New(20, 10, 2, 0)
 | |
| 		assert.Equal(t, 10, p.PagingNum())
 | |
| 		assert.False(t, p.IsFirst())
 | |
| 		assert.True(t, p.HasPrevious())
 | |
| 		assert.False(t, p.HasNext())
 | |
| 		assert.True(t, p.IsLast())
 | |
| 
 | |
| 		p = New(25, 10, 2, 0)
 | |
| 		assert.Equal(t, 10, p.PagingNum())
 | |
| 		assert.False(t, p.IsFirst())
 | |
| 		assert.True(t, p.HasPrevious())
 | |
| 		assert.True(t, p.HasNext())
 | |
| 		assert.False(t, p.IsLast())
 | |
| 	})
 | |
| 
 | |
| 	t.Run("Generate pages", func(t *testing.T) {
 | |
| 		p := New(0, 10, 1, 0)
 | |
| 		pages := p.Pages()
 | |
| 		assert.Empty(t, pages)
 | |
| 	})
 | |
| 
 | |
| 	t.Run("Only current page", func(t *testing.T) {
 | |
| 		p := New(0, 10, 1, 1)
 | |
| 		pages := p.Pages()
 | |
| 		assert.Len(t, pages, 1)
 | |
| 		assert.Equal(t, 1, pages[0].Num())
 | |
| 		assert.True(t, pages[0].IsCurrent())
 | |
| 
 | |
| 		p = New(1, 10, 1, 1)
 | |
| 		pages = p.Pages()
 | |
| 		assert.Len(t, pages, 1)
 | |
| 		assert.Equal(t, 1, pages[0].Num())
 | |
| 		assert.True(t, pages[0].IsCurrent())
 | |
| 	})
 | |
| 
 | |
| 	t.Run("Total page number is less or equal", func(t *testing.T) {
 | |
| 		p := New(1, 10, 1, 2)
 | |
| 		pages := p.Pages()
 | |
| 		assert.Len(t, pages, 1)
 | |
| 		assert.Equal(t, 1, pages[0].Num())
 | |
| 		assert.True(t, pages[0].IsCurrent())
 | |
| 
 | |
| 		p = New(11, 10, 1, 2)
 | |
| 		pages = p.Pages()
 | |
| 		assert.Len(t, pages, 2)
 | |
| 		assert.Equal(t, 1, pages[0].Num())
 | |
| 		assert.True(t, pages[0].IsCurrent())
 | |
| 		assert.Equal(t, 2, pages[1].Num())
 | |
| 		assert.False(t, pages[1].IsCurrent())
 | |
| 
 | |
| 		p = New(11, 10, 2, 2)
 | |
| 		pages = p.Pages()
 | |
| 		assert.Len(t, pages, 2)
 | |
| 		assert.Equal(t, 1, pages[0].Num())
 | |
| 		assert.False(t, pages[0].IsCurrent())
 | |
| 		assert.Equal(t, 2, pages[1].Num())
 | |
| 		assert.True(t, pages[1].IsCurrent())
 | |
| 
 | |
| 		p = New(25, 10, 2, 3)
 | |
| 		pages = p.Pages()
 | |
| 		assert.Len(t, pages, 3)
 | |
| 		assert.Equal(t, 1, pages[0].Num())
 | |
| 		assert.False(t, pages[0].IsCurrent())
 | |
| 		assert.Equal(t, 2, pages[1].Num())
 | |
| 		assert.True(t, pages[1].IsCurrent())
 | |
| 		assert.Equal(t, 3, pages[2].Num())
 | |
| 		assert.False(t, pages[2].IsCurrent())
 | |
| 	})
 | |
| 
 | |
| 	t.Run("Has more previous pages ", func(t *testing.T) {
 | |
| 		// ... 2
 | |
| 		p := New(11, 10, 2, 1)
 | |
| 		pages := p.Pages()
 | |
| 		assert.Len(t, pages, 2)
 | |
| 		assert.Equal(t, -1, pages[0].Num())
 | |
| 		assert.False(t, pages[0].IsCurrent())
 | |
| 		assert.Equal(t, 2, pages[1].Num())
 | |
| 		assert.True(t, pages[1].IsCurrent())
 | |
| 
 | |
| 		// ... 2 3
 | |
| 		p = New(21, 10, 2, 2)
 | |
| 		pages = p.Pages()
 | |
| 		assert.Len(t, pages, 3)
 | |
| 		assert.Equal(t, -1, pages[0].Num())
 | |
| 		assert.False(t, pages[0].IsCurrent())
 | |
| 		assert.Equal(t, 2, pages[1].Num())
 | |
| 		assert.True(t, pages[1].IsCurrent())
 | |
| 		assert.Equal(t, 3, pages[2].Num())
 | |
| 		assert.False(t, pages[2].IsCurrent())
 | |
| 
 | |
| 		// ... 2 3 4
 | |
| 		p = New(31, 10, 3, 3)
 | |
| 		pages = p.Pages()
 | |
| 		assert.Len(t, pages, 4)
 | |
| 		assert.Equal(t, -1, pages[0].Num())
 | |
| 		assert.False(t, pages[0].IsCurrent())
 | |
| 		assert.Equal(t, 2, pages[1].Num())
 | |
| 		assert.False(t, pages[1].IsCurrent())
 | |
| 		assert.Equal(t, 3, pages[2].Num())
 | |
| 		assert.True(t, pages[2].IsCurrent())
 | |
| 		assert.Equal(t, 4, pages[3].Num())
 | |
| 		assert.False(t, pages[3].IsCurrent())
 | |
| 
 | |
| 		// ... 3 4 5
 | |
| 		p = New(41, 10, 4, 3)
 | |
| 		pages = p.Pages()
 | |
| 		assert.Len(t, pages, 4)
 | |
| 		assert.Equal(t, -1, pages[0].Num())
 | |
| 		assert.False(t, pages[0].IsCurrent())
 | |
| 		assert.Equal(t, 3, pages[1].Num())
 | |
| 		assert.False(t, pages[1].IsCurrent())
 | |
| 		assert.Equal(t, 4, pages[2].Num())
 | |
| 		assert.True(t, pages[2].IsCurrent())
 | |
| 		assert.Equal(t, 5, pages[3].Num())
 | |
| 		assert.False(t, pages[3].IsCurrent())
 | |
| 
 | |
| 		// ... 4 5 6 7 8 9 10
 | |
| 		p = New(100, 10, 9, 7)
 | |
| 		pages = p.Pages()
 | |
| 		assert.Len(t, pages, 8)
 | |
| 		assert.Equal(t, -1, pages[0].Num())
 | |
| 		assert.False(t, pages[0].IsCurrent())
 | |
| 		assert.Equal(t, 4, pages[1].Num())
 | |
| 		assert.False(t, pages[1].IsCurrent())
 | |
| 		assert.Equal(t, 5, pages[2].Num())
 | |
| 		assert.False(t, pages[2].IsCurrent())
 | |
| 		assert.Equal(t, 6, pages[3].Num())
 | |
| 		assert.False(t, pages[3].IsCurrent())
 | |
| 		assert.Equal(t, 7, pages[4].Num())
 | |
| 		assert.False(t, pages[4].IsCurrent())
 | |
| 		assert.Equal(t, 8, pages[5].Num())
 | |
| 		assert.False(t, pages[5].IsCurrent())
 | |
| 		assert.Equal(t, 9, pages[6].Num())
 | |
| 		assert.True(t, pages[6].IsCurrent())
 | |
| 		assert.Equal(t, 10, pages[7].Num())
 | |
| 		assert.False(t, pages[7].IsCurrent())
 | |
| 	})
 | |
| 
 | |
| 	t.Run("Has more next pages", func(t *testing.T) {
 | |
| 		// 1 ...
 | |
| 		p := New(21, 10, 1, 1)
 | |
| 		pages := p.Pages()
 | |
| 		assert.Len(t, pages, 2)
 | |
| 		assert.Equal(t, 1, pages[0].Num())
 | |
| 		assert.True(t, pages[0].IsCurrent())
 | |
| 		assert.Equal(t, -1, pages[1].Num())
 | |
| 		assert.False(t, pages[1].IsCurrent())
 | |
| 
 | |
| 		// 1 2 ...
 | |
| 		p = New(21, 10, 1, 2)
 | |
| 		pages = p.Pages()
 | |
| 		assert.Len(t, pages, 3)
 | |
| 		assert.Equal(t, 1, pages[0].Num())
 | |
| 		assert.True(t, pages[0].IsCurrent())
 | |
| 		assert.Equal(t, 2, pages[1].Num())
 | |
| 		assert.False(t, pages[1].IsCurrent())
 | |
| 		assert.Equal(t, -1, pages[2].Num())
 | |
| 		assert.False(t, pages[2].IsCurrent())
 | |
| 
 | |
| 		// 1 2 3 ...
 | |
| 		p = New(31, 10, 2, 3)
 | |
| 		pages = p.Pages()
 | |
| 		assert.Len(t, pages, 4)
 | |
| 		assert.Equal(t, 1, pages[0].Num())
 | |
| 		assert.False(t, pages[0].IsCurrent())
 | |
| 		assert.Equal(t, 2, pages[1].Num())
 | |
| 		assert.True(t, pages[1].IsCurrent())
 | |
| 		assert.Equal(t, 3, pages[2].Num())
 | |
| 		assert.False(t, pages[2].IsCurrent())
 | |
| 		assert.Equal(t, -1, pages[3].Num())
 | |
| 		assert.False(t, pages[3].IsCurrent())
 | |
| 
 | |
| 		// 1 2 3 ...
 | |
| 		p = New(41, 10, 2, 3)
 | |
| 		pages = p.Pages()
 | |
| 		assert.Len(t, pages, 4)
 | |
| 		assert.Equal(t, 1, pages[0].Num())
 | |
| 		assert.False(t, pages[0].IsCurrent())
 | |
| 		assert.Equal(t, 2, pages[1].Num())
 | |
| 		assert.True(t, pages[1].IsCurrent())
 | |
| 		assert.Equal(t, 3, pages[2].Num())
 | |
| 		assert.False(t, pages[2].IsCurrent())
 | |
| 		assert.Equal(t, -1, pages[3].Num())
 | |
| 		assert.False(t, pages[3].IsCurrent())
 | |
| 
 | |
| 		// 1 2 3 4 5 6 7 ...
 | |
| 		p = New(100, 10, 1, 7)
 | |
| 		pages = p.Pages()
 | |
| 		assert.Len(t, pages, 8)
 | |
| 		assert.Equal(t, 1, pages[0].Num())
 | |
| 		assert.True(t, pages[0].IsCurrent())
 | |
| 		assert.Equal(t, 2, pages[1].Num())
 | |
| 		assert.False(t, pages[1].IsCurrent())
 | |
| 		assert.Equal(t, 3, pages[2].Num())
 | |
| 		assert.False(t, pages[2].IsCurrent())
 | |
| 		assert.Equal(t, 4, pages[3].Num())
 | |
| 		assert.False(t, pages[3].IsCurrent())
 | |
| 		assert.Equal(t, 5, pages[4].Num())
 | |
| 		assert.False(t, pages[4].IsCurrent())
 | |
| 		assert.Equal(t, 6, pages[5].Num())
 | |
| 		assert.False(t, pages[5].IsCurrent())
 | |
| 		assert.Equal(t, 7, pages[6].Num())
 | |
| 		assert.False(t, pages[6].IsCurrent())
 | |
| 		assert.Equal(t, -1, pages[7].Num())
 | |
| 		assert.False(t, pages[7].IsCurrent())
 | |
| 
 | |
| 		// 1 2 3 4 5 6 7 ...
 | |
| 		p = New(100, 10, 2, 7)
 | |
| 		pages = p.Pages()
 | |
| 		assert.Len(t, pages, 8)
 | |
| 		assert.Equal(t, 1, pages[0].Num())
 | |
| 		assert.False(t, pages[0].IsCurrent())
 | |
| 		assert.Equal(t, 2, pages[1].Num())
 | |
| 		assert.True(t, pages[1].IsCurrent())
 | |
| 		assert.Equal(t, 3, pages[2].Num())
 | |
| 		assert.False(t, pages[2].IsCurrent())
 | |
| 		assert.Equal(t, 4, pages[3].Num())
 | |
| 		assert.False(t, pages[3].IsCurrent())
 | |
| 		assert.Equal(t, 5, pages[4].Num())
 | |
| 		assert.False(t, pages[4].IsCurrent())
 | |
| 		assert.Equal(t, 6, pages[5].Num())
 | |
| 		assert.False(t, pages[5].IsCurrent())
 | |
| 		assert.Equal(t, 7, pages[6].Num())
 | |
| 		assert.False(t, pages[6].IsCurrent())
 | |
| 		assert.Equal(t, -1, pages[7].Num())
 | |
| 		assert.False(t, pages[7].IsCurrent())
 | |
| 	})
 | |
| 
 | |
| 	t.Run("Has both more previous and next pages", func(t *testing.T) {
 | |
| 		// ... 2 3 ...
 | |
| 		p := New(35, 10, 2, 2)
 | |
| 		pages := p.Pages()
 | |
| 		assert.Len(t, pages, 4)
 | |
| 		assert.Equal(t, -1, pages[0].Num())
 | |
| 		assert.False(t, pages[0].IsCurrent())
 | |
| 		assert.Equal(t, 2, pages[1].Num())
 | |
| 		assert.True(t, pages[1].IsCurrent())
 | |
| 		assert.Equal(t, 3, pages[2].Num())
 | |
| 		assert.False(t, pages[2].IsCurrent())
 | |
| 		assert.Equal(t, -1, pages[3].Num())
 | |
| 		assert.False(t, pages[3].IsCurrent())
 | |
| 
 | |
| 		// ... 2 3 4 ...
 | |
| 		p = New(49, 10, 3, 3)
 | |
| 		pages = p.Pages()
 | |
| 		assert.Len(t, pages, 5)
 | |
| 		assert.Equal(t, -1, pages[0].Num())
 | |
| 		assert.False(t, pages[0].IsCurrent())
 | |
| 		assert.Equal(t, 2, pages[1].Num())
 | |
| 		assert.False(t, pages[1].IsCurrent())
 | |
| 		assert.Equal(t, 3, pages[2].Num())
 | |
| 		assert.True(t, pages[2].IsCurrent())
 | |
| 		assert.Equal(t, 4, pages[3].Num())
 | |
| 		assert.False(t, pages[3].IsCurrent())
 | |
| 		assert.Equal(t, -1, pages[4].Num())
 | |
| 		assert.False(t, pages[4].IsCurrent())
 | |
| 	})
 | |
| }
 |