mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-04 02:04:03 +00:00
Lina JodoinandGitHub
3ae9d6bf43
## What changed? - Addresses a few follow-ups from the initial PR after merge: - Makes `Memo` provider use deterministic proto marshalling - Fixes a few style nitpicks - Removes some unneeded TODOs
30 lines
969 B
Go
30 lines
969 B
Go
package scheduler_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"go.temporal.io/server/common/testing/protorequire"
|
|
"google.golang.org/protobuf/types/known/timestamppb"
|
|
)
|
|
|
|
func TestListInfo(t *testing.T) {
|
|
scheduler, ctx, _ := setupSchedulerForTest(t)
|
|
|
|
// Generator maintains the FutureActionTimes list, set that up first.
|
|
generator := scheduler.Generator.Get(ctx)
|
|
expectedFutureTimes := []*timestamppb.Timestamp{timestamppb.Now(), timestamppb.Now()}
|
|
generator.FutureActionTimes = expectedFutureTimes
|
|
|
|
listInfo := scheduler.ListInfo(ctx)
|
|
|
|
// Should return a populated info block.
|
|
require.NotNil(t, listInfo)
|
|
require.NotNil(t, listInfo.Spec)
|
|
require.NotEmpty(t, listInfo.Spec.Interval)
|
|
protorequire.ProtoEqual(t, listInfo.Spec.Interval[0], scheduler.Schedule.Spec.Interval[0])
|
|
require.NotNil(t, listInfo.WorkflowType)
|
|
require.NotEmpty(t, listInfo.FutureActionTimes)
|
|
require.Equal(t, expectedFutureTimes, listInfo.FutureActionTimes)
|
|
}
|