diff options
Diffstat (limited to 'service/test/support/integration/model.py')
| -rw-r--r-- | service/test/support/integration/model.py | 12 |
1 files changed, 12 insertions, 0 deletions
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) |