From a0bc490dc985f88d8a7f9a835e6cd202e82cc760 Mon Sep 17 00:00:00 2001
From: Patrick Maia <patrickjourdanmaia@gmail.com>
Date: Fri, 12 Dec 2014 16:20:12 +0000
Subject: Card #168 - introduces performance tests for contacts controller

---
 service/test/support/integration/model.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'service/test/support/integration/model.py')

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)
-- 
cgit v1.2.3