1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81
|
import pytest
from celery.utils.text import abbr, abbrtask, ensure_newlines, indent, pretty, truncate
RANDTEXT = """\
The quick brown
fox jumps
over the
lazy dog\
"""
RANDTEXT_RES = """\
The quick brown
fox jumps
over the
lazy dog\
"""
QUEUES = {
'queue1': {
'exchange': 'exchange1',
'exchange_type': 'type1',
'routing_key': 'bind1',
},
'queue2': {
'exchange': 'exchange2',
'exchange_type': 'type2',
'routing_key': 'bind2',
},
}
QUEUE_FORMAT1 = '.> queue1 exchange=exchange1(type1) key=bind1'
QUEUE_FORMAT2 = '.> queue2 exchange=exchange2(type2) key=bind2'
class test_Info:
def test_textindent(self):
assert indent(RANDTEXT, 4) == RANDTEXT_RES
def test_format_queues(self, app):
app.amqp.queues = app.amqp.Queues(QUEUES)
assert (sorted(app.amqp.queues.format().split('\n')) ==
sorted([QUEUE_FORMAT1, QUEUE_FORMAT2]))
def test_ensure_newlines(self):
assert len(ensure_newlines('foo\nbar\nbaz\n').splitlines()) == 3
assert len(ensure_newlines('foo\nbar').splitlines()) == 2
@pytest.mark.parametrize('s,maxsize,expected', [
('ABCDEFGHI', 3, 'ABC...'),
('ABCDEFGHI', 10, 'ABCDEFGHI'),
])
def test_truncate_text(s, maxsize, expected):
assert truncate(s, maxsize) == expected
@pytest.mark.parametrize('args,expected', [
((None, 3), '???'),
(('ABCDEFGHI', 6), 'ABC...'),
(('ABCDEFGHI', 20), 'ABCDEFGHI'),
(('ABCDEFGHI', 6, None), 'ABCDEF'),
])
def test_abbr(args, expected):
assert abbr(*args) == expected
@pytest.mark.parametrize('s,maxsize,expected', [
(None, 3, '???'),
('feeds.tasks.refresh', 10, '[.]refresh'),
('feeds.tasks.refresh', 30, 'feeds.tasks.refresh'),
])
def test_abbrtask(s, maxsize, expected):
assert abbrtask(s, maxsize) == expected
def test_pretty():
assert pretty(('a', 'b', 'c'))
|