def test_consumer_decrement_eventually(self): consumer = Mock() qos = QoS(consumer, 10) qos.decrement_eventually() self.assertEqual(qos.value, 9) qos.value = 0 qos.decrement_eventually() self.assertEqual(qos.value, 0)
def test_consumer_decrement_eventually(self): consumer = Mock() qos = QoS(consumer, 10, current_app.log.get_default_logger()) qos.decrement_eventually() self.assertEqual(qos.value, 9) qos.value = 0 qos.decrement_eventually() self.assertEqual(qos.value, 0)
def test_consumer_increment_decrement(self): consumer = Mock() qos = QoS(consumer, 10, current_app.log.get_default_logger()) qos.update() self.assertEqual(qos.value, 10) self.assertIn({"prefetch_count": 10}, consumer.qos.call_args) qos.decrement() self.assertEqual(qos.value, 9) self.assertIn({"prefetch_count": 9}, consumer.qos.call_args) qos.decrement_eventually() self.assertEqual(qos.value, 8) self.assertIn({"prefetch_count": 9}, consumer.qos.call_args) # Does not decrement 0 value qos.value = 0 qos.decrement() self.assertEqual(qos.value, 0) qos.increment() self.assertEqual(qos.value, 0)
def test_consumer_increment_decrement(self): consumer = self.MockConsumer() qos = QoS(consumer, 10, current_app.log.get_default_logger()) qos.update() self.assertEqual(qos.value, 10) self.assertEqual(consumer.prefetch_count, 10) qos.decrement() self.assertEqual(qos.value, 9) self.assertEqual(consumer.prefetch_count, 9) qos.decrement_eventually() self.assertEqual(qos.value, 8) self.assertEqual(consumer.prefetch_count, 9) # Does not decrement 0 value qos.value = 0 qos.decrement() self.assertEqual(qos.value, 0) qos.increment() self.assertEqual(qos.value, 0)
def test_consumer_increment_decrement(self): consumer = Mock() qos = QoS(consumer, 10) qos.update() self.assertEqual(qos.value, 10) consumer.qos.assert_called_with(prefetch_count=10) qos.decrement() self.assertEqual(qos.value, 9) consumer.qos.assert_called_with(prefetch_count=9) qos.decrement_eventually() self.assertEqual(qos.value, 8) consumer.qos.assert_called_with(prefetch_count=9) self.assertIn({"prefetch_count": 9}, consumer.qos.call_args) # Does not decrement 0 value qos.value = 0 qos.decrement() self.assertEqual(qos.value, 0) qos.increment() self.assertEqual(qos.value, 0)
def test_consumer_increment_decrement(self): consumer = Mock() qos = QoS(consumer, 10) qos.update() self.assertEqual(qos.value, 10) consumer.qos.assert_called_with(prefetch_count=10) qos.decrement() self.assertEqual(qos.value, 9) consumer.qos.assert_called_with(prefetch_count=9) qos.decrement_eventually() self.assertEqual(qos.value, 8) consumer.qos.assert_called_with(prefetch_count=9) self.assertIn({'prefetch_count': 9}, consumer.qos.call_args) # Does not decrement 0 value qos.value = 0 qos.decrement() self.assertEqual(qos.value, 0) qos.increment() self.assertEqual(qos.value, 0)