def test_advance_time_delta(self): timeutils.set_time_override(self.one_minute_before) timeutils.advance_time_delta(datetime.timedelta(seconds=60)) self.assertEqual(timeutils.utcnow(), self.skynet_self_aware_time)
def test_message_counter(self): queue_name = self.queue_name iterations = 10 seed_marker1 = self.queue_controller._get_counter(queue_name, self.project) self.assertEqual(seed_marker1, 1, 'First marker is 1') for i in range(iterations): self.controller.post(queue_name, [{'ttl': 60}], 'uuid', project=self.project) marker1 = self.queue_controller._get_counter(queue_name, self.project) marker2 = self.queue_controller._get_counter(queue_name, self.project) marker3 = self.queue_controller._get_counter(queue_name, self.project) self.assertEqual(marker1, marker2) self.assertEqual(marker2, marker3) self.assertEqual(marker1, i + 2) new_value = self.queue_controller._inc_counter(queue_name, self.project) self.assertIsNotNone(new_value) value_before = self.queue_controller._get_counter(queue_name, project=self.project) new_value = self.queue_controller._inc_counter(queue_name, project=self.project) self.assertIsNotNone(new_value) value_after = self.queue_controller._get_counter(queue_name, project=self.project) self.assertEqual(value_after, value_before + 1) value_before = value_after new_value = self.queue_controller._inc_counter(queue_name, project=self.project, amount=7) value_after = self.queue_controller._get_counter(queue_name, project=self.project) self.assertEqual(value_after, value_before + 7) self.assertEqual(value_after, new_value) reference_value = value_after unchanged = self.queue_controller._inc_counter(queue_name, project=self.project, window=10) self.assertIsNone(unchanged) timeutils.set_time_override() timeutils.advance_time_delta(datetime.timedelta(seconds=10)) changed = self.queue_controller._inc_counter(queue_name, project=self.project, window=5) self.assertEqual(changed, reference_value + 1) timeutils.clear_time_override()
def advance_time_delta(self, timedelta): """Advance overridden time using a datetime.timedelta.""" timeutils.advance_time_delta(timedelta)
def test_message_counter(self): queue_name = self.queue_name iterations = 10 m = mock.MagicMock(controllers.QueueController) self.controller._queue_ctrl = m del self.controller._queue_ctrl._get_counter del self.controller._queue_ctrl._inc_counter seed_marker1 = self.controller._get_counter(queue_name, self.project) self.assertEqual(0, seed_marker1, 'First marker is 0') for i in range(iterations): self.controller.post(queue_name, [{'ttl': 60}], 'uuid', project=self.project) marker1 = self.controller._get_counter(queue_name, self.project) marker2 = self.controller._get_counter(queue_name, self.project) marker3 = self.controller._get_counter(queue_name, self.project) self.assertEqual(marker1, marker2) self.assertEqual(marker2, marker3) self.assertEqual(i + 1, marker1) new_value = self.controller._inc_counter(queue_name, self.project) self.assertIsNotNone(new_value) value_before = self.controller._get_counter(queue_name, project=self.project) new_value = self.controller._inc_counter(queue_name, project=self.project) self.assertIsNotNone(new_value) value_after = self.controller._get_counter(queue_name, project=self.project) self.assertEqual(value_before + 1, value_after) value_before = value_after new_value = self.controller._inc_counter(queue_name, project=self.project, amount=7) value_after = self.controller._get_counter(queue_name, project=self.project) self.assertEqual(value_before + 7, value_after) self.assertEqual(new_value, value_after) reference_value = value_after unchanged = self.controller._inc_counter(queue_name, project=self.project, window=10) self.assertIsNone(unchanged) timeutils.set_time_override() timeutils.advance_time_delta(datetime.timedelta(seconds=10)) changed = self.controller._inc_counter(queue_name, project=self.project, window=5) self.assertEqual(reference_value + 1, changed) timeutils.clear_time_override()
def test_message_counter(self): queue_name = self.queue_name iterations = 10 seed_marker1 = self.queue_controller._get_counter( queue_name, self.project) self.assertEqual(seed_marker1, 1, 'First marker is 1') for i in range(iterations): self.controller.post(queue_name, [{ 'ttl': 60 }], 'uuid', project=self.project) marker1 = self.queue_controller._get_counter( queue_name, self.project) marker2 = self.queue_controller._get_counter( queue_name, self.project) marker3 = self.queue_controller._get_counter( queue_name, self.project) self.assertEqual(marker1, marker2) self.assertEqual(marker2, marker3) self.assertEqual(marker1, i + 2) new_value = self.queue_controller._inc_counter(queue_name, self.project) self.assertIsNotNone(new_value) value_before = self.queue_controller._get_counter(queue_name, project=self.project) new_value = self.queue_controller._inc_counter(queue_name, project=self.project) self.assertIsNotNone(new_value) value_after = self.queue_controller._get_counter(queue_name, project=self.project) self.assertEqual(value_after, value_before + 1) value_before = value_after new_value = self.queue_controller._inc_counter(queue_name, project=self.project, amount=7) value_after = self.queue_controller._get_counter(queue_name, project=self.project) self.assertEqual(value_after, value_before + 7) self.assertEqual(value_after, new_value) reference_value = value_after unchanged = self.queue_controller._inc_counter(queue_name, project=self.project, window=10) self.assertIsNone(unchanged) timeutils.set_time_override() timeutils.advance_time_delta(datetime.timedelta(seconds=10)) changed = self.queue_controller._inc_counter(queue_name, project=self.project, window=5) self.assertEqual(changed, reference_value + 1) timeutils.clear_time_override()