From 509669e260e8019902f3b08c44a07836af434860 Mon Sep 17 00:00:00 2001
From: irenarindos <irena.rindos@gmail.com>
Date: Tue, 8 Jul 2025 15:22:13 -0400
Subject: [PATCH] rdp: add proxy certs to protocol context at auth conn

---
 internal/session/testing.go | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/internal/session/testing.go b/internal/session/testing.go
index e9237e978..5e347dac9 100644
--- a/internal/session/testing.go
+++ b/internal/session/testing.go
@@ -152,6 +152,22 @@ func TestSession(t testing.TB, conn *db.DB, rootWrapper wrapping.Wrapper, c Comp
 		require.NoError(err)
 	}
 
+	if opts.withProxyCertificate != nil {
+		sessionProxyCertificate := opts.withProxyCertificate
+		sessionProxyCertificate.SessionId = s.PublicId
+
+		if len(sessionProxyCertificate.PrivateKey) == 0 || len(sessionProxyCertificate.Certificate) == 0 {
+			t.Fatalf("proxy certificate and private key must both be set")
+		}
+		err := sessionProxyCertificate.Encrypt(ctx, wrapper)
+		if err != nil {
+			require.NoError(err)
+		}
+		if err = rw.Create(ctx, sessionProxyCertificate); err != nil {
+			require.NoError(err)
+		}
+	}
+
 	ss, err := fetchStates(ctx, rw, s.PublicId, opts.withDbOpts...)
 	require.NoError(err)
 	s.States = ss