Function TestCache_DeploymentStats

coderd/metricscache/metricscache_test.go:295–343

(t *testing.T)

}

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

	var (
		ctx   = testutil.Context(t, testutil.WaitShort)
		log   = testutil.Logger(t)
		clock = quartz.NewMock(t)
	)

	tickerTrap := clock.Trap().TickerFunc("metricscache")
	defer tickerTrap.Close()

	cache, db := newMetricsCache(t, log, clock, metricscache.Intervals{
		DeploymentStats: time.Minute,
	}, false)

	err := db.InsertWorkspaceAgentStats(context.Background(), database.InsertWorkspaceAgentStatsParams{
		ID:                 []uuid.UUID{uuid.New()},
		CreatedAt:          []time.Time{clock.Now()},
		WorkspaceID:        []uuid.UUID{uuid.New()},
		UserID:             []uuid.UUID{uuid.New()},
		TemplateID:         []uuid.UUID{uuid.New()},
		AgentID:            []uuid.UUID{uuid.New()},
		ConnectionsByProto: json.RawMessage(`[{}]`),

		RxPackets:                   []int64{0},
		RxBytes:                     []int64{1},
		TxPackets:                   []int64{0},
		TxBytes:                     []int64{1},
		ConnectionCount:             []int64{1},
		SessionCountVSCode:          []int64{1},
		SessionCountJetBrains:       []int64{0},
		SessionCountReconnectingPTY: []int64{0},
		SessionCountSSH:             []int64{0},
		ConnectionMedianLatencyMS:   []float64{10},
		Usage:                       []bool{false},
	})
	require.NoError(t, err)

	// Wait for both ticker functions to be created (template build times and deployment stats)
	tickerTrap.MustWait(ctx).MustRelease(ctx)
	tickerTrap.MustWait(ctx).MustRelease(ctx)

	clock.Advance(time.Minute).MustWait(ctx)

	stat, ok := cache.DeploymentStats()
	require.True(t, ok, "cache should be populated after refresh")
	require.Equal(t, int64(1), stat.SessionCount.VSCode)
}