Function TestUserAIProviderKeys

enterprise/dbcrypt/dbcrypt_internal_test.go:1309–1469

(t *testing.T)

}

func TestUserAIProviderKeys(t *testing.T) {
	t.Parallel()
	ctx := context.Background()

	const (
		//nolint:gosec // test credentials
		initialAPIKey = "sk-initial-ai-provider-key-value"
		//nolint:gosec // test credentials
		updatedAPIKey = "sk-updated-ai-provider-key-value"
		//nolint:gosec // test credentials
		rotatedAPIKey = "sk-rotated-ai-provider-key-value"
	)

	insertProviderAndKey := func(
		t *testing.T,
		crypt *dbCrypt,
		ciphers []Cipher,
	) (database.AIProvider, database.UserAiProviderKey) {
		t.Helper()
		user := dbgen.User(t, crypt, database.User{})
		provider := dbgen.AIProvider(t, crypt, database.AIProvider{})
		now := dbtime.Now()

		key, err := crypt.UpsertUserAIProviderKey(ctx, database.UpsertUserAIProviderKeyParams{
			ID:           uuid.New(),
			UserID:       user.ID,
			AIProviderID: provider.ID,
			APIKey:       initialAPIKey,
			CreatedAt:    now,
			UpdatedAt:    now,
		})
		require.NoError(t, err)
		require.Equal(t, initialAPIKey, key.APIKey)
		require.Equal(t, ciphers[0].HexDigest(), key.ApiKeyKeyID.String)
		return provider, key
	}

	getRawUserAIProviderKey := func(t *testing.T, store database.Store, userID uuid.UUID, providerID uuid.UUID) database.UserAiProviderKey {
		t.Helper()
		key, err := store.GetUserAIProviderKeyByProviderID(ctx, database.GetUserAIProviderKeyByProviderIDParams{
			UserID:       userID,
			AIProviderID: providerID,
		})
		require.NoError(t, err)
		return key
	}

	t.Run("UpsertUserAIProviderKeyCreatesValue", func(t *testing.T) {
		t.Parallel()
		db, crypt, ciphers := setup(t)
		provider, key := insertProviderAndKey(t, crypt, ciphers)

		got, err := crypt.GetUserAIProviderKeyByProviderID(ctx, database.GetUserAIProviderKeyByProviderIDParams{
			UserID:       key.UserID,
			AIProviderID: provider.ID,
		})
		require.NoError(t, err)
		require.Equal(t, key.ID, got.ID)