def test_rate_limit_messages_matching_substring(self):
config = {'match': ['rate limited']}
f = RateLimitingFilter(rate=1, per=1, burst=1, **config)
mock_matching_record = Mock()
mock_matching_record.msg = 'a rate limited test message'
mock_non_matching_record = Mock()
mock_non_matching_record.msg = 'a different test message'
result = []
for _ in range(20):
if f.filter(mock_matching_record):
result.append(
mock_matching_record.msg
) # Only 2 of these get logged as they match the substring
if f.filter(mock_non_matching_record):
result.append(mock_non_matching_record.msg
) # 20 of these get logged as they don't match
self.mock_time.return_value += 0.1
self.assertEqual(
len([m for m in result if 'a rate limited test message' in m]), 2)
self.assertEqual(result.count('a different test message'), 20)
def test_rate_limit_messages_automatically(self):
config = {'match': 'auto'}
f = RateLimitingFilter(rate=1, per=1, burst=1, **config)
result = []
for i in range(20):
mock_varying_record = Mock()
mock_varying_record.msg = 'a rate limited varying message: {varying}'.format(
varying=i)
mock_rate_limited_record = Mock()
mock_rate_limited_record.msg = 'a completely different message'
if f.filter(mock_varying_record):
# Only 2 of these get logged as they are considered the same message,
# even though they are not identical
result.append(mock_varying_record.msg)
if f.filter(mock_rate_limited_record):
# Only 2 of these get logged as they are the all identical
result.append(mock_rate_limited_record.msg)
self.mock_time.return_value += 0.1
self.assertEqual(
len([m for m in result if 'a rate limited varying message' in m]),
2)
self.assertEqual(
len([m for m in result if 'a completely different message' in m]),
2)
def test_append_num_limited_records_to_message(self):
filtered = []
f = RateLimitingFilter(rate=1, per=1, burst=4)
for _ in range(30):
mock_record = Mock()
mock_record.msg = 'test message'
if f.filter(mock_record):
filtered.append(mock_record)
self.mock_time.return_value += 0.1
self.assertEqual(len(filtered), 6)
self.assertTrue(
filtered[4].msg.endswith(os.linesep +
'... 6 additional messages suppressed'))
self.assertTrue(
filtered[5].msg.endswith(os.linesep +
'... 9 additional messages suppressed'))