Esempio n. 1
0
 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)
Esempio n. 2
0
 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)
Esempio n. 3
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)
Esempio n. 4
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)
Esempio n. 5
0
 def test_exceeds_short(self):
     qos = QoS(Mock(), PREFETCH_COUNT_MAX - 1)
     qos.update()
     self.assertEqual(qos.value, PREFETCH_COUNT_MAX - 1)
     qos.increment_eventually()
     self.assertEqual(qos.value, PREFETCH_COUNT_MAX)
     qos.increment_eventually()
     self.assertEqual(qos.value, PREFETCH_COUNT_MAX + 1)
     qos.decrement_eventually()
     self.assertEqual(qos.value, PREFETCH_COUNT_MAX)
     qos.decrement_eventually()
     self.assertEqual(qos.value, PREFETCH_COUNT_MAX - 1)
Esempio n. 6
0
 def test_decrement(self):
     consumer = self.MockConsumer()
     qos = QoS(consumer, 10, app_or_default().log.get_default_logger())
     qos.update()
     self.assertEqual(int(qos.value), 10)
     self.assertEqual(consumer.prefetch_count, 10)
     qos.decrement()
     self.assertEqual(int(qos.value), 9)
     self.assertEqual(consumer.prefetch_count, 9)
     qos.decrement_eventually()
     self.assertEqual(int(qos.value), 8)
     self.assertEqual(consumer.prefetch_count, 9)
Esempio n. 7
0
 def test_exceeds_short(self):
     qos = QoS(Mock(), PREFETCH_COUNT_MAX - 1)
     qos.update()
     self.assertEqual(qos.value, PREFETCH_COUNT_MAX - 1)
     qos.increment_eventually()
     self.assertEqual(qos.value, PREFETCH_COUNT_MAX)
     qos.increment_eventually()
     self.assertEqual(qos.value, PREFETCH_COUNT_MAX + 1)
     qos.decrement_eventually()
     self.assertEqual(qos.value, PREFETCH_COUNT_MAX)
     qos.decrement_eventually()
     self.assertEqual(qos.value, PREFETCH_COUNT_MAX - 1)
Esempio n. 8
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)
Esempio n. 9
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)
Esempio n. 10
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)
Esempio n. 11
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)
Esempio n. 12
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)
Esempio n. 13
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)