Fix problem with mocks

xmtp · Oct 24, 2023 · 24c8c2a · 24c8c2a
1 parent a50e30d
commit 24c8c2a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pkg/api/message/v3/service_test.go b/pkg/api/message/v3/service_test.go
@@ -25,7 +25,12 @@ type mockedMlsValidationService struct {
 func (m *mockedMlsValidationService) ValidateKeyPackages(ctx context.Context, keyPackages [][]byte) ([]mlsvalidate.IdentityValidationResult, error) {
 	args := m.Called(ctx, keyPackages)
 
-	return args.Get(0).([]mlsvalidate.IdentityValidationResult), args.Error(1)
+	response := args.Get(0)
+	if response == nil {
+		return nil, args.Error(1)
+	}
+
+	return response.([]mlsvalidate.IdentityValidationResult), args.Error(1)
 }
 
 func (m *mockedMlsValidationService) ValidateGroupMessages(ctx context.Context, groupMessages [][]byte) ([]mlsvalidate.GroupMessageValidationResult, error) {