def client(redis_client, request):
def cleanup():
"""Remove everything from redis to ensure a clean slate between tests."""
redis_client.flushall()
request.addfinalizer(cleanup)
return testing.TestClient(create(DbClient(client=redis_client)))
def test_get_unread_returns_stored_messages(mocker):
mockdb = MockDb()
messages = mock_get(mocker, mockdb, 'get_unread')
client = testing.TestClient(create(mockdb))
result = client.simulate_get('/messages/1')
assert result.status_code == 200
assert json.loads(result.text) == messages
def test_get_with_no_messages(mocker):
mockdb = MockDb()
mocker.patch.object(mockdb, 'get_unread')
client = testing.TestClient(create(mockdb))
result = client.simulate_get('/messages/1')
assert result.status_code == 200
print('text', result.text)
assert json.loads(result.text) == []
def test_delete_no_messages(mocker):
mockdb = MockDb()
patch = mocker.patch.object(mockdb, 'remove')
patch.return_value = []
client = testing.TestClient(create(mockdb))
result = client.simulate_delete('/messages/1', body=json.dumps([]))
assert result.status_code == 200
assert json.loads(result.text) == {'removed': 0}
def test_post_fails_if_add_fails(mocker):
mockdb = MockDb()
patch = mocker.patch.object(mockdb, 'add')
patch.return_value = False
client = testing.TestClient(create(mockdb))
result = client.simulate_post('/messages/1',
body=json.dumps({'foo': 'bar'}))
assert result.status_code == 500
assert result.text == ''
def test_post_adds_message_with_uuid(mocker):
mockdb = MockDb()
spy = mocker.spy(mockdb, 'add')
client = testing.TestClient(create(mockdb))
client.simulate_post('/messages/1', body=json.dumps({'foo': 'bar'}))
(id, data) = spy.call_args[0]
assert spy.call_count == 1
assert id == '1'
assert msgpack.unpackb(data)[b'uuid']
def test_get_defaults_with_no_end(mocker):
mockdb = MockDb()
spy = mocker.spy(mockdb, 'get_range')
client = testing.TestClient(create(mockdb))
client.simulate_get('/messages/1', params={'start': '0'})
(id, start, end) = spy.call_args[0]
assert spy.call_count == 1
assert id == '1'
assert start == 0
assert end == -1
def test_get_returns_stored_messages(mocker):
mockdb = MockDb()
messages = mock_get(mocker, mockdb, 'get_range')
client = testing.TestClient(create(mockdb))
result = client.simulate_get('/messages/1',
params={
'start': '0',
'end': 2
})
assert result.status_code == 200
assert json.loads(result.text) == messages
def test_delete_packs_messages(mocker):
mockdb = MockDb()
spy = mocker.spy(mockdb, 'remove')
client = testing.TestClient(create(mockdb))
messages = [{'uuid': '0', 'data': 'foo'}, {'uuid': '3', 'data': {}}]
client.simulate_delete('/messages/1', body=json.dumps(messages))
(id, elements) = spy.call_args[0]
assert spy.call_count == 1
assert id == '1'
assert elements == [
msgpack.packb(msg, use_bin_type=True) for msg in messages
]
import logging
import colorlog
from msgr.app import create
from msgr.db import DbClient
app = create(DbClient())
def _configure_logging():
log = logging.getLogger('msgr')
handler = colorlog.StreamHandler()
formatter = colorlog.ColoredFormatter(
fmt=('%(log_color)s[%(asctime)s %(levelname)8s] --'
' %(message)s (%(filename)s:%(lineno)s)'),
datefmt='%Y-%m-%d %H:%M:%S')
handler.setFormatter(formatter)
log.addHandler(handler)
log.setLevel(logging.DEBUG)
