def test_long_url(self, mocked_time):
# This test does nothing on SQLite - max length of a varchar is not enforced
# The test is here as an example, or for a future where we run tests on
# another database engine.
url = 'https://testserver/api/resource?' + 'longurlparams' * 100
throttle_1 = CacheDBThrottle(throttle_at=2, timeframe=5, expiration=2)
throttle_1.accessed(identifier='accessor', url=url)
access = ApiAccess.objects.filter(identifier='accessor').first()
self.assertEqual(access.url, url)
def test_long_url(self, mocked_time):
# This test does nothing on SQLite - max length of a varchar is not enforced
# The test is here as an example, or for a future where we run tests on
# another database engine.
url = 'https://testserver/api/resource?' + 'longurlparams' * 100
throttle_1 = CacheDBThrottle(throttle_at=2, timeframe=5, expiration=2)
throttle_1.accessed(identifier='accessor', url=url)
access = ApiAccess.objects.filter(identifier='accessor').first()
self.assertEqual(access.url, url)
def test_throttling(self, mocked_time):
mocked_time.time.return_value = time.time()
throttle_1 = CacheDBThrottle(throttle_at=2, timeframe=5, expiration=2)
self.assertEqual(throttle_1.should_be_throttled('daniel'), False)
self.assertEqual(len(cache.get('daniel_accesses')), 0)
self.assertEqual(ApiAccess.objects.count(), 0)
self.assertEqual(
ApiAccess.objects.filter(identifier='daniel').count(), 0)
self.assertEqual(throttle_1.accessed('daniel'), None)
self.assertEqual(throttle_1.should_be_throttled('daniel'), False)
self.assertEqual(len(cache.get('daniel_accesses')), 1)
self.assertEqual(cache.get('cody_accesses'), None)
self.assertEqual(ApiAccess.objects.count(), 1)
self.assertEqual(
ApiAccess.objects.filter(identifier='daniel').count(), 1)
self.assertEqual(throttle_1.accessed('daniel'), None)
self.assertEqual(throttle_1.accessed('cody'), None)
self.assertEqual(throttle_1.should_be_throttled('cody'), False)
self.assertEqual(len(cache.get('daniel_accesses')), 2)
self.assertEqual(len(cache.get('cody_accesses')), 1)
self.assertEqual(ApiAccess.objects.count(), 3)
self.assertEqual(
ApiAccess.objects.filter(identifier='daniel').count(), 2)
self.assertEqual(throttle_1.accessed('cody'), None)
# THROTTLE'D!
self.assertEqual(throttle_1.accessed('daniel'), None)
self.assertEqual(throttle_1.should_be_throttled('daniel'), 5)
self.assertEqual(len(cache.get('daniel_accesses')), 3)
self.assertEqual(ApiAccess.objects.count(), 5)
self.assertEqual(
ApiAccess.objects.filter(identifier='daniel').count(), 3)
self.assertEqual(throttle_1.accessed('daniel'), None)
self.assertEqual(throttle_1.should_be_throttled('daniel'), 5)
self.assertEqual(len(cache.get('daniel_accesses')), 4)
self.assertEqual(ApiAccess.objects.count(), 6)
self.assertEqual(
ApiAccess.objects.filter(identifier='daniel').count(), 4)
# Should be no interplay.
self.assertEqual(throttle_1.should_be_throttled('cody'), 5)
self.assertEqual(throttle_1.accessed('cody'), None)
self.assertEqual(ApiAccess.objects.count(), 7)
self.assertEqual(
ApiAccess.objects.filter(identifier='daniel').count(), 4)
# Test the timeframe.
mocked_time.time.return_value += throttle_1.timeframe + 1
self.assertEqual(throttle_1.should_be_throttled('daniel'), False)
self.assertEqual(len(cache.get('daniel_accesses')), 0)
self.assertEqual(ApiAccess.objects.count(), 7)
self.assertEqual(
ApiAccess.objects.filter(identifier='daniel').count(), 4)
def test_throttling(self):
throttle_1 = CacheDBThrottle(throttle_at=2, timeframe=5, expiration=2)
self.assertEqual(throttle_1.should_be_throttled('daniel'), False)
self.assertEqual(len(cache.get('daniel_accesses')), 0)
self.assertEqual(ApiAccess.objects.count(), 0)
self.assertEqual(ApiAccess.objects.filter(identifier='daniel').count(), 0)
self.assertEqual(throttle_1.accessed('daniel'), None)
self.assertEqual(throttle_1.should_be_throttled('daniel'), False)
self.assertEqual(len(cache.get('daniel_accesses')), 1)
self.assertEqual(cache.get('cody_accesses'), None)
self.assertEqual(ApiAccess.objects.count(), 1)
self.assertEqual(ApiAccess.objects.filter(identifier='daniel').count(), 1)
self.assertEqual(throttle_1.accessed('daniel'), None)
self.assertEqual(throttle_1.accessed('cody'), None)
self.assertEqual(throttle_1.should_be_throttled('cody'), False)
self.assertEqual(len(cache.get('daniel_accesses')), 2)
self.assertEqual(len(cache.get('cody_accesses')), 1)
self.assertEqual(ApiAccess.objects.count(), 3)
self.assertEqual(ApiAccess.objects.filter(identifier='daniel').count(), 2)
self.assertEqual(throttle_1.accessed('cody'), None)
# THROTTLE'D!
self.assertEqual(throttle_1.accessed('daniel'), None)
self.assertEqual(throttle_1.should_be_throttled('daniel'), True)
self.assertEqual(len(cache.get('daniel_accesses')), 3)
self.assertEqual(ApiAccess.objects.count(), 5)
self.assertEqual(ApiAccess.objects.filter(identifier='daniel').count(), 3)
self.assertEqual(throttle_1.accessed('daniel'), None)
self.assertEqual(throttle_1.should_be_throttled('daniel'), True)
self.assertEqual(len(cache.get('daniel_accesses')), 4)
self.assertEqual(ApiAccess.objects.count(), 6)
self.assertEqual(ApiAccess.objects.filter(identifier='daniel').count(), 4)
# Should be no interplay.
self.assertEqual(throttle_1.should_be_throttled('cody'), True)
self.assertEqual(throttle_1.accessed('cody'), None)
self.assertEqual(ApiAccess.objects.count(), 7)
self.assertEqual(ApiAccess.objects.filter(identifier='daniel').count(), 4)
# Test the timeframe.
ret_time = time.time() + throttle_1.timeframe + 1
with mock.patch('tastypie.throttle.time') as mocked_time:
mocked_time.time.return_value = ret_time
self.assertEqual(throttle_1.should_be_throttled('daniel'), False)
self.assertEqual(len(cache.get('daniel_accesses')), 0)
self.assertEqual(ApiAccess.objects.count(), 7)
self.assertEqual(ApiAccess.objects.filter(identifier='daniel').count(), 4)
def test_throttling(self):
throttle_1 = CacheDBThrottle(throttle_at=2, timeframe=5, expiration=2)
self.assertEqual(throttle_1.should_be_throttled('daniel'), False)
self.assertEqual(len(cache.get('daniel_accesses')), 0)
self.assertEqual(ApiAccess.objects.count(), 0)
self.assertEqual(ApiAccess.objects.filter(identifier='daniel').count(), 0)
self.assertEqual(throttle_1.accessed('daniel'), None)
self.assertEqual(throttle_1.should_be_throttled('daniel'), False)
self.assertEqual(len(cache.get('daniel_accesses')), 1)
self.assertEqual(cache.get('cody_accesses'), None)
self.assertEqual(ApiAccess.objects.count(), 1)
self.assertEqual(ApiAccess.objects.filter(identifier='daniel').count(), 1)
self.assertEqual(throttle_1.accessed('daniel'), None)
self.assertEqual(throttle_1.accessed('cody'), None)
self.assertEqual(throttle_1.should_be_throttled('cody'), False)
self.assertEqual(len(cache.get('daniel_accesses')), 2)
self.assertEqual(len(cache.get('cody_accesses')), 1)
self.assertEqual(ApiAccess.objects.count(), 3)
self.assertEqual(ApiAccess.objects.filter(identifier='daniel').count(), 2)
self.assertEqual(throttle_1.accessed('cody'), None)
# THROTTLE'D!
self.assertEqual(throttle_1.accessed('daniel'), None)
self.assertEqual(throttle_1.should_be_throttled('daniel'), True)
self.assertEqual(len(cache.get('daniel_accesses')), 3)
self.assertEqual(ApiAccess.objects.count(), 5)
self.assertEqual(ApiAccess.objects.filter(identifier='daniel').count(), 3)
self.assertEqual(throttle_1.accessed('daniel'), None)
self.assertEqual(throttle_1.should_be_throttled('daniel'), True)
self.assertEqual(len(cache.get('daniel_accesses')), 4)
self.assertEqual(ApiAccess.objects.count(), 6)
self.assertEqual(ApiAccess.objects.filter(identifier='daniel').count(), 4)
# Should be no interplay.
self.assertEqual(throttle_1.should_be_throttled('cody'), True)
self.assertEqual(throttle_1.accessed('cody'), None)
self.assertEqual(ApiAccess.objects.count(), 7)
self.assertEqual(ApiAccess.objects.filter(identifier='daniel').count(), 4)
# Test the timeframe.
time.sleep(3)
self.assertEqual(throttle_1.should_be_throttled('daniel'), False)
self.assertEqual(len(cache.get('daniel_accesses')), 0)
self.assertEqual(ApiAccess.objects.count(), 7)
self.assertEqual(ApiAccess.objects.filter(identifier='daniel').count(), 4)
