Function TestCron

enterprise/coderd/usage/cron_test.go:25–81

(t *testing.T)

)

func TestCron(t *testing.T) {
	t.Parallel()

	t.Run("BasicTick", func(t *testing.T) {
		t.Parallel()

		ctx := testutil.Context(t, testutil.WaitLong)
		ctrl := gomock.NewController(t)
		db := dbmock.NewMockStore(ctrl)
		clock := quartz.NewMock(t)

		// The existence check should return false so the event gets
		// inserted.
		db.EXPECT().UsageEventExistsByID(gomock.Any(), gomock.Any()).
			Return(false, nil).AnyTimes()

		inserted := make(chan database.InsertUsageEventParams, 1)
		db.EXPECT().InsertUsageEvent(gomock.Any(), gomock.Any()).
			DoAndReturn(func(_ context.Context, params database.InsertUsageEventParams) error {
				inserted <- params
				return nil
			}).AnyTimes()

		inserter := usage.NewDBInserter(usage.InserterWithClock(clock))
		cron := usage.NewCron(clock, slogtest.Make(t, nil), db, inserter)
		require.NoError(t, cron.Register(usage.CronJob{
			Name:      "test-job",
			Interval:  5 * time.Minute,
			EventType: usagetypes.UsageEventTypeHBAISeatsV1,
			Fn: func(_ context.Context) (usagetypes.HeartbeatEvent, error) {
				return usagetypes.HBAISeats{Count: 42}, nil
			},
		}))

		timerTrap := clock.Trap().NewTimer("test-job")

		cron.Start(ctx)
		defer cron.Close()
		defer timerTrap.Close()

		// Wait for timer creation, then fire it. The delay is the
		// time until the next epoch-aligned boundary for the 5-minute
		// interval — we don't assert the exact value since it depends
		// on the mock clock's current time.
		timerCall := timerTrap.MustWait(ctx)
		timerCall.MustRelease(ctx)
		clock.Advance(timerCall.Duration)

		// Verify the event was inserted with an epoch-aligned ID.
		select {
		case params := <-inserted:
			assert.Contains(t, params.ID, "hb_ai_seats_v1:")
		case <-ctx.Done():
			t.Fatal("timed out waiting for insert")
		}
	})
}