Function TestGetUserStatusCounts

coderd/database/querier_test.go:5410–6004

(t *testing.T)

}

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

	type testCase struct {
		timezone    string
		location    *time.Location
		reportFrom  time.Time
		reportUntil time.Time
	}
	testCases := []testCase{}

	// GetUserStatusCounts is sensitive to DST transitions, because it generates timestamps exactly
	// one day apart from one another, and specific days can have varying lengths depending on the timezone.
	// Therefore, we test with a variety of timezones.
	timezones := []string{
		"America/St_Johns",
		"Africa/Johannesburg",
		"America/New_York",
		"Europe/London",
		"Asia/Tokyo",
		"Australia/Sydney",
	}

	// assemble test cases
	for _, tz := range timezones {
		location, err := time.LoadLocation(tz)
		if err != nil {
			t.Fatalf("failed to load location: %v", err)
		}

		// Testing based on the current system date will flake due to DST transitions.
		// Instead, we test with a fixed range of dates that is large enough to span multiple DST transitions.
		startOfTestDateRange := time.Date(2025, 1, 1, 0, 0, 0, 0, location)
		endOfTestDateRange := time.Date(2026, 1, 1, 0, 0, 0, 0, location)
		// To keep the number of test cases manageable given the large date range,
		// we test with a suitable large interval. This interval is also the length of each report.
		// this ensures we have full coverage of the date range.
		testDateRangeInterval := 60

		for reportFrom := startOfTestDateRange; !reportFrom.After(endOfTestDateRange); reportFrom = reportFrom.AddDate(0, 0, testDateRangeInterval) {
			testCases = append(testCases, testCase{
				timezone:    tz,
				location:    location,
				reportFrom:  dbtime.Time(reportFrom),
				reportUntil: dbtime.Time(reportFrom.AddDate(0, 0, testDateRangeInterval)),
			})
		}
	}

	for _, tc := range testCases {
		t.Run(fmt.Sprintf("%s/%s", tc.timezone, tc.reportUntil.Format("2006-01-02T15:04:05Z")), func(t *testing.T) {
			t.Parallel()

			userCreatedAt := tc.reportUntil.AddDate(0, 0, -60)
			firstStatusChange := userCreatedAt.AddDate(0, 0, 29)
			secondStatusChange := firstStatusChange.AddDate(0, 0, 29)

			t.Run("No Users", func(t *testing.T) {