def setUp(self): super(BannedAdressCacheTestCase, self).setUp() self.cache = BannedAddressesCache() self.address = '127.0.0.1' self.ban_timestamp = datetime.datetime(2016, 12, 10, 21, 0, 0) self.expected_ban_expiry = self.ban_timestamp + datetime.timedelta( seconds=60)
class BannedAdressCacheTestCase(TestCase): def setUp(self): super(BannedAdressCacheTestCase, self).setUp() self.cache = BannedAddressesCache() self.address = '127.0.0.1' self.ban_timestamp = datetime.datetime(2016, 12, 10, 21, 0, 0) self.expected_ban_expiry = self.ban_timestamp + datetime.timedelta( seconds=60) def test_correct_withdrawal_is_set(self): self.cache.add(self.address) self.assertTrue(self.cache.contains(self.address)) expected_expiry = datetime.datetime.now() + datetime.timedelta( seconds=60) self.assertEqual(self.cache._cache[self.address].year, expected_expiry.year) self.assertEqual(self.cache._cache[self.address].month, expected_expiry.month) self.assertEqual(self.cache._cache[self.address].day, expected_expiry.day) self.assertEqual(self.cache._cache[self.address].hour, expected_expiry.hour) self.assertEqual(self.cache._cache[self.address].minute, expected_expiry.minute) self.assertEqual(self.cache._cache[self.address].second, expected_expiry.second) def test_flush_after_withdrawal(self): with mock_now(self.ban_timestamp): self.cache.add(self.address) with mock_now(self.expected_ban_expiry): self.assertFalse(self.cache.contains(self.address))
class BannedAdressCacheTestCase(TestCase): def setUp(self): super(BannedAdressCacheTestCase, self).setUp() self.cache = BannedAddressesCache() self.address = '127.0.0.1' self.ban_timestamp = datetime.datetime(2016, 12, 10, 21, 0, 0) self.expected_ban_expiry = self.ban_timestamp + datetime.timedelta(seconds=60) def test_correct_withdrawal_is_set(self): self.cache.add(self.address) self.assertTrue(self.cache.contains(self.address)) expected_expiry = datetime.datetime.now() + datetime.timedelta(seconds=60) self.assertEqual(self.cache._cache[self.address].year, expected_expiry.year) self.assertEqual(self.cache._cache[self.address].month, expected_expiry.month) self.assertEqual(self.cache._cache[self.address].day, expected_expiry.day) self.assertEqual(self.cache._cache[self.address].hour, expected_expiry.hour) self.assertEqual(self.cache._cache[self.address].minute, expected_expiry.minute) self.assertEqual(self.cache._cache[self.address].second, expected_expiry.second) def test_flush_after_withdrawal(self): with mock_now(self.ban_timestamp): self.cache.add(self.address) with mock_now(self.expected_ban_expiry): self.assertFalse(self.cache.contains(self.address))
def setUp(self): super(BannedAdressCacheTestCase, self).setUp() self.cache = BannedAddressesCache() self.address = '127.0.0.1' self.ban_timestamp = datetime.datetime(2016, 12, 10, 21, 0, 0) self.expected_ban_expiry = self.ban_timestamp + datetime.timedelta(seconds=60)