summaryrefslogtreecommitdiff
path: root/service/test/adapter/test_helper.py
diff options
context:
space:
mode:
authorDuda Dornelles <ddornell@thoughtworks.com>2014-09-29 09:55:53 -0300
committerDuda Dornelles <ddornell@thoughtworks.com>2014-09-29 09:59:19 -0300
commit34df7f4dde3f481ff9c50dbb2338507ab70b04dd (patch)
tree860bb7e22d051ef609978319d73a59acdba14279 /service/test/adapter/test_helper.py
parent363196b8f860fb2ed487a07c5b7c36aa6cec9268 (diff)
fixing tests and pep8
Diffstat (limited to 'service/test/adapter/test_helper.py')
-rw-r--r--service/test/adapter/test_helper.py10
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
-