feat(IMEX-9): introduce a typedef to be able to mock ImportMessages.

ProtonMail · May 30, 2024 · 6fe543d · 6fe543d
1 parent a616326
commit 6fe543d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/message_import.go b/message_import.go
@@ -26,7 +26,9 @@ const (
 var ErrImportEncrypt = errors.New("failed to encrypt message")
 var ErrImportSizeExceeded = errors.New("message exceeds maximum import size of 30MB")
 
-func (c *Client) ImportMessages(ctx context.Context, addrKR *crypto.KeyRing, workers, buffer int, req ...ImportReq) (stream.Stream[ImportRes], error) {
+type ImportResStream stream.Stream[ImportRes] // gomock does not support generics. In order to be able to mock ImportMessages, we introduce a typedef.
+
+func (c *Client) ImportMessages(ctx context.Context, addrKR *crypto.KeyRing, workers, buffer int, req ...ImportReq) (ImportResStream, error) {
 	// Encrypt each message.
 	for idx := range req {
 		enc, err := EncryptRFC822(addrKR, req[idx].Message)