Function TestPublisherClaimExpiry

enterprise/coderd/usage/publisher_test.go:276–350

TestPublisherClaimExpiry tests the claim query to ensure that events are not claimed if they've recently been claimed by another publisher.

(t *testing.T)

// TestPublisherClaimExpiry tests the claim query to ensure that events are not
// claimed if they've recently been claimed by another publisher.
func TestPublisherClaimExpiry(t *testing.T) {
	t.Parallel()
	ctx := testutil.Context(t, testutil.WaitLong)
	log := slogtest.Make(t, nil)
	db, _ := dbtestutil.NewDB(t)
	clock := quartz.NewMock(t)
	deploymentID, licenseJWT := configureDeployment(ctx, t, db)
	now := time.Now()

	var calls int
	ingestURL := fakeServer(t, tallymanHandler(t, deploymentID.String(), licenseJWT, func(req usagetypes.TallymanV1IngestRequest) any {
		calls++
		return tallymanAcceptAllHandler(req)
	}))

	inserter := usage.NewDBInserter(
		usage.InserterWithClock(clock),
	)

	publisher := usage.NewTallymanPublisher(ctx, log, db, coderdenttest.Keys,
		usage.PublisherWithClock(clock),
		usage.PublisherWithIngestURL(ingestURL),
		usage.PublisherWithInitialDelay(17*time.Minute),
	)
	defer publisher.Close()

	// Create an event that was claimed 1h-18m ago. The ticker has a forced
	// delay of 17m in this test.
	clock.Set(now)
	err := inserter.InsertDiscreteUsageEvent(ctx, db, usagetypes.DCManagedAgentsV1{
		Count: 1,
	})
	require.NoError(t, err)
	// Claim the event in the past. Claiming it this way via the database
	// directly means it won't be marked as published or unclaimed.
	events, err := db.SelectUsageEventsForPublishing(ctx, now.Add(-42*time.Minute))
	require.NoError(t, err)
	require.Len(t, events, 1)

	// Start the publisher with a trap.
	tickerTrap := clock.Trap().NewTicker()
	defer tickerTrap.Close()
	startErr := make(chan error)
	go func() {
		err := publisher.Start()
		testutil.RequireSend(ctx, t, startErr, err)
	}()
	tickerCall := tickerTrap.MustWait(ctx)
	require.Equal(t, 17*time.Minute, tickerCall.Duration)
	tickerCall.MustRelease(ctx)
	require.NoError(t, testutil.RequireReceive(ctx, t, startErr))

	// Set up a trap for the ticker.Reset call.
	tickerResetTrap := clock.Trap().TickerReset()
	defer tickerResetTrap.Close()

	// Advance the clock to the initial tick, which should trigger the first
	// publish, then wait for the reset call. The duration will always be 17m