mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-04 02:04:03 +00:00
Alex ShtinandGitHub
8e8f1b0f7c
## What changed?
<!-- Describe what has changed in this PR -->
Remove license header from every file. Because it is really hard to
follow in this PR here is the summary:
1. License header is removed from all `.go` and `.proto` files
:fireworks::fireworks:🎆.
2. `LICENSE` file in the root directory has only Temporal and Uber
copyrights.
3. 5 other `LICENSE` files added to the packages which have copyrights
different from Temporal and Uber: Datadog, Xargin, "Mat Ryer, Tyler
Bunnell and contributors".
4. `license_file` flag is removed from all code generation tools.
5. `copyright_file` flag is removed from `go:generate mockgen`
directive.
6. All copyright related targets are removed from `Makefile`.
7. Updated Temporal copyright year to 2025 everywhere.
## Why?
<!-- Tell your future self why have you made these changes -->
I double checked with legal department that it is not needed to have
license header in every file. One file per repo is enough. I put all
copyrights to the root `LICENSE` file and removed header from all other
files. Also updated tools and `Makefile`.
139 lines
2.7 KiB
Go
139 lines
2.7 KiB
Go
package collection
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
type (
|
|
pagingIteratorSuite struct {
|
|
suite.Suite
|
|
}
|
|
)
|
|
|
|
func TestPagingIteratorSuite(t *testing.T) {
|
|
s := new(pagingIteratorSuite)
|
|
suite.Run(t, s)
|
|
}
|
|
|
|
func (s *pagingIteratorSuite) SetupSuite() {
|
|
}
|
|
|
|
func (s *pagingIteratorSuite) TearDownSuite() {
|
|
|
|
}
|
|
|
|
func (s *pagingIteratorSuite) SetupTest() {
|
|
|
|
}
|
|
|
|
func (s *pagingIteratorSuite) TearDownTest() {
|
|
|
|
}
|
|
|
|
func (s *pagingIteratorSuite) TestIteration_NoErr() {
|
|
phase := 0
|
|
outputs := [][]int{
|
|
{1, 2, 3, 4, 5},
|
|
{},
|
|
{6},
|
|
{},
|
|
}
|
|
tokens := [][]byte{
|
|
[]byte("some random token 1"),
|
|
[]byte("some random token 2"),
|
|
[]byte("some random token 3"),
|
|
[]byte(nil),
|
|
}
|
|
pagingFn := func(token []byte) ([]int, []byte, error) {
|
|
switch phase {
|
|
case 0:
|
|
s.Equal(0, len(token))
|
|
defer func() { phase++ }()
|
|
return outputs[phase], tokens[phase], nil
|
|
case 1:
|
|
s.Equal(tokens[0], token)
|
|
defer func() { phase++ }()
|
|
return outputs[phase], tokens[phase], nil
|
|
case 2:
|
|
s.Equal(tokens[1], token)
|
|
defer func() { phase++ }()
|
|
return outputs[phase], tokens[phase], nil
|
|
case 3:
|
|
s.Equal(tokens[2], token)
|
|
defer func() { phase++ }()
|
|
return outputs[phase], tokens[phase], nil
|
|
default:
|
|
panic("should not reach here during test")
|
|
}
|
|
}
|
|
|
|
result := []int{}
|
|
ite := NewPagingIterator(pagingFn)
|
|
for ite.HasNext() {
|
|
num, err := ite.Next()
|
|
s.Nil(err)
|
|
result = append(result, num)
|
|
}
|
|
s.Equal([]int{1, 2, 3, 4, 5, 6}, result)
|
|
}
|
|
|
|
func (s *pagingIteratorSuite) TestIteration_Err_Beginging() {
|
|
phase := 0
|
|
ite := NewPagingIterator(func(token []byte) ([]interface{}, []byte, error) {
|
|
switch phase {
|
|
case 0:
|
|
defer func() { phase++ }()
|
|
return nil, nil, errors.New("some random error")
|
|
default:
|
|
panic("should not reach here during test")
|
|
}
|
|
})
|
|
|
|
s.True(ite.HasNext())
|
|
item, err := ite.Next()
|
|
s.Nil(item)
|
|
s.NotNil(err)
|
|
s.False(ite.HasNext())
|
|
}
|
|
|
|
func (s *pagingIteratorSuite) TestIteration_Err_NotBegining() {
|
|
|
|
phase := 0
|
|
outputs := [][]interface{}{
|
|
{1, 2, 3, 4, 5},
|
|
}
|
|
tokens := [][]byte{
|
|
[]byte("some random token 1"),
|
|
}
|
|
pagingFn := func(token []byte) ([]interface{}, []byte, error) {
|
|
switch phase {
|
|
case 0:
|
|
s.Equal(0, len(token))
|
|
defer func() { phase++ }()
|
|
return outputs[phase], tokens[phase], nil
|
|
case 1:
|
|
s.Equal(tokens[0], token)
|
|
defer func() { phase++ }()
|
|
return nil, nil, errors.New("some random error")
|
|
default:
|
|
panic("should not reach here during test")
|
|
}
|
|
}
|
|
|
|
result := []int{}
|
|
ite := NewPagingIterator(pagingFn)
|
|
for ite.HasNext() {
|
|
item, err := ite.Next()
|
|
if err != nil {
|
|
break
|
|
}
|
|
num, ok := item.(int)
|
|
s.True(ok)
|
|
result = append(result, num)
|
|
}
|
|
s.Equal([]int{1, 2, 3, 4, 5}, result)
|
|
}
|