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'))