Card #168 - introduces performance tests for contacts controller

2 files changed, 14 insertions, 2 deletions

diff --git a/service/test/support/integration/app_test_client.py b/service/test/support/integration/app_test_client.py
index 2a35a89f..82b34d69 100644
--- a/service/test/support/integration/app_test_client.py
+++ b/service/test/support/integration/app_test_client.py
@@ -122,10 +122,10 @@ class AppTestClient:
         mail.update_tags(input_mail.tags)
         self.search_engine.index_mail(mail)
 
-    def add_multiple_to_mailbox(self, num, mailbox='', flags=[], tags=[]):
+    def add_multiple_to_mailbox(self, num, mailbox='', flags=[], tags=[], to='recipient@to.com', cc='recipient@cc.com', bcc='recipient@bcc.com'):
         mails = []
         for _ in range(num):
-            input_mail = MailBuilder().with_status(flags).with_tags(tags).build_input_mail()
+            input_mail = MailBuilder().with_status(flags).with_tags(tags).with_to(to).with_cc(cc).with_bcc(bcc).build_input_mail()
             mail = self.mailboxes._create_or_get(mailbox).add(input_mail)
             mails.append(mail)
             mail.update_tags(input_mail.tags)
diff --git a/service/test/support/integration/model.py b/service/test/support/integration/model.py
index 79552dad..e05d74bb 100644
--- a/service/test/support/integration/model.py
+++ b/service/test/support/integration/model.py
@@ -44,6 +44,18 @@ class MailBuilder:
         self.mail['header']['subject'] = subject
         return self
 
+    def with_to(self, to):
+        self.mail['header']['to'] = to
+        return self
+
+    def with_cc(self, cc):
+        self.mail['header']['cc'] = cc
+        return self
+
+    def with_bcc(self, bcc):
+        self.mail['header']['bcc'] = bcc
+        return self
+
     def with_status(self, flags):
         for status in Status.from_flags(flags):
             self.mail['status'].append(status)