tests: use a method to test a method
Change-Id: I14ea6bbd55512e8798e5cadf4f6e68f95961206b
This commit is contained in:
parent
7d3a32d107
commit
6a633d06fb
@ -7692,88 +7692,82 @@ class TestLoggerStatsdClientDelegation(unittest.TestCase):
|
|||||||
class TestTimingStatsDecorators(unittest.TestCase):
|
class TestTimingStatsDecorators(unittest.TestCase):
|
||||||
def test_timing_stats(self):
|
def test_timing_stats(self):
|
||||||
class MockController(object):
|
class MockController(object):
|
||||||
def __init__(self, status):
|
def __init__(mock_self, status):
|
||||||
self.status = status
|
mock_self.status = status
|
||||||
with mock.patch(
|
mock_self.logger = debug_logger()
|
||||||
'swift.common.statsd_client.StatsdClient',
|
|
||||||
FakeStatsdClient):
|
|
||||||
self.logger = utils.get_logger({}, 'name')
|
|
||||||
|
|
||||||
@utils.timing_stats()
|
@utils.timing_stats()
|
||||||
def METHOD(controller):
|
def METHOD(mock_self):
|
||||||
return Response(status=controller.status)
|
return Response(status=mock_self.status)
|
||||||
|
|
||||||
now = time.time()
|
now = time.time()
|
||||||
mock_controller = MockController(200)
|
mock_controller = MockController(200)
|
||||||
with mock.patch('swift.common.utils.time.time', return_value=now):
|
with mock.patch('swift.common.utils.time.time', return_value=now):
|
||||||
METHOD(mock_controller)
|
mock_controller.METHOD()
|
||||||
self.assertEqual({'timing_since': [(('METHOD.timing', now), {})]},
|
self.assertEqual({'timing_since': [(('METHOD.timing', now), {})]},
|
||||||
mock_controller.logger.logger.statsd_client.calls)
|
mock_controller.logger.statsd_client.calls)
|
||||||
|
|
||||||
mock_controller = MockController(400)
|
mock_controller = MockController(400)
|
||||||
with mock.patch('swift.common.utils.time.time', return_value=now):
|
with mock.patch('swift.common.utils.time.time', return_value=now):
|
||||||
METHOD(mock_controller)
|
mock_controller.METHOD()
|
||||||
self.assertEqual({'timing_since': [(('METHOD.timing', now), {})]},
|
self.assertEqual({'timing_since': [(('METHOD.timing', now), {})]},
|
||||||
mock_controller.logger.logger.statsd_client.calls)
|
mock_controller.logger.statsd_client.calls)
|
||||||
|
|
||||||
mock_controller = MockController(404)
|
mock_controller = MockController(404)
|
||||||
with mock.patch('swift.common.utils.time.time', return_value=now):
|
with mock.patch('swift.common.utils.time.time', return_value=now):
|
||||||
METHOD(mock_controller)
|
mock_controller.METHOD()
|
||||||
self.assertEqual({'timing_since': [(('METHOD.timing', now), {})]},
|
self.assertEqual({'timing_since': [(('METHOD.timing', now), {})]},
|
||||||
mock_controller.logger.logger.statsd_client.calls)
|
mock_controller.logger.statsd_client.calls)
|
||||||
|
|
||||||
mock_controller = MockController(412)
|
mock_controller = MockController(412)
|
||||||
with mock.patch('swift.common.utils.time.time', return_value=now):
|
with mock.patch('swift.common.utils.time.time', return_value=now):
|
||||||
METHOD(mock_controller)
|
mock_controller.METHOD()
|
||||||
self.assertEqual({'timing_since': [(('METHOD.timing', now), {})]},
|
self.assertEqual({'timing_since': [(('METHOD.timing', now), {})]},
|
||||||
mock_controller.logger.logger.statsd_client.calls)
|
mock_controller.logger.statsd_client.calls)
|
||||||
|
|
||||||
mock_controller = MockController(416)
|
mock_controller = MockController(416)
|
||||||
with mock.patch('swift.common.utils.time.time', return_value=now):
|
with mock.patch('swift.common.utils.time.time', return_value=now):
|
||||||
METHOD(mock_controller)
|
mock_controller.METHOD()
|
||||||
self.assertEqual({'timing_since': [(('METHOD.timing', now), {})]},
|
self.assertEqual({'timing_since': [(('METHOD.timing', now), {})]},
|
||||||
mock_controller.logger.logger.statsd_client.calls)
|
mock_controller.logger.statsd_client.calls)
|
||||||
|
|
||||||
mock_controller = MockController(500)
|
mock_controller = MockController(500)
|
||||||
with mock.patch('swift.common.utils.time.time', return_value=now):
|
with mock.patch('swift.common.utils.time.time', return_value=now):
|
||||||
METHOD(mock_controller)
|
mock_controller.METHOD()
|
||||||
self.assertEqual(
|
self.assertEqual(
|
||||||
{'timing_since': [(('METHOD.errors.timing', now), {})]},
|
{'timing_since': [(('METHOD.errors.timing', now), {})]},
|
||||||
mock_controller.logger.logger.statsd_client.calls)
|
mock_controller.logger.statsd_client.calls)
|
||||||
|
|
||||||
mock_controller = MockController(507)
|
mock_controller = MockController(507)
|
||||||
with mock.patch('swift.common.utils.time.time', return_value=now):
|
with mock.patch('swift.common.utils.time.time', return_value=now):
|
||||||
METHOD(mock_controller)
|
mock_controller.METHOD()
|
||||||
self.assertEqual(
|
self.assertEqual(
|
||||||
{'timing_since': [(('METHOD.errors.timing', now), {})]},
|
{'timing_since': [(('METHOD.errors.timing', now), {})]},
|
||||||
mock_controller.logger.logger.statsd_client.calls)
|
mock_controller.logger.statsd_client.calls)
|
||||||
|
|
||||||
def test_memcached_timing_stats(self):
|
def test_memcached_timing_stats(self):
|
||||||
class MockMemcached(object):
|
class MockMemcached(object):
|
||||||
def __init__(self):
|
def __init__(mock_self):
|
||||||
with mock.patch(
|
mock_self.logger = debug_logger()
|
||||||
'swift.common.statsd_client.StatsdClient',
|
|
||||||
FakeStatsdClient):
|
|
||||||
self.logger = utils.get_logger({}, 'name')
|
|
||||||
|
|
||||||
@utils.memcached_timing_stats()
|
@utils.memcached_timing_stats()
|
||||||
def set(cache):
|
def set(mock_self):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
@utils.memcached_timing_stats()
|
@utils.memcached_timing_stats()
|
||||||
def get(cache):
|
def get(mock_self):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
mock_cache = MockMemcached()
|
mock_cache = MockMemcached()
|
||||||
with patch('time.time', return_value=1000.99):
|
with patch('time.time', return_value=1000.99):
|
||||||
set(mock_cache)
|
mock_cache.set()
|
||||||
self.assertEqual(
|
self.assertEqual(
|
||||||
{'timing_since': [(('memcached.set.timing', 1000.99), {})]},
|
{'timing_since': [(('memcached.set.timing', 1000.99), {})]},
|
||||||
mock_cache.logger.logger.statsd_client.calls)
|
mock_cache.logger.statsd_client.calls)
|
||||||
|
|
||||||
mock_cache = MockMemcached()
|
mock_cache = MockMemcached()
|
||||||
with patch('time.time', return_value=2000.99):
|
with patch('time.time', return_value=2000.99):
|
||||||
get(mock_cache)
|
mock_cache.get()
|
||||||
self.assertEqual(
|
self.assertEqual(
|
||||||
{'timing_since': [(('memcached.get.timing', 2000.99), {})]},
|
{'timing_since': [(('memcached.get.timing', 2000.99), {})]},
|
||||||
mock_cache.logger.logger.statsd_client.calls)
|
mock_cache.logger.statsd_client.calls)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user