Function TestUsageLimitFeatures

enterprise/coderd/license/license_test.go:1726–1801

(t *testing.T)

}

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

	// Ensures that usage limit features are ranked by issued at, not by
	// values.
	t.Run("IssuedAtRanking", func(t *testing.T) {
		t.Parallel()

		// Generate 2 real licenses both with managed agent limit
		// features. lic2 should trump lic1 even though it has a lower
		// limit, because it was issued later.
		lic1 := database.License{
			ID:         1,
			UploadedAt: time.Now(),
			Exp:        time.Now().Add(time.Hour),
			UUID:       uuid.New(),
			JWT: coderdenttest.GenerateLicense(t, coderdenttest.LicenseOptions{
				IssuedAt:  time.Now().Add(-time.Minute * 2),
				NotBefore: dbtime.Now().Add(-time.Minute * 2),
				ExpiresAt: time.Now().Add(time.Hour * 2),
				Features: license.Features{
					codersdk.FeatureManagedAgentLimit: 100,
				},
			}),
		}
		lic2Iat := time.Now().Add(-time.Minute * 1)
		lic2Nbf := lic2Iat.Add(-time.Minute)
		lic2Exp := lic2Iat.Add(time.Hour)
		lic2 := database.License{
			ID:         2,
			UploadedAt: time.Now(),
			Exp:        lic2Exp,
			UUID:       uuid.New(),
			JWT: coderdenttest.GenerateLicense(t, coderdenttest.LicenseOptions{
				IssuedAt:  lic2Iat,
				NotBefore: lic2Nbf,
				ExpiresAt: lic2Exp,
				Features: license.Features{
					codersdk.FeatureManagedAgentLimit: 50,
				},
			}),
		}

		const actualAgents = 10
		arguments := license.FeatureArguments{
			ActiveUserCount:   10,
			ReplicaCount:      0,
			ExternalAuthCount: 0,
			ManagedAgentCountFn: func(ctx context.Context, from time.Time, to time.Time) (int64, error) {
				return actualAgents, nil
			},
		}

		// Load the licenses in both orders to ensure the correct
		// behavior is observed no matter the order.
		for _, order := range [][]database.License{
			{lic1, lic2},
			{lic2, lic1},