Exemple #1
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)
 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)
Exemple #3
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_eventually()
        qos.update()
        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_eventually()
        self.assertEqual(qos.value, 0)
        qos.increment_eventually()
        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_eventually()
        qos.update()
        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_eventually()
        self.assertEqual(qos.value, 0)
        qos.increment_eventually()
        self.assertEqual(qos.value, 0)