|
9 | 9 | from tests.contrib.config import REDIS_CONFIG |
10 | 10 | from tests.test_tracer import DummyWriter |
11 | 11 |
|
12 | | -from nose.tools import eq_, ok_ |
13 | | - |
14 | 12 | if __name__ == '__main__': |
15 | 13 | r = redis.Redis(port=REDIS_CONFIG['port']) |
16 | 14 | pin = Pin.get_from(r) |
17 | | - ok_(pin) |
18 | | - eq_(pin.app, 'redis') |
19 | | - eq_(pin.service, 'redis') |
| 15 | + assert pin |
| 16 | + assert pin.app == 'redis' |
| 17 | + assert pin.service == 'redis' |
20 | 18 |
|
21 | 19 | pin.tracer.writer = DummyWriter() |
22 | 20 | r.flushall() |
23 | 21 | spans = pin.tracer.writer.pop() |
24 | 22 |
|
25 | | - eq_(len(spans), 1) |
26 | | - eq_(spans[0].service, 'redis') |
27 | | - eq_(spans[0].resource, 'FLUSHALL') |
| 23 | + assert len(spans) == 1 |
| 24 | + assert spans[0].service == 'redis' |
| 25 | + assert spans[0].resource == 'FLUSHALL' |
28 | 26 |
|
29 | 27 | long_cmd = "mget %s" % " ".join(map(str, range(1000))) |
30 | 28 | us = r.execute_command(long_cmd) |
31 | 29 |
|
32 | 30 | spans = pin.tracer.writer.pop() |
33 | | - eq_(len(spans), 1) |
| 31 | + assert len(spans) == 1 |
34 | 32 | span = spans[0] |
35 | | - eq_(span.service, 'redis') |
36 | | - eq_(span.name, 'redis.command') |
37 | | - eq_(span.span_type, 'redis') |
38 | | - eq_(span.error, 0) |
| 33 | + assert span.service == 'redis' |
| 34 | + assert span.name == 'redis.command' |
| 35 | + assert span.span_type == 'redis' |
| 36 | + assert span.error == 0 |
39 | 37 | meta = { |
40 | 38 | 'out.host': u'localhost', |
41 | 39 | 'out.port': str(REDIS_CONFIG['port']), |
42 | 40 | 'out.redis_db': u'0', |
43 | 41 | } |
44 | 42 | for k, v in meta.items(): |
45 | | - eq_(span.get_tag(k), v) |
| 43 | + assert span.get_tag(k) == v |
46 | 44 |
|
47 | 45 | assert span.get_tag('redis.raw_command').startswith(u'mget 0 1 2 3') |
48 | 46 | assert span.get_tag('redis.raw_command').endswith(u'...') |
|
0 commit comments