synced/coder
1
0
Fork 0
mirror of https://github.com/coder/coder.git synced 2025-07-09 11:45:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(enterprise/coderd/license): fix time-related flake in license expiration warning test (#15228)

Browse Source 
Fixes a time-related test flake in enterprise/coderd/license/license_test.go
This commit is contained in:
Cian Johnston
2024-10-25 12:19:50 +01:00
committed by GitHub
parent b62f3e6af5
commit 5bcaa93198
1 changed files with 16 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
enterprise/coderd/license/license_test.go
View File

@ -54,7 +54,7 @@ func TestEntitlements(t *testing.T) {
db := dbmem.New() db := dbmem.New()
db.InsertLicense(context.Background(), database.InsertLicenseParams{ db.InsertLicense(context.Background(), database.InsertLicenseParams{
JWT: coderdenttest.GenerateLicense(t, coderdenttest.LicenseOptions{}), JWT: coderdenttest.GenerateLicense(t, coderdenttest.LicenseOptions{}),
Exp: time.Now().Add(time.Hour), Exp: dbtime.Now().Add(time.Hour),
}) })
entitlements, err := license.Entitlements(context.Background(), db, 1, 1, coderdenttest.Keys, empty) entitlements, err := license.Entitlements(context.Background(), db, 1, 1, coderdenttest.Keys, empty)
require.NoError(t, err) require.NoError(t, err)
@ -78,7 +78,7 @@ func TestEntitlements(t *testing.T) {
return f return f
}(), }(),
}), }),
Exp: time.Now().Add(time.Hour), Exp: dbtime.Now().Add(time.Hour),
}) })
entitlements, err := license.Entitlements(context.Background(), db, 1, 1, coderdenttest.Keys, empty) entitlements, err := license.Entitlements(context.Background(), db, 1, 1, coderdenttest.Keys, empty)
require.NoError(t, err) require.NoError(t, err)
@ -98,10 +98,10 @@ func TestEntitlements(t *testing.T) {
codersdk.FeatureAuditLog: 1, codersdk.FeatureAuditLog: 1,
}, },
GraceAt: time.Now().Add(-time.Hour), GraceAt: dbtime.Now().Add(-time.Hour),
ExpiresAt: time.Now().Add(time.Hour), ExpiresAt: dbtime.Now().Add(time.Hour),
}), }),
Exp: time.Now().Add(time.Hour), Exp: dbtime.Now().Add(time.Hour),
}) })
entitlements, err := license.Entitlements(context.Background(), db, 1, 1, coderdenttest.Keys, all) entitlements, err := license.Entitlements(context.Background(), db, 1, 1, coderdenttest.Keys, all)
require.NoError(t, err) require.NoError(t, err)
@ -124,10 +124,10 @@ func TestEntitlements(t *testing.T) {
codersdk.FeatureAuditLog: 1, codersdk.FeatureAuditLog: 1,
}, },
GraceAt: time.Now().AddDate(0, 0, 2), GraceAt: dbtime.Now().AddDate(0, 0, 2),
ExpiresAt: time.Now().AddDate(0, 0, 5), ExpiresAt: dbtime.Now().AddDate(0, 0, 5),
}), }),
Exp: time.Now().AddDate(0, 0, 5), Exp: dbtime.Now().AddDate(0, 0, 5),
}) })
entitlements, err := license.Entitlements(context.Background(), db, 1, 1, coderdenttest.Keys, all) entitlements, err := license.Entitlements(context.Background(), db, 1, 1, coderdenttest.Keys, all)
@ -153,8 +153,8 @@ func TestEntitlements(t *testing.T) {
codersdk.FeatureAuditLog: 1, codersdk.FeatureAuditLog: 1,
}, },
GraceAt: time.Now().AddDate(0, 0, 1), GraceAt: dbtime.Now().AddDate(0, 0, 1),
ExpiresAt: time.Now().AddDate(0, 0, 5), ExpiresAt: dbtime.Now().AddDate(0, 0, 5),
}), }),
Exp: time.Now().AddDate(0, 0, 5), Exp: time.Now().AddDate(0, 0, 5),
}) })
@ -183,10 +183,10 @@ func TestEntitlements(t *testing.T) {
}, },
Trial: true, Trial: true,
GraceAt: time.Now().AddDate(0, 0, 8), GraceAt: dbtime.Now().AddDate(0, 0, 8),
ExpiresAt: time.Now().AddDate(0, 0, 5), ExpiresAt: dbtime.Now().AddDate(0, 0, 5),
}), }),
Exp: time.Now().AddDate(0, 0, 5), Exp: dbtime.Now().AddDate(0, 0, 5),
}) })
entitlements, err := license.Entitlements(context.Background(), db, 1, 1, coderdenttest.Keys, all) entitlements, err := license.Entitlements(context.Background(), db, 1, 1, coderdenttest.Keys, all)
@ -212,10 +212,10 @@ func TestEntitlements(t *testing.T) {
codersdk.FeatureAuditLog: 1, codersdk.FeatureAuditLog: 1,
}, },
GraceAt: time.Now().AddDate(0, 0, 30), GraceAt: dbtime.Now().AddDate(0, 0, 30),
ExpiresAt: time.Now().AddDate(0, 0, 5), ExpiresAt: dbtime.Now().AddDate(0, 0, 5),
}), }),
Exp: time.Now().AddDate(0, 0, 5), Exp: dbtime.Now().AddDate(0, 0, 5),
}) })
entitlements, err := license.Entitlements(context.Background(), db, 1, 1, coderdenttest.Keys, all) entitlements, err := license.Entitlements(context.Background(), db, 1, 1, coderdenttest.Keys, all)