def testGetUserBlockRate_WasMax(self):
test_utils.CreateEvent(
self.blockable, last_blocked_dt=datetime.datetime.utcnow(),
parent=utils.ConcatenateKeys(self.user1.key, self.host.key))
was_max, _ = self.host.GetUserBlockRate(
self.user1, max_events_to_fetch=1)
self.assertTrue(was_max)
def setUp(self):
super(EventTest, self).setUp()
self.earlier = datetime.datetime.utcnow()
self.middle = self.earlier + datetime.timedelta(seconds=1)
self.later = self.earlier + datetime.timedelta(seconds=2)
self.blockable = test_utils.CreateBlockable()
self.user = test_utils.CreateUser()
self.other_user = test_utils.CreateUser()
self.event_1 = test_utils.CreateEvent(
self.blockable, first_blocked_dt=self.earlier,
last_blocked_dt=self.earlier, executing_user=self.user.nickname)
self.event_2 = test_utils.CreateEvent(
self.blockable, first_blocked_dt=self.later, last_blocked_dt=self.later,
executing_user=self.other_user.nickname)
self.PatchSetting('EVENT_CREATION', constants.EVENT_CREATION.EXECUTING_USER)
def setUp(self):
super(GetEventKeysToInsertTest, self).setUp()
now = datetime.datetime.utcnow()
self.user = test_utils.CreateUser()
self.event = test_utils.CreateEvent(
test_utils.CreateBlockable(), first_blocked_dt=now,
last_blocked_dt=now, executing_user=self.user.nickname)
self.PatchSetting('EVENT_CREATION', constants.EVENT_CREATION.EXECUTING_USER)
def testGetUserBlockRate(self):
test_utils.CreateEvent(
self.blockable, last_blocked_dt=datetime.datetime.utcnow(),
parent=utils.ConcatenateKeys(self.user1.key, self.host.key))
was_max, block_rate = self.host.GetUserBlockRate(
self.user1, duration_to_fetch=datetime.timedelta(days=7))
self.assertFalse(was_max)
self.assertEqual(1. / 5, block_rate)
