diff options
author | Duda Dornelles <ddornell@thoughtworks.com> | 2014-09-29 09:55:53 -0300 |
---|---|---|
committer | Duda Dornelles <ddornell@thoughtworks.com> | 2014-09-29 09:59:19 -0300 |
commit | 34df7f4dde3f481ff9c50dbb2338507ab70b04dd (patch) | |
tree | 860bb7e22d051ef609978319d73a59acdba14279 /service/test/adapter/test_helper.py | |
parent | 363196b8f860fb2ed487a07c5b7c36aa6cec9268 (diff) |
fixing tests and pep8
Diffstat (limited to 'service/test/adapter/test_helper.py')
-rw-r--r-- | service/test/adapter/test_helper.py | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/service/test/adapter/test_helper.py b/service/test/adapter/test_helper.py index 86552008..af0fed95 100644 --- a/service/test/adapter/test_helper.py +++ b/service/test/adapter/test_helper.py @@ -27,6 +27,7 @@ LEAP_FLAGS = ['\\Seen', DEFAULT_HEADERS = {'date': str(datetime.now())} + def mail_dict(): return { 'header': { @@ -40,16 +41,20 @@ def mail_dict(): 'tags': [] } + class TestDoc: def __init__(self, content): self.content = content + def leap_mail(uid=0, flags=LEAP_FLAGS, headers=None, extra_headers={}, mbox='INBOX', body='body', chash='chash'): fdoc = TestDoc({'flags': flags, 'mbox': mbox, 'type': 'flags', 'uid': uid, 'chash': chash}) - if headers is None: headers = {} - if not (headers.get('received') or headers.get('date')): headers.update(DEFAULT_HEADERS) + if headers is None: + headers = {} + if not (headers.get('received') or headers.get('date')): + headers.update(DEFAULT_HEADERS) headers['headers'] = extra_headers hdoc = TestDoc(headers) @@ -64,4 +69,3 @@ def input_mail(): mail._chash = "123" mail.as_dict = lambda: None return mail - |