Function TestMCPServerConfigs

enterprise/dbcrypt/dbcrypt_internal_test.go:919–1056

(t *testing.T)

}

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

	const (
		//nolint:gosec // test credentials
		oauthSecret   = "my-oauth-secret"
		apiKeyValue   = "my-api-key"
		customHeaders = `{"X-Custom":"header-value"}`
	)
	// insertConfig is a small helper that creates an MCP server
	// config through the encrypted store with secret fields set.
	insertConfig := func(t *testing.T, crypt *dbCrypt, ciphers []Cipher) database.MCPServerConfig {
		t.Helper()
		cfg := dbgen.MCPServerConfig(t, crypt, database.MCPServerConfig{
			Description:        "test description",
			AuthType:           "oauth2",
			OAuth2ClientID:     "client-id",
			OAuth2ClientSecret: oauthSecret,
			APIKeyValue:        apiKeyValue,
			CustomHeaders:      customHeaders,
			Availability:       "force_on",
		})
		requireMCPServerConfigDecrypted(t, cfg, ciphers, oauthSecret, apiKeyValue, customHeaders)
		return cfg
	}

	t.Run("InsertMCPServerConfig", func(t *testing.T) {
		t.Parallel()
		db, crypt, ciphers := setup(t)
		cfg := insertConfig(t, crypt, ciphers)
		requireMCPServerConfigRawEncrypted(ctx, t, db, cfg.ID, ciphers, oauthSecret, apiKeyValue, customHeaders)
	})

	t.Run("GetMCPServerConfigByID", func(t *testing.T) {
		t.Parallel()
		db, crypt, ciphers := setup(t)
		cfg := insertConfig(t, crypt, ciphers)

		got, err := crypt.GetMCPServerConfigByID(ctx, cfg.ID)
		require.NoError(t, err)
		requireMCPServerConfigDecrypted(t, got, ciphers, oauthSecret, apiKeyValue, customHeaders)
		requireMCPServerConfigRawEncrypted(ctx, t, db, cfg.ID, ciphers, oauthSecret, apiKeyValue, customHeaders)
	})

	t.Run("GetMCPServerConfigBySlug", func(t *testing.T) {
		t.Parallel()
		db, crypt, ciphers := setup(t)
		cfg := insertConfig(t, crypt, ciphers)

		got, err := crypt.GetMCPServerConfigBySlug(ctx, cfg.Slug)
		require.NoError(t, err)
		requireMCPServerConfigDecrypted(t, got, ciphers, oauthSecret, apiKeyValue, customHeaders)
		requireMCPServerConfigRawEncrypted(ctx, t, db, cfg.ID, ciphers, oauthSecret, apiKeyValue, customHeaders)
	})

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