Function TestRotator

coderd/cryptokeys/rotate_test.go:17–107

(t *testing.T)

)

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

	t.Run("NoKeysOnInit", func(t *testing.T) {
		t.Parallel()

		var (
			db, _  = dbtestutil.NewDB(t)
			clock  = quartz.NewMock(t)
			logger = testutil.Logger(t)
			ctx    = testutil.Context(t, testutil.WaitShort)
		)

		dbkeys, err := db.GetCryptoKeys(ctx)
		require.NoError(t, err)
		require.Len(t, dbkeys, 0)

		cryptokeys.StartRotator(ctx, logger, db, cryptokeys.WithClock(clock))

		// Fetch the keys from the database and ensure they
		// are as expected.
		dbkeys, err = db.GetCryptoKeys(ctx)
		require.NoError(t, err)
		require.Len(t, dbkeys, len(database.AllCryptoKeyFeatureValues()))
		requireContainsAllFeatures(t, dbkeys)
	})

	t.Run("RotateKeys", func(t *testing.T) {
		t.Parallel()

		var (
			db, _  = dbtestutil.NewDB(t)
			clock  = quartz.NewMock(t)
			logger = testutil.Logger(t)
			ctx    = testutil.Context(t, testutil.WaitShort)
		)

		now := clock.Now().UTC()

		rotatingKey := dbgen.CryptoKey(t, db, database.CryptoKey{
			Feature:  database.CryptoKeyFeatureWorkspaceAppsAPIKey,
			StartsAt: now.Add(-cryptokeys.DefaultKeyDuration + time.Hour + time.Minute),
			Sequence: 12345,
		})

		trap := clock.Trap().TickerFunc()
		t.Cleanup(trap.Close)

		cryptokeys.StartRotator(ctx, logger, db, cryptokeys.WithClock(clock))

		initialKeyLen := len(database.AllCryptoKeyFeatureValues())
		// Fetch the keys from the database and ensure they
		// are as expected.
		dbkeys, err := db.GetCryptoKeys(ctx)
		require.NoError(t, err)
		require.Len(t, dbkeys, initialKeyLen)
		requireContainsAllFeatures(t, dbkeys)