Python KeyedAIOProducer.__repr__の例

コード例 #1

ファイル: test_producer.py プロジェクト: benjolitz/aiokafka

    def test_keyed_producer_ctor(self):
        client = mock.Mock()
        ack = KeyedAIOProducer.ACK_AFTER_LOCAL_WRITE

        producer = KeyedAIOProducer(client,
                                    partitioner=RoundRobinPartitioner,
                                    req_acks=ack)
        name = producer.__class__.__name__
        self.assertTrue(name in producer.__repr__())
        self.assertEqual(producer._req_acks, ack)

コード例 #2

    def test_keyed_producer_ctor(self):
        client = mock.Mock()
        ack = KeyedAIOProducer.ACK_AFTER_LOCAL_WRITE

        producer = KeyedAIOProducer(client,
                                    partitioner=RoundRobinPartitioner,
                                    req_acks=ack)
        name = producer.__class__.__name__
        self.assertTrue(name in producer.__repr__())
        self.assertEqual(producer._req_acks, ack)