From 460e6066b2638f7ccbd7268c51631bed7ecfe444 Mon Sep 17 00:00:00 2001
From: Nicholas Molnar <65710+neekolas@users.noreply.github.com>
Date: Fri, 26 Jan 2024 12:05:21 -0800
Subject: [PATCH] Add HpkePublicKey to stream (#348)

* Add HpkePublicKey to stream

* Remove useless assertion
---
 pkg/mls/api/v1/service.go      | 1 +
 pkg/mls/api/v1/service_test.go | 1 +
 2 files changed, 2 insertions(+)

diff --git a/pkg/mls/api/v1/service.go b/pkg/mls/api/v1/service.go
index c83cb1e0..426f229c 100644
--- a/pkg/mls/api/v1/service.go
+++ b/pkg/mls/api/v1/service.go
@@ -310,6 +310,7 @@ func (s *Service) SendWelcomeMessages(ctx context.Context, req *mlsv1.SendWelcom
 					CreatedNs:       uint64(msg.CreatedAt.UnixNano()),
 					InstallationKey: msg.InstallationKey,
 					Data:            msg.Data,
+					HpkePublicKey:   msg.HpkePublicKey,
 				},
 			},
 		})
diff --git a/pkg/mls/api/v1/service_test.go b/pkg/mls/api/v1/service_test.go
index 45308261..4d4565ad 100644
--- a/pkg/mls/api/v1/service_test.go
+++ b/pkg/mls/api/v1/service_test.go
@@ -409,6 +409,7 @@ func TestSubscribeWelcomeMessages(t *testing.T) {
 					CreatedNs:       uint64(i + 1),
 					InstallationKey: installationKey,
 					Data:            []byte(fmt.Sprintf("data%d", i+1)),
+					HpkePublicKey:   []byte(fmt.Sprintf("hpke%d", i+1)),
 				},
 			},
 		}