tasks/service/worker/scheduler/buffer_test.go at release/v1.28.x

mirror of https://github.com/hanzoai/tasks.git synced 2026-10-04 02:04:03 +00:00

Files

T

Alex ShtinandGitHub 8e8f1b0f7c Remove license header from every file (#7689 )

## 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`.

2025-05-01 18:50:21 -07:00

172 lines

6.8 KiB

Go

Raw Permalink Blame History

 package scheduler
 import (
 	"testing"
 	"github.com/stretchr/testify/suite"
 	enumspb "go.temporal.io/api/enums/v1"
 )
 type (
 	processBufferSuite struct {
 		suite.Suite
 	}
 	job struct {
 		id     int
 		policy enumspb.ScheduleOverlapPolicy
 	}
 )
 func (j *job) GetOverlapPolicy() enumspb.ScheduleOverlapPolicy { return j.policy }
 func jobIds(jobs []*job) (out []int) {
 	for _, j := range jobs {
 		out = append(out, j.id)
 	}
 	return
 }
 func identity[T any](v T) T { return v }
 func TestProcessBuffer(t *testing.T) {
 	suite.Run(t, new(processBufferSuite))
 }
 func (s *processBufferSuite) TestProcessBufferEmpty() {
 	buffer := []*job{}
 	action := ProcessBuffer(buffer, false, identity[enumspb.ScheduleOverlapPolicy])
 	s.Empty(action.OverlappingStarts)
 	s.Nil(action.NonOverlappingStart)
 	s.Empty(action.NewBuffer)
 	s.False(action.NeedCancel)
 	s.False(action.NeedTerminate)
 }
 func (s *processBufferSuite) TestProcessSkipRunning() {
 	buffer := []*job{{3, enumspb.SCHEDULE_OVERLAP_POLICY_SKIP}, {5, enumspb.SCHEDULE_OVERLAP_POLICY_SKIP}, {7, enumspb.SCHEDULE_OVERLAP_POLICY_SKIP}}
 	action := ProcessBuffer(buffer, true, identity[enumspb.ScheduleOverlapPolicy])
 	s.Empty(action.OverlappingStarts)
 	s.Nil(action.NonOverlappingStart)
 	s.Empty(action.NewBuffer)
 	s.False(action.NeedCancel)
 	s.False(action.NeedTerminate)
 }
 func (s *processBufferSuite) TestProcessSkipNotRunning() {
 	buffer := []*job{{3, enumspb.SCHEDULE_OVERLAP_POLICY_SKIP}, {5, enumspb.SCHEDULE_OVERLAP_POLICY_SKIP}, {7, enumspb.SCHEDULE_OVERLAP_POLICY_SKIP}}
 	action := ProcessBuffer(buffer, false, identity[enumspb.ScheduleOverlapPolicy])
 	s.Empty(action.OverlappingStarts)
 	s.Equal(3, action.NonOverlappingStart.id)
 	s.Empty(action.NewBuffer)
 	s.False(action.NeedCancel)
 	s.False(action.NeedTerminate)
 }
 func (s *processBufferSuite) TestProcessBufferOneRunning() {
 	buffer := []*job{{3, enumspb.SCHEDULE_OVERLAP_POLICY_BUFFER_ONE}, {5, enumspb.SCHEDULE_OVERLAP_POLICY_BUFFER_ONE}, {7, enumspb.SCHEDULE_OVERLAP_POLICY_BUFFER_ONE}}
 	action := ProcessBuffer(buffer, true, identity[enumspb.ScheduleOverlapPolicy])
 	s.Empty(action.OverlappingStarts)
 	s.Nil(action.NonOverlappingStart)
 	s.Equal([]int{3}, jobIds(action.NewBuffer))
 	s.False(action.NeedCancel)
 	s.False(action.NeedTerminate)
 }
 func (s *processBufferSuite) TestProcessBufferOneNotRunning() {
 	buffer := []*job{{3, enumspb.SCHEDULE_OVERLAP_POLICY_BUFFER_ONE}, {5, enumspb.SCHEDULE_OVERLAP_POLICY_BUFFER_ONE}, {7, enumspb.SCHEDULE_OVERLAP_POLICY_BUFFER_ONE}}
 	action := ProcessBuffer(buffer, false, identity[enumspb.ScheduleOverlapPolicy])
 	s.Empty(action.OverlappingStarts)
 	s.Equal(3, action.NonOverlappingStart.id)
 	s.Equal([]int{5}, jobIds(action.NewBuffer))
 	s.False(action.NeedCancel)
 	s.False(action.NeedTerminate)
 }
 func (s *processBufferSuite) TestProcessBufferAllRunning() {
 	buffer := []*job{{3, enumspb.SCHEDULE_OVERLAP_POLICY_BUFFER_ALL}, {5, enumspb.SCHEDULE_OVERLAP_POLICY_BUFFER_ALL}, {7, enumspb.SCHEDULE_OVERLAP_POLICY_BUFFER_ALL}}
 	action := ProcessBuffer(buffer, true, identity[enumspb.ScheduleOverlapPolicy])
 	s.Empty(action.OverlappingStarts)
 	s.Nil(action.NonOverlappingStart)
 	s.Equal([]int{3, 5, 7}, jobIds(action.NewBuffer))
 	s.False(action.NeedCancel)
 	s.False(action.NeedTerminate)
 }
 func (s *processBufferSuite) TestProcessBufferAllNotRunning() {
 	buffer := []*job{{3, enumspb.SCHEDULE_OVERLAP_POLICY_BUFFER_ALL}, {5, enumspb.SCHEDULE_OVERLAP_POLICY_BUFFER_ALL}, {7, enumspb.SCHEDULE_OVERLAP_POLICY_BUFFER_ALL}}
 	action := ProcessBuffer(buffer, false, identity[enumspb.ScheduleOverlapPolicy])
 	s.Empty(action.OverlappingStarts)
 	s.Equal(3, action.NonOverlappingStart.id)
 	s.Equal([]int{5, 7}, jobIds(action.NewBuffer))
 	s.False(action.NeedCancel)
 	s.False(action.NeedTerminate)
 }
 func (s *processBufferSuite) TestProcessCancelRunning() {
 	buffer := []*job{{3, enumspb.SCHEDULE_OVERLAP_POLICY_CANCEL_OTHER}, {5, enumspb.SCHEDULE_OVERLAP_POLICY_CANCEL_OTHER}, {7, enumspb.SCHEDULE_OVERLAP_POLICY_CANCEL_OTHER}}
 	action := ProcessBuffer(buffer, true, identity[enumspb.ScheduleOverlapPolicy])
 	s.Empty(action.OverlappingStarts)
 	s.Nil(action.NonOverlappingStart)
 	s.Equal([]int{3, 5, 7}, jobIds(action.NewBuffer))
 	s.True(action.NeedCancel)
 	s.False(action.NeedTerminate)
 }
 func (s *processBufferSuite) TestProcessCancelNotRunning() {
 	buffer := []*job{{3, enumspb.SCHEDULE_OVERLAP_POLICY_CANCEL_OTHER}, {5, enumspb.SCHEDULE_OVERLAP_POLICY_CANCEL_OTHER}, {7, enumspb.SCHEDULE_OVERLAP_POLICY_CANCEL_OTHER}}
 	action := ProcessBuffer(buffer, false, identity[enumspb.ScheduleOverlapPolicy])
 	s.Empty(action.OverlappingStarts)
 	// optimization: 3 and 5 don't even get started since they would be immediately cancelled
 	s.Equal(7, action.NonOverlappingStart.id)
 	s.Empty(action.NewBuffer)
 	s.False(action.NeedCancel)
 	s.False(action.NeedTerminate)
 }
 func (s *processBufferSuite) TestProcessTerminateRunning() {
 	buffer := []*job{{3, enumspb.SCHEDULE_OVERLAP_POLICY_TERMINATE_OTHER}, {5, enumspb.SCHEDULE_OVERLAP_POLICY_TERMINATE_OTHER}, {7, enumspb.SCHEDULE_OVERLAP_POLICY_TERMINATE_OTHER}}
 	action := ProcessBuffer(buffer, true, identity[enumspb.ScheduleOverlapPolicy])
 	s.Empty(action.OverlappingStarts)
 	s.Nil(action.NonOverlappingStart)
 	s.Equal([]int{3, 5, 7}, jobIds(action.NewBuffer))
 	s.False(action.NeedCancel)
 	s.True(action.NeedTerminate)
 }
 func (s *processBufferSuite) TestProcessTerminateNotRunning() {
 	buffer := []*job{{3, enumspb.SCHEDULE_OVERLAP_POLICY_TERMINATE_OTHER}, {5, enumspb.SCHEDULE_OVERLAP_POLICY_TERMINATE_OTHER}, {7, enumspb.SCHEDULE_OVERLAP_POLICY_TERMINATE_OTHER}}
 	action := ProcessBuffer(buffer, false, identity[enumspb.ScheduleOverlapPolicy])
 	s.Empty(action.OverlappingStarts)
 	// optimization: 3 and 5 don't even get started since they would be immediately terminated
 	s.Equal(7, action.NonOverlappingStart.id)
 	s.Empty(action.NewBuffer)
 	s.False(action.NeedCancel)
 	s.False(action.NeedTerminate)
 }
 func (s *processBufferSuite) TestProcessAllowAll() {
 	buffer := []*job{{3, enumspb.SCHEDULE_OVERLAP_POLICY_ALLOW_ALL}, {5, enumspb.SCHEDULE_OVERLAP_POLICY_ALLOW_ALL}, {7, enumspb.SCHEDULE_OVERLAP_POLICY_ALLOW_ALL}}
 	action := ProcessBuffer(buffer, false, identity[enumspb.ScheduleOverlapPolicy])
 	s.Equal([]int{3, 5, 7}, jobIds(action.OverlappingStarts))
 	s.Nil(action.NonOverlappingStart)
 	s.Empty(action.NewBuffer)
 	s.False(action.NeedCancel)
 	s.False(action.NeedTerminate)
 }
 func (s *processBufferSuite) TestProcessWithResolve() {
 	buffer := []*job{{3, enumspb.SCHEDULE_OVERLAP_POLICY_UNSPECIFIED}, {5, enumspb.SCHEDULE_OVERLAP_POLICY_UNSPECIFIED}, {7, enumspb.SCHEDULE_OVERLAP_POLICY_UNSPECIFIED}}
 	terminate := func(enumspb.ScheduleOverlapPolicy) enumspb.ScheduleOverlapPolicy {
 		return enumspb.SCHEDULE_OVERLAP_POLICY_TERMINATE_OTHER
 	}
 	action := ProcessBuffer(buffer, false, terminate)
 	s.Empty(action.OverlappingStarts)
 	s.Equal(7, action.NonOverlappingStart.id)
 	s.Empty(action.NewBuffer)
 	s.False(action.NeedCancel)
 	s.False(action.NeedTerminate)
 }
 // TODO: add test cases for mixed policies

172 lines 6.8 KiB Go Raw Permalink Blame History

172 lines

6.8 KiB

Go

Raw Permalink Blame History