def test_fetch_by_chats(self):
"""Test if it returns only those messages that belong to the given chats"""
_ = setup_http_server()
tlg = Telegram(TELEGRAM_BOT, TELEGRAM_TOKEN)
chats = [8, -1]
messages = [msg for msg in tlg.fetch(chats=chats)]
self.assertEqual(len(messages), 3)
expected = [(31, '5a5457aec04237ac3fab30031e84c745a3bdd157', 1467289325.0, 319280318),
(33, '9d03eeea7e3186ca8e5c150b4cbf18c8283cca9d', 1467289371.0, 319280320),
(34, '2e61e72b64c9084f3c5a36671c3119641c3ae42f', 1467370372.0, 319280321)]
for x in range(len(messages)):
message = messages[x]
self.assertEqual(message['data']['message']['message_id'], expected[x][0])
self.assertEqual(message['origin'], 'https://telegram.org/' + TELEGRAM_BOT)
self.assertEqual(message['uuid'], expected[x][1])
self.assertEqual(message['updated_on'], expected[x][2])
self.assertEqual(message['offset'], expected[x][3])
self.assertEqual(message['category'], 'message')
self.assertEqual(message['tag'], 'https://telegram.org/' + TELEGRAM_BOT)
# Empty list of chats will return no messages
chats = []
messages = [msg for msg in tlg.fetch(chats=chats)]
self.assertEqual(len(messages), 0)
def test_fetch_from_offset(self):
"""Test whether it fetches and parses messages from the given offset"""
http_requests = setup_http_server()
tlg = Telegram(TELEGRAM_BOT, TELEGRAM_TOKEN)
messages = [msg for msg in tlg.fetch(offset=319280321)]
self.assertEqual(len(messages), 1)
msg = messages[0]
self.assertEqual(msg['data']['message']['message_id'], 34)
self.assertEqual(msg['origin'], 'https://telegram.org/' + TELEGRAM_BOT)
self.assertEqual(msg['uuid'],
'2e61e72b64c9084f3c5a36671c3119641c3ae42f')
self.assertEqual(msg['updated_on'], 1467370372.0)
self.assertEqual(msg['offset'], 319280321)
self.assertEqual(msg['category'], 'message')
self.assertEqual(msg['tag'], 'https://telegram.org/' + TELEGRAM_BOT)
# Check requests
expected = [{'offset': ['319280321']}, {'offset': ['319280322']}]
self.assertEqual(len(http_requests), len(expected))
for i in range(len(expected)):
self.assertDictEqual(http_requests[i].querystring, expected[i])
def test_fetch_from_offset(self):
"""Test whether it fetches and parses messages from the given offset"""
http_requests = setup_http_server()
tlg = Telegram(TELEGRAM_BOT, TELEGRAM_TOKEN)
messages = [msg for msg in tlg.fetch(offset=319280321)]
self.assertEqual(len(messages), 1)
msg = messages[0]
self.assertEqual(msg['data']['message']['message_id'], 34)
self.assertEqual(msg['origin'], 'https://telegram.org/' + TELEGRAM_BOT)
self.assertEqual(msg['uuid'], '2e61e72b64c9084f3c5a36671c3119641c3ae42f')
self.assertEqual(msg['updated_on'], 1467370372.0)
self.assertEqual(msg['offset'], 319280321)
self.assertEqual(msg['category'], 'message')
self.assertEqual(msg['tag'], 'https://telegram.org/' + TELEGRAM_BOT)
# Check requests
expected = [
{'offset' : ['319280321']},
{'offset' : ['319280322']}
]
self.assertEqual(len(http_requests), len(expected))
for i in range(len(expected)):
self.assertDictEqual(http_requests[i].querystring, expected[i])
def test_fetch(self):
"""Test whether a list of messages is returned"""
http_requests = setup_http_server()
tlg = Telegram(TELEGRAM_BOT, TELEGRAM_TOKEN)
messages = [msg for msg in tlg.fetch()]
expected = [(31, '5a5457aec04237ac3fab30031e84c745a3bdd157', 1467289325.0, 319280318),
(32, '16a59e93e919174fcd4e70e5b3289201c1016c72', 1467289329.0, 319280319),
(33, '9d03eeea7e3186ca8e5c150b4cbf18c8283cca9d', 1467289371.0, 319280320),
(34, '2e61e72b64c9084f3c5a36671c3119641c3ae42f', 1467370372.0, 319280321)]
self.assertEqual(len(messages), len(expected))
for x in range(len(messages)):
message = messages[x]
self.assertEqual(message['data']['message']['message_id'], expected[x][0])
self.assertEqual(message['origin'], 'https://telegram.org/' + TELEGRAM_BOT)
self.assertEqual(message['uuid'], expected[x][1])
self.assertEqual(message['updated_on'], expected[x][2])
self.assertEqual(message['offset'], expected[x][3])
self.assertEqual(message['category'], 'message')
# Check requests
expected = [
{'offset' : ['1']},
{'offset' : ['319280321']},
{'offset' : ['319280322']}
]
self.assertEqual(len(http_requests), len(expected))
for i in range(len(expected)):
self.assertDictEqual(http_requests[i].querystring, expected[i])
def test_fetch(self):
"""Test whether a list of messages is returned"""
http_requests = setup_http_server()
tlg = Telegram(TELEGRAM_BOT, TELEGRAM_TOKEN)
messages = [msg for msg in tlg.fetch()]
expected = [(31, '5a5457aec04237ac3fab30031e84c745a3bdd157', 1467289325.0, 319280318),
(32, '16a59e93e919174fcd4e70e5b3289201c1016c72', 1467289329.0, 319280319),
(33, '9d03eeea7e3186ca8e5c150b4cbf18c8283cca9d', 1467289371.0, 319280320),
(34, '2e61e72b64c9084f3c5a36671c3119641c3ae42f', 1467370372.0, 319280321)]
self.assertEqual(len(messages), len(expected))
for x in range(len(messages)):
message = messages[x]
self.assertEqual(message['data']['message']['message_id'], expected[x][0])
self.assertEqual(message['origin'], 'https://telegram.org/' + TELEGRAM_BOT)
self.assertEqual(message['uuid'], expected[x][1])
self.assertEqual(message['updated_on'], expected[x][2])
self.assertEqual(message['offset'], expected[x][3])
# Check requests
expected = [
{'offset' : ['1']},
{'offset' : ['319280321']},
{'offset' : ['319280322']}
]
self.assertEqual(len(http_requests), len(expected))
for i in range(len(expected)):
self.assertDictEqual(http_requests[i].querystring, expected[i])
def test_fetch_empty(self):
"""Test whether it works when there are no messages to fetch"""
http_requests = setup_http_server()
tlg = Telegram(TELEGRAM_BOT, TELEGRAM_TOKEN)
messages = [msg for msg in tlg.fetch(offset=319280322)]
self.assertEqual(len(messages), 0)
# Check requests
self.assertEqual(len(http_requests), 1)
self.assertDictEqual(http_requests[0].querystring,
{'offset': ['319280322']})
def test_fetch_empty(self):
"""Test whether it works when there are no messages to fetch"""
http_requests = setup_http_server()
tlg = Telegram(TELEGRAM_BOT, TELEGRAM_TOKEN)
messages = [msg for msg in tlg.fetch(offset=319280322)]
self.assertEqual(len(messages), 0)
# Check requests
self.assertEqual(len(http_requests), 1)
self.assertDictEqual(http_requests[0].querystring,
{'offset' : ['319280322']})
def test_fetch_from_cache(self):
"""Test whether the cache works"""
http_requests = setup_http_server()
# First, we fetch the messages from the server,
# storing them in a cache
cache = Cache(self.tmp_path)
tlg = Telegram(TELEGRAM_BOT, TELEGRAM_TOKEN, cache=cache)
messages = [msg for msg in tlg.fetch()]
self.assertEqual(len(http_requests), 3)
# Now, we get the messages from the cache.
# The contents should be the same and there won't be
# any new request to the server
cached_messages = [msg for msg in tlg.fetch_from_cache()]
self.assertEqual(len(cached_messages), len(messages))
expected = [(31, '5a5457aec04237ac3fab30031e84c745a3bdd157',
1467289325.0, 319280318),
(32, '16a59e93e919174fcd4e70e5b3289201c1016c72',
1467289329.0, 319280319),
(33, '9d03eeea7e3186ca8e5c150b4cbf18c8283cca9d',
1467289371.0, 319280320),
(34, '2e61e72b64c9084f3c5a36671c3119641c3ae42f',
1467370372.0, 319280321)]
self.assertEqual(len(cached_messages), len(expected))
for x in range(len(cached_messages)):
message = cached_messages[x]
self.assertEqual(message['data']['message']['message_id'],
expected[x][0])
self.assertEqual(message['origin'],
'https://telegram.org/' + TELEGRAM_BOT)
self.assertEqual(message['uuid'], expected[x][1])
self.assertEqual(message['updated_on'], expected[x][2])
self.assertEqual(message['offset'], expected[x][3])
self.assertEqual(message['category'], 'message')
self.assertEqual(message['tag'],
'https://telegram.org/' + TELEGRAM_BOT)
# No more requests were sent
self.assertEqual(len(http_requests), 3)
def test_fetch_from_cache(self):
"""Test whether the cache works"""
http_requests = setup_http_server()
# First, we fetch the messages from the server,
# storing them in a cache
cache = Cache(self.tmp_path)
tlg = Telegram(TELEGRAM_BOT, TELEGRAM_TOKEN, cache=cache)
messages = [msg for msg in tlg.fetch()]
self.assertEqual(len(http_requests), 3)
# Now, we get the messages from the cache.
# The contents should be the same and there won't be
# any new request to the server
cached_messages = [msg for msg in tlg.fetch_from_cache()]
self.assertEqual(len(cached_messages), len(messages))
expected = [(31, '5a5457aec04237ac3fab30031e84c745a3bdd157', 1467289325.0, 319280318),
(32, '16a59e93e919174fcd4e70e5b3289201c1016c72', 1467289329.0, 319280319),
(33, '9d03eeea7e3186ca8e5c150b4cbf18c8283cca9d', 1467289371.0, 319280320),
(34, '2e61e72b64c9084f3c5a36671c3119641c3ae42f', 1467370372.0, 319280321)]
self.assertEqual(len(cached_messages), len(expected))
for x in range(len(cached_messages)):
message = cached_messages[x]
self.assertEqual(message['data']['message']['message_id'], expected[x][0])
self.assertEqual(message['origin'], 'https://telegram.org/' + TELEGRAM_BOT)
self.assertEqual(message['uuid'], expected[x][1])
self.assertEqual(message['updated_on'], expected[x][2])
self.assertEqual(message['offset'], expected[x][3])
self.assertEqual(message['category'], 'message')
self.assertEqual(message['tag'], 'https://telegram.org/' + TELEGRAM_BOT)
# No more requests were sent
self.assertEqual(len(http_requests), 3)
