def setUp(self):
super(StoreTest, self).setUp()
self.store = Store(self.config)
def setUp(self):
super(StoreTest,self).setUp()
self.store = Store(self.config)
class StoreTest(BaseTest):
def setUp(self):
super(StoreTest, self).setUp()
self.store = Store(self.config)
def _get_msg(self, id):
return get_msg(self.store, id)
def _store_msg(self, fileName):
return archive_msg(self.store, fileName)
def test_message_exists_false(self):
msg = load_msg('sample')
self.assertFalse(self.store.message_exists(msg.id))
def test_message_exists_true(self):
msg = load_msg('sample')
self.store.archive(msg)
self.assertTrue(self.store.message_exists(msg.id))
def test_archive(self):
msg = load_msg('sample')
self.assertEqual(self.store.archive(msg), msg.id)
self.assertEqual(self._get_msg(msg.id).content, msg.as_string())
def test_archive_msg(self):
id, msg = self._store_msg('sample')
parsed_msg = self._get_msg(id).parsed_content
self.assertEqual(parsed_msg['Subject'], u'BETA 2.0')
self.assertEqual(parsed_msg['To'], u'*****@*****.**')
def _date_is_today(self, *dates):
today = datetime.now().date()
for d in dates:
self.assertEqual(d.date(), today)
def test_archive_date(self):
id, msg = self._store_msg('sample')
parsed_msg = self._get_msg(id)
self.assertTrue(parsed_msg.sent_at is None)
self._date_is_today(parsed_msg.received_at)
def test_mark_as_sent(self):
id, msg = self._store_msg('sample')
self.store.mark_as_sent(id)
parsed_msg = self._get_msg(id)
self._date_is_today(parsed_msg.sent_at)
# This test requires sqlite 3.6.19, which seems is not that widely distributed.
# def test_digest_no_msg(self):
# msg = load_msg('sample')
# # for some reason the assert is not working
# try:
# self.assertRaises(IntegrityError,
# self.store.digest(msg.id, u'*****@*****.**'))
# except IntegrityError:
# return
# self.fail()
def _digest_msg(self, fileName, *addresses):
return archive_msg(self.store, fileName, *addresses)
def test_digest(self):
id, msg = self._digest_msg('sample', u'*****@*****.**')
parsed_msg = self._get_msg(id)
self.assertEqual(len(parsed_msg.digests), 1)
digest = parsed_msg.digests[0]
self.assertEqual(digest.send_to, u'*****@*****.**')
self.assertEqual(digest.msg_id, id)
self.assertTrue(digest.sent_at is None)
self.assertEqual(digest.msg.content, msg.as_string())
def test_digest_empty(self):
id, msg = self._digest_msg('sample')
parsed_msg = self._get_msg(id)
self.assertEqual(len(parsed_msg.digests), 0)
def test_digest_multiple(self):
addresses = [u'*****@*****.**', u'*****@*****.**']
id, msg = self._digest_msg('sample', *addresses)
self._store_msg('sample2')
parsed_msg = self._get_msg(id)
self.assertEqual([d.send_to for d in parsed_msg.digests], addresses)
def test_messages_for_user_empty(self):
self.assertEqual(self.store.messages_for_user(u'*****@*****.**'), [])
def _digest_msgs(self, fileList, *addresses):
return [self._digest_msg(f, *addresses) for f in fileList]
def test_messages_for_user(self):
'''
Generate digests for two users and four messages. The first
two messages are marked as sent, so the pending messages
should be the last two.
'''
addresses = [u'*****@*****.**', u'*****@*****.**']
self._digest_msgs(['sample', 'sample2'], *addresses)
self.store.mark_digest_as_sent(addresses[0])
msgs = self._digest_msgs(['sample3', 'sample4'], *addresses)
parsed_msgs = self.store.messages_for_user(addresses[0])
self.assertEqual([m.id for m in parsed_msgs], [id for id, _ in msgs])
self.assertEqual(parsed_msgs[1].as_string(), msgs[1][1].as_string())
def test_users_with_pending_digests_empty(self):
self.assertEqual(self.store.users_with_pending_digests(), set())
def test_users_with_pending_digests(self):
'''
Generate digests for 3 users. Mark as read for first and
last. Add more digests for first user. Users 1 and 2 should
have pending digests.
'''
addresses = [
u'*****@*****.**', u'*****@*****.**', u'*****@*****.**'
]
self._digest_msgs(['sample', 'sample2'], *addresses)
self.store.mark_digest_as_sent(addresses[0])
self.store.mark_digest_as_sent(addresses[2])
self._digest_msgs(['sample3', 'sample4'], addresses[0])
self.assertEqual(self.store.users_with_pending_digests(),
set(addresses[0:2]))
def test_mark_digest_as_sent_empty(self):
self.assertEqual(self.store.mark_digest_as_sent(u'*****@*****.**'), [])
def test_mark_digest_as_sent(self):
addresses = [u'*****@*****.**', u'*****@*****.**']
msgs = self._digest_msgs(['sample', 'sample2'], *addresses)
self.assertEqual(self.store.mark_digest_as_sent(addresses[0]),
[id for id, _ in msgs])
self._date_is_today(
*[d.sent_at for d in get_digests(self.store, addresses[0])])
for digest in get_digests(self.store, addresses[1]):
self.assertTrue(digest.sent_at is None)
def test_discard_old_digests_empty(self):
self.assertEqual(self.store.discard_old_digests(3), [])
def test_discard_old_digests(self):
'''
Scenario: we have digests that have been already sent, new
digests and old digests.
If the discard method is executed, it should only mark the old
digests. Afterwards, the new digests should still be found by
messages_for_user.
'''
addresses = [u'*****@*****.**', u'*****@*****.**']
msgs = self._digest_msgs(['sample', 'sample2'], *addresses)
self.store.mark_digest_as_sent(addresses[1])
get_digests(self.store,
addresses[0])[0].scheduled_at = datetime.now() - timedelta(
days=100)
self.store._db.session.flush()
self.assertEqual(self.store.discard_old_digests(3), [msgs[0][0]])
self.assertEqual(self.store.users_with_pending_digests(),
set([addresses[0]]))
parsed_msgs = self.store.messages_for_user(addresses[0])
self.assertEqual(len(parsed_msgs), 1)
self.assertEquals(parsed_msgs[0].as_string(), msgs[1][1].as_string())
class StoreTest(BaseTest):
def setUp(self):
super(StoreTest,self).setUp()
self.store = Store(self.config)
def _get_msg(self, id):
return get_msg(self.store, id)
def _store_msg(self, fileName):
return archive_msg(self.store, fileName)
def test_message_exists_false(self):
msg = load_msg('sample')
self.assertFalse(self.store.message_exists(msg.id))
def test_message_exists_true(self):
msg = load_msg('sample')
self.store.archive(msg)
self.assertTrue(self.store.message_exists(msg.id))
def test_archive(self):
msg = load_msg('sample')
self.assertEqual(self.store.archive(msg), msg.id)
self.assertEqual(self._get_msg(msg.id).content, msg.as_string())
def test_archive_msg(self):
id, msg = self._store_msg('sample')
parsed_msg = self._get_msg(id).parsed_content
self.assertEqual(parsed_msg['Subject'], u'BETA 2.0')
self.assertEqual(parsed_msg['To'], u'*****@*****.**')
def _date_is_today(self, *dates):
today = datetime.now().date()
for d in dates:
self.assertEqual(d.date(), today)
def test_archive_date(self):
id, msg = self._store_msg('sample')
parsed_msg = self._get_msg(id)
self.assertTrue(parsed_msg.sent_at is None)
self._date_is_today(parsed_msg.received_at)
def test_mark_as_sent(self):
id, msg = self._store_msg('sample')
self.store.mark_as_sent(id)
parsed_msg = self._get_msg(id)
self._date_is_today(parsed_msg.sent_at)
# This test requires sqlite 3.6.19, which seems is not that widely distributed.
# def test_digest_no_msg(self):
# msg = load_msg('sample')
# # for some reason the assert is not working
# try:
# self.assertRaises(IntegrityError,
# self.store.digest(msg.id, u'*****@*****.**'))
# except IntegrityError:
# return
# self.fail()
def _digest_msg(self, fileName, *addresses):
return archive_msg(self.store, fileName, *addresses)
def test_digest(self):
id, msg = self._digest_msg('sample', u'*****@*****.**')
parsed_msg = self._get_msg(id)
self.assertEqual(len(parsed_msg.digests), 1)
digest = parsed_msg.digests[0]
self.assertEqual(digest.send_to, u'*****@*****.**')
self.assertEqual(digest.msg_id, id)
self.assertTrue(digest.sent_at is None)
self.assertEqual(digest.msg.content, msg.as_string())
def test_digest_empty(self):
id, msg = self._digest_msg('sample')
parsed_msg = self._get_msg(id)
self.assertEqual(len(parsed_msg.digests), 0)
def test_digest_multiple(self):
addresses = [u'*****@*****.**', u'*****@*****.**']
id, msg = self._digest_msg('sample', *addresses)
self._store_msg('sample2')
parsed_msg = self._get_msg(id)
self.assertEqual([d.send_to for d in parsed_msg.digests], addresses)
def test_messages_for_user_empty(self):
self.assertEqual(self.store.messages_for_user(u'*****@*****.**'), [])
def _digest_msgs(self, fileList, *addresses):
return [self._digest_msg(f, *addresses) for f in fileList]
def test_messages_for_user(self):
'''
Generate digests for two users and four messages. The first
two messages are marked as sent, so the pending messages
should be the last two.
'''
addresses = [u'*****@*****.**', u'*****@*****.**']
self._digest_msgs(['sample', 'sample2'], *addresses)
self.store.mark_digest_as_sent(addresses[0])
msgs = self._digest_msgs(['sample3', 'sample4'], *addresses)
parsed_msgs = self.store.messages_for_user(addresses[0])
self.assertEqual([m.id for m in parsed_msgs], [id for id, _ in msgs])
self.assertEqual(parsed_msgs[1].as_string(), msgs[1][1].as_string())
def test_users_with_pending_digests_empty(self):
self.assertEqual(self.store.users_with_pending_digests(), set())
def test_users_with_pending_digests(self):
'''
Generate digests for 3 users. Mark as read for first and
last. Add more digests for first user. Users 1 and 2 should
have pending digests.
'''
addresses = [u'*****@*****.**', u'*****@*****.**', u'*****@*****.**']
self._digest_msgs(['sample', 'sample2'], *addresses)
self.store.mark_digest_as_sent(addresses[0])
self.store.mark_digest_as_sent(addresses[2])
self._digest_msgs(['sample3', 'sample4'], addresses[0])
self.assertEqual(self.store.users_with_pending_digests(), set(addresses[0:2]))
def test_mark_digest_as_sent_empty(self):
self.assertEqual(self.store.mark_digest_as_sent(u'*****@*****.**'), [])
def test_mark_digest_as_sent(self):
addresses = [u'*****@*****.**', u'*****@*****.**']
msgs = self._digest_msgs(['sample', 'sample2'], *addresses)
self.assertEqual(self.store.mark_digest_as_sent(addresses[0]),
[id for id, _ in msgs])
self._date_is_today(*[d.sent_at for d in get_digests(self.store, addresses[0])])
for digest in get_digests(self.store, addresses[1]):
self.assertTrue(digest.sent_at is None)
def test_discard_old_digests_empty(self):
self.assertEqual(self.store.discard_old_digests(3), [])
def test_discard_old_digests(self):
'''
Scenario: we have digests that have been already sent, new
digests and old digests.
If the discard method is executed, it should only mark the old
digests. Afterwards, the new digests should still be found by
messages_for_user.
'''
addresses = [u'*****@*****.**', u'*****@*****.**']
msgs = self._digest_msgs(['sample', 'sample2'], *addresses)
self.store.mark_digest_as_sent(addresses[1])
get_digests(self.store, addresses[0])[0].scheduled_at = datetime.now() - timedelta(days=100)
self.store._db.session.flush()
self.assertEqual(self.store.discard_old_digests(3), [msgs[0][0]])
self.assertEqual(self.store.users_with_pending_digests(), set([addresses[0]]))
parsed_msgs = self.store.messages_for_user(addresses[0])
self.assertEqual(len(parsed_msgs), 1)
self.assertEquals(parsed_msgs[0].as_string(), msgs[1][1].as_string())
